chore(js): make sure we use "type": "text" for invocation params trac…

…king
langchain-ai · May 20, 2024 · efe2972 · efe2972
1 parent 7fc116a
commit efe2972
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/js/src/schemas.ts b/js/src/schemas.ts
@@ -397,7 +397,7 @@ export type RetrieverOutput = Array<{
 export interface InvocationParamsSchema {
   ls_provider?: string;
   ls_model_name?: string;
-  ls_model_type: "chat";
+  ls_model_type: "chat" | "text";
   ls_temperature?: number;
   ls_max_tokens?: number;
   ls_stop?: string[];

diff --git a/js/src/wrappers/openai.ts b/js/src/wrappers/openai.ts
@@ -263,7 +263,7 @@ export const wrapOpenAI = <T extends OpenAIType>(
 
         return {
           ls_provider: "openai",
-          ls_model_type: "chat",
+          ls_model_type: "text",
           ls_model_name: params.model,
           ls_max_tokens: params.max_tokens ?? undefined,
           ls_temperature: params.temperature ?? undefined,