jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

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

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/components/io/ChatInput.py
View file

@ -34,7 +34,7 @@ class ChatInput(CustomComponent):
def build( def build(
self, self,
sender: Optional[str] = "User", sender: Optional[str] = "User",
sender_name: Optional[str] = "You", sender_name: Optional[str] = "User",
message: Optional[str] = None, message: Optional[str] = None,
as_record: Optional[bool] = False, as_record: Optional[bool] = False,
session_id: Optional[str] = None, session_id: Optional[str] = None,

4
src/backend/langflow/components/vectorstores/Chroma.py
View file

@ -28,7 +28,6 @@ class ChromaComponent(CustomComponent):
""" """
return { return {
"collection_name": {"display_name": "Collection Name", "value": "langflow"}, "collection_name": {"display_name": "Collection Name", "value": "langflow"},
"persist": {"display_name": "Persist"},
"index_directory": {"display_name": "Persist Directory"}, "index_directory": {"display_name": "Persist Directory"},
"code": {"advanced": True, "display_name": "Code"}, "code": {"advanced": True, "display_name": "Code"},
"documents": {"display_name": "Documents", "is_list": True}, "documents": {"display_name": "Documents", "is_list": True},
@ -52,7 +51,6 @@ class ChromaComponent(CustomComponent):
def build( def build(
self, self,
collection_name: str, collection_name: str,
persist: bool,
embedding: Embeddings, embedding: Embeddings,
chroma_server_ssl_enabled: bool, chroma_server_ssl_enabled: bool,
index_directory: Optional[str] = None, index_directory: Optional[str] = None,
@ -107,7 +105,7 @@ class ChromaComponent(CustomComponent):
) )
chroma = Chroma.from_documents( chroma = Chroma.from_documents(
documents=documents, # type: ignore documents=documents, # type: ignore
persist_directory=index_directory if persist else None, persist_directory=index_directory,
collection_name=collection_name, collection_name=collection_name,
embedding=embedding, embedding=embedding,
client_settings=chroma_settings, client_settings=chroma_settings,