diff --git a/js/src/schemas.ts b/js/src/schemas.ts index f33ee1f80..6cba693ef 100644 --- 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 index e43c55775..0ea56f882 100644 --- a/js/src/wrappers/openai.ts +++ b/js/src/wrappers/openai.ts @@ -263,7 +263,7 @@ export const wrapOpenAI = ( 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,