lfx-nightly 0.1.13.dev2__py3-none-any.whl → 0.1.13.dev4__py3-none-any.whl

lfx/components/datastax/create_assistant.py

@@ -10,6 +10,7 @@ class AssistantsCreateAssistant(ComponentWithCache):
     icon = "AstraDB"
     display_name = "Create Assistant"
     description = "Creates an Assistant and returns it's id"
+    legacy = True
 
     inputs = [
         StrInput(

lfx/components/datastax/create_thread.py

@@ -9,6 +9,7 @@ class AssistantsCreateThread(ComponentWithCache):
     display_name = "Create Assistant Thread"
     description = "Creates a thread and returns the thread id"
     icon = "AstraDB"
+    legacy = True
     inputs = [
         MultilineInput(
             name="env_set",

lfx/components/datastax/dotenv.py

@@ -12,6 +12,7 @@ class Dotenv(Component):
     display_name = "Dotenv"
     description = "Load .env file into env vars"
     icon = "AstraDB"
+    legacy = True
     inputs = [
         MultilineSecretInput(
             name="dotenv_file_content",

lfx/components/datastax/get_assistant.py

@@ -9,6 +9,7 @@ class AssistantsGetAssistantName(ComponentWithCache):
     display_name = "Get Assistant name"
     description = "Assistant by id"
     icon = "AstraDB"
+    legacy = True
     inputs = [
         StrInput(
             name="assistant_id",

lfx/components/datastax/getenvvar.py

@@ -10,6 +10,7 @@ class GetEnvVar(Component):
     display_name = "Get Environment Variable"
     description = "Gets the value of an environment variable from the system."
     icon = "AstraDB"
+    legacy = True
 
     inputs = [
         StrInput(

lfx/components/datastax/graph_rag.py

@@ -44,7 +44,7 @@ class GraphRAGComponent(LCVectorStoreComponent):
 
     display_name: str = "Graph RAG"
     description: str = "Graph RAG traversal for vector store."
-    name = "Graph RAG"
+    name = "GraphRAG"
     icon: str = "AstraDB"
 
     inputs = [

lfx/components/datastax/list_assistants.py

@@ -8,6 +8,7 @@ class AssistantsListAssistants(ComponentWithCache):
     display_name = "List Assistants"
     description = "Returns a list of assistant id's"
     icon = "AstraDB"
+    legacy = True
     outputs = [
         Output(display_name="Assistants", name="assistants", method="process_inputs"),
     ]

lfx/components/datastax/run.py

@@ -14,6 +14,7 @@ class AssistantsRun(ComponentWithCache):
     display_name = "Run Assistant"
     description = "Executes an Assistant Run against a thread"
     icon = "AstraDB"
+    legacy = True
 
     def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)

lfx/components/knowledge_bases/ingestion.py

@@ -48,12 +48,20 @@ HUGGINGFACE_MODEL_NAMES = [
 ]
 COHERE_MODEL_NAMES = ["embed-english-v3.0", "embed-multilingual-v3.0"]
 
-settings = get_settings_service().settings
-knowledge_directory = settings.knowledge_bases_dir
-if not knowledge_directory:
-    msg = "Knowledge bases directory is not set in the settings."
-    raise ValueError(msg)
-KNOWLEDGE_BASES_ROOT_PATH = Path(knowledge_directory).expanduser()
+_KNOWLEDGE_BASES_ROOT_PATH: Path | None = None
+
+
+def _get_knowledge_bases_root_path() -> Path:
+    """Lazy load the knowledge bases root path from settings."""
+    global _KNOWLEDGE_BASES_ROOT_PATH  # noqa: PLW0603
+    if _KNOWLEDGE_BASES_ROOT_PATH is None:
+        settings = get_settings_service().settings
+        knowledge_directory = settings.knowledge_bases_dir
+        if not knowledge_directory:
+            msg = "Knowledge bases directory is not set in the settings."
+            raise ValueError(msg)
+        _KNOWLEDGE_BASES_ROOT_PATH = Path(knowledge_directory).expanduser()
+    return _KNOWLEDGE_BASES_ROOT_PATH
 
 
 class KnowledgeIngestionComponent(Component):
@@ -203,7 +211,7 @@ class KnowledgeIngestionComponent(Component):
     # ------ Internal helpers ---------------------------------------------
     def _get_kb_root(self) -> Path:
         """Return the root directory for knowledge bases."""
-        return KNOWLEDGE_BASES_ROOT_PATH
+        return _get_knowledge_bases_root_path()
 
     def _validate_column_config(self, df_source: pd.DataFrame) -> list[dict[str, Any]]:
         """Validate column configuration using Structured Output patterns."""
@@ -662,7 +670,7 @@ class KnowledgeIngestionComponent(Component):
                     raise ValueError(msg) from e
 
                 # Create the new knowledge base directory
-                kb_path = KNOWLEDGE_BASES_ROOT_PATH / kb_user / field_value["01_new_kb_name"]
+                kb_path = _get_knowledge_bases_root_path() / kb_user / field_value["01_new_kb_name"]
                 kb_path.mkdir(parents=True, exist_ok=True)
 
                 # Save the embedding metadata
@@ -675,7 +683,7 @@ class KnowledgeIngestionComponent(Component):
 
             # Update the knowledge base options dynamically
             build_config["knowledge_base"]["options"] = await get_knowledge_bases(
-                KNOWLEDGE_BASES_ROOT_PATH,
+                _get_knowledge_bases_root_path(),
                 user_id=self.user_id,
             )
 

lfx/components/knowledge_bases/retrieval.py

@@ -16,12 +16,20 @@ from lfx.schema.data import Data
 from lfx.schema.dataframe import DataFrame
 from lfx.services.deps import get_settings_service, session_scope
 
-settings = get_settings_service().settings
-knowledge_directory = settings.knowledge_bases_dir
-if not knowledge_directory:
-    msg = "Knowledge bases directory is not set in the settings."
-    raise ValueError(msg)
-KNOWLEDGE_BASES_ROOT_PATH = Path(knowledge_directory).expanduser()
+_KNOWLEDGE_BASES_ROOT_PATH: Path | None = None
+
+
+def _get_knowledge_bases_root_path() -> Path:
+    """Lazy load the knowledge bases root path from settings."""
+    global _KNOWLEDGE_BASES_ROOT_PATH  # noqa: PLW0603
+    if _KNOWLEDGE_BASES_ROOT_PATH is None:
+        settings = get_settings_service().settings
+        knowledge_directory = settings.knowledge_bases_dir
+        if not knowledge_directory:
+            msg = "Knowledge bases directory is not set in the settings."
+            raise ValueError(msg)
+        _KNOWLEDGE_BASES_ROOT_PATH = Path(knowledge_directory).expanduser()
+    return _KNOWLEDGE_BASES_ROOT_PATH
 
 
 class KnowledgeRetrievalComponent(Component):
@@ -90,7 +98,7 @@ class KnowledgeRetrievalComponent(Component):
         if field_name == "knowledge_base":
             # Update the knowledge base options dynamically
             build_config["knowledge_base"]["options"] = await get_knowledge_bases(
-                KNOWLEDGE_BASES_ROOT_PATH,
+                _get_knowledge_bases_root_path(),
                 user_id=self.user_id,  # Use the user_id from the component context
             )
 
@@ -186,7 +194,7 @@ class KnowledgeRetrievalComponent(Component):
                 msg = f"User with ID {self.user_id} not found."
                 raise ValueError(msg)
             kb_user = current_user.username
-        kb_path = KNOWLEDGE_BASES_ROOT_PATH / kb_user / self.knowledge_base
+        kb_path = _get_knowledge_bases_root_path() / kb_user / self.knowledge_base
 
         metadata = self._get_kb_metadata(kb_path)
         if not metadata:

lfx/components/mistral/mistral_embeddings.py

@@ -1,4 +1,4 @@
-from langchain_mistralai.embeddings import MistralAIEmbeddings
+from langchain_mistralai import MistralAIEmbeddings
 from pydantic.v1 import SecretStr
 
 from lfx.base.models.model import LCModelComponent

lfx/components/openrouter/openrouter.py

@@ -1,6 +1,3 @@
-from collections import defaultdict
-from typing import Any
-
 import httpx
 from langchain_openai import ChatOpenAI
 from pydantic.v1 import SecretStr
@@ -8,13 +5,7 @@ from pydantic.v1 import SecretStr
 from lfx.base.models.model import LCModelComponent
 from lfx.field_typing import LanguageModel
 from lfx.field_typing.range_spec import RangeSpec
-from lfx.inputs.inputs import (
-    DropdownInput,
-    IntInput,
-    SecretStrInput,
-    SliderInput,
-    StrInput,
-)
+from lfx.inputs.inputs import DropdownInput, IntInput, SecretStrInput, SliderInput, StrInput
 
 
 class OpenRouterComponent(LCModelComponent):
@@ -28,36 +19,13 @@ class OpenRouterComponent(LCModelComponent):
 
     inputs = [
         *LCModelComponent.get_base_inputs(),
-        SecretStrInput(
-            name="api_key", display_name="OpenRouter API Key", required=True, info="Your OpenRouter API key"
-        ),
-        StrInput(
-            name="site_url",
-            display_name="Site URL",
-            info="Your site URL for OpenRouter rankings",
-            advanced=True,
-        ),
-        StrInput(
-            name="app_name",
-            display_name="App Name",
-            info="Your app name for OpenRouter rankings",
-            advanced=True,
-        ),
-        DropdownInput(
-            name="provider",
-            display_name="Provider",
-            info="The AI model provider",
-            options=["Loading providers..."],
-            value="Loading providers...",
-            real_time_refresh=True,
-            required=True,
-        ),
+        SecretStrInput(name="api_key", display_name="API Key", required=True),
         DropdownInput(
             name="model_name",
             display_name="Model",
-            info="The model to use for chat completion",
-            options=["Select a provider first"],
-            value="Select a provider first",
+            options=[],
+            value="",
+            refresh_button=True,
             real_time_refresh=True,
             required=True,
         ),
@@ -66,137 +34,71 @@ class OpenRouterComponent(LCModelComponent):
             display_name="Temperature",
             value=0.7,
             range_spec=RangeSpec(min=0, max=2, step=0.01),
-            info="Controls randomness. Lower values are more deterministic, higher values are more creative.",
-            advanced=True,
-        ),
-        IntInput(
-            name="max_tokens",
-            display_name="Max Tokens",
-            info="Maximum number of tokens to generate",
             advanced=True,
         ),
+        IntInput(name="max_tokens", display_name="Max Tokens", advanced=True),
+        StrInput(name="site_url", display_name="Site URL", advanced=True),
+        StrInput(name="app_name", display_name="App Name", advanced=True),
     ]
 
-    def fetch_models(self) -> dict[str, list]:
-        """Fetch available models from OpenRouter API and organize them by provider."""
-        url = "https://openrouter.ai/api/v1/models"
-
+    def fetch_models(self) -> list[dict]:
+        """Fetch available models from OpenRouter."""
         try:
-            with httpx.Client() as client:
-                response = client.get(url)
-                response.raise_for_status()
-
-                models_data = response.json().get("data", [])
-                provider_models = defaultdict(list)
-
-                for model in models_data:
-                    model_id = model.get("id", "")
-                    if "/" in model_id:
-                        provider = model_id.split("/")[0].title()
-                        provider_models[provider].append(
-                            {
-                                "id": model_id,
-                                "name": model.get("name", ""),
-                                "description": model.get("description", ""),
-                                "context_length": model.get("context_length", 0),
-                            }
-                        )
-
-                return dict(provider_models)
-
-        except httpx.HTTPError as e:
-            self.log(f"Error fetching models: {e!s}")
-            return {"Error": [{"id": "error", "name": f"Error fetching models: {e!s}"}]}
+            response = httpx.get("https://openrouter.ai/api/v1/models", timeout=10.0)
+            response.raise_for_status()
+            models = response.json().get("data", [])
+            return sorted(
+                [
+                    {
+                        "id": m["id"],
+                        "name": m.get("name", m["id"]),
+                        "context": m.get("context_length", 0),
+                    }
+                    for m in models
+                    if m.get("id")
+                ],
+                key=lambda x: x["name"],
+            )
+        except (httpx.RequestError, httpx.HTTPStatusError) as e:
+            self.log(f"Error fetching models: {e}")
+            return []
+
+    def update_build_config(self, build_config: dict, field_value: str, field_name: str | None = None) -> dict:  # noqa: ARG002
+        """Update model options."""
+        models = self.fetch_models()
+        if models:
+            build_config["model_name"]["options"] = [m["id"] for m in models]
+            build_config["model_name"]["tooltips"] = {m["id"]: f"{m['name']} ({m['context']:,} tokens)" for m in models}
+        else:
+            build_config["model_name"]["options"] = ["Failed to load models"]
+            build_config["model_name"]["value"] = "Failed to load models"
+        return build_config
 
     def build_model(self) -> LanguageModel:
-        """Build and return the OpenRouter language model."""
-        model_not_selected = "Please select a model"
-        api_key_required = "API key is required"
-
-        if not self.model_name or self.model_name == "Select a provider first":
-            raise ValueError(model_not_selected)
-
+        """Build the OpenRouter model."""
         if not self.api_key:
-            raise ValueError(api_key_required)
-
-        api_key = SecretStr(self.api_key).get_secret_value()
+            msg = "API key is required"
+            raise ValueError(msg)
+        if not self.model_name or self.model_name == "Loading...":
+            msg = "Please select a model"
+            raise ValueError(msg)
 
-        # Build base configuration
-        kwargs: dict[str, Any] = {
+        kwargs = {
             "model": self.model_name,
-            "openai_api_key": api_key,
+            "openai_api_key": SecretStr(self.api_key).get_secret_value(),
             "openai_api_base": "https://openrouter.ai/api/v1",
             "temperature": self.temperature if self.temperature is not None else 0.7,
         }
 
-        # Add optional parameters
         if self.max_tokens:
-            kwargs["max_tokens"] = self.max_tokens
+            kwargs["max_tokens"] = int(self.max_tokens)
 
         headers = {}
         if self.site_url:
             headers["HTTP-Referer"] = self.site_url
         if self.app_name:
             headers["X-Title"] = self.app_name
-
         if headers:
             kwargs["default_headers"] = headers
 
-        try:
-            return ChatOpenAI(**kwargs)
-        except (ValueError, httpx.HTTPError) as err:
-            error_msg = f"Failed to build model: {err!s}"
-            self.log(error_msg)
-            raise ValueError(error_msg) from err
-
-    def _get_exception_message(self, e: Exception) -> str | None:
-        """Get a message from an OpenRouter exception.
-
-        Args:
-            e (Exception): The exception to get the message from.
-
-        Returns:
-            str | None: The message from the exception, or None if no specific message can be extracted.
-        """
-        try:
-            from openai import BadRequestError
-
-            if isinstance(e, BadRequestError):
-                message = e.body.get("message")
-                if message:
-                    return message
-        except ImportError:
-            pass
-        return None
-
-    def update_build_config(self, build_config: dict, field_value: str, field_name: str | None = None) -> dict:
-        """Update build configuration based on field updates."""
-        try:
-            if field_name is None or field_name == "provider":
-                provider_models = self.fetch_models()
-                build_config["provider"]["options"] = sorted(provider_models.keys())
-                if build_config["provider"]["value"] not in provider_models:
-                    build_config["provider"]["value"] = build_config["provider"]["options"][0]
-
-            if field_name == "provider" and field_value in self.fetch_models():
-                provider_models = self.fetch_models()
-                models = provider_models[field_value]
-
-                build_config["model_name"]["options"] = [model["id"] for model in models]
-                if models:
-                    build_config["model_name"]["value"] = models[0]["id"]
-
-                tooltips = {
-                    model["id"]: (f"{model['name']}\nContext Length: {model['context_length']}\n{model['description']}")
-                    for model in models
-                }
-                build_config["model_name"]["tooltips"] = tooltips
-
-        except httpx.HTTPError as e:
-            self.log(f"Error updating build config: {e!s}")
-            build_config["provider"]["options"] = ["Error loading providers"]
-            build_config["provider"]["value"] = "Error loading providers"
-            build_config["model_name"]["options"] = ["Error loading models"]
-            build_config["model_name"]["value"] = "Error loading models"
-
-        return build_config
+        return ChatOpenAI(**kwargs)

lfx/components/vectorstores/__init__.py

@@ -5,20 +5,14 @@ from typing import TYPE_CHECKING, Any
 from lfx.components._importing import import_mod
 
 if TYPE_CHECKING:
-    from .astradb import AstraDBVectorStoreComponent
     from .local_db import LocalDBComponent
-    from .mongodb_atlas import MongoVectorStoreComponent
 
 _dynamic_imports = {
     "LocalDBComponent": "local_db",
-    "AstraDBVectorStoreComponent": "astradb",
-    "MongoVectorStoreComponent": "mongodb_atlas",
 }
 
 __all__ = [
-    "AstraDBVectorStoreComponent",
     "LocalDBComponent",
-    "MongoVectorStoreComponent",
 ]
 
 

lfx/custom/custom_component/component.py

@@ -154,7 +154,7 @@ class Component(CustomComponent):
             self.trace_type = "chain"
 
         # Setup inputs and outputs
-        self._reset_all_output_values()
+        self.reset_all_output_values()
         if self.inputs is not None:
             self.map_inputs(self.inputs)
         self.map_outputs()
@@ -330,7 +330,8 @@ class Component(CustomComponent):
     def set_event_manager(self, event_manager: EventManager | None = None) -> None:
         self._event_manager = event_manager
 
-    def _reset_all_output_values(self) -> None:
+    def reset_all_output_values(self) -> None:
+        """Reset all output values to UNDEFINED."""
         if isinstance(self._outputs_map, dict):
             for output in self._outputs_map.values():
                 output.value = UNDEFINED

lfx/graph/edge/base.py

@@ -63,8 +63,8 @@ class Edge:
             # target_param is documents
             if isinstance(self._target_handle, str):
                 self.target_param = self._target_handle.split("|")[1]
-                self.source_handle = None
-                self.target_handle = None
+                self.source_handle = None  # type: ignore[assignment]
+                self.target_handle = None  # type: ignore[assignment]
             else:
                 msg = "Target handle is not a string"
                 raise ValueError(msg)

lfx/graph/graph/base.py

@@ -402,7 +402,7 @@ class Graph:
         for vertex in self.vertices:
             if vertex.custom_component is None:
                 continue
-            vertex.custom_component._reset_all_output_values()
+            vertex.custom_component.reset_all_output_values()
 
     def start(
         self,

lfx/graph/graph/schema.py

@@ -4,11 +4,12 @@ from typing import TYPE_CHECKING, NamedTuple, Protocol
 
 from typing_extensions import NotRequired, TypedDict
 
+from lfx.graph.edge.schema import EdgeData
+from lfx.graph.vertex.schema import NodeData
+
 if TYPE_CHECKING:
-    from lfx.graph.edge.schema import EdgeData
     from lfx.graph.schema import ResultData
     from lfx.graph.vertex.base import Vertex
-    from lfx.graph.vertex.schema import NodeData
     from lfx.schema.log import LoggableType
 
 

lfx/graph/vertex/vertex_types.py

@@ -65,7 +65,7 @@ class ComponentVertex(Vertex):
                 self.built_object, self.artifacts = result
             elif len(result) == 3:  # noqa: PLR2004
                 self.custom_component, self.built_object, self.artifacts = result
-                self.logs = self.custom_component._output_logs
+                self.logs = self.custom_component.get_output_logs()
                 for key in self.artifacts:
                     if self.artifacts_raw is None:
                         self.artifacts_raw = {}

{lfx_nightly-0.1.13.dev2.dist-info → lfx_nightly-0.1.13.dev4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lfx-nightly
-Version: 0.1.13.dev2
+Version: 0.1.13.dev4
 Summary: Langflow Executor - A lightweight CLI tool for executing and serving Langflow AI flows
 Author-email: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
 Requires-Python: <3.14,>=3.10