From 4c3b54f5efdb159a2e8f758acbc0d132f747bfda Mon Sep 17 00:00:00 2001 From: igorrCarvalho Date: Thu, 22 Feb 2024 12:57:06 -0300 Subject: [PATCH] Refactor: Change chat input sender name default value and remove persist from Chroma --- src/backend/langflow/components/io/ChatInput.py | 2 +- src/backend/langflow/components/vectorstores/Chroma.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py index 3e03642b3..42f2715e6 100644 --- a/src/backend/langflow/components/io/ChatInput.py +++ b/src/backend/langflow/components/io/ChatInput.py @@ -34,7 +34,7 @@ class ChatInput(CustomComponent): def build( self, sender: Optional[str] = "User", - sender_name: Optional[str] = "You", + sender_name: Optional[str] = "User", message: Optional[str] = None, as_record: Optional[bool] = False, session_id: Optional[str] = None, diff --git a/src/backend/langflow/components/vectorstores/Chroma.py b/src/backend/langflow/components/vectorstores/Chroma.py index a031f090f..ed0a096f3 100644 --- a/src/backend/langflow/components/vectorstores/Chroma.py +++ b/src/backend/langflow/components/vectorstores/Chroma.py @@ -28,7 +28,6 @@ class ChromaComponent(CustomComponent): """ return { "collection_name": {"display_name": "Collection Name", "value": "langflow"}, - "persist": {"display_name": "Persist"}, "index_directory": {"display_name": "Persist Directory"}, "code": {"advanced": True, "display_name": "Code"}, "documents": {"display_name": "Documents", "is_list": True}, @@ -52,7 +51,6 @@ class ChromaComponent(CustomComponent): def build( self, collection_name: str, - persist: bool, embedding: Embeddings, chroma_server_ssl_enabled: bool, index_directory: Optional[str] = None, @@ -107,7 +105,7 @@ class ChromaComponent(CustomComponent): ) chroma = Chroma.from_documents( documents=documents, # type: ignore - persist_directory=index_directory if persist else None, + persist_directory=index_directory, collection_name=collection_name, embedding=embedding, client_settings=chroma_settings,