Update agent.py

agent.py

@@ -20,6 +20,8 @@ import io
 import os
 import tempfile
 from pathlib import Path
+from litellm import LiteLLMModel
+import os
 
 
 # Load environment variablesTuple