fix: fix lint errors

Author: winstxnhdw

pyproject.toml

@@ -40,8 +40,11 @@ package-mode = false
 
 [tool.poetry.dependencies]
 python = "^3.12"
-pydantic-settings = "^2.3.4"
+granian = "^1.6.0"
+litestar = "^2.11.0"
 uvloop = "^0.19.0"
+msgspec = "^0.18.6"
+pydantic-settings = "^2.3.4"
 ctranslate2 = "^4.3.1"
 transformers = "^4.43.3"
 sentencepiece = "^0.2.0"
@@ -53,11 +56,8 @@ hiredis = "^3.0.0"
 tesserocr = "^2.7.0"
 torch = "^2.4.0"
 nltk = "^3.8.1"
-granian = "^1.6.0"
-litestar = "^2.11.0"
 picologging = "^0.9.3"
 fastnanoid = "^0.4.1"
-msgspec = "^0.18.6"
 
 [tool.poetry.group.dev.dependencies]
 pyright = "^1.1.373"

server/api/debug/redis.py

@@ -6,7 +6,7 @@
 
 from server.config import Config
 from server.databases.redis.wrapper import RedisAsync
-from server.dependencies import embedder, redis_client
+from server.dependencies import embedder_model, redis_client
 from server.features.embeddings import Embedder
 from server.schemas.v1 import Query
 
@@ -21,7 +21,7 @@ class RedisController(Controller):
     path = '/redis'
     dependencies = {
         'redis': Provide(redis_client),
-        'embedder': Provide(embedder),
+        'embedder': Provide(embedder_model),
     }
 
     @delete()

server/api/v1/chat.py

@@ -10,7 +10,7 @@
 
 from server.databases.redis.features import store_chunks
 from server.databases.redis.wrapper import RedisAsync
-from server.dependencies import embedder, redis_client
+from server.dependencies import embedder_model, redis_client
 from server.features.chunking import SentenceSplitter, chunk_document
 from server.features.embeddings import Embedder
 from server.features.extraction import extract_documents_from_pdfs
@@ -29,7 +29,7 @@ class ChatController(Controller):
     path = '/chats'
     dependencies = {
         'redis': Provide(redis_client),
-        'embedder': Provide(embedder),
+        'embedder': Provide(embedder_model),
     }
 
     @get()

server/dependencies/__init__.py

@@ -1,2 +1,2 @@
-from server.dependencies.embedder import embedder as embedder
+from server.dependencies.embedder import embedder_model as embedder_model
 from server.dependencies.redis import redis_client as redis_client

server/dependencies/embedder.py

@@ -3,7 +3,7 @@
 from server.features.embeddings import Embedder
 
 
-def embedder() -> Iterator[Embedder]:
+def embedder_model() -> Iterator[Embedder]:
     """
     Summary
     -------

server/features/chat/model.py

@@ -42,9 +42,9 @@ def __init__(
         max_context_length: int,
         max_generation_length: int,
     ):
-        self.max_query_length = self.max_context_length - self.max_generation_length
+        self.max_query_length = max_context_length - max_generation_length
 
-        if self.max_query_length < self.min_query_length:
+        if self.max_query_length < min_query_length:
             raise ValueError('The minimum query length cannot be greater than the maximum query length!')
 
         self.generator = generator

server/lifespans/download_nltk.py

@@ -24,7 +24,7 @@ def find_or_download_nltk(download_directory: str):
 
 
 @asynccontextmanager
-async def download_nltk(app: Litestar) -> AsyncIterator[None]:
+async def download_nltk(_: Litestar) -> AsyncIterator[None]:
     """
     Summary
     -------

server/state.py

@@ -5,5 +5,16 @@
 
 
 class AppState(State):
+    """
+    Summary
+    -------
+    the Litestar application state
+
+    Attributes
+    ----------
+    redis_pool (ConnectionPool) : the global Redis connection pool
+    chat (ChatModel) : the LLM chat model
+    """
+
     redis_pool: ConnectionPool
     chat: ChatModel

tests/test_embedding.py

@@ -5,15 +5,15 @@
 from numpy import array_equal
 from pytest import fixture
 
-from server.dependencies import embedder
+from server.dependencies import embedder_model
 from server.features.embeddings import Embedder
 
 type Text = Literal['Hello world!']
 
 
 @fixture()
 def embedding() -> Iterable[Embedder]:
-    return embedder()
+    return embedder_model()
 
 
 @fixture()