diff --git a/src/backend/base/langflow/components/astra_assistants/create_assistant.py b/src/backend/base/langflow/components/astra_assistants/create_assistant.py
index 50df68e78..37725a93f 100644
--- a/src/backend/base/langflow/components/astra_assistants/create_assistant.py
+++ b/src/backend/base/langflow/components/astra_assistants/create_assistant.py
@@ -1,6 +1,9 @@
-from langflow.custom import CustomComponent
+from typing import Optional
+
+from astra_assistants import patch  # type: ignore
 from openai import OpenAI
-from astra_assistants import patch
+
+from langflow.custom import CustomComponent
 
 
 class AssistantsCreateAssistant(CustomComponent):
@@ -35,8 +38,7 @@ class AssistantsCreateAssistant(CustomComponent):
             },
         }
 
-    def build(self, name: str, instructions: str, model: str, env_set: str = None) -> str:
-        print(f"env_set is {env_set}")
+    def build(self, name: str, instructions: str, model: str, env_set: Optional[str] = None) -> str:
         if env_set is None:
             raise Exception("Environment variables not set")
         client = patch(OpenAI())
diff --git a/src/backend/base/langflow/components/astra_assistants/create_thread.py b/src/backend/base/langflow/components/astra_assistants/create_thread.py
index 5c6b26aa2..3d68e6e3e 100644
--- a/src/backend/base/langflow/components/astra_assistants/create_thread.py
+++ b/src/backend/base/langflow/components/astra_assistants/create_thread.py
@@ -1,6 +1,9 @@
-from langflow.custom import CustomComponent
+from typing import Optional
+
+from astra_assistants import patch  # type: ignore
 from openai import OpenAI
-from astra_assistants import patch
+
+from langflow.custom import CustomComponent
 
 
 class AssistantsCreateThread(CustomComponent):
@@ -16,7 +19,7 @@ class AssistantsCreateThread(CustomComponent):
             },
         }
 
-    def build(self, env_set: str = None) -> str:
+    def build(self, env_set: Optional[str] = None) -> str:
         client = patch(OpenAI())
 
         thread = client.beta.threads.create()
diff --git a/src/backend/base/langflow/components/astra_assistants/dotenv.py b/src/backend/base/langflow/components/astra_assistants/dotenv.py
index d4abf8147..d9fc27a85 100644
--- a/src/backend/base/langflow/components/astra_assistants/dotenv.py
+++ b/src/backend/base/langflow/components/astra_assistants/dotenv.py
@@ -1,5 +1,7 @@
 import io
+
 from dotenv import load_dotenv
+
 from langflow.custom import CustomComponent
 
 
@@ -23,6 +25,6 @@ class Dotenv(CustomComponent):
         try:
             fake_file = io.StringIO(dotenv_file_content)
             result = load_dotenv(stream=fake_file, override=True)
-            return result
+            return "Loaded .env" if result else "No variables found in .env"
         except Exception as e:
             raise e
diff --git a/src/backend/base/langflow/components/astra_assistants/get_assistant.py b/src/backend/base/langflow/components/astra_assistants/get_assistant.py
index ab8b7f0da..0d2d193da 100644
--- a/src/backend/base/langflow/components/astra_assistants/get_assistant.py
+++ b/src/backend/base/langflow/components/astra_assistants/get_assistant.py
@@ -1,6 +1,8 @@
-from langflow.custom import CustomComponent
+from typing import Optional
+from astra_assistants import patch  # type: ignore
 from openai import OpenAI
-from astra_assistants import patch
+
+from langflow.custom import CustomComponent
 
 
 class AssistantsGetAssistantName(CustomComponent):
@@ -20,7 +22,7 @@ class AssistantsGetAssistantName(CustomComponent):
             },
         }
 
-    def build(self, assistant_id: str, env_set: str = None) -> str:
+    def build(self, assistant_id: str, env_set: Optional[str] = None) -> str:
         client = patch(OpenAI())
         assistant = client.beta.assistants.retrieve(
             assistant_id=assistant_id,
diff --git a/src/backend/base/langflow/components/astra_assistants/list_assistants.py b/src/backend/base/langflow/components/astra_assistants/list_assistants.py
index e6608a63c..f0cb5ba5f 100644
--- a/src/backend/base/langflow/components/astra_assistants/list_assistants.py
+++ b/src/backend/base/langflow/components/astra_assistants/list_assistants.py
@@ -1,7 +1,9 @@
 from typing import List
-from langflow.custom import CustomComponent
+
+from astra_assistants import patch  # type: ignore
 from openai import OpenAI
-from astra_assistants import patch
+
+from langflow.custom import CustomComponent
 
 
 class AssistantsListAssistants(CustomComponent):
diff --git a/src/backend/base/langflow/components/astra_assistants/run.py b/src/backend/base/langflow/components/astra_assistants/run.py
index 5afaaf716..d50246312 100644
--- a/src/backend/base/langflow/components/astra_assistants/run.py
+++ b/src/backend/base/langflow/components/astra_assistants/run.py
@@ -1,7 +1,9 @@
-from langflow.custom import CustomComponent
+from typing import Optional
+from astra_assistants import patch  # type: ignore
 from openai import OpenAI
 from openai.lib.streaming import AssistantEventHandler
-from astra_assistants import patch
+
+from langflow.custom import CustomComponent
 
 
 class AssistantsRun(CustomComponent):
@@ -35,7 +37,9 @@ class AssistantsRun(CustomComponent):
             },
         }
 
-    def build(self, assistant_id: str, user_message: str, thread_id: str = None, env_set: str = None) -> str:
+    def build(
+        self, assistant_id: str, user_message: str, thread_id: Optional[str] = None, env_set: Optional[str] = None
+    ) -> str:
         text = ""
         client = patch(OpenAI())