Skip to content

Commit 89f37a0

Browse files
chore: update langchain examples to use top-level get_ai_metrics_from_response
Replace LangChainProvider.get_ai_metrics_from_response with the top-level get_ai_metrics_from_response function from ldai_langchain, which is the new API in launchdarkly-server-sdk-ai-langchain 0.4.0. Co-Authored-By: rlamb@launchdarkly.com <4955475+kinyoklion@users.noreply.github.com>
1 parent ce84fe5 commit 89f37a0

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

examples/langchain_example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from ldclient import Context
55
from ldclient.config import Config
66
from ldai.client import LDAIClient
7-
from ldai_langchain import LangChainProvider
7+
from ldai_langchain import get_ai_metrics_from_response
88
from langchain.chat_models import init_chat_model
99

1010
# Set sdk_key to your LaunchDarkly SDK key.
@@ -89,7 +89,7 @@ async def async_main():
8989
# Track the LangChain completion with LaunchDarkly metrics using the LD LangChain provider's extractor
9090
completion = await tracker.track_metrics_of(
9191
lambda: llm.ainvoke(messages),
92-
LangChainProvider.get_ai_metrics_from_response,
92+
get_ai_metrics_from_response,
9393
)
9494
ai_response = completion.content
9595

examples/langgraph_agent_example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from ldclient.config import Config
66
from ldai.client import LDAIClient
77
from ldai.tracker import TokenUsage
8-
from ldai_langchain import LangChainProvider
8+
from ldai_langchain import get_ai_metrics_from_response
99
from langchain.chat_models import init_chat_model
1010
from langgraph.prebuilt import create_react_agent
1111

@@ -36,7 +36,7 @@ def track_langgraph_metrics(tracker, func):
3636
total_tokens = 0
3737
if "messages" in result:
3838
for message in result["messages"]:
39-
metrics = LangChainProvider.get_ai_metrics_from_response(message)
39+
metrics = get_ai_metrics_from_response(message)
4040
if metrics.usage:
4141
total_input_tokens += metrics.usage.input
4242
total_output_tokens += metrics.usage.output

examples/langgraph_multi_agent_example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from ldclient.config import Config
55
from ldai.client import LDAIClient
66
from ldai.tracker import TokenUsage
7-
from ldai_langchain import LangChainProvider
7+
from ldai_langchain import get_ai_metrics_from_response
88
from langchain.chat_models import init_chat_model
99
from langgraph.prebuilt import create_react_agent
1010
from langgraph.graph import StateGraph, END
@@ -47,7 +47,7 @@ def track_langgraph_metrics(tracker, func, prev_message_count=0):
4747
if "messages" in result:
4848
new_messages = result["messages"][prev_message_count:]
4949
for message in new_messages:
50-
metrics = LangChainProvider.get_ai_metrics_from_response(message)
50+
metrics = get_ai_metrics_from_response(message)
5151
if metrics.usage:
5252
total_input_tokens += metrics.usage.input
5353
total_output_tokens += metrics.usage.output

0 commit comments

Comments
 (0)