Infer o1 model strings in initChatModel

langchain-ai · Dec 11, 2024 · a6e99b0 · a6e99b0
1 parent 1b7e34e
commit a6e99b0
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";