File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
examples/pydantic_ai_examples Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 2424
2525import asyncio
2626from dataclasses import dataclass
27- from datetime import UTC , datetime
27+ from datetime import datetime , timezone
2828from enum import Enum
2929from typing import Any
3030
3333
3434from pydantic_ai import Agent , ModelHTTPError , RunContext
3535
36- MODEL = 'openai:gpt-4.1-mini'
36+ # MODEL = 'openai:gpt-4.1-mini'
37+ MODEL = 'cohere:command-r7b-12-2024'
3738
3839
3940# Structured Outputs
@@ -220,7 +221,7 @@ def __init__(self):
220221 async def handle_patient (
221222 self , complaint : str , patient : PatientInfo
222223 ) -> dict [str , str ]:
223- timestamp = datetime .now (UTC ).isoformat ()
224+ timestamp = datetime .now (tz = timezone . utc ).isoformat ()
224225 print (f'\n [{ timestamp } ] Processing complaint: { complaint } ' )
225226
226227 triage_prompt = (
You can’t perform that action at this time.
0 commit comments