admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

model_manager import

Browse source
This commit is contained in:
LUIS NOVO 2024-11-01 22:32:03 -03:00
parent 212d3a33b0
commit 4f586ad513
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
pages/3_🔍_Search.py
View file

@ -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

2
pages/stream_app/utils.py
View file

@ -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,