From 0b2924303605de0ee9eda6bfd21369d13e9601c7 Mon Sep 17 00:00:00 2001 From: Oleksandr Kozachuk Date: Sat, 9 Sep 2023 21:37:48 +0200 Subject: [PATCH] Fix to use ID instead of name in message.ai. --- chatmastermind/ais/openai.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/chatmastermind/ais/openai.py b/chatmastermind/ais/openai.py index 1db4d20..a388a7a 100644 --- a/chatmastermind/ais/openai.py +++ b/chatmastermind/ais/openai.py @@ -45,14 +45,14 @@ class OpenAI(AI): presence_penalty=self.config.presence_penalty) question.answer = Answer(response['choices'][0]['message']['content']) question.tags = otags - question.ai = self.name + question.ai = self.ID question.model = self.config.model answers: list[Message] = [question] for choice in response['choices'][1:]: # type: ignore answers.append(Message(question=question.question, answer=Answer(choice['message']['content']), tags=otags, - ai=self.name, + ai=self.ID, model=self.config.model)) return AIResponse(answers, Tokens(response['usage']['prompt_tokens'], response['usage']['completion_tokens'],