File tree 1 file changed +15
-9
lines changed
src/promplate_pyodide/utils/openai
1 file changed +15
-9
lines changed Original file line number Diff line number Diff line change 4
4
5
5
6
6
def patch_sync_apis ():
7
- from promplate . llm . openai import v0 , v1
7
+ from openai . version import VERSION
8
8
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
13
11
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__ )
You can’t perform that action at this time.
0 commit comments