lfx-cohere 0.1.0__py3-none-any.whl

lfx_cohere/__init__.py

@@ -0,0 +1,17 @@
+"""lfx-cohere: Cohere bundle.
+
+Distribution unit ``lfx-cohere``.  At runtime Langflow's loader
+discovers ``extension.json`` shipped alongside this ``__init__.py`` and
+registers the bundle's components under the namespaced IDs
+``ext:cohere:<Class>@official``.
+"""
+
+from lfx_cohere.components.cohere.cohere_embeddings import CohereEmbeddingsComponent
+from lfx_cohere.components.cohere.cohere_models import CohereComponent
+from lfx_cohere.components.cohere.cohere_rerank import CohereRerankComponent
+
+__all__ = [
+    "CohereComponent",
+    "CohereEmbeddingsComponent",
+    "CohereRerankComponent",
+]

lfx_cohere/components/cohere/__init__.py

@@ -0,0 +1,48 @@
+"""Lazy component re-exports for the ``cohere`` bundle.
+
+Mirrors the pre-extraction layout of ``lfx.components.cohere`` so saved
+flows that referenced the module-level class
+(e.g. ``lfx.components.cohere.<Class>``) keep resolving via the
+migration table after rewrite to
+``lfx_cohere.components.cohere.<Class>``.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from lfx.utils.lazy_import import import_mod
+
+if TYPE_CHECKING:
+    from .cohere_embeddings import CohereEmbeddingsComponent
+    from .cohere_models import CohereComponent
+    from .cohere_rerank import CohereRerankComponent
+
+_dynamic_imports = {
+    "CohereComponent": "cohere_models",
+    "CohereEmbeddingsComponent": "cohere_embeddings",
+    "CohereRerankComponent": "cohere_rerank",
+}
+
+__all__ = [
+    "CohereComponent",
+    "CohereEmbeddingsComponent",
+    "CohereRerankComponent",
+]
+
+
+def __getattr__(attr_name: str) -> Any:
+    if attr_name not in _dynamic_imports:
+        msg = f"module {__name__!r} has no attribute {attr_name!r}"
+        raise AttributeError(msg)
+    try:
+        result = import_mod(attr_name, _dynamic_imports[attr_name], __spec__.parent)
+    except (ModuleNotFoundError, ImportError, AttributeError) as e:
+        msg = f"Could not import {attr_name!r} from {__name__!r}: {e}"
+        raise AttributeError(msg) from e
+    globals()[attr_name] = result
+    return result
+
+
+def __dir__() -> list[str]:
+    return list(__all__)

lfx_cohere/components/cohere/cohere_embeddings.py

@@ -0,0 +1,80 @@
+from typing import Any
+
+import cohere
+from langchain_cohere import CohereEmbeddings
+from lfx.base.models.model import LCModelComponent
+from lfx.field_typing import Embeddings
+from lfx.io import DropdownInput, FloatInput, IntInput, MessageTextInput, Output, SecretStrInput
+
+HTTP_STATUS_OK = 200
+
+
+class CohereEmbeddingsComponent(LCModelComponent):
+    display_name = "Cohere Embeddings"
+    description = "Generate embeddings using Cohere models."
+    icon = "Cohere"
+    name = "CohereEmbeddings"
+
+    inputs = [
+        SecretStrInput(name="api_key", display_name="Cohere API Key", required=True, real_time_refresh=True),
+        DropdownInput(
+            name="model_name",
+            display_name="Model",
+            advanced=False,
+            options=[
+                "embed-english-v2.0",
+                "embed-multilingual-v2.0",
+                "embed-english-light-v2.0",
+                "embed-multilingual-light-v2.0",
+            ],
+            value="embed-english-v2.0",
+            refresh_button=True,
+            combobox=True,
+        ),
+        MessageTextInput(name="truncate", display_name="Truncate", advanced=True),
+        IntInput(name="max_retries", display_name="Max Retries", value=3, advanced=True),
+        MessageTextInput(name="user_agent", display_name="User Agent", advanced=True, value="langchain"),
+        FloatInput(name="request_timeout", display_name="Request Timeout", advanced=True),
+    ]
+
+    outputs = [
+        Output(display_name="Embeddings", name="embeddings", method="build_embeddings"),
+    ]
+
+    def build_embeddings(self) -> Embeddings:
+        data = None
+        try:
+            data = CohereEmbeddings(
+                cohere_api_key=self.api_key,
+                model=self.model_name,
+                truncate=self.truncate,
+                max_retries=self.max_retries,
+                user_agent=self.user_agent,
+                request_timeout=self.request_timeout or None,
+            )
+        except Exception as e:
+            msg = (
+                "Unable to create Cohere Embeddings. ",
+                "Please verify the API key and model parameters, and try again.",
+            )
+            raise ValueError(msg) from e
+        # added status if not the return data would be serialised to create the status
+        return data
+
+    def get_model(self):
+        try:
+            co = cohere.ClientV2(self.api_key)
+            response = co.models.list(endpoint="embed")
+            models = response.models
+            return [model.name for model in models]
+        except Exception as e:
+            msg = f"Failed to fetch Cohere models. Error: {e}"
+            raise ValueError(msg) from e
+
+    async def update_build_config(self, build_config: dict, field_value: Any, field_name: str | None = None):
+        if field_name in {"model_name", "api_key"}:
+            if build_config.get("api_key", {}).get("value", None):
+                build_config["model_name"]["options"] = self.get_model()
+        else:
+            build_config["model_name"]["options"] = field_value
+        return build_config

lfx_cohere/components/cohere/cohere_models.py

@@ -0,0 +1,45 @@
+from langchain_cohere import ChatCohere
+from lfx.base.models.model import LCModelComponent
+from lfx.field_typing import LanguageModel
+from lfx.field_typing.range_spec import RangeSpec
+from lfx.io import SecretStrInput, SliderInput
+from pydantic.v1 import SecretStr
+
+
+class CohereComponent(LCModelComponent):
+    display_name = "Cohere Language Models"
+    description = "Generate text using Cohere LLMs."
+    documentation = "https://python.langchain.com/docs/integrations/llms/cohere/"
+    icon = "Cohere"
+    name = "CohereModel"
+
+    inputs = [
+        *LCModelComponent.get_base_inputs(),
+        SecretStrInput(
+            name="cohere_api_key",
+            display_name="Cohere API Key",
+            info="The Cohere API Key to use for the Cohere model.",
+            advanced=False,
+            value="COHERE_API_KEY",
+            required=True,
+        ),
+        SliderInput(
+            name="temperature",
+            display_name="Temperature",
+            value=0.75,
+            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,
+        ),
+    ]
+
+    def build_model(self) -> LanguageModel:  # type: ignore[type-var]
+        cohere_api_key = self.cohere_api_key
+        temperature = self.temperature
+
+        api_key = SecretStr(cohere_api_key).get_secret_value() if cohere_api_key else None
+
+        return ChatCohere(
+            temperature=temperature or 0.75,
+            cohere_api_key=api_key,
+        )

lfx_cohere/components/cohere/cohere_rerank.py

@@ -0,0 +1,51 @@
+from lfx.base.compressors.model import LCCompressorComponent
+from lfx.field_typing import BaseDocumentCompressor
+from lfx.inputs.inputs import SecretStrInput
+from lfx.io import DropdownInput
+from lfx.template.field.base import Output
+
+
+class CohereRerankComponent(LCCompressorComponent):
+    display_name = "Cohere Rerank"
+    description = "Rerank documents using the Cohere API."
+    name = "CohereRerank"
+    icon = "Cohere"
+
+    inputs = [
+        *LCCompressorComponent.inputs,
+        SecretStrInput(
+            name="api_key",
+            display_name="Cohere API Key",
+        ),
+        DropdownInput(
+            name="model",
+            display_name="Model",
+            options=[
+                "rerank-english-v3.0",
+                "rerank-multilingual-v3.0",
+                "rerank-english-v2.0",
+                "rerank-multilingual-v2.0",
+            ],
+            value="rerank-english-v3.0",
+        ),
+    ]
+
+    outputs = [
+        Output(
+            display_name="Reranked Documents",
+            name="reranked_documents",
+            method="compress_documents",
+        ),
+    ]
+
+    def build_compressor(self) -> BaseDocumentCompressor:  # type: ignore[type-var]
+        try:
+            from langchain_cohere import CohereRerank
+        except ImportError as e:
+            msg = "Please install langchain-cohere to use the Cohere model."
+            raise ImportError(msg) from e
+        return CohereRerank(
+            cohere_api_key=self.api_key,
+            model=self.model,
+            top_n=self.top_n,
+        )

lfx_cohere/extension.json

@@ -0,0 +1,16 @@
+{
+  "$schema": "https://schemas.langflow.org/extension/v1.json",
+  "id": "lfx-cohere",
+  "version": "0.1.0",
+  "name": "Cohere",
+  "description": "Cohere component(s) as a standalone Langflow Extension Bundle.",
+  "lfx": {
+    "compat": ["1"]
+  },
+  "bundles": [
+    {
+      "name": "cohere",
+      "path": "components/cohere"
+    }
+  ]
+}

lfx_cohere-0.1.0.dist-info/METADATA

@@ -0,0 +1,44 @@
+Metadata-Version: 2.4
+Name: lfx-cohere
+Version: 0.1.0
+Summary: Cohere component(s) as a standalone Langflow Extension Bundle.
+Project-URL: Homepage, https://github.com/langflow-ai/langflow
+Project-URL: Documentation, https://docs.langflow.org/extensions
+Project-URL: Repository, https://github.com/langflow-ai/langflow
+Author-email: Langflow <contact@langflow.org>
+License: MIT
+Keywords: bundle,cohere,extension,langflow,lfx
+Requires-Python: <3.15,>=3.10
+Requires-Dist: langchain-cohere~=0.5.0
+Requires-Dist: lfx<2.0.0,>=1.11.0.dev0
+Description-Content-Type: text/markdown
+
+# lfx-cohere
+
+Cohere component(s) as a standalone Langflow Extension Bundle.
+
+## Install
+
+```bash
+pip install lfx-cohere
+```
+
+The bundle is registered automatically via the `langflow.extensions`
+entry-point.  After install, restart your Langflow server; the bundle's
+components will appear in the palette under the `cohere` group with
+the namespaced IDs `ext:cohere:<Class>@official`.
+
+## Develop
+
+```bash
+cd src/bundles/cohere
+pip install -e .
+lfx extension validate src/lfx_cohere
+```
+
+## Migration
+
+Saved flows referencing the legacy class name(s) or the old import paths
+under `lfx.components.cohere.*` are rewritten to the new namespaced
+IDs by the migration table in
+`src/lfx/src/lfx/extension/migration/migration_table.json`.

lfx_cohere-0.1.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+lfx_cohere/__init__.py,sha256=MczmAux1tSkNDyj5F8k_wNaZX9d0AWU2aPEizz-pxHI,597
+lfx_cohere/extension.json,sha256=S4-cbGeBtBsOnyvxtRbiizcAtbGxqEV1rsK2kWrcCwA,339
+lfx_cohere/components/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+lfx_cohere/components/cohere/__init__.py,sha256=chs8Z0J4iFw1_YGuamgXqa3KmvlLZL1ZBYiGHZ3oV-A,1449
+lfx_cohere/components/cohere/cohere_embeddings.py,sha256=z1_t2epH_XuzdswW9_jeMw0hBqsHMeU2CdNjn3ycNag,3082
+lfx_cohere/components/cohere/cohere_models.py,sha256=HDHfM5ghJ9w1QUn4rwGNV0SOaygAfX0iVll7XRS5Xwg,1562
+lfx_cohere/components/cohere/cohere_rerank.py,sha256=qUoNEe6sjUnvkTHkCzwayBuLDoH957BBEgb-Qu_k9Yk,1554
+lfx_cohere-0.1.0.dist-info/METADATA,sha256=JvctMF5B4v_cKkVqMubJJ1ezm1DHV0chO8mnV0_frKs,1307
+lfx_cohere-0.1.0.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+lfx_cohere-0.1.0.dist-info/entry_points.txt,sha256=yEwxbwRG6SIkWSCREU5a8uBQQnlMfB3bt2FhSrydYjo,46
+lfx_cohere-0.1.0.dist-info/RECORD,,

lfx_cohere-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.30.1
+Root-Is-Purelib: true
+Tag: py3-none-any

lfx_cohere-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[langflow.extensions]
+lfx-cohere = lfx_cohere