chore: set status code appropriately

winstxnhdw · Sep 23, 2024 · 107d655 · 107d655
1 parent f6a3718
commit 107d655
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/server/api/debug/generate.py b/server/api/debug/generate.py
@@ -2,6 +2,7 @@
 
 from litestar import Controller, post
 from litestar.response import ServerSentEvent
+from litestar.status_codes import HTTP_200_OK
 
 from server.features.chat.types import Message
 from server.schemas.v1 import Benchmark, Query
@@ -17,7 +18,7 @@ class LLMController(Controller):
 
     path = '/llm'
 
-    @post(sync_to_thread=True)
+    @post(status_code=HTTP_200_OK, sync_to_thread=True)
     def generate(self, state: AppState, data: Query) -> ServerSentEvent:
         """
         Summary

diff --git a/server/api/v1/files_to_text.py b/server/api/v1/files_to_text.py
@@ -6,11 +6,12 @@
 from litestar.enums import RequestEncodingType
 from litestar.params import Body
 from litestar.response import ServerSentEvent
+from litestar.status_codes import HTTP_200_OK
 
 from server.features.extraction import file_to_text
 
 
-@post('/files/text')
+@post('/files/text', status_code=HTTP_200_OK)
 async def files_to_text(
     data: Annotated[list[UploadFile], Body(media_type=RequestEncodingType.MULTI_PART)],
 ) -> ServerSentEvent: