From 0ed7fab2318b8bc5b575e644409c010693bb7016 Mon Sep 17 00:00:00 2001 From: Umberto Griffo <1609440+umbertogriffo@users.noreply.github.com> Date: Sat, 11 Jan 2025 11:36:44 +0000 Subject: [PATCH] fix: cli dependencies --- chatbot/cli/chatbot.py | 2 +- chatbot/cli/rag_chatbot.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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