@friskai/frisk-js 0.3.1 → 0.3.2

package/dist/adapters/langchain/frisk-callback-handler.d.ts

@@ -72,12 +72,12 @@ export declare class FriskCallbackHandler extends BaseCallbackHandler {
      * Called at the start of a Chat Model run. Opens an LLM call span and records
      * the prompt messages it is being asked to respond to.
      */
-    handleChatModelStart(_llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string): void;
+    handleChatModelStart(_llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): void;
     /**
      * Called at the start of a (completion) LLM run. Opens an LLM call span and
      * records the raw prompt strings.
      */
-    handleLLMStart(_llm: Serialized, prompts: string[], runId: string, parentRunId?: string): void;
+    handleLLMStart(_llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): void;
     /**
      * Called at the end of an LLM/Chat Model run. Records responses and requested
      * tool calls, then closes the LLM call span.

package/dist/adapters/langchain/frisk-callback-handler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"frisk-callback-handler.d.ts","sourceRoot":"","sources":["../../../src/adapters/langchain/frisk-callback-handler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAA;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAA;AACnE,OAAO,KAAK,EAEV,WAAW,EAEZ,MAAM,0BAA0B,CAAA;AACjC,OAAO,KAAK,EAAkB,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACxE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAA;AAC9D,OAAO,EAAE,YAAY,EAAkB,MAAM,0BAA0B,CAAA;AAOvE,OAAO,EAEL,sBAAsB,EAEvB,MAAM,eAAe,CAAA;AACtB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AAElD,MAAM,MAAM,uBAAuB,GAAG,OAAO,GAAG,SAAS,CAAA;AAezD;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,qBAAa,oBAAqB,SAAQ,mBAAmB;IAC3D,IAAI,SAAyB;IAE7B,OAAO,CAAC,OAAO,CAAc;IAI7B,OAAO,CAAC,YAAY,CAAwB;IAI5C,OAAO,CAAC,yBAAyB,CAAqB;IACtD,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,MAAM,CAAW;gBAEb,EACV,OAAO,EACP,KAAK,EACL,OAAO,GACR,EAAE;QACD,OAAO,EAAE,YAAY,CAAA;QACrB,KAAK,EAAE,cAAc,CAAA;QACrB,OAAO,CAAC,EAAE,sBAAsB,CAAA;KACjC;IAYD;;;;;;;;;OASG;IACH,gBAAgB,CACd,MAAM,EAAE,UAAU,EAClB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,IAAI;IAeP;;;;;OAKG;IACH,cAAc,CACZ,QAAQ,EAAE,WAAW,EACrB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,EAAE,EAChB,OAAO,CAAC,EAAE;QACR,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KACjC,GACA,IAAI;IAoBP;;;OAGG;IACH,oBAAoB,CAClB,IAAI,EAAE,UAAU,EAChB,QAAQ,EAAE,WAAW,EAAE,EAAE,EACzB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,IAAI;IAWP;;;OAGG;IACH,cAAc,CACZ,IAAI,EAAE,UAAU,EAChB,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,IAAI;IAQP;;;OAGG;IACH,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAIpD;;;OAGG;IACH,cAAc,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAIzC,eAAe,CACnB,IAAI,EAAE,UAAU,EAChB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAkCV,aAAa,CACjB,MAAM,EAAE,GAAG,EACX,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,IAAI,CAAC;IAwBV,eAAe,CACnB,GAAG,EAAE,KAAK,EACV,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,IAAI,CAAC;IAqBhB;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAOzB;;;OAGG;IACH,OAAO,CAAC,eAAe;IAYvB;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;IAczB;;;;;;;;;OASG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;;;OAIG;IACH,OAAO,CAAC,eAAe;IAmHvB;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAc1B,6DAA6D;IAC7D,OAAO,CAAC,OAAO;IAQf,OAAO,CAAC,iBAAiB;CAwB1B"}
+{"version":3,"file":"frisk-callback-handler.d.ts","sourceRoot":"","sources":["../../../src/adapters/langchain/frisk-callback-handler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAA;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAA;AACnE,OAAO,KAAK,EAEV,WAAW,EAEZ,MAAM,0BAA0B,CAAA;AACjC,OAAO,KAAK,EAAkB,SAAS,EAAE,MAAM,yBAAyB,CAAA;AACxE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,6BAA6B,CAAA;AAC9D,OAAO,EAAE,YAAY,EAAkB,MAAM,0BAA0B,CAAA;AAOvE,OAAO,EAEL,sBAAsB,EAEvB,MAAM,eAAe,CAAA;AACtB,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAA;AAElD,MAAM,MAAM,uBAAuB,GAAG,OAAO,GAAG,SAAS,CAAA;AAezD;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,qBAAa,oBAAqB,SAAQ,mBAAmB;IAC3D,IAAI,SAAyB;IAE7B,OAAO,CAAC,OAAO,CAAc;IAI7B,OAAO,CAAC,YAAY,CAAwB;IAI5C,OAAO,CAAC,yBAAyB,CAAqB;IACtD,OAAO,CAAC,KAAK,CAAgB;IAC7B,OAAO,CAAC,MAAM,CAAW;gBAEb,EACV,OAAO,EACP,KAAK,EACL,OAAO,GACR,EAAE;QACD,OAAO,EAAE,YAAY,CAAA;QACrB,KAAK,EAAE,cAAc,CAAA;QACrB,OAAO,CAAC,EAAE,sBAAsB,CAAA;KACjC;IAYD;;;;;;;;;OASG;IACH,gBAAgB,CACd,MAAM,EAAE,UAAU,EAClB,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,GACnB,IAAI;IAeP;;;;;OAKG;IACH,cAAc,CACZ,QAAQ,EAAE,WAAW,EACrB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,EAAE,EAChB,OAAO,CAAC,EAAE;QACR,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;KACjC,GACA,IAAI;IAoBP;;;OAGG;IACH,oBAAoB,CAClB,IAAI,EAAE,UAAU,EAChB,QAAQ,EAAE,WAAW,EAAE,EAAE,EACzB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,IAAI;IAqBP;;;OAGG;IACH,cAAc,CACZ,IAAI,EAAE,UAAU,EAChB,OAAO,EAAE,MAAM,EAAE,EACjB,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACrC,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACjC,IAAI;IAmBP;;;OAGG;IACH,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAIpD;;;OAGG;IACH,cAAc,CAAC,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAIzC,eAAe,CACnB,IAAI,EAAE,UAAU,EAChB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,EACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,OAAO,CAAC,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAkCV,aAAa,CACjB,MAAM,EAAE,GAAG,EACX,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,IAAI,CAAC;IAwBV,eAAe,CACnB,GAAG,EAAE,KAAK,EACV,KAAK,EAAE,MAAM,EACb,WAAW,CAAC,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,MAAM,EAAE,GACd,OAAO,CAAC,IAAI,CAAC;IAqBhB;;;;;;OAMG;IACH,OAAO,CAAC,iBAAiB;IAOzB;;;OAGG;IACH,OAAO,CAAC,eAAe;IAYvB;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;IAczB;;;;;;;;;OASG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;;;OAIG;IACH,OAAO,CAAC,eAAe;IA0FvB;;;OAGG;IACH,OAAO,CAAC,kBAAkB;IAc1B,6DAA6D;IAC7D,OAAO,CAAC,OAAO;IAQf,OAAO,CAAC,iBAAiB;CAwB1B"}

package/dist/adapters/langchain/index.js

@@ -48741,19 +48741,35 @@ class FriskCallbackHandler extends BaseCallbackHandler {
     }
     this.endSpan(runId);
   }
-  handleChatModelStart(_llm, messages, runId, parentRunId) {
+  handleChatModelStart(_llm, messages, runId, parentRunId, extraParams, tags, metadata) {
     const span = this.openLlmCallSpan(runId, parentRunId);
     const promptsJson = this.safeJsonStringify(this.getPromptMessages(messages), "chat prompt messages");
     if (promptsJson !== null) {
       span.setAttributes({ promptsJson });
     }
+    const modelName = metadata?.ls_model_name;
+    const modelProvider = metadata?.ls_provider;
+    if (modelName) {
+      span.setAttributes({ modelName });
+    }
+    if (modelProvider) {
+      span.setAttributes({ modelProvider });
+    }
   }
-  handleLLMStart(_llm, prompts, runId, parentRunId) {
+  handleLLMStart(_llm, prompts, runId, parentRunId, extraParams, tags, metadata) {
     const span = this.openLlmCallSpan(runId, parentRunId);
     const promptsJson = this.safeJsonStringify(prompts, "LLM prompts");
     if (promptsJson !== null) {
       span.setAttributes({ promptsJson });
     }
+    const modelName = metadata?.ls_model_name;
+    const modelProvider = metadata?.ls_provider;
+    if (modelName) {
+      span.setAttributes({ modelName });
+    }
+    if (modelProvider) {
+      span.setAttributes({ modelProvider });
+    }
   }
   handleLLMEnd(output, runId) {
     this.finalizeLlmCall(output, runId, null);
@@ -48872,8 +48888,6 @@ class FriskCallbackHandler extends BaseCallbackHandler {
     const generations = output?.generations ?? [];
     const responses = [];
     const toolCalls = [];
-    let modelName;
-    let modelProvider;
     let inputTokens = 0;
     let outputTokens = 0;
     let totalTokens = 0;
@@ -48881,17 +48895,13 @@ class FriskCallbackHandler extends BaseCallbackHandler {
     for (const generationList of generations) {
       for (const generation of generationList) {
         const message = generation.message;
-        const generationInfo = generation.generationInfo ?? {};
-        const responseMetadata = message?.response_metadata ?? {};
-        modelName = modelName || generationInfo.model || generationInfo.model_name || responseMetadata.model;
-        modelProvider = modelProvider || generationInfo.model_provider || responseMetadata.model_provider;
         if (!message || message.getType() !== "ai") {
           if (!message && generation.text) {
             responses.push({ content: generation.text });
           }
           continue;
         }
-        responses.push({ content: message.content });
+        responses.push({ content: message.text });
         const usage = message.usage_metadata;
         if (usage) {
           inputTokens += usage.input_tokens ?? 0;
@@ -48920,8 +48930,6 @@ class FriskCallbackHandler extends BaseCallbackHandler {
       const responsesJson = responses.length > 0 ? this.safeJsonStringify(responses, "LLM responses") : null;
       const toolCallsJson = toolCalls.length > 0 ? this.safeJsonStringify(toolCalls, "LLM tool calls") : null;
       llmCallSpan.setAttributes({
-        modelName,
-        modelProvider,
         ...hasUsage ? { inputTokens, outputTokens, totalTokens } : {},
         ...responsesJson !== null ? { responsesJson } : {},
         ...toolCallsJson !== null ? { toolCallsJson } : {}
@@ -49425,7 +49433,7 @@ import { fileURLToPath } from "node:url";
 
 // src/generated/sdk-meta.ts
 var SDK_NAME = "@friskai/frisk-js";
-var SDK_VERSION = "0.3.1";
+var SDK_VERSION = "0.3.2";
 
 // src/core/sdk-attributes/read-sdk-meta.ts
 function getSdkMeta() {
@@ -50649,5 +50657,5 @@ export {
   FRISK_SESSION_ID_KEY
 };
 
-//# debugId=B3DA4933179DBD8B64756E2164756E21
+//# debugId=7D7BF0CEC9ECF9F864756E2164756E21
 //# sourceMappingURL=index.js.map