Weather Agent Sample: Only include tool_call_id to chatbot.append if …

…it's set (#771) Co-authored-by: Sydney Runkle <[email protected]>
pydantic · Feb 3, 2025 · fa1625d · fa1625d
1 parent 42c70c0
commit fa1625d
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/examples/pydantic_ai_examples/weather_agent_gradio.py b/examples/pydantic_ai_examples/weather_agent_gradio.py
@@ -38,13 +38,16 @@ async def stream_from_agent(prompt: str, chatbot: list[dict], past_messages: lis
                         if hasattr(call.args, 'args_json')
                         else json.dumps(call.args.args_dict)
                     )
+                    metadata = {
+                        'title': f'🛠️ Using {TOOL_TO_DISPLAY_NAME[call.tool_name]}',
+                    }
+                    if call.tool_call_id is not None:
+                        metadata['id'] = {call.tool_call_id}
+
                     gr_message = {
                         'role': 'assistant',
                         'content': 'Parameters: ' + call_args,
-                        'metadata': {
-                            'title': f'🛠️ Using {TOOL_TO_DISPLAY_NAME[call.tool_name]}',
-                            'id': call.tool_call_id,
-                        },
+                        'metadata': metadata,
                     }
                     chatbot.append(gr_message)
                 if isinstance(call, ToolReturnPart):