fix: add missing await

promplate · Nov 26, 2023 · 9156414 · 9156414
1 parent 27abf08
commit 9156414
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/promplate/llm/openai/v1.py b/python/promplate/llm/openai/v1.py
@@ -61,7 +61,7 @@ def __call__(self, text: str, /, **config):
 class AsyncTextGenerate(Config):
     async def __call__(self, text: str, /, **config):
         config = config | {"stream": True, "prompt": text}
-        stream = self.aclient.completions.create(**config)
+        stream = await self.aclient.completions.create(**config)
         async for event in stream:
             yield event.choices[0].text
 
@@ -99,6 +99,6 @@ class AsyncChatGenerate(Config):
     async def __call__(self, messages: list[Message] | str, /, **config):
         messages = ensure(messages)
         config = config | {"stream": True, "messages": messages}
-        stream = self.aclient.chat.completions.create(**config)
+        stream = await self.aclient.chat.completions.create(**config)
         async for event in stream:
             yield event.choices[0].delta.content
diff --git a/python/pyproject.toml b/python/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promplate"
-version = "0.3.0a1"
+version = "0.3.0a2"
 description = "cross-language prompt engineering framework"
 homepage = "https://promplate.dev/"
 repository = "https://github.com/promplate/core"