updated __init__.py

Pythagora-io · Oct 4, 2023 · d0c8db2 · d0c8db2
1 parent c1e47ce
commit d0c8db2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/pilot/helpers/exceptions/ApiKeyNotDefinedError.py b/pilot/helpers/exceptions/ApiKeyNotDefinedError.py
@@ -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}")
diff --git a/pilot/helpers/exceptions/__init__.py b/pilot/helpers/exceptions/__init__.py
@@ -1,2 +1,3 @@
-from .AgentConvo import AgentConvo
-from .Project import Project
+from .ApiKeyNotDefinedError import ApiKeyNotDefinedError
+from .TokenLimitError import TokenLimitError
+from .TooDeepRecursionError import TooDeepRecursionError