chore: bump and linters

Author: skoob13

CHANGELOG.md

@@ -1,3 +1,11 @@
+## 3.10.0 - 2025-01-27
+
+1. Add the `$ai_span` event to the LangChain callback handler to capture the input and output of intermediary chains.
+
+## 3.9.3 - 2025-01-23
+
+1. Fix capturing of multiple traces in the LangChain callback handler.
+
 ## 3.9.2 - 2025-01-22
 
 1. Fix importing of LangChain callback handler under certain circumstances.

posthog/ai/langchain/callbacks.py

@@ -6,6 +6,7 @@
 import logging
 import time
 import uuid
+from dataclasses import dataclass
 from typing import (
     Any,
     Dict,
@@ -17,13 +18,12 @@
     cast,
 )
 from uuid import UUID
-from dataclasses import dataclass
 
 from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema.agent import AgentAction, AgentFinish
+from langchain_core.documents import Document
 from langchain_core.messages import AIMessage, BaseMessage, FunctionMessage, HumanMessage, SystemMessage, ToolMessage
 from langchain_core.outputs import ChatGeneration, LLMResult
-from langchain_core.documents import Document
 from pydantic import BaseModel
 
 from posthog import default_client

posthog/test/ai/langchain/test_callbacks.py

@@ -4,20 +4,20 @@
 import os
 import time
 import uuid
-from typing import List, Optional, TypedDict, Union, Literal
+from typing import List, Literal, Optional, TypedDict, Union
 from unittest.mock import patch
 
 import pytest
 from langchain_anthropic.chat_models import ChatAnthropic
 from langchain_community.chat_models.fake import FakeMessagesListChatModel
 from langchain_community.llms.fake import FakeListLLM, FakeStreamingListLLM
-from langchain_core.messages import AIMessage, HumanMessage, ToolCall
+from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.runnables import RunnableLambda
+from langchain_core.tools import tool
 from langchain_openai.chat_models import ChatOpenAI
 from langgraph.graph.state import END, START, StateGraph
 from langgraph.prebuilt import create_react_agent
-from langchain_core.tools import tool
 
 from posthog.ai.langchain import CallbackHandler
 from posthog.ai.langchain.callbacks import GenerationMetadata, SpanMetadata
@@ -1257,7 +1257,7 @@ def get_weather(city: Literal["nyc", "sf"]):
     graph = create_react_agent(model, tools=tools)
     inputs = {"messages": [("user", "what is the weather in sf")]}
     cb = CallbackHandler(mock_client, trace_id="test-trace-id", distinct_id="test-distinct-id")
-    res = graph.invoke(inputs, config={"callbacks": [cb]})
+    graph.invoke(inputs, config={"callbacks": [cb]})
     calls = [call[1] for call in mock_client.capture.call_args_list]
     assert len(calls) == 21
     for call in calls:

posthog/version.py

@@ -1,4 +1,4 @@
-VERSION = "3.9.3"
+VERSION = "3.10.0"
 
 if __name__ == "__main__":
     print(VERSION, end="")  # noqa: T201