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

minor formatting

Browse source
This commit is contained in:
Atharva J 2023-11-13 19:58:43 +05:30
commit 68d6fae606
1 changed files with 15 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/backend/langflow/components/vectorstores/Vectara.py
View file

@ -22,25 +22,25 @@ class VectaraComponent(CustomComponent):
"vectara_customer_id": { "vectara_customer_id": {
"display_name": "Vectara Customer ID", "display_name": "Vectara Customer ID",
"required": True, "required": True,
}, },
"vectara_corpus_id": { "vectara_corpus_id": {
"display_name": "Vectara Corpus ID", "display_name": "Vectara Corpus ID",
"required": True, "required": True,
}, },
"vectara_api_key": { "vectara_api_key": {
"display_name": "Vectara API Key", "display_name": "Vectara API Key",
"password": True, "password": True,
"required": True, "required": True,
}, },
"code": {"show": False}, "code": {"show": False},
"documents": { "documents": {
"display_name": "Documents", "display_name": "Documents",
"info": "Pass in either for Self Query Retriever or for making a Vectara Object" "info": "Pass in either for Self Query Retriever or for making a Vectara Object",
}, },
'files_url':{ "files_url": {
"display_name": "Files Url", "display_name": "Files Url",
"info": "Make vectara object using url of files(documents not needed)", "info": "Make vectara object using url of files(documents not needed)",
}, },
} }
def build( def build(
@ -48,11 +48,10 @@ class VectaraComponent(CustomComponent):
vectara_customer_id: str, vectara_customer_id: str,
vectara_corpus_id: str, vectara_corpus_id: str,
vectara_api_key: str, vectara_api_key: str,
files_url: Optional[List[str]] = None, files_url: Optional[List[str]] = None,
documents: Optional[Document] = None, documents: Optional[Document] = None,
) -> Union[VectorStore, BaseRetriever]: ) -> Union[VectorStore, BaseRetriever]:
if documents is not None:
if documents is not None :
return Vectara.from_documents( return Vectara.from_documents(
documents=documents, documents=documents,
embedding=FakeEmbeddings(size=768), embedding=FakeEmbeddings(size=768),
@ -60,8 +59,8 @@ class VectaraComponent(CustomComponent):
vectara_corpus_id=vectara_corpus_id, vectara_corpus_id=vectara_corpus_id,
vectara_api_key=vectara_api_key, vectara_api_key=vectara_api_key,
) )
if files_url is not None : if files_url is not None:
files_list = [] files_list = []
for url in files_url: for url in files_url:
name = tempfile.NamedTemporaryFile().name name = tempfile.NamedTemporaryFile().name
@ -80,4 +79,4 @@ class VectaraComponent(CustomComponent):
vectara_customer_id=vectara_customer_id, vectara_customer_id=vectara_customer_id,
vectara_corpus_id=vectara_corpus_id, vectara_corpus_id=vectara_corpus_id,
vectara_api_key=vectara_api_key, vectara_api_key=vectara_api_key,
) )