cr

src/agent/open-canvas/nodes/rewriteArtifact.ts

@@ -40,25 +40,25 @@ export const rewriteArtifact = async (
       .describe(
         "The new title to give the artifact. ONLY update this if the user is making a request which changes the subject/topic of the artifact."
       ),
-    programmingLanguage: z
+    language: z
       .enum(
         PROGRAMMING_LANGUAGES.map((lang) => lang.language) as [
           string,
           ...string[],
         ]
       )
-      .optional()
       .describe(
-        "The programming language of the code artifact. ONLY update this if the user is making a request which changes the programming language of the code artifact, or is asking for a code artifact to be generated."
+        "The language of the code artifact. This should be populated with the programming language if the user is requesting code to be written, or 'other', in all other cases."
       ),
   });
   const { modelName, modelProvider } =
     getModelNameAndProviderFromConfig(config);
-  const model = await initChatModel(modelName, {
-    temperature: 0,
-    modelProvider,
-  });
-  const toolCallingModel = model
+  const toolCallingModel = (
+    await initChatModel(modelName, {
+      temperature: 0,
+      modelProvider,
+    })
+  )
     .bindTools(
       [
         {
@@ -71,6 +71,15 @@ export const rewriteArtifact = async (
     )
     .withConfig({ runName: "optionally_update_artifact_meta" });
 
+  const smallModelWithConfig = (
+    await initChatModel(modelName, {
+      temperature: 0,
+      modelProvider,
+    })
+  ).withConfig({
+    runName: "rewrite_artifact_model_call",
+  });
+
   const store = ensureStoreInConfig(config);
   const assistantId = config.configurable?.assistant_id;
   if (!assistantId) {
@@ -136,10 +145,6 @@ export const rewriteArtifact = async (
         : ""
     );
 
-  const smallModelWithConfig = model.withConfig({
-    runName: "rewrite_artifact_model_call",
-  });
-
   const newArtifactResponse = await smallModelWithConfig.invoke([
     { role: "system", content: formattedPrompt },
     recentHumanMessage,

src/components/artifacts/ArtifactRenderer.tsx

@@ -288,7 +288,7 @@ export function ArtifactRenderer(props: ArtifactRendererProps) {
     ? getArtifactContent(props.artifact)
     : undefined;
 
-  if (!props.artifact || (!currentArtifactContent && props.isStreaming)) {
+  if (!props.artifact && props.isStreaming) {
     return <ArtifactLoading />;
   }
 

src/constants.ts

@@ -1,5 +1,5 @@
 export const LANGGRAPH_API_URL =
-  process.env.LANGGRAPH_API_URL ?? "http://localhost:52375";
+  process.env.LANGGRAPH_API_URL ?? "http://localhost:53529";
 // v2 is tied to the 'open-canvas-prod' deployment.
 export const ASSISTANT_ID_COOKIE = "oc_assistant_id_v2";
 // export const ASSISTANT_ID_COOKIE = "oc_assistant_id";
@@ -43,16 +43,16 @@ export const FIREWORKS_MODELS = [
     label: "Fireworks Llama 70B",
   },
   {
-    name: "accounts/fireworks/models/llama-v3p1-8b-instruct",
-    label: "Fireworks Llama 8B",
+    name: "accounts/fireworks/models/llama-v3p1-405b-instruct",
+    label: "Fireworks Llama 405B",
   },
 ];
 
 export const GEMINI_MODELS = [
-  {
-    name: "gemini-1.5-flash",
-    label: "Gemini 1.5 Flash",
-  },
+  // {
+  //   name: "gemini-1.5-flash",
+  //   label: "Gemini 1.5 Flash",
+  // },
 ];
 export const DEFAULT_MODEL_NAME: ALL_MODEL_NAMES = "gpt-4o-mini";
 export type OPENAI_MODEL_NAMES = (typeof OPENAI_MODELS)[number]["name"];

src/hooks/use-graph/useGraph.tsx

@@ -342,13 +342,8 @@ export function useGraph(useGraphInput: UseGraphInput) {
                   type: newArtifactText.type ?? "",
                 };
               } catch (_) {
-                console.error(
-                  "Failed to parse new artifact text",
-                  generateArtifactToolCallStr
-                );
                 continue;
               }
-              console.log("newArtifactText", newArtifactText);
 
               if (
                 newArtifactText.artifact &&
@@ -539,7 +534,6 @@ export function useGraph(useGraphInput: UseGraphInput) {
               chunk.data.name === "rewrite_artifact_model_call" &&
               rewriteArtifactMeta
             ) {
-              console.log("IN THIS NODE");
               if (!artifact) {
                 toast({
                   title: "Error",
@@ -557,12 +551,12 @@ export function useGraph(useGraphInput: UseGraphInput) {
               if (
                 !artifactLanguage &&
                 rewriteArtifactMeta.type === "code" &&
-                rewriteArtifactMeta.programmingLanguage
+                rewriteArtifactMeta.language
               ) {
                 // If the type is `code` we should have a programming language populated
                 // in the rewriteArtifactMeta and can use that.
                 artifactLanguage =
-                  rewriteArtifactMeta.programmingLanguage as ProgrammingLanguageOptions;
+                  rewriteArtifactMeta.language as ProgrammingLanguageOptions;
               } else if (!artifactLanguage) {
                 artifactLanguage =
                   (prevCurrentContent?.title as ProgrammingLanguageOptions) ??
@@ -657,7 +651,7 @@ export function useGraph(useGraphInput: UseGraphInput) {
                   rewriteArtifactMeta: {
                     type: artifactType,
                     title: prevCurrentContent.title,
-                    programmingLanguage: artifactLanguage,
+                    language: artifactLanguage,
                   },
                   prevCurrentContent,
                   newArtifactIndex,

src/types.ts

@@ -258,9 +258,10 @@ export type RewriteArtifactMetaToolResponse =
   | {
       type: "text";
       title?: string;
+      language: ProgrammingLanguageOptions;
     }
   | {
       type: "code";
       title: string;
-      programmingLanguage: string;
+      language: ProgrammingLanguageOptions;
     };