fix(openai): include model_name in response_metadata (#7403)

langchain-ai · Dec 20, 2024 · 62bcabd · 62bcabd
1 parent 5afc890
commit 62bcabd
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/libs/langchain-openai/src/chat_models.ts b/libs/langchain-openai/src/chat_models.ts
@@ -173,13 +173,15 @@ function openAIResponseToChatMessage(
       if (includeRawResponse !== undefined) {
         additional_kwargs.__raw_response = rawResponse;
       }
-      let response_metadata: Record<string, unknown> | undefined;
-      if (rawResponse.system_fingerprint) {
-        response_metadata = {
-          usage: { ...rawResponse.usage },
-          system_fingerprint: rawResponse.system_fingerprint,
-        };
-      }
+      const response_metadata: Record<string, unknown> | undefined = {
+        model_name: rawResponse.model,
+        ...(rawResponse.system_fingerprint
+          ? {
+              usage: { ...rawResponse.usage },
+              system_fingerprint: rawResponse.system_fingerprint,
+            }
+          : {}),
+      };
 
       if (message.audio) {
         additional_kwargs.audio = message.audio;
@@ -1443,6 +1445,7 @@ export class ChatOpenAI<
         // Only include system fingerprint in the last chunk for now
         // to avoid concatenation issues
         generationInfo.system_fingerprint = data.system_fingerprint;
+        generationInfo.model_name = data.model;
       }
       if (this.logprobs) {
         generationInfo.logprobs = choice.logprobs;