Run Black & Flake8

Author: KennyVaneetvelde

atomic-examples/fastapi-memory/fastapi_memory/client.py

@@ -95,10 +95,7 @@ def _delete_session(user_id: str, session_id: str) -> bool:
         True if successful, False otherwise
     """
     try:
-        response = httpx.delete(
-            f"{BASE_URL}/users/{user_id}/sessions/{session_id}",
-            timeout=REQUEST_TIMEOUT
-        )
+        response = httpx.delete(f"{BASE_URL}/users/{user_id}/sessions/{session_id}", timeout=REQUEST_TIMEOUT)
         response.raise_for_status()
         return True
     except httpx.HTTPStatusError as e:
@@ -212,10 +209,7 @@ def _fetch_conversation_history(user_id: str, session_id: str) -> Optional[List[
         List of message dicts with 'role', 'content', 'timestamp', or None if request failed
     """
     try:
-        response = httpx.get(
-            f"{BASE_URL}/users/{user_id}/sessions/{session_id}/history",
-            timeout=REQUEST_TIMEOUT
-        )
+        response = httpx.get(f"{BASE_URL}/users/{user_id}/sessions/{session_id}/history", timeout=REQUEST_TIMEOUT)
         response.raise_for_status()
         data = response.json()
         return data.get("messages", [])
@@ -244,7 +238,7 @@ def _display_conversation_history(messages: List[dict]) -> None:
         elif role == "assistant":
             console.print(Text("Agent:", style="bold green"), end=" ")
             console.print(Text(content, style="green"))
-            
+
             if msg.get("suggested_questions"):
                 _display_suggested_questions(msg["suggested_questions"])
         console.print()
@@ -289,7 +283,7 @@ def chat_non_streaming(user_id: str, session_id: str) -> None:
         initial_questions = [
             "What can you help me with?",
             "Tell me about your capabilities",
-            "How does this chat system work?"
+            "How does this chat system work?",
         ]
         _display_suggested_questions(initial_questions)
         console.print()
@@ -347,7 +341,7 @@ async def chat_streaming_async(user_id: str, session_id: str) -> None:
         initial_questions = [
             "What can you help me with?",
             "Tell me about your capabilities",
-            "How does this chat system work?"
+            "How does this chat system work?",
         ]
         _display_suggested_questions(initial_questions)
         console.print()
@@ -380,25 +374,19 @@ async def chat_streaming_async(user_id: str, session_id: str) -> None:
                                     data = json.loads(data_str)
 
                                     if "error" in data:
-                                        console.print(
-                                            f"\n[bold red]Error:[/bold red] {data['error']}\n"
-                                        )
+                                        console.print(f"\n[bold red]Error:[/bold red] {data['error']}\n")
                                         break
 
                                     if data.get("response"):
                                         current_response = data["response"]
                                     if data.get("suggested_questions"):
                                         current_questions = data["suggested_questions"]
 
-                                    display_text = Text.assemble(
-                                        ("Agent: ", "bold green"), (current_response, "green")
-                                    )
+                                    display_text = Text.assemble(("Agent: ", "bold green"), (current_response, "green"))
 
                                     if current_questions:
                                         display_text.append("\n\n")
-                                        display_text.append(
-                                            "Suggested questions:\n", style="bold cyan"
-                                        )
+                                        display_text.append("Suggested questions:\n", style="bold cyan")
                                         for i, question in enumerate(current_questions, 1):
                                             display_text.append(f"{i}. {question}\n", style="cyan")
 
@@ -461,11 +449,11 @@ def manage_sessions(user_id: str) -> None:
                 confirm = Prompt.ask(
                     f"\n[bold yellow]Delete session {session_to_delete[:8]}...?[/bold yellow]",
                     choices=["yes", "no"],
-                    default="no"
+                    default="no",
                 )
                 if confirm == "yes":
                     if _delete_session(user_id, session_to_delete):
-                        console.print(f"\n[bold green]✓ Session deleted[/bold green]")
+                        console.print("\n[bold green]✓ Session deleted[/bold green]")
             else:
                 console.print("[bold red]Invalid selection[/bold red]")
         except ValueError:

atomic-examples/fastapi-memory/fastapi_memory/lib/config.py

@@ -36,10 +36,7 @@ def get_api_key() -> str:
             except Exception:
                 continue
 
-    raise ValueError(
-        "OPENAI_API_KEY environment variable is required. "
-        "Please set it or create a .env file."
-    )
+    raise ValueError("OPENAI_API_KEY environment variable is required. " "Please set it or create a .env file.")
 
 
 # Constants

atomic-examples/fastapi-memory/fastapi_memory/lib/schemas.py

@@ -11,9 +11,7 @@ class ChatRequest(BaseIOSchema):
 
     message: str = Field(..., description="User message")
     user_id: str = Field(..., description="User identifier")
-    session_id: Optional[str] = Field(
-        None, description="Session identifier for conversation continuity"
-    )
+    session_id: Optional[str] = Field(None, description="Session identifier for conversation continuity")
 
 
 class ChatResponse(BaseIOSchema):

atomic-examples/fastapi-memory/fastapi_memory/main.py

@@ -16,7 +16,6 @@
     ChatResponse,
     ConversationHistory,
     ConversationMessage,
-    SessionCreateRequest,
     SessionCreateResponse,
     SessionDeleteResponse,
     SessionInfo,
