Fixing complete

seanchatmangpt · Feb 19, 2024 · 683dd7e · 683dd7e
1 parent 77cfe69
commit 683dd7e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/experiments/complete.py b/src/experiments/complete.py
@@ -9,7 +9,6 @@
 
 from utils.models import get_model
 
-client = AsyncOpenAI()
 
 
 @dataclass
@@ -188,6 +187,8 @@ async def acreate(*, config: Optional[LLMConfig] = None, **kwargs):
         )
 
     else:
+        client = AsyncOpenAI()
+
         response = await client.completions.create(
             model=model,
             prompt=prompt,
@@ -525,6 +526,8 @@ def _create_messages(sys_msg, prompt, msgs):
 
 
 async def main():
+    client = AsyncOpenAI()
+
     await acreate(
         prompt="Convert the CLI to Typer and use jinja withing the echo",
         model="chatgpt",