Refactor: Change chat input sender name default value and remove persist from Chroma

This commit is contained in:
igorrCarvalho 2024-02-22 12:57:06 -03:00
commit 4c3b54f5ef
2 changed files with 2 additions and 4 deletions

View file

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

View file

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