diff --git a/ui/litellm-dashboard/src/components/chat_ui.tsx b/ui/litellm-dashboard/src/components/chat_ui.tsx
index ed6744754a8..edab67b2bad 100644
--- a/ui/litellm-dashboard/src/components/chat_ui.tsx
+++ b/ui/litellm-dashboard/src/components/chat_ui.tsx
@@ -22,9 +22,12 @@ interface ChatUIProps {
 
 async function generateModelResponse(inputMessage: string, updateUI: (chunk: string) => void, selectedModel: string, accessToken: string) {
     // base url should be the current base_url 
+    const isLocal = process.env.NODE_ENV === "development";
+    console.log("isLocal:", isLocal);
+    const proxyBaseUrl = isLocal ? "http://localhost:4000" : window.location.origin;
     const client = new openai.OpenAI({
     apiKey: accessToken, // Replace with your OpenAI API key
-    baseURL: 'http://0.0.0.0:4000', // Replace with your OpenAI API base URL
+    baseURL: proxyBaseUrl, // Replace with your OpenAI API base URL
     dangerouslyAllowBrowser: true, // using a temporary litellm proxy key
   });