@@ -174,8 +173,7 @@ async def chat(request: ChatRequest) -> ChatResponse:
     try:
         if not request.session_id:
             raise HTTPException(
-                status_code=400,
-                detail="session_id is required. Create a session first using POST /users/{user_id}/sessions"
+                status_code=400, detail="session_id is required. Create a session first using POST /users/{user_id}/sessions"
             )
 
         # Store user message in history
@@ -225,8 +223,7 @@ async def chat_stream(request: ChatRequest) -> StreamingResponse:
     try:
         if not request.session_id:
             raise HTTPException(
-                status_code=400,
-                detail="session_id is required. Create a session first using POST /users/{user_id}/sessions"
+                status_code=400, detail="session_id is required. Create a session first using POST /users/{user_id}/sessions"
             )
 
         # Store user message in history
@@ -239,13 +236,11 @@ async def generate():
             full_response = ""
             final_suggested_questions = []
             try:
-                async for chunk in agent.run_async_stream(
-                    ChatRequest(message=request.message, user_id=request.user_id)
-                ):
+                async for chunk in agent.run_async_stream(ChatRequest(message=request.message, user_id=request.user_id)):
                     chunk_dict = chunk.model_dump() if hasattr(chunk, "model_dump") else {}
                     response_text = chunk_dict.get("response", "")
                     full_response = response_text  # Keep updating with latest full text
-                    
+
                     if chunk_dict.get("suggested_questions"):
                         final_suggested_questions = chunk_dict.get("suggested_questions")
 
@@ -299,10 +294,7 @@ async def create_session(user_id: str) -> SessionCreateResponse:
         session_id = _generate_session_id()
         session_metadata[user_id][session_id] = datetime.now().isoformat()
 
-        return SessionCreateResponse(
-            session_id=session_id,
-            message=f"Session '{session_id}' created successfully"
-        )
+        return SessionCreateResponse(session_id=session_id, message=f"Session '{session_id}' created successfully")
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Failed to create session: {str(e)}")
@@ -327,17 +319,10 @@ async def get_user_sessions(user_id: str) -> UserSessionsResponse:
 
     # Build session info list
     session_list = [
-        SessionInfo(
-            session_id=sid,
-            created_at=session_metadata.get(user_id, {}).get(sid)
-        )
-        for sid in sorted(all_session_ids)
+        SessionInfo(session_id=sid, created_at=session_metadata.get(user_id, {}).get(sid)) for sid in sorted(all_session_ids)
     ]
 
-    return UserSessionsResponse(
-        user_id=user_id,
-        sessions=session_list
-    )
+    return UserSessionsResponse(user_id=user_id, sessions=session_list)
 
 
 @app.get("/users/{user_id}/sessions/{session_id}/history", response_model=ConversationHistory, tags=["Sessions"])
@@ -358,10 +343,7 @@ async def get_conversation_history(user_id: str, session_id: str) -> Conversatio
 
     messages = conversation_history.get(user_id, {}).get(session_id, [])
 
-    return ConversationHistory(
-        session_id=session_id,
-        messages=[ConversationMessage(**msg) for msg in messages]
-    )
+    return ConversationHistory(session_id=session_id, messages=[ConversationMessage(**msg) for msg in messages])
 
 
 @app.delete("/users/{user_id}/sessions/{session_id}", response_model=SessionDeleteResponse, tags=["Sessions"])
@@ -395,10 +377,7 @@ async def delete_session(user_id: str, session_id: str) -> SessionDeleteResponse
         del conversation_history[user_id][session_id]
 
     if not found:
-        raise HTTPException(
-            status_code=404,
-            detail=f"Session '{session_id}' not found for user '{user_id}'"
-        )
+        raise HTTPException(status_code=404, detail=f"Session '{session_id}' not found for user '{user_id}'")
 
     return SessionDeleteResponse(message=f"Session '{session_id}' deleted successfully")
 

atomic-examples/fastapi-memory/test_api.py

@@ -4,6 +4,7 @@
 
 BASE_URL = "http://localhost:8000"
 
+
 def test_api():
     """Test the basic API flow."""
     print("Testing FastAPI Memory API...\n")
@@ -29,26 +30,14 @@ def test_api():
     # 3. Send a chat message
     print("3. Sending first chat message...")
     response = httpx.post(
-        f"{BASE_URL}/chat",
-        json={
-            "message": "Hello, how are you?",
-            "user_id": user_id,
-            "session_id": session_id
-        }
+        f"{BASE_URL}/chat", json={"message": "Hello, how are you?", "user_id": user_id, "session_id": session_id}
     )
     print(f"   Status: {response.status_code}")
     print(f"   Response: {response.json()}\n")
 
     # 3b. Send another message to build conversation
     print("3b. Sending second chat message...")
-    response = httpx.post(
-        f"{BASE_URL}/chat",
-        json={
-            "message": "Tell me a joke",
-            "user_id": user_id,
-            "session_id": session_id
-        }
-    )
+    response = httpx.post(f"{BASE_URL}/chat", json={"message": "Tell me a joke", "user_id": user_id, "session_id": session_id})
     print(f"   Status: {response.status_code}")
     print(f"   Response: {response.json()}\n")
 
@@ -100,6 +89,7 @@ def test_api():
 
     print("✅ All tests completed!")
 
+
 if __name__ == "__main__":
     try:
         test_api()