From 6f2db272c4e1c1dbbb8481cb305537a729b78752 Mon Sep 17 00:00:00 2001
From: grillazz <the@grillazz.com>
Date: Sat, 3 May 2025 08:47:28 +0200
Subject: [PATCH] format code

---
 app/api/ml.py | 9 +++------
 tests/chat.py | 6 ++++--
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/app/api/ml.py b/app/api/ml.py
index e8d2c77..ca600a9 100644
--- a/app/api/ml.py
+++ b/app/api/ml.py
@@ -11,9 +11,6 @@ logger = AppLogger().get_logger()
 router = APIRouter()
 
 
-@router.post('/chat/')
-async def chat(
-    prompt: Annotated[str, Form()],
-    llm_service = Depends(get_llm_service)
-):
-    return StreamingResponse(llm_service.stream_chat(prompt), media_type="text/plain")
\ No newline at end of file
+@router.post("/chat/")
+async def chat(prompt: Annotated[str, Form()], llm_service=Depends(get_llm_service)):
+    return StreamingResponse(llm_service.stream_chat(prompt), media_type="text/plain")
diff --git a/tests/chat.py b/tests/chat.py
index f99c819..617ee14 100644
--- a/tests/chat.py
+++ b/tests/chat.py
@@ -14,7 +14,9 @@ async def chat_with_endpoint():
 
             print("\nModel: ", end="", flush=True)
             try:
-                async with client.stream("POST", API_URL, data={"prompt": prompt}, timeout=60) as response:
+                async with client.stream(
+                    "POST", API_URL, data={"prompt": prompt}, timeout=60
+                ) as response:
                     async for chunk in response.aiter_lines():
                         if not chunk:
                             continue
@@ -28,4 +30,4 @@ async def chat_with_endpoint():
 
 
 if __name__ == "__main__":
-    anyio.run(chat_with_endpoint)
\ No newline at end of file
+    anyio.run(chat_with_endpoint)