diff --git a/js/src/schemas.ts b/js/src/schemas.ts index 4d73f29aa..a4f62aef0 100644 --- a/js/src/schemas.ts +++ b/js/src/schemas.ts @@ -409,7 +409,7 @@ export type RetrieverOutput = Array<{ export interface InvocationParamsSchema { ls_provider?: string; ls_model_name?: string; - ls_model_type: "chat" | "text"; + ls_model_type: "chat" | "llm"; 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 05fae4d5d..fa5af83ce 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: "text", + ls_model_type: "llm", ls_model_name: params.model, ls_max_tokens: params.max_tokens ?? undefined, ls_temperature: params.temperature ?? undefined,