Formatting

tests/test_api.py

@@ -1,6 +1,7 @@
 """Tests for FastAPI endpoints."""
-import pytest
-from unittest.mock import Mock, patch, MagicMock
+
+from unittest.mock import patch
+
 from fastapi.testclient import TestClient
 
 
@@ -10,6 +11,7 @@ class TestHealthEndpoint:
     def test_health_check(self, memory):
         """Should return healthy status."""
         from app import app
+
         client = TestClient(app)
 
         response = client.get("/health")
@@ -24,6 +26,7 @@ class TestModelsEndpoint:
     def test_list_models(self, memory):
         """Should return model list."""
         from app import app
+
         client = TestClient(app)
 
         response = client.get("/v1/models")
@@ -41,6 +44,7 @@ class TestMemoryEndpoints:
     def test_get_memory_state(self, memory):
         """Should return full memory state."""
         from app import app
+
         client = TestClient(app)
 
         response = client.get("/memory/state")
@@ -54,6 +58,7 @@ class TestMemoryEndpoints:
     def test_get_search_results_empty(self, memory):
         """Should return empty when no search results."""
         from app import app
+
         client = TestClient(app)
 
         response = client.get("/memory/episodic/search-results")
@@ -65,6 +70,7 @@ class TestMemoryEndpoints:
     def test_get_search_results_with_data(self, memory_with_search_results):
         """Should return search results when available."""
         from app import app
+
         client = TestClient(app)
 
         response = client.get("/memory/episodic/search-results")
@@ -78,6 +84,7 @@ class TestMemoryEndpoints:
     def test_clear_session(self, memory_with_search_results):
         """Should clear session memories."""
         from app import app
+
         client = TestClient(app)
 
         response = client.post("/memory/clear-session")
@@ -96,14 +103,18 @@ class TestChatCompletionsEndpoint:
     def test_chat_completion_success(self, memory):
         """Should return chat completion."""
         from app import app
+
         # Patch the agent's step method directly
         with patch("app.agent.step", return_value="Hello! How can I help?"):
             client = TestClient(app)
 
-            response = client.post("/v1/chat/completions", json={
-                "model": "agent-media",
-                "messages": [{"role": "user", "content": "Hello"}],
-            })
+            response = client.post(
+                "/v1/chat/completions",
+                json={
+                    "model": "agent-media",
+                    "messages": [{"role": "user", "content": "Hello"}],
+                },
+            )
 
             assert response.status_code == 200
             data = response.json()
@@ -113,12 +124,16 @@ class TestChatCompletionsEndpoint:
     def test_chat_completion_no_user_message(self, memory):
         """Should return error if no user message."""
         from app import app
+
         client = TestClient(app)
 
-        response = client.post("/v1/chat/completions", json={
-            "model": "agent-media",
-            "messages": [{"role": "system", "content": "You are helpful"}],
-        })
+        response = client.post(
+            "/v1/chat/completions",
+            json={
+                "model": "agent-media",
+                "messages": [{"role": "system", "content": "You are helpful"}],
+            },
+        )
 
         assert response.status_code == 422
         detail = response.json()["detail"]
@@ -132,18 +147,23 @@ class TestChatCompletionsEndpoint:
     def test_chat_completion_empty_messages(self, memory):
         """Should return error for empty messages."""
         from app import app
+
         client = TestClient(app)
 
-        response = client.post("/v1/chat/completions", json={
-            "model": "agent-media",
-            "messages": [],
-        })
+        response = client.post(
+            "/v1/chat/completions",
+            json={
+                "model": "agent-media",
+                "messages": [],
+            },
+        )
 
         assert response.status_code == 422
 
     def test_chat_completion_invalid_json(self, memory):
         """Should return error for invalid JSON."""
         from app import app
+
         client = TestClient(app)
 
         response = client.post(
@@ -157,14 +177,18 @@ class TestChatCompletionsEndpoint:
     def test_chat_completion_streaming(self, memory):
         """Should support streaming mode."""
         from app import app
+
         with patch("app.agent.step", return_value="Streaming response"):
             client = TestClient(app)
 
-            response = client.post("/v1/chat/completions", json={
-                "model": "agent-media",
-                "messages": [{"role": "user", "content": "Hello"}],
-                "stream": True,
-            })
+            response = client.post(
+                "/v1/chat/completions",
+                json={
+                    "model": "agent-media",
+                    "messages": [{"role": "user", "content": "Hello"}],
+                    "stream": True,
+                },
+            )
 
             assert response.status_code == 200
             assert "text/event-stream" in response.headers["content-type"]
@@ -172,17 +196,21 @@ class TestChatCompletionsEndpoint:
     def test_chat_completion_extracts_last_user_message(self, memory):
         """Should use last user message."""
         from app import app
+
         with patch("app.agent.step", return_value="Response") as mock_step:
             client = TestClient(app)
 
-            response = client.post("/v1/chat/completions", json={
-                "model": "agent-media",
-                "messages": [
-                    {"role": "user", "content": "First message"},
-                    {"role": "assistant", "content": "Response"},
-                    {"role": "user", "content": "Second message"},
-                ],
-            })
+            response = client.post(
+                "/v1/chat/completions",
+                json={
+                    "model": "agent-media",
+                    "messages": [
+                        {"role": "user", "content": "First message"},
+                        {"role": "assistant", "content": "Response"},
+                        {"role": "user", "content": "Second message"},
+                    ],
+                },
+            )
 
             assert response.status_code == 200
             # Verify the agent received the last user message
@@ -191,13 +219,17 @@ class TestChatCompletionsEndpoint:
     def test_chat_completion_response_format(self, memory):
         """Should return OpenAI-compatible format."""
         from app import app
+
         with patch("app.agent.step", return_value="Test response"):
             client = TestClient(app)
 
-            response = client.post("/v1/chat/completions", json={
-                "model": "agent-media",
-                "messages": [{"role": "user", "content": "Test"}],
-            })
+            response = client.post(
+                "/v1/chat/completions",
+                json={
+                    "model": "agent-media",
+                    "messages": [{"role": "user", "content": "Test"}],
+                },
+            )
 
             data = response.json()
             assert "id" in data