diff --git a/chatbot/cli/chatbot.py b/chatbot/cli/chatbot.py index 24d11d9..1fa22fa 100644 --- 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 index 8ed85d0..20cb847 100644 --- 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