diff --git a/core/pydantic_ai_agent.py b/core/pydantic_ai_agent.py
index dc036f9..4fa2016 100755
--- a/core/pydantic_ai_agent.py
+++ b/core/pydantic_ai_agent.py
@@ -10,7 +10,7 @@ import glob
 import asyncio
 import random
 import hashlib
-from typing import Optional, Dict, List, Any
+from typing import Optional, Dict, List, Any, Tuple
 from datetime import datetime
 from pydantic import BaseModel, Field
 from pydantic_ai import Agent, RunContext