feat(langchain): Infer o1 model strings in initChatModel (langchain-a…

…i#7350)
aks-456 · Dec 13, 2024 · 14f8e20 · 14f8e20
1 parent ba56d19
commit 14f8e20
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/langchain/src/chat_models/universal.ts b/langchain/src/chat_models/universal.ts
@@ -175,7 +175,11 @@ async function _initChatModelHelper(
  * _inferModelProvider("unknown-model"); // returns undefined
  */
 export function _inferModelProvider(modelName: string): string | undefined {
-  if (modelName.startsWith("gpt-3") || modelName.startsWith("gpt-4")) {
+  if (
+    modelName.startsWith("gpt-3") ||
+    modelName.startsWith("gpt-4") ||
+    modelName.startsWith("o1")
+  ) {
     return "openai";
   } else if (modelName.startsWith("claude")) {
     return "anthropic";