update type annotation

This commit is contained in:
youkaichao
2023-04-15 22:04:05 +08:00
parent 1073954fb7
commit afd2c5e2c6
3 changed files with 14 additions and 13 deletions

View File

@@ -14,7 +14,7 @@ class AgentManager(metaclass=Singleton):
# Create new GPT agent
# TODO: Centralise use of create_chat_completion() to globally enforce token limit
def create_agent(self, task: str, prompt: str, model: str) -> tuple[int, str]:
def create_agent(self, task: str, prompt: str, model: str) -> Tuple[int, str]:
"""Create a new agent and return its key
Args: