updated __init__.py

This commit is contained in:
Nicholas Albion
2023-10-04 16:09:41 +11:00
parent c1e47ceb8b
commit d0c8db238c
2 changed files with 7 additions and 7 deletions

View File

@@ -1,5 +1,4 @@
class TokenLimitError(Exception):
def __init__(self, tokens_in_messages, max_tokens):
self.tokens_in_messages = tokens_in_messages
self.max_tokens = max_tokens
super().__init__(f"Token limit error happened with {tokens_in_messages}/{max_tokens} tokens in messages!")
class ApiKeyNotDefinedError(Exception):
def __init__(self, env_key: str):
self.env_key = env_key
super().__init__(f"API Key has not been configured: {env_key}")

View File

@@ -1,2 +1,3 @@
from .AgentConvo import AgentConvo
from .Project import Project
from .ApiKeyNotDefinedError import ApiKeyNotDefinedError
from .TokenLimitError import TokenLimitError
from .TooDeepRecursionError import TooDeepRecursionError