fix: Allow basic Astra DB Collection creation in combobox (#5958)
* fix: Allow basic creation colleciton in combobox * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes (attempt 2/3) * Add new collection to list if created * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
parent
cc592e8ad1
commit
76764b52d8
2 changed files with 161 additions and 159 deletions
|
|
@ -478,9 +478,10 @@ class AstraDBVectorStoreComponent(LCVectorStoreComponent):
|
||||||
# Define variables for common database conditions a user may experience
|
# Define variables for common database conditions a user may experience
|
||||||
is_hosted = os.getenv("LANGFLOW_HOST") is not None
|
is_hosted = os.getenv("LANGFLOW_HOST") is not None
|
||||||
no_databases = "options" not in build_config["api_endpoint"] or not build_config["api_endpoint"]["options"]
|
no_databases = "options" not in build_config["api_endpoint"] or not build_config["api_endpoint"]["options"]
|
||||||
|
no_api_endpoint = not build_config["api_endpoint"]["value"]
|
||||||
|
|
||||||
# Refresh the database name options
|
# Refresh the database name options
|
||||||
if not is_hosted and (field_name in ["token", "environment"] or no_databases):
|
if not is_hosted and (field_name in ["token", "environment"] or (no_databases and no_api_endpoint)):
|
||||||
# Get the list of options we have based on the token provided
|
# Get the list of options we have based on the token provided
|
||||||
database_options = self._initialize_database_options()
|
database_options = self._initialize_database_options()
|
||||||
|
|
||||||
|
|
@ -524,11 +525,8 @@ class AstraDBVectorStoreComponent(LCVectorStoreComponent):
|
||||||
] = self.map_cloud_providers()[cloud_provider]["regions"]
|
] = self.map_cloud_providers()[cloud_provider]["regions"]
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# Define variables for common collection conditions a user may experience
|
|
||||||
no_collections = not build_config["collection_name"]["options"]
|
|
||||||
|
|
||||||
# Refresh the collection name options
|
# Refresh the collection name options
|
||||||
if field_name == "api_endpoint" or (field_name == "collection_name" and no_collections):
|
if field_name == "api_endpoint":
|
||||||
# Reset the selected collection
|
# Reset the selected collection
|
||||||
build_config["collection_name"]["value"] = ""
|
build_config["collection_name"]["value"] = ""
|
||||||
|
|
||||||
|
|
@ -539,11 +537,15 @@ class AstraDBVectorStoreComponent(LCVectorStoreComponent):
|
||||||
{k: v for k, v in col.items() if k not in ["name"]} for col in collection_options
|
{k: v for k, v in col.items() if k not in ["name"]} for col in collection_options
|
||||||
]
|
]
|
||||||
|
|
||||||
# Define variables for common collection choice conditions a user may experience
|
|
||||||
collection_chosen = field_value and build_config["collection_name"]["options"]
|
|
||||||
|
|
||||||
# Hide embedding model option if opriona_metadata provider is not null
|
# Hide embedding model option if opriona_metadata provider is not null
|
||||||
if field_name == "collection_name" and collection_chosen:
|
if field_name == "collection_name" and field_value:
|
||||||
|
# Set the options for collection name to be the field value if its a new collection
|
||||||
|
if not is_hosted and field_value not in build_config["collection_name"]["options"]:
|
||||||
|
build_config["collection_name"]["options"].append(field_value)
|
||||||
|
build_config["collection_name"]["options_metadata"].append(
|
||||||
|
{"records": 0, "provider": None, "icon": "", "model": None}
|
||||||
|
)
|
||||||
|
|
||||||
# Find location of the name in the options list
|
# Find location of the name in the options list
|
||||||
index_of_name = build_config["collection_name"]["options"].index(field_value)
|
index_of_name = build_config["collection_name"]["options"].index(field_value)
|
||||||
value_of_provider = build_config["collection_name"]["options_metadata"][index_of_name]["provider"]
|
value_of_provider = build_config["collection_name"]["options_metadata"][index_of_name]["provider"]
|
||||||
|
|
|
||||||
File diff suppressed because one or more lines are too long
Loading…
Add table
Add a link
Reference in a new issue