Skip to content

Commit 5169152

Browse files
fix: patch v0 api only when openai v0 is installed
1 parent ab54dde commit 5169152

File tree

1 file changed

+15
-9
lines changed
  • src/promplate_pyodide/utils/openai

1 file changed

+15
-9
lines changed

src/promplate_pyodide/utils/openai/sync.py

+15-9
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,20 @@
44

55

66
def patch_sync_apis():
7-
from promplate.llm.openai import v0, v1
7+
from openai.version import VERSION
88

9-
v1.SyncTextOpenAI.complete = v1.TextComplete.__call__ = to_sync(v1.AsyncTextComplete.__call__)
10-
v1.SyncTextOpenAI.generate = v1.TextGenerate.__call__ = to_sync(v1.AsyncTextGenerate.__call__)
11-
v1.SyncChatOpenAI.complete = v1.ChatComplete.__call__ = to_sync(v1.AsyncChatComplete.__call__)
12-
v1.SyncChatOpenAI.generate = v1.ChatGenerate.__call__ = to_sync(v1.AsyncChatGenerate.__call__)
9+
if VERSION.startswith("0"):
10+
from promplate.llm.openai import v0
1311

14-
v0.TextComplete.__call__ = to_sync(v0.AsyncTextComplete.__call__)
15-
v0.TextGenerate.__call__ = to_sync(v0.AsyncTextGenerate.__call__)
16-
v0.ChatComplete.__call__ = to_sync(v0.AsyncChatComplete.__call__)
17-
v0.ChatGenerate.__call__ = to_sync(v0.AsyncChatGenerate.__call__)
12+
v0.TextComplete.__call__ = to_sync(v0.AsyncTextComplete.__call__)
13+
v0.TextGenerate.__call__ = to_sync(v0.AsyncTextGenerate.__call__)
14+
v0.ChatComplete.__call__ = to_sync(v0.AsyncChatComplete.__call__)
15+
v0.ChatGenerate.__call__ = to_sync(v0.AsyncChatGenerate.__call__)
16+
17+
else:
18+
from promplate.llm.openai import v1
19+
20+
v1.SyncTextOpenAI.complete = v1.TextComplete.__call__ = to_sync(v1.AsyncTextComplete.__call__)
21+
v1.SyncTextOpenAI.generate = v1.TextGenerate.__call__ = to_sync(v1.AsyncTextGenerate.__call__)
22+
v1.SyncChatOpenAI.complete = v1.ChatComplete.__call__ = to_sync(v1.AsyncChatComplete.__call__)
23+
v1.SyncChatOpenAI.generate = v1.ChatGenerate.__call__ = to_sync(v1.AsyncChatGenerate.__call__)

0 commit comments

Comments
 (0)