Move ModelConfig to types

vSpaceWallet · Nov 4, 2024 · d925dd5 · d925dd5
1 parent 04879b2
commit d925dd5
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 13 deletions.
diff --git a/src/agent/utils.ts b/src/agent/utils.ts
@@ -1,20 +1,13 @@
 import { isArtifactCodeContent } from "@/lib/artifact_content_types";
 import { BaseStore, LangGraphRunnableConfig } from "@langchain/langgraph";
-import { ArtifactCodeV3, ArtifactMarkdownV3, Reflections } from "../types";
+import {
+  ArtifactCodeV3,
+  ArtifactMarkdownV3,
+  Reflections,
+  ModelConfig,
+} from "../types";
 import { initChatModel } from "langchain/chat_models/universal";
 
-type ModelConfig = {
-  temperature?: number;
-  modelProvider: string;
-  maxTokens?: number;
-  azureConfig?: {
-    azureOpenAIApiKey: string;
-    azureOpenAIApiInstanceName: string;
-    azureOpenAIApiDeploymentName: string;
-    azureOpenAIApiVersion: string;
-  };
-};
-
 /**
  * Wrapper around initChatModel
  */

diff --git a/src/types.ts b/src/types.ts
@@ -265,3 +265,16 @@ export type RewriteArtifactMetaToolResponse =
       title: string;
       language: ProgrammingLanguageOptions;
     };
+
+export interface ModelConfig {
+  temperature?: number;
+  modelProvider: string;
+  maxTokens?: number;
+  azureConfig?: {
+    azureOpenAIApiKey: string;
+    azureOpenAIApiInstanceName: string;
+    azureOpenAIApiDeploymentName: string;
+    azureOpenAIApiVersion: string;
+    azureOpenAIBasePath?: string;
+  };
+}