Skip to content

Commit 9c79622

Browse files
jsonbaileyclaude
andcommitted
feat: Update examples to use create_tracker() factory pattern
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 7e56e1f commit 9c79622

6 files changed

Lines changed: 10 additions & 6 deletions

File tree

examples/bedrock/bedrock_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,12 +58,13 @@ def main():
5858
context,
5959
variables={'myUserVariable': "Testing Variable"}
6060
)
61-
tracker = config_value.tracker
6261

6362
if not config_value.enabled:
6463
print("AI Config is disabled")
6564
return
6665

66+
tracker = config_value.create_tracker()
67+
6768
# Map the messages to the format expected by Bedrock
6869
chat_messages = [{'role': msg.role, 'content': [{'text': msg.content}]} for msg in config_value.messages if msg.role != 'system']
6970
system_messages = [{'text': msg.content} for msg in config_value.messages if msg.role == 'system']

examples/gemini/gemini_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,12 +128,13 @@ def main():
128128
context,
129129
variables={'myUserVariable': "Testing Variable"}
130130
)
131-
tracker = config_value.tracker
132131

133132
if not config_value.enabled:
134133
print("AI Config is disabled")
135134
return
136135

136+
tracker = config_value.create_tracker()
137+
137138
# Configure Google Generative AI
138139
client = genai.Client(
139140
api_key=google_api_key,

examples/langchain/langchain_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,13 @@ async def async_main():
6666
context,
6767
variables={'myUserVariable': "Testing Variable"}
6868
)
69-
tracker = config_value.tracker
7069

7170
if not config_value.enabled:
7271
print("AI Config is disabled")
7372
return
7473

74+
tracker = config_value.create_tracker()
75+
7576
try:
7677
# Create LangChain model instance using init_chat_model
7778
# Map the provider from config_value to LangChain format

examples/langgraph_agent/langgraph_agent_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def main():
115115

116116
try:
117117
# Track and execute the agent
118-
response = track_langgraph_metrics(agent_config.tracker, lambda: agent.invoke({
118+
response = track_langgraph_metrics(agent_config.create_tracker(), lambda: agent.invoke({
119119
"messages": [{"role": "user", "content": "What is the weather in Tokyo?"}]
120120
}))
121121

examples/langgraph_multi_agent/langgraph_multi_agent_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def create_agent_with_config(aiclient, config_key, context):
9595
# Create a React agent with the LLM
9696
agent = create_react_agent(llm, [], prompt=agent_config.instructions)
9797

98-
return agent, agent_config.tracker, False
98+
return agent, agent_config.create_tracker(), False
9999

100100
def ai_node(
101101
state: CodeReviewState,

examples/openai/openai_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,11 +60,12 @@ def main():
6060
context,
6161
variables={'myUserVariable': "Testing Variable"}
6262
)
63-
tracker = config_value.tracker
6463

6564
if not config_value.enabled:
6665
print("AI Config is disabled")
6766
return
67+
68+
tracker = config_value.create_tracker()
6869

6970
messages = [message.to_dict() for message in (config_value.messages or [])]
7071

0 commit comments

Comments
 (0)