Fix tests

deepsense-ai · Dec 4, 2024 · 8fd2bd1 · 8fd2bd1
1 parent d5718c6
commit 8fd2bd1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/ragbits-core/src/ragbits/core/llms/clients/litellm.py b/packages/ragbits-core/src/ragbits/core/llms/clients/litellm.py
@@ -181,7 +181,7 @@ async def _get_litellm_response(
         options: LiteLLMOptions,
         response_format: type[BaseModel] | dict | None,
         stream: bool = False,
-    ) -> "ModelResponse" | "CustomStreamWrapper":
+    ) -> "ModelResponse | CustomStreamWrapper":
         try:
             response = await litellm.acompletion(
                 messages=conversation,