xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-25 11:37:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: 功能完成

Browse Source
This commit is contained in:
Rock Chin
2023-03-03 00:21:16 +08:00
parent adc2290fc1
commit a1c167fb7f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/modelmgr.py
+2 -2
View File
@@ -65,7 +65,7 @@ class ChatCompletionModel(ModelRequest):
super().__init__(model_name, user_name, request_fun) super().__init__(model_name, user_name, request_fun)
def request(self, prompts, **kwargs): def request(self, prompts, **kwargs):
ret = self.request_fun(messages = self.__msg_handle__(prompts), **kwargs, user=self.user_name) self.ret = self.request_fun(messages = self.__msg_handle__(prompts), **kwargs, user=self.user_name)
self.ret_handle() self.ret_handle()
self.message = self.ret["choices"][0]["message"]['content'] self.message = self.ret["choices"][0]["message"]['content']
@@ -87,7 +87,7 @@ class CompletionModel(ModelRequest):
super().__init__(model_name, user_name, request_fun) super().__init__(model_name, user_name, request_fun)
def request(self, prompts, **kwargs): def request(self, prompts, **kwargs):
ret = self.request_fun(prompt = self.__msg_handle__(prompts), **kwargs) self.ret = self.request_fun(prompt = self.__msg_handle__(prompts), **kwargs)
self.ret_handle() self.ret_handle()
self.message = self.ret["choices"][0]["text"] self.message = self.ret["choices"][0]["text"]