XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Add providers in api

0ea720c1
alvarosoares01@gmail.com <alvarosoares01@gmail.com>
提交于

代码差异

1 个文件 +6 -0
Modified g4f/api/__init__.py +6 -0
@@ -84,6 +84,10 @@ class Api:
84 84 stream = True if item_data.get("stream") == "True" else False
85 85 messages = item_data.get('messages')
86 86 conversation = item_data.get('conversation') if item_data.get('conversation') != None else None
87 provider = item_data.get('provider').replace('g4f.Provider.', '')
88 provider = provider if provider and provider != "Auto" else None
89 if provider != None:
90 provider = g4f.Provider.ProviderUtils.convert.get(provider)
87 91
88 92 try:
89 93 if model == 'pi':
@@ -92,12 +96,14 @@ class Api:
92 96 stream=stream,
93 97 messages=messages,
94 98 conversation=conversation,
99 provider = provider,
95 100 ignored=self.list_ignored_providers)
96 101 else:
97 102 response = g4f.ChatCompletion.create(
98 103 model=model,
99 104 stream=stream,
100 105 messages=messages,
106 provider = provider,
101 107 ignored=self.list_ignored_providers)
102 108 except Exception as e:
103 109 logging.exception(e)