kaizenkodo/ChatMastermind
2
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

openai: stores AI.ID instead of AI.name in message

Browse Source
This commit is contained in:
juk0de 2023-09-10 07:55:47 +02:00
parent 1442026d58
commit 68ac4bd60d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
chatmastermind/ais/openai.py
View File

@ -45,14 +45,14 @@ class OpenAI(AI):
presence_penalty=self.config.presence_penalty) presence_penalty=self.config.presence_penalty)
question.answer = Answer(response['choices'][0]['message']['content']) question.answer = Answer(response['choices'][0]['message']['content'])
question.tags = otags question.tags = otags
question.ai = self.name question.ai = self.ID
question.model = self.config.model question.model = self.config.model
answers: list[Message] = [question] answers: list[Message] = [question]
for choice in response['choices'][1:]: # type: ignore for choice in response['choices'][1:]: # type: ignore
answers.append(Message(question=question.question, answers.append(Message(question=question.question,
answer=Answer(choice['message']['content']), answer=Answer(choice['message']['content']),
tags=otags, tags=otags,
ai=self.name, ai=self.ID,
model=self.config.model)) model=self.config.model))
return AIResponse(answers, Tokens(response['usage']['prompt_tokens'], return AIResponse(answers, Tokens(response['usage']['prompt_tokens'],
response['usage']['completion_tokens'], response['usage']['completion_tokens'],