Code cleaning

src/agent/plan_and_execute.py

@@ -20,14 +20,14 @@ import re
 from collections.abc import Generator
 from typing import TypedDict
 
-from langchain_core.messages import AIMessage, HumanMessage, SystemMessage, ToolMessage
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 from langchain_core.runnables import Runnable
 from langgraph.graph import END, StateGraph
 from langgraph.prebuilt import create_react_agent
 
 from src.agent.memory import ConversationMemory
 from src.agent.tools import ToolResultStore, detect_document_languages, make_retrieval_tools
-from src.models import GenerationResponse, IntentType, PipelineDetails, QueryResult
+from src.models import GenerationResponse, IntentType, PipelineDetails
 from src.retrieval.hybrid import HybridRetriever
 from src.retrieval.reranker import Reranker
 from src.retrieval.vector_store import VectorStore
@@ -570,7 +570,7 @@ def _parse_plan(raw: str) -> list[PlanStep]:
     if cleaned.startswith("```"):
         lines = cleaned.splitlines()
         # Remove opening and closing fences
-        lines = [l for l in lines if not l.strip().startswith("```")]
+        lines = [line for line in lines if not line.strip().startswith("```")]
         cleaned = "\n".join(lines).strip()
 
     try:

src/ui/app.py

@@ -268,7 +268,7 @@ elif "session_id" not in st.session_state:
 # through innerHTML never execute (HTML5 spec). We instead use a tiny iframe
 # bootstrap (via components.html) that attaches the real Umami script to the
 # parent document, so analytics track the actual Streamlit page URL.
-import streamlit.components.v1 as components
+import streamlit.components.v1 as components  # noqa: E402
 
 components.html(
     """

tests/test_api.py

@@ -5,7 +5,7 @@ from unittest.mock import MagicMock, patch
 import pytest
 from fastapi.testclient import TestClient
 
-from src.api.routes import router, set_dependencies, QueryResponse, HealthResponse, ReadinessResponse
+from src.api.routes import router, set_dependencies
 from src.models import DocumentChunk, GenerationResponse, IntentType, QueryResult
 
 

tests/test_config.py

@@ -1,6 +1,6 @@
 """Tests for src.config."""
 
-from src.config import Settings, load_settings
+from src.config import Settings, load_settings  # noqa: F401
 
 
 class TestSettings:

tests/test_embedder.py

@@ -2,8 +2,6 @@
 
 from unittest.mock import MagicMock
 
-import pytest
-
 from src.retrieval.embedder import Embedder
 
 EMBEDDING_DIM = 384

tests/test_hybrid.py

@@ -4,7 +4,7 @@ from unittest.mock import MagicMock
 
 import pytest
 
-from src.models import ChunkStrategy, DocumentChunk, QueryResult
+from src.models import DocumentChunk, QueryResult
 from src.retrieval.hybrid import HybridRetriever
 
 

tests/test_memory.py

@@ -1,7 +1,5 @@
 """Tests for conversation memory."""
 
-import pytest
-
 from src.agent.memory import ConversationMemory, Turn
 from src.models import DocumentChunk, QueryResult
 

tests/test_models.py

@@ -1,6 +1,6 @@
 """Tests for src.models."""
 
-from src.models import DocumentChunk, QueryResult, GenerationResponse, ChunkStrategy, IntentType
+from src.models import DocumentChunk, QueryResult, GenerationResponse, ChunkStrategy, IntentType  # noqa: F401
 
 
 class TestDocumentChunk:

tests/test_plan_and_execute.py

@@ -3,8 +3,6 @@
 from unittest.mock import MagicMock, patch
 import json
 
-import pytest
-
 from src.agent.plan_and_execute import (
     PlanAndExecuteRouter,
     PlanExecState,

tests/test_reranker.py

@@ -1,6 +1,5 @@
 """Tests for the cross-encoder reranker."""
 
-import math
 from unittest.mock import MagicMock
 
 import numpy as np

tests/test_session_store.py

@@ -1,11 +1,7 @@
 """Tests for the SQLite-backed session store."""
 
 import os
-import tempfile
 
-import pytest
-
-from src.agent.memory import ConversationMemory
 from src.agent.session_store import SessionStore
 from src.models import DocumentChunk, QueryResult
 

tests/test_vector_store.py

@@ -1,7 +1,5 @@
 """Tests for the Qdrant vector store."""
 
-import tempfile
-
 import pytest
 
 from src.models import ChunkStrategy, DocumentChunk