From aae7c0e46865d86eb05b679ede27360d55d17d82 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Thu, 30 Nov 2023 11:56:51 -0500 Subject: [PATCH] fix type annotation --- backend/llm.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/llm.py b/backend/llm.py index 82a765e..fdb1ba0 100644 --- 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)