fix tests

upstash · Jul 4, 2024 · 2007399 · 2007399
1 parent 7aa3cad
commit 2007399
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/asyncio/test_chat.py b/tests/asyncio/test_chat.py
@@ -188,7 +188,7 @@ async def test_prompt_custom_provider_async(async_qstash: AsyncQStash) -> None:
     res = await async_qstash.chat.prompt(
         model="gpt-3.5-turbo",
         user="just say hello",
-        provider=openai(token="OPENAI_API_KEY"),
+        provider=openai(token=OPENAI_API_KEY),  # type:ignore[arg-type]
     )
 
     assert isinstance(res, ChatCompletion)

diff --git a/tests/test_chat.py b/tests/test_chat.py
@@ -158,7 +158,7 @@ def test_prompt_custom_provider(qstash: QStash) -> None:
     res = qstash.chat.prompt(
         model="gpt-3.5-turbo",
         user="just say hello",
-        provider=openai(token="OPENAI_API_KEY"),
+        provider=openai(token=OPENAI_API_KEY),  # type:ignore[arg-type]
     )
 
     assert isinstance(res, ChatCompletion)