move types

Author: RulaKhaled

packages/core/src/tracing/langchain/types.ts

@@ -47,6 +47,10 @@ export interface LangChainMessage {
   // LangChain serialized format
   lc?: number;
   id?: string[];
+  response_metadata?: {
+    model_name?: string;
+    finish_reason?: string;
+  };
   kwargs?: {
     [key: string]: unknown;
     content?: string;
@@ -70,6 +74,7 @@ export interface LangChainLLMResult {
         finish_reason?: string;
         logprobs?: unknown;
       };
+      // v1+ uses generationInfo instead of generation_info
       generationInfo?: {
         [key: string]: unknown;
         finish_reason?: string;

packages/core/src/tracing/langchain/utils.ts

@@ -419,19 +419,11 @@ export function extractLlmResponseAttributes(
 
   addTokenUsageAttributes(llmResult.llmOutput, attrs);
 
-  const llmOutput = llmResult.llmOutput as { model_name?: string; model?: string; id?: string; stop_reason?: string };
+  const llmOutput = llmResult.llmOutput;
 
   // Extract from v1 generations structure if available
   const firstGeneration = llmResult.generations?.[0]?.[0];
-  const v1Message = firstGeneration?.message as
-    | {
-        id?: string;
-        response_metadata?: {
-          model_name?: string;
-          finish_reason?: string;
-        };
-      }
-    | undefined;
+  const v1Message = firstGeneration?.message;
 
   // Provider model identifier: `model_name` (OpenAI-style) or `model` (others)
   // v1 stores this in message.response_metadata.model_name