From 2f07f0ab498b6f47187b7c45a78f4d970d652db5 Mon Sep 17 00:00:00 2001 From: LUIS NOVO Date: Wed, 30 Oct 2024 14:46:02 -0300 Subject: [PATCH] fix search embedding --- pages/3_🔍_Search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pages/3_🔍_Search.py b/pages/3_🔍_Search.py index 6c69ea2..3d6977c 100644 --- a/pages/3_🔍_Search.py +++ b/pages/3_🔍_Search.py @@ -1,7 +1,7 @@ import streamlit as st +from open_notebook.config import EMBEDDING_MODEL from open_notebook.domain.notebook import text_search, vector_search -from open_notebook.utils import get_embedding from stream_app.note import note_list_item from stream_app.source import source_list_item from stream_app.utils import version_sidebar @@ -33,7 +33,7 @@ with st.container(border=True): ) elif search_type == "Vector Search": st.write(f"Searching for {search_term}") - embed_query = get_embedding(search_term) + embed_query = EMBEDDING_MODEL.embed(search_term) st.session_state["search_results"] = vector_search( embed_query, 100, search_sources, search_notes )