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

[autofix.ci] apply automated fixes

Browse source
This commit is contained in:
autofix-ci[bot] 2024-06-26 19:29:41 +00:00 • committed by GitHub
commit 7bd03336bf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 49 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

43
src/backend/base/langflow/components/embeddings/AstraVectorize.py
View file

@ -13,20 +13,39 @@ class AstraVectorize(Component):
VECTORIZE_PROVIDERS_MAPPING = { VECTORIZE_PROVIDERS_MAPPING = {
"Azure OpenAI": ["azureOpenAI", ["text-embedding-3-small", "text-embedding-3-large", "text-embedding-ada-002"]], "Azure OpenAI": ["azureOpenAI", ["text-embedding-3-small", "text-embedding-3-large", "text-embedding-ada-002"]],
"Hugging Face - Dedicated": ["huggingfaceDedicated", ["endpoint-defined-model"]], "Hugging Face - Dedicated": ["huggingfaceDedicated", ["endpoint-defined-model"]],
"Hugging Face - Serverless": ["huggingface", "Hugging Face - Serverless": [
["sentence-transformers/all-MiniLM-L6-v2", "intfloat/multilingual-e5-large", "huggingface",
"intfloat/multilingual-e5-large-instruct", "BAAI/bge-small-en-v1.5", [
"BAAI/bge-base-en-v1.5", "BAAI/bge-large-en-v1.5"]], "sentence-transformers/all-MiniLM-L6-v2",
"Jina AI": ["jinaAI", ["jina-embeddings-v2-base-en", "jina-embeddings-v2-base-de", "jina-embeddings-v2-base-es", "intfloat/multilingual-e5-large",
"jina-embeddings-v2-base-code", "jina-embeddings-v2-base-zh"]], "intfloat/multilingual-e5-large-instruct",
"BAAI/bge-small-en-v1.5",
"BAAI/bge-base-en-v1.5",
"BAAI/bge-large-en-v1.5",
],
],
"Jina AI": [
"jinaAI",
[
"jina-embeddings-v2-base-en",
"jina-embeddings-v2-base-de",
"jina-embeddings-v2-base-es",
"jina-embeddings-v2-base-code",
"jina-embeddings-v2-base-zh",
],
],
"Mistral AI": ["mistral", ["mistral-embed"]], "Mistral AI": ["mistral", ["mistral-embed"]],
"NVIDIA": ["nvidia", ["NV-Embed-QA"]], "NVIDIA": ["nvidia", ["NV-Embed-QA"]],
"OpenAI": ["openai", ["text-embedding-3-small", "text-embedding-3-large", "text-embedding-ada-002"]], "OpenAI": ["openai", ["text-embedding-3-small", "text-embedding-3-large", "text-embedding-ada-002"]],
"Upstage": ["upstageAI", ["solar-embedding-1-large"]], "Upstage": ["upstageAI", ["solar-embedding-1-large"]],
"Voyage AI": ["voyageAI", "Voyage AI": [
["voyage-large-2-instruct", "voyage-law-2", "voyage-code-2", "voyage-large-2", "voyage-2"]] "voyageAI",
["voyage-large-2-instruct", "voyage-law-2", "voyage-code-2", "voyage-large-2", "voyage-2"],
],
} }
VECTORIZE_MODELS_STR = "\n\n".join([provider + ": " + (', '.join(models[1])) for provider, models in VECTORIZE_PROVIDERS_MAPPING.items()]) VECTORIZE_MODELS_STR = "\n\n".join(
[provider + ": " + (", ".join(models[1])) for provider, models in VECTORIZE_PROVIDERS_MAPPING.items()]
)
inputs = [ inputs = [
DropdownInput( DropdownInput(
@ -39,13 +58,13 @@ class AstraVectorize(Component):
name="model_name", name="model_name",
display_name="Model name", display_name="Model name",
info=f"The embedding model to use for the selected provider. Each provider has a different set of models " info=f"The embedding model to use for the selected provider. Each provider has a different set of models "
f"available (full list at https://docs.datastax.com/en/astra-db-serverless/databases/embedding-generation.html):\n\n{VECTORIZE_MODELS_STR}", f"available (full list at https://docs.datastax.com/en/astra-db-serverless/databases/embedding-generation.html):\n\n{VECTORIZE_MODELS_STR}",
required=True required=True,
), ),
MessageTextInput( MessageTextInput(
name="api_key_name", name="api_key_name",
display_name="API Key name", display_name="API Key name",
info="The name of the embeddings provider API key stored on Astra. If set, it will override the 'ProviderKey' in the authentication parameters." info="The name of the embeddings provider API key stored on Astra. If set, it will override the 'ProviderKey' in the authentication parameters.",
), ),
DictInput( DictInput(
name="authentication", name="authentication",

7
src/backend/base/langflow/components/langchain_utilities/FirecrawlCrawlApi.py
View file

@ -4,6 +4,7 @@ from langflow.custom import CustomComponent
from langflow.schema import Data from langflow.schema import Data
import uuid import uuid
class FirecrawlCrawlApi(CustomComponent): class FirecrawlCrawlApi(CustomComponent):
display_name: str = "FirecrawlCrawlApi" display_name: str = "FirecrawlCrawlApi"
description: str = "Firecrawl Crawl API." description: str = "Firecrawl Crawl API."
@ -53,12 +54,12 @@ class FirecrawlCrawlApi(CustomComponent):
idempotency_key: Optional[str] = None, idempotency_key: Optional[str] = None,
) -> Data: ) -> Data:
if crawlerOptions: if crawlerOptions:
crawler_options_dict = crawlerOptions.__dict__['data']['text'] crawler_options_dict = crawlerOptions.__dict__["data"]["text"]
else: else:
crawler_options_dict = {} crawler_options_dict = {}
if pageOptions: if pageOptions:
page_options_dict = pageOptions.__dict__['data']['text'] page_options_dict = pageOptions.__dict__["data"]["text"]
else: else:
page_options_dict = {} page_options_dict = {}
@ -74,7 +75,7 @@ class FirecrawlCrawlApi(CustomComponent):
}, },
True, True,
int(timeout / 1000), int(timeout / 1000),
idempotency_key idempotency_key,
) )
records = Data(data={"results": crawl_result}) records = Data(data={"results": crawl_result})

