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

19
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(
@ -51,8 +51,7 @@ class VectaraComponent(CustomComponent):
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),
@ -61,7 +60,7 @@ class VectaraComponent(CustomComponent):
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