fix: cli dependencies

umbertogriffo · Jan 11, 2025 · 0ed7fab · 0ed7fab
1 parent 7204cad
commit 0ed7fab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/chatbot/cli/chatbot.py b/chatbot/cli/chatbot.py
@@ -4,7 +4,7 @@
 from pathlib import Path
 
 from bot.client.lama_cpp_client import LamaCppClient
-from bot.model.model_settings import get_model_settings, get_models
+from bot.model.model_registry import get_model_settings, get_models
 from helpers.log import get_logger
 from helpers.reader import read_input
 from pyfiglet import Figlet

diff --git a/chatbot/cli/rag_chatbot.py b/chatbot/cli/rag_chatbot.py
@@ -8,7 +8,7 @@
 from bot.conversation.ctx_strategy import get_ctx_synthesis_strategies, get_ctx_synthesis_strategy
 from bot.memory.embedder import Embedder
 from bot.memory.vector_database.chroma import Chroma
-from bot.model.model_settings import get_model_settings, get_models
+from bot.model.model_registry import get_model_settings, get_models
 from helpers.log import get_logger
 from helpers.prettier import prettify_source
 from helpers.reader import read_input