xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-19 19:06:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

修复使用文本完成模型生成对话型文本时输出随机AI名的问题

Browse Source
This commit is contained in:
LINSTCL
2023-03-02 23:50:51 +08:00
parent 2234e9db0e
commit 77df3d1ae5
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/openai/modelmgr.py
+15 -6
View File
@@ -42,8 +42,8 @@ class ModelRequest():
def msg_handle(self, msg): def msg_handle(self, msg):
return msg return msg
def ret_handle(self, ret): def ret_handle(self):
return ret return
def get_total_tokens(self): def get_total_tokens(self):
return self.ret['usage']['total_tokens'] return self.ret['usage']['total_tokens']
@@ -64,8 +64,8 @@ class ChatCompletionModel(ModelRequest):
super().__init__(model_name, user_name, request_fun) super().__init__(model_name, user_name, request_fun)
def request(self, messages, **kwargs): def request(self, messages, **kwargs):
ret = self.request_fun(messages = self.msg_handle(messages), **kwargs, user=self.user_name) self.ret = self.request_fun(messages = self.msg_handle(messages), **kwargs, user=self.user_name)
self.ret = self.ret_handle(ret) self.ret_handle()
self.message = self.ret["choices"][0]["message"]['content'] self.message = self.ret["choices"][0]["message"]['content']
def msg_handle(self, msgs): def msg_handle(self, msgs):
@@ -87,8 +87,8 @@ class CompletionModel(ModelRequest):
super().__init__(model_name, user_name, request_fun) super().__init__(model_name, user_name, request_fun)
def request(self, prompt, **kwargs): def request(self, prompt, **kwargs):
ret = self.request_fun(prompt = self.msg_handle(prompt), **kwargs) self.ret = self.request_fun(prompt = self.msg_handle(prompt), **kwargs)
self.ret = self.ret_handle(ret) self.ret_handle()
self.message = self.ret["choices"][0]["text"] self.message = self.ret["choices"][0]["text"]
def msg_handle(self, msgs): def msg_handle(self, msgs):
@@ -100,6 +100,15 @@ class CompletionModel(ModelRequest):
prompt = prompt + "{}:{}\n".format(msg['role'] , msg['content']) prompt = prompt + "{}:{}\n".format(msg['role'] , msg['content'])
return prompt return prompt
def ret_handle(self):
temp_text:str = self.ret["choices"][0]["text"]
texts = temp_text.split(':')
if len(texts) >= 1:
temp_text = ""
for text in texts[1:]:
temp_text = temp_text + text
self.ret["choices"][0]["text"] = temp_text
def get_text(self): def get_text(self):
return self.message return self.message