Delete Message typed dict

This commit is contained in:
fiatrete
2023-06-16 15:58:57 +08:00
parent 62d31b68ae
commit 88e3ed0648
6 changed files with 5 additions and 36 deletions
+1 -2
View File
@@ -4,7 +4,6 @@ import openai
from openai.error import RateLimitError, APIError, Timeout
from jarvis import CFG
from jarvis.gpt.message import Message
from jarvis.logger import logger
from typing import Callable
@@ -60,7 +59,7 @@ async def acreate_chat_completion_once(
# Overly simple abstraction until we create something better
# simple retry mechanism when getting a rate error or a bad gateway
async def acreate_chat_completion(
messages: list[Message], # type: ignore
messages: list[dict],
model: str = None,
temperature: float = CFG.temperature,
max_tokens: int = None,