Simplify and clarify

This commit is contained in:
Anton Osika
2023-06-03 23:19:19 +02:00
parent 6a31d875c7
commit de046d39e9
6 changed files with 66 additions and 28 deletions

3
ai.py
View File

@@ -20,12 +20,13 @@ class AI:
def fuser(self, msg):
return {"role": "user", "content": msg}
def next(self, messages, prompt=None):
def next(self, messages: list[dict[str, str]], prompt=None):
if prompt:
messages = messages + [{"role": "user", "content": prompt}]
response = openai.ChatCompletion.create(
messages=messages,
stream=True,
**self.kwargs
)