From e1816246b53c9d8fdcb026ec5c3681c693792e1a Mon Sep 17 00:00:00 2001
From: Mike Fortman <michael.fortman@datastax.com>
Date: Fri, 27 Sep 2024 16:00:31 -0500
Subject: [PATCH] feat: add env support for Astra Db components (#3908)

* add env support for Astra Db components

* [autofix.ci] apply automated fixes

* use parse_api_endpoint

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes

* quick cleanup

* merge

* package lock

* reset templates

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
---
 .../base/langflow/components/memories/AstraDBChatMemory.py     | 3 +++
 src/backend/base/langflow/components/vectorstores/AstraDB.py   | 2 ++
 src/frontend/package-lock.json                                 | 1 +
 3 files changed, 6 insertions(+)

diff --git a/src/backend/base/langflow/components/memories/AstraDBChatMemory.py b/src/backend/base/langflow/components/memories/AstraDBChatMemory.py
index 7a528a7ba..3141a52e6 100644
--- a/src/backend/base/langflow/components/memories/AstraDBChatMemory.py
+++ b/src/backend/base/langflow/components/memories/AstraDBChatMemory.py
@@ -1,5 +1,7 @@
 import os
 
+from astrapy.admin import parse_api_endpoint
+
 from langflow.base.memory.model import LCChatMemoryComponent
 from langflow.field_typing import BaseChatMessageHistory
 from langflow.inputs import MessageTextInput, SecretStrInput, StrInput
@@ -62,5 +64,6 @@ class AstraDBChatMemory(LCChatMemoryComponent):
             token=self.token,
             api_endpoint=self.api_endpoint,
             namespace=self.namespace or None,
+            environment=parse_api_endpoint(self.api_endpoint).environment,
         )
         return memory
diff --git a/src/backend/base/langflow/components/vectorstores/AstraDB.py b/src/backend/base/langflow/components/vectorstores/AstraDB.py
index eace7d0d3..2e151c8b8 100644
--- a/src/backend/base/langflow/components/vectorstores/AstraDB.py
+++ b/src/backend/base/langflow/components/vectorstores/AstraDB.py
@@ -1,5 +1,6 @@
 import os
 
+from astrapy.admin import parse_api_endpoint
 from loguru import logger
 
 from langflow.base.vectorstores.model import LCVectorStoreComponent, check_cached_vector_store
@@ -397,6 +398,7 @@ class AstraVectorStoreComponent(LCVectorStoreComponent):
             "token": self.token,
             "api_endpoint": self.api_endpoint,
             "namespace": self.namespace or None,
+            "environment": parse_api_endpoint(self.api_endpoint).environment,
             "metric": self.metric or None,
             "batch_size": self.batch_size or None,
             "bulk_insert_batch_concurrency": self.bulk_insert_batch_concurrency or None,
diff --git a/src/frontend/package-lock.json b/src/frontend/package-lock.json
index e54783c66..b07ae39d7 100644
--- a/src/frontend/package-lock.json
+++ b/src/frontend/package-lock.json
@@ -858,6 +858,7 @@
     },
     "node_modules/@clack/prompts/node_modules/is-unicode-supported": {
       "version": "1.3.0",
+      "extraneous": true,
       "inBundle": true,
       "license": "MIT",
       "engines": {