fix circular import

AbanteAI · Apr 5, 2024 · 4f42e1c · 4f42e1c
1 parent f8a61fd
commit 4f42e1c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mentat/llm_api_handler.py b/mentat/llm_api_handler.py
@@ -38,7 +38,6 @@
 
 from mentat.errors import MentatError, ReturnToUser
 from mentat.session_context import SESSION_CONTEXT
-from mentat.session_input import collect_user_input
 from mentat.utils import mentat_dir_path
 
 TOKEN_COUNT_WARNING = 32000
@@ -334,6 +333,8 @@ async def initialize_client(self):
         try:
             self.spice.load_provider(OPEN_AI)
         except NoAPIKeyError:
+            from mentat.session_input import collect_user_input
+
             ctx.stream.send(
                 "No OpenAI api key detected. To avoid entering your api key on startup, create a .env file in"
                 " ~/.mentat/.env or in your workspace root.",