Polish codebase: Fix linting errors identified by ruff and cleanup unused imports

Author: gabaoun

src/api/app.py

@@ -4,10 +4,9 @@
 from src.pipeline.retrieval import RetrievalWorkflow
 from src.config.settings import settings
 from src.utils.logger import logger
-from src.db.session import SessionLocal, get_db
+from src.db.session import get_db
 from src.models.db import IngestionJob
 from src.infra.queue import get_queue
-import os
 from contextlib import asynccontextmanager
 from sqlalchemy.orm import Session
 

src/jobs/ingestion.py

@@ -1,6 +1,6 @@
 import asyncio
 from src.db.session import SessionLocal
-from src.models.db import IngestionJob, Document
+from src.models.db import IngestionJob
 from src.pipeline.ingestion import IngestionWorkflow
 from src.utils.logger import logger
 from src.config.settings import settings

src/pipeline/retrieval.py

@@ -1,5 +1,5 @@
-from typing import List, Optional, Union
-from llama_index.core import VectorStoreIndex, QueryBundle
+from typing import List, Union
+from llama_index.core import QueryBundle
 from llama_index.core.schema import NodeWithScore
 from llama_index.core.workflow import (
     Workflow,

src/services/chroma.py

@@ -1,9 +1,8 @@
 import chromadb
 from chromadb.config import Settings
 from chromadb.api.models.Collection import Collection
-from typing import List, Dict, Any, Optional
+from typing import List, Dict, Any
 import uuid
-import math
 from src.config.settings import settings
 from src.utils.logger import logger
 

src/worker.py

@@ -6,7 +6,7 @@ def run_worker():
     redis_conn = get_redis_connection()
     queue = get_queue()
 
-    logger.info(f"Starting worker connected to Redis")
+    logger.info("Starting worker connected to Redis")
 
     # In recent RQ versions, connection is passed to the Worker constructor
     worker = Worker([queue], connection=redis_conn)

tests/test_ingestion.py

@@ -1,5 +1,5 @@
 import pytest
-from unittest.mock import MagicMock, AsyncMock
+from unittest.mock import MagicMock
 from llama_index.core.schema import Document
 from src.pipeline.ingestion import IngestionWorkflow, DocumentsLoadedEvent
 from llama_index.core.workflow import StartEvent

tests/test_retrieval.py

@@ -1,5 +1,5 @@
 import pytest
-from unittest.mock import MagicMock, AsyncMock
+from unittest.mock import MagicMock
 from src.pipeline.retrieval import RetrievalWorkflow, QueryTransformedEvent
 from llama_index.core.workflow import StartEvent, StopEvent