coco-framework 0.1.0__py3-none-any.whl

coco_core/runnables/__init__.py

@@ -0,0 +1,136 @@
+"""Coco **Runnable** and the **Coco Expression Language (LCEL)**.
+
+The Coco Expression Language (LCEL) offers a declarative method to build
+production-grade programs that harness the power of LLMs.
+
+Programs created using LCEL and Coco `Runnable` objects inherently support
+synchronous asynchronous, batch, and streaming operations.
+
+Support for **async** allows servers hosting LCEL based programs to scale bette for
+higher concurrent loads.
+
+**Batch** operations allow for processing multiple inputs in parallel.
+
+**Streaming** of intermediate outputs, as they're being generated, allows for creating
+more responsive UX.
+
+This module contains schema and implementation of Coco `Runnable` object
+primitives.
+"""
+
+from typing import TYPE_CHECKING
+
+from coco_core._import_utils import import_attr
+
+if TYPE_CHECKING:
+    from coco_core.runnables.base import (
+        Runnable,
+        RunnableBinding,
+        RunnableGenerator,
+        RunnableLambda,
+        RunnableMap,
+        RunnableParallel,
+        RunnableSequence,
+        RunnableSerializable,
+        chain,
+    )
+    from coco_core.runnables.branch import RunnableBranch
+    from coco_core.runnables.config import (
+        RunnableConfig,
+        ensure_config,
+        get_config_list,
+        patch_config,
+        run_in_executor,
+    )
+    from coco_core.runnables.fallbacks import RunnableWithFallbacks
+    from coco_core.runnables.history import RunnableWithMessageHistory
+    from coco_core.runnables.passthrough import (
+        RunnableAssign,
+        RunnablePassthrough,
+        RunnablePick,
+    )
+    from coco_core.runnables.router import RouterInput, RouterRunnable
+    from coco_core.runnables.utils import (
+        AddableDict,
+        ConfigurableField,
+        ConfigurableFieldMultiOption,
+        ConfigurableFieldSingleOption,
+        ConfigurableFieldSpec,
+        aadd,
+        add,
+    )
+
+__all__ = (
+    "AddableDict",
+    "ConfigurableField",
+    "ConfigurableFieldMultiOption",
+    "ConfigurableFieldSingleOption",
+    "ConfigurableFieldSpec",
+    "RouterInput",
+    "RouterRunnable",
+    "Runnable",
+    "RunnableAssign",
+    "RunnableBinding",
+    "RunnableBranch",
+    "RunnableConfig",
+    "RunnableGenerator",
+    "RunnableLambda",
+    "RunnableMap",
+    "RunnableParallel",
+    "RunnablePassthrough",
+    "RunnablePick",
+    "RunnableSequence",
+    "RunnableSerializable",
+    "RunnableWithFallbacks",
+    "RunnableWithMessageHistory",
+    "aadd",
+    "add",
+    "chain",
+    "ensure_config",
+    "get_config_list",
+    "patch_config",
+    "run_in_executor",
+)
+
+_dynamic_imports = {
+    "chain": "base",
+    "Runnable": "base",
+    "RunnableBinding": "base",
+    "RunnableGenerator": "base",
+    "RunnableLambda": "base",
+    "RunnableMap": "base",
+    "RunnableParallel": "base",
+    "RunnableSequence": "base",
+    "RunnableSerializable": "base",
+    "RunnableBranch": "branch",
+    "RunnableConfig": "config",
+    "ensure_config": "config",
+    "get_config_list": "config",
+    "patch_config": "config",
+    "run_in_executor": "config",
+    "RunnableWithFallbacks": "fallbacks",
+    "RunnableWithMessageHistory": "history",
+    "RunnableAssign": "passthrough",
+    "RunnablePassthrough": "passthrough",
+    "RunnablePick": "passthrough",
+    "RouterInput": "router",
+    "RouterRunnable": "router",
+    "AddableDict": "utils",
+    "ConfigurableField": "utils",
+    "ConfigurableFieldMultiOption": "utils",
+    "ConfigurableFieldSingleOption": "utils",
+    "ConfigurableFieldSpec": "utils",
+    "aadd": "utils",
+    "add": "utils",
+}
+
+
+def __getattr__(attr_name: str) -> object:
+    module_name = _dynamic_imports.get(attr_name)
+    result = import_attr(attr_name, module_name, __spec__.parent)
+    globals()[attr_name] = result
+    return result
+
+
+def __dir__() -> list[str]:
+    return list(__all__)