Skip to content

Commit 137cf62

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 aeda568 commit 137cf62

6 files changed

Lines changed: 10 additions & 6 deletions

File tree

examples/bedrock_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,12 +55,13 @@ def main():
5555
context,
5656
variables={'myUserVariable': "Testing Variable"}
5757
)
58-
tracker = config_value.tracker
5958

6059
if not config_value.enabled:
6160
print("AI Config is disabled")
6261
return
6362

63+
tracker = config_value.create_tracker()
64+
6465
# Map the messages to the format expected by Bedrock
6566
chat_messages = [{'role': msg.role, 'content': [{'text': msg.content}]} for msg in config_value.messages if msg.role != 'system']
6667
system_messages = [{'text': msg.content} for msg in config_value.messages if msg.role == 'system']

examples/gemini_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,12 +125,13 @@ def main():
125125
context,
126126
variables={'myUserVariable': "Testing Variable"}
127127
)
128-
tracker = config_value.tracker
129128

130129
if not config_value.enabled:
131130
print("AI Config is disabled")
132131
return
133132

133+
tracker = config_value.create_tracker()
134+
134135
# Configure Google Generative AI
135136
client = genai.Client(
136137
api_key=google_api_key,

examples/langchain_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,12 +63,13 @@ async def async_main():
6363
context,
6464
variables={'myUserVariable': "Testing Variable"}
6565
)
66-
tracker = config_value.tracker
6766

6867
if not config_value.enabled:
6968
print("AI Config is disabled")
7069
return
7170

71+
tracker = config_value.create_tracker()
72+
7273
try:
7374
# Create LangChain model instance using init_chat_model
7475
# Map the provider from config_value to LangChain format

examples/langgraph_agent_example.py

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

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

examples/langgraph_multi_agent_example.py

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

95-
return agent, agent_config.tracker, False
95+
return agent, agent_config.create_tracker(), False
9696

9797
def ai_node(
9898
state: CodeReviewState,

examples/openai_example.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,12 @@ def main():
5656
context,
5757
variables={'myUserVariable': "Testing Variable"}
5858
)
59-
tracker = config_value.tracker
6059

6160
if not config_value.enabled:
6261
print("AI Config is disabled")
6362
return
63+
64+
tracker = config_value.create_tracker()
6465

6566
messages = [message.to_dict() for message in (config_value.messages or [])]
6667

0 commit comments

Comments
 (0)