From e38af369e51bfe5e07224a8b97e51c6460f72a0f Mon Sep 17 00:00:00 2001 From: Eric Hare Date: Wed, 27 Nov 2024 16:39:23 -0800 Subject: [PATCH] fix: Use global var if available for api endpoint (#4909) * fix: Use global var if available for api endpoint * Update astradb.py --- src/backend/base/langflow/components/tools/astradb.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/tools/astradb.py b/src/backend/base/langflow/components/tools/astradb.py index 2153b8082..432add9de 100644 --- a/src/backend/base/langflow/components/tools/astradb.py +++ b/src/backend/base/langflow/components/tools/astradb.py @@ -1,3 +1,4 @@ +import os from typing import Any from astrapy import Collection, DataAPIClient, Database @@ -10,7 +11,7 @@ from langflow.schema import Data class AstraDBToolComponent(LCToolComponent): - display_name: str = "Astra DB" + display_name: str = "Astra DB Tool" description: str = "Create a tool to get transactional data from DataStax Astra DB Collection" documentation: str = "https://docs.langflow.org/Components/components-tools#astra-db-tool" icon: str = "AstraDB" @@ -48,9 +49,9 @@ class AstraDBToolComponent(LCToolComponent): value="ASTRA_DB_APPLICATION_TOKEN", required=True, ), - StrInput( + SecretStrInput( name="api_endpoint", - display_name="API Endpoint", + display_name="Database" if os.getenv("ASTRA_ENHANCED", "false").lower() == "true" else "API Endpoint", info="API endpoint URL for the Astra DB service.", value="ASTRA_DB_API_ENDPOINT", required=True,