From 4f586ad513b34704a919c8c6e3743852e171b04e Mon Sep 17 00:00:00 2001 From: LUIS NOVO Date: Fri, 1 Nov 2024 22:32:03 -0300 Subject: [PATCH] model_manager import --- pages/3_🔍_Search.py | 3 ++- pages/stream_app/utils.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/pages/3_🔍_Search.py b/pages/3_🔍_Search.py index c0a7b84..a9abad5 100644 --- a/pages/3_🔍_Search.py +++ b/pages/3_🔍_Search.py @@ -1,7 +1,8 @@ import streamlit as st +from open_notebook.domain.models import model_manager from open_notebook.domain.notebook import text_search, vector_search -from open_notebook.models import EmbeddingModel, model_manager +from open_notebook.models import EmbeddingModel from pages.stream_app.note import note_list_item from pages.stream_app.source import source_list_item from pages.stream_app.utils import setup_page diff --git a/pages/stream_app/utils.py b/pages/stream_app/utils.py index 69bf78e..88f8849 100644 --- a/pages/stream_app/utils.py +++ b/pages/stream_app/utils.py @@ -2,8 +2,8 @@ import streamlit as st from loguru import logger from open_notebook.database.migrate import MigrationManager +from open_notebook.domain.models import model_manager from open_notebook.graphs.chat import ThreadState, graph -from open_notebook.models import model_manager from open_notebook.utils import ( compare_versions, get_installed_version,