fix oai provider bug (#696)

SciPhi-AI · Jul 14, 2024 · 2be5fac · 2be5fac
1 parent c61cf66
commit 2be5fac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "r2r"
-version = "0.2.59"
+version = "0.2.60"
 description = "SciPhi R2R"
 authors = ["Owen Colegrove <[email protected]>"]
 license = "MIT"

diff --git a/r2r/providers/llms/openai/base_openai.py b/r2r/providers/llms/openai/base_openai.py
@@ -29,7 +29,7 @@ def __init__(
                 "The provided config must be an instance of OpenAIConfig."
             )
         try:
-            from openai import OpenAI  # noqa
+            from openai import AsyncOpenAI, OpenAI  # noqa
         except ImportError:
             raise ImportError(
                 "Error, `openai` is required to run an OpenAILLM. Please install it using `pip install openai`."
@@ -44,6 +44,7 @@ def __init__(
             )
         super().__init__(config)
         self.config: LLMConfig = config
+        self.async_client = AsyncOpenAI()
         self.client = OpenAI()
 
     def get_completion(
@@ -141,4 +142,4 @@ async def _aget_completion(
 
         args = {**args, **kwargs}
         # Create the chat completion
-        return await self.client.chat.completions.create(**args)
+        return await self.async_client.chat.completions.create(**args)