fix type annotation

2023-11-30 11:56:51 -05:00 · 2023-11-30 11:56:51 -05:00 · aae7c0e468
commit aae7c0e468
parent 572dedbd01
1 changed files with 4 additions and 1 deletions
--- a/backend/llm.py
+++ b/backend/llm.py
@ -6,7 +6,10 @@ MODEL_GPT_4_VISION = "gpt-4-vision-preview"


 async def stream_openai_response(
-    messages, api_key: str, base_url:str,  callback: Callable[[str], Awaitable[None]]
+    messages,
+    api_key: str,
+    base_url: str | None,
+    callback: Callable[[str], Awaitable[None]],
 ):
    client = AsyncOpenAI(api_key=api_key, base_url=base_url)