5
src/backend/base/langflow/components/langchain_utilities/FirecrawlScrapeApi.py
View file

@ -3,6 +3,7 @@ from firecrawl.firecrawl import FirecrawlApp
from langflow.custom import CustomComponent from langflow.custom import CustomComponent
from langflow.schema import Data from langflow.schema import Data
class FirecrawlScrapeApi(CustomComponent): class FirecrawlScrapeApi(CustomComponent):
display_name: str = "FirecrawlScrapeApi" display_name: str = "FirecrawlScrapeApi"
description: str = "Firecrawl Scrape API." description: str = "Firecrawl Scrape API."
@ -47,12 +48,12 @@ class FirecrawlScrapeApi(CustomComponent):
extractorOptions: Optional[Data] = None, extractorOptions: Optional[Data] = None,
) -> Data: ) -> Data:
if extractorOptions: if extractorOptions:
extractor_options_dict = extractorOptions.__dict__['data']['text'] extractor_options_dict = extractorOptions.__dict__["data"]["text"]
else: else:
extractor_options_dict = {} extractor_options_dict = {}
if pageOptions: if pageOptions:
page_options_dict = pageOptions.__dict__['data']['text'] page_options_dict = pageOptions.__dict__["data"]["text"]
else: else:
page_options_dict = {} page_options_dict = {}

20
src/backend/base/langflow/components/vectorstores/Cassandra.py
View file

@ -24,18 +24,20 @@ class CassandraVectorStoreComponent(LCVectorStoreComponent):
icon = "Cassandra" icon = "Cassandra"
inputs = [ inputs = [
MessageTextInput(name="database_ref", MessageTextInput(
display_name="Contact Points / Astra Database ID", name="database_ref",
info="Contact points for the database (or AstraDB database ID)", display_name="Contact Points / Astra Database ID",
required=True), info="Contact points for the database (or AstraDB database ID)",
MessageTextInput(name="username", required=True,
display_name="Username", ),
info="Username for the database (leave empty for AstraDB)."), MessageTextInput(
name="username", display_name="Username", info="Username for the database (leave empty for AstraDB)."
),
SecretStrInput( SecretStrInput(
name="token", name="token",
display_name="Password / AstraDB Token", display_name="Password / AstraDB Token",
info="User password for the database (or AstraDB token).", info="User password for the database (or AstraDB token).",
required=True required=True,
), ),
MessageTextInput( MessageTextInput(
name="keyspace", name="keyspace",
@ -81,7 +83,7 @@ class CassandraVectorStoreComponent(LCVectorStoreComponent):
display_name="Cluster arguments", display_name="Cluster arguments",
info="Optional dictionary of additional keyword arguments for the Cassandra cluster.", info="Optional dictionary of additional keyword arguments for the Cassandra cluster.",
advanced=True, advanced=True,
is_list=True is_list=True,
), ),
MultilineInput(name="search_query", display_name="Search Query"), MultilineInput(name="search_query", display_name="Search Query"),
DataInput( DataInput(