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

Fix some of the commands.

Browse Source
This commit is contained in:
Oleksandr Kozachuk 2023-11-10 10:42:46 +01:00
parent 75a123eb72
commit 5d1bb1f9e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
chatmastermind/ais/openai.py
View File

@ -68,6 +68,7 @@ class OpenAI(AI):
self.ID = config.ID self.ID = config.ID
self.name = config.name self.name = config.name
self.config = config self.config = config
openai.api_key = self.config.api_key
def request(self, def request(self,
question: Message, question: Message,
@ -80,7 +81,6 @@ class OpenAI(AI):
nr. of messages in the 'AIResponse'. nr. of messages in the 'AIResponse'.
""" """
self.encoding = tiktoken.encoding_for_model(self.config.model) self.encoding = tiktoken.encoding_for_model(self.config.model)
openai.api_key = self.config.api_key
oai_chat, prompt_tokens = self.openai_chat(chat, self.config.system, question) oai_chat, prompt_tokens = self.openai_chat(chat, self.config.system, question)
tokens: Tokens = Tokens(prompt_tokens, 0, prompt_tokens) tokens: Tokens = Tokens(prompt_tokens, 0, prompt_tokens)
response = openai.ChatCompletion.create( response = openai.ChatCompletion.create(