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 -3
View File
@@ -5,11 +5,9 @@ from typing import List
import tiktoken_async
from jarvis.gpt.message import Message
async def count_message_tokens(
messages: List[Message], model: str = "gpt-3.5-turbo-0301"
messages: List[dict], model: str = "gpt-3.5-turbo-0301"
) -> int:
"""
Returns the number of tokens used by a list of messages.