Change the 'text' model type to 'llm' to match other libraries (#1019)

langchain-ai · Sep 19, 2024 · e8ce685 · e8ce685
1 parent 5c43494
commit e8ce685
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -263,7 +263,7 @@ export const wrapOpenAI = <T extends OpenAIType>(
 
         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,

diff --git a/python/langsmith/wrappers/_openai.py b/python/langsmith/wrappers/_openai.py
@@ -243,6 +243,6 @@ def wrap_openai(
         completions_name,
         _reduce_completions,
         tracing_extra=tracing_extra,
-        invocation_params_fn=functools.partial(_infer_invocation_params, "text"),
+        invocation_params_fn=functools.partial(_infer_invocation_params, "llm"),
     )
     return client