@posthog/ai 5.2.0 → 5.2.2

package/src/langchain/callbacks.ts

@@ -3,10 +3,11 @@ import { withPrivacyMode, getModelParams } from '../utils'
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base'
 import type { Serialized } from '@langchain/core/load/serializable'
 import type { ChainValues } from '@langchain/core/utils/types'
-import type { BaseMessage } from '@langchain/core/messages'
 import type { LLMResult } from '@langchain/core/outputs'
 import type { AgentAction, AgentFinish } from '@langchain/core/agents'
 import type { DocumentInterface } from '@langchain/core/documents'
+import { ToolCall } from '@langchain/core/messages/tool'
+import { BaseMessage } from '@langchain/core/messages'
 
 interface SpanMetadata {
   /** Name of the trace/span (e.g. chain name) */
@@ -498,11 +499,21 @@ export class LangChainCallbackHandler extends BaseCallbackHandler {
     return undefined
   }
 
+  private _convertLcToolCallsToOai(toolCalls: ToolCall[]): Record<string, any>[] {
+    return toolCalls.map((toolCall: ToolCall) => ({
+      type: 'function',
+      id: toolCall.id,
+      function: {
+        name: toolCall.name,
+        arguments: JSON.stringify(toolCall.args),
+      },
+    }))
+  }
+
   private _convertMessageToDict(message: any): Record<string, any> {
     let messageDict: Record<string, any> = {}
 
-    // Check the _getType() method or type property instead of instanceof
-    const messageType = message._getType?.() || message.type
+    const messageType: string = message.getType()
 
     switch (messageType) {
       case 'human':
@@ -510,6 +521,11 @@ export class LangChainCallbackHandler extends BaseCallbackHandler {
         break
       case 'ai':
         messageDict = { role: 'assistant', content: message.content }
+
+        if (message.tool_calls) {
+          messageDict.tool_calls = this._convertLcToolCallsToOai(message.tool_calls)
+        }
+
         break
       case 'system':
         messageDict = { role: 'system', content: message.content }
@@ -521,12 +537,14 @@ export class LangChainCallbackHandler extends BaseCallbackHandler {
         messageDict = { role: 'function', content: message.content }
         break
       default:
-        messageDict = { role: messageType || 'unknown', content: String(message.content) }
+        messageDict = { role: messageType, content: String(message.content) }
+        break
     }
 
     if (message.additional_kwargs) {
       messageDict = { ...messageDict, ...message.additional_kwargs }
     }
+
     return messageDict
   }
 

package/src/openai/index.ts

@@ -1,4 +1,4 @@
-import OpenAIOrignal, { ClientOptions } from 'openai'
+import { OpenAI as OpenAIOrignal, ClientOptions } from 'openai'
 import { PostHog } from 'posthog-node'
 import { v4 as uuidv4 } from 'uuid'
 import { formatResponseOpenAI, MonitoringParams, sendEventToPosthog } from '../utils'
@@ -6,6 +6,10 @@ import type { APIPromise } from 'openai'
 import type { Stream } from 'openai/streaming'
 import type { ParsedResponse } from 'openai/resources/responses/responses'
 
+const Chat = OpenAIOrignal.Chat
+const Completions = Chat.Completions
+const Responses = OpenAIOrignal.Responses
+
 type ChatCompletion = OpenAIOrignal.ChatCompletion
 type ChatCompletionChunk = OpenAIOrignal.ChatCompletionChunk
 type ChatCompletionCreateParamsBase = OpenAIOrignal.Chat.Completions.ChatCompletionCreateParams
@@ -37,7 +41,7 @@ export class PostHogOpenAI extends OpenAIOrignal {
   }
 }
 
-export class WrappedChat extends OpenAIOrignal.Chat {
+export class WrappedChat extends Chat {
   constructor(parentClient: PostHogOpenAI, phClient: PostHog) {
     super(parentClient)
     this.completions = new WrappedCompletions(parentClient, phClient)
@@ -46,7 +50,7 @@ export class WrappedChat extends OpenAIOrignal.Chat {
   public completions: WrappedCompletions
 }
 
-export class WrappedCompletions extends OpenAIOrignal.Chat.Completions {
+export class WrappedCompletions extends Completions {
   private readonly phClient: PostHog
 
   constructor(client: OpenAIOrignal, phClient: PostHog) {
@@ -223,7 +227,7 @@ export class WrappedCompletions extends OpenAIOrignal.Chat.Completions {
   }
 }
 
-export class WrappedResponses extends OpenAIOrignal.Responses {
+export class WrappedResponses extends Responses {
   private readonly phClient: PostHog
 
   constructor(client: OpenAIOrignal, phClient: PostHog) {

package/tests/callbacks.test.ts

@@ -0,0 +1,48 @@
+import { LangChainCallbackHandler } from '../src/langchain/callbacks'
+import { PostHog } from 'posthog-node'
+import { AIMessage } from '@langchain/core/messages'
+
+const mockPostHogClient = {
+  capture: jest.fn(),
+} as unknown as PostHog
+
+describe('LangChainCallbackHandler', () => {
+  let handler: LangChainCallbackHandler
+
+  beforeEach(() => {
+    handler = new LangChainCallbackHandler({
+      client: mockPostHogClient,
+    })
+    jest.clearAllMocks()
+  })
+
+  it('should convert AIMessage with tool calls to dict format', () => {
+    const toolCalls = [
+      {
+        id: 'call_123',
+        name: 'get_weather',
+        args: { city: 'San Francisco', units: 'celsius' },
+      },
+    ]
+
+    const aiMessage = new AIMessage({
+      content: "I'll check the weather for you.",
+      tool_calls: toolCalls,
+    })
+
+    const result = (handler as any)._convertMessageToDict(aiMessage)
+
+    expect(result.role).toBe('assistant')
+    expect(result.content).toBe("I'll check the weather for you.")
+    expect(result.tool_calls).toEqual([
+      {
+        type: 'function',
+        id: 'call_123',
+        function: {
+          name: 'get_weather',
+          arguments: '{"city":"San Francisco","units":"celsius"}',
+        },
+      },
+    ])
+  })
+})