more pr changes

aadesh18 · aadesh18 · commit 48fdc6e5c56b · 2026-03-12T17:13:04.000-07:00
diff --git a/apps/dashboard/src/components/assistant-ui/markdown-text.tsx b/apps/dashboard/src/components/assistant-ui/markdown-text.tsx
@@ -25,6 +25,7 @@ const MarkdownTextImpl = () => {
 export const MarkdownText = memo(MarkdownTextImpl);
 
 const CodeHeader: FC<CodeHeaderProps> = ({ language }) => {
+  if (!language) return null;
   return (
     <div className="flex items-center rounded-t-lg bg-zinc-900 px-4 py-2 text-sm font-semibold text-white">
       <span className="lowercase [&>span]:text-xs">{language}</span>
diff --git a/apps/dashboard/src/components/commands/create-dashboard/dashboard-sandbox-host.tsx b/apps/dashboard/src/components/commands/create-dashboard/dashboard-sandbox-host.tsx
@@ -107,13 +107,20 @@ function getDependencyScripts(esmVersion: string, esmFallbackVersion: string, da
     </script>`;
 }
 
+function escapeScriptContent(code: string): string {
+  return code
+    .replace(/<\/script/gi, "<\\/script")
+    .replace(/<!--/g, "<\\!--")
+    .replace(/-->/g, "--\\>");
+}
+
 function getSandboxDocument(artifact: DashboardArtifact, baseUrl: string, dashboardUrl: string, initialTheme: "light" | "dark", showControls: boolean, initialChatOpen: boolean): string {
-  const sourceCode = artifact.runtimeCodegen.uiRuntimeSourceCode;
+  const sourceCode = escapeScriptContent(artifact.runtimeCodegen.uiRuntimeSourceCode);
   const darkClass = initialTheme === "dark" ? "dark" : "";
   const esmVersion = packageJson.version;
   const esmFallbackVersion = "2.8.71";
   const devScriptSrc = isDev ? ` ${dashboardUrl}` : '';
-  const devConnectSrc = isDev ? ` ${dashboardUrl} http://127.0.0.1:7322` : '';
+  const devConnectSrc = isDev ? ` ${dashboardUrl}` : '';
 
   return html`<!doctype html>
 <html class="${darkClass}">
@@ -445,7 +452,9 @@ export const DashboardSandboxHost = memo(function DashboardSandboxHost({
   const { resolvedTheme } = useTheme();
 
   const baseUrl = useMemo(() => {
-    return getPublicEnvVar("NEXT_PUBLIC_STACK_API_URL") ?? 'http://localhost:8102';
+    const url = getPublicEnvVar("NEXT_PUBLIC_STACK_API_URL");
+    if (!url) throw new Error("NEXT_PUBLIC_STACK_API_URL is not set");
+    return url;
   }, []);
 
   const dashboardUrl = useMemo(() => {
