Trying out timeouts

pyproject.toml

@@ -11,7 +11,7 @@ dependencies = [
     # Core Dependencies
     "pydantic>=2.4.2",
     "openai>=1.13.3",
-    "litellm>=1.56.4",
+    "litellm>=1.44.22",
     "instructor>=1.3.3",
 
     # Text Processing

tests/agent_test.py

@@ -1457,7 +1457,7 @@ def test_agent_with_ollama_llama3():
     assert agent.llm.model == "ollama/llama3.2:3b"
     assert agent.llm.base_url == "http://localhost:11434"
 
-    task = "Respond in 20 words. Who are you?"
+    task = "Respond in 20 words. Which model are you?"
     response = agent.llm.call([{"role": "user", "content": task}])
 
     assert response
@@ -1473,7 +1473,9 @@ def test_llm_call_with_ollama_llama3():
         temperature=0.7,
         max_tokens=30,
     )
-    messages = [{"role": "user", "content": "Respond in 20 words. Who are you?"}]
+    messages = [
+        {"role": "user", "content": "Respond in 20 words. Which model are you?"}
+    ]
 
     response = llm.call(messages)
 

tests/cli/tools/test_main.py

@@ -28,9 +28,10 @@ def test_create_success(mock_subprocess):
     with in_temp_dir():
         tool_command = ToolCommand()
 
-        with patch.object(tool_command, "login") as mock_login, patch(
-            "sys.stdout", new=StringIO()
-        ) as fake_out:
+        with (
+            patch.object(tool_command, "login") as mock_login,
+            patch("sys.stdout", new=StringIO()) as fake_out,
+        ):
             tool_command.create("test-tool")
             output = fake_out.getvalue()
 
@@ -82,7 +83,7 @@ def test_install_success(mock_get, mock_subprocess_run):
         capture_output=False,
         text=True,
         check=True,
-        env=unittest.mock.ANY
+        env=unittest.mock.ANY,
     )
 
     assert "Successfully installed sample-tool" in output