diff --git a/open_notebook/config.py b/open_notebook/config.py index 9b15710..7a45dc9 100644 --- a/open_notebook/config.py +++ b/open_notebook/config.py @@ -6,7 +6,6 @@ from loguru import logger from open_notebook.domain.models import DefaultModels from open_notebook.models import get_model -# todo: enable config file overwrite with env vars current_dir = os.path.dirname(os.path.abspath(__file__)) project_root = os.path.dirname(current_dir) config_path = os.path.join(project_root, "open_notebook_config.yaml") diff --git a/open_notebook/models/embedding_models.py b/open_notebook/models/embedding_models.py index ac9466c..43119cf 100644 --- a/open_notebook/models/embedding_models.py +++ b/open_notebook/models/embedding_models.py @@ -39,7 +39,6 @@ class OllamaEmbeddingModel(EmbeddingModel): """ Embeds the content using Open AI embedding """ - # todo: make this Singleton text = text.replace("\n", " ") response = requests.post( f"{self.base_url}/api/embed", diff --git a/stream_app/chat.py b/stream_app/chat.py index ae2e732..00d8bd5 100644 --- a/stream_app/chat.py +++ b/stream_app/chat.py @@ -54,8 +54,6 @@ def execute_chat(txt_input, session_id): 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): context = build_context(session_id=session_id) tokens = token_count(str(context) + str(st.session_state[session_id]["messages"]))