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

cleanup todos

Browse source
This commit is contained in:
LUIS NOVO 2024-10-30 17:16:06 -03:00
parent c01b8255fc
commit 9390ea82f2
3 changed files with 0 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
open_notebook/config.py
View file

@ -6,7 +6,6 @@ from loguru import logger
from open_notebook.domain.models import DefaultModels from open_notebook.domain.models import DefaultModels
from open_notebook.models import get_model from open_notebook.models import get_model
# todo: enable config file overwrite with env vars
current_dir = os.path.dirname(os.path.abspath(__file__)) current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.dirname(current_dir) project_root = os.path.dirname(current_dir)
config_path = os.path.join(project_root, "open_notebook_config.yaml") config_path = os.path.join(project_root, "open_notebook_config.yaml")

1
open_notebook/models/embedding_models.py
View file

@ -39,7 +39,6 @@ class OllamaEmbeddingModel(EmbeddingModel):
""" """
Embeds the content using Open AI embedding Embeds the content using Open AI embedding
""" """
# todo: make this Singleton
text = text.replace("\n", " ") text = text.replace("\n", " ")
response = requests.post( response = requests.post(
f"{self.base_url}/api/embed", f"{self.base_url}/api/embed",

2
stream_app/chat.py
View file

@ -54,8 +54,6 @@ def execute_chat(txt_input, session_id):
return result return result
# todo: se eu for usar o token count, preciso deixar configuravel
# seria bom ter um total de tokens no admin em algum lugar
def chat_sidebar(session_id): def chat_sidebar(session_id):
context = build_context(session_id=session_id) context = build_context(session_id=session_id)
tokens = token_count(str(context) + str(st.session_state[session_id]["messages"])) tokens = token_count(str(context) + str(st.session_state[session_id]["messages"]))