gllm-pipeline-binary 0.4.23__cp312-cp312-win_amd64.whl → 0.4.24__cp312-cp312-win_amd64.whl

gllm_pipeline/pipeline/pipeline.pyi

@@ -1,4 +1,5 @@
 from _typeshed import Incomplete
+from gllm_core.schema.tool import Tool
 from gllm_datastore.cache.cache import BaseCache as BaseCache
 from gllm_pipeline.alias import PipelineState as PipelineState
 from gllm_pipeline.exclusions import ExclusionManager as ExclusionManager, ExclusionSet as ExclusionSet
@@ -8,11 +9,13 @@ from gllm_pipeline.steps.pipeline_step import BasePipelineStep as BasePipelineSt
 from gllm_pipeline.steps.terminator_step import TerminatorStep as TerminatorStep
 from gllm_pipeline.utils.graph import create_edge as create_edge
 from gllm_pipeline.utils.mermaid import MERMAID_HEADER as MERMAID_HEADER, combine_mermaid_diagrams as combine_mermaid_diagrams, extract_step_diagrams as extract_step_diagrams
+from gllm_pipeline.utils.typing_compat import TypedDict as TypedDict, is_typeddict as is_typeddict
 from langgraph.graph import StateGraph
 from langgraph.graph.state import CompiledStateGraph as CompiledStateGraph
 from pydantic import BaseModel
-from typing import Any, TypedDict
+from typing import Any
 
+DEFAULT_TOOL_NAME: str
 INDENTATION: str
 
 class Pipeline:
@@ -192,6 +195,43 @@ class Pipeline:
         Returns:
             Composer: A composer instance that manages this pipeline.
         """
+    def as_tool(self, description: str | None = None) -> Tool:
+        '''Convert the pipeline to a Tool instance.
+
+        This method allows a Pipeline instance to be used as a tool, with the input schema
+        being derived from the pipeline\'s input schema. The pipeline must have an input_type
+        defined to be convertible as a tool.
+
+        Args:
+            description (str | None, optional): Optional description to associate with the tool.
+                Defaults to None, in which case a description will be generated automatically.
+
+        Returns:
+            Tool: A Tool instance that wraps the pipeline.
+
+        Raises:
+            ValueError: If the pipeline does not have an input schema defined.
+
+        Examples:
+            ```python
+            class InputState(TypedDict):
+                user_query: str
+                context: str
+
+            class OutputState(TypedDict):
+                result: str
+
+            pipeline = Pipeline(
+                [retrieval_step, generation_step],
+                input_type=InputState,
+                output_type=OutputState,
+                name="rag_pipeline"
+            )
+
+            tool = pipeline.as_tool(description="Process user questions")
+            result = await tool.invoke(user_query="What is AI?", context="")
+            ```
+        '''
     def clear(self) -> None:
         """Clears the pipeline by resetting steps, graph, and app to their initial state.
 

gllm_pipeline/pipeline/states.pyi

@@ -1,7 +1,8 @@
 from _typeshed import Incomplete
 from gllm_core.event.event_emitter import EventEmitter
+from gllm_pipeline.utils.typing_compat import TypedDict as TypedDict
 from pydantic import BaseModel
-from typing import Any, TypedDict
+from typing import Any
 
 class RAGState(TypedDict):
     '''A TypedDict representing the state of a Retrieval-Augmented Generation (RAG) pipeline.

gllm_pipeline/steps/_func.pyi

@@ -665,7 +665,7 @@ def subgraph(subgraph: Pipeline, input_state_map: dict[str, str] | None = None,
 
     Examples:
         ```python
-        from typing import TypedDict
+        from gllm_pipeline.utils.typing_compat import TypedDict
         from gllm_pipeline.pipeline.pipeline import Pipeline
 
         # Define state schemas using TypedDict

gllm_pipeline/utils/typing_compat.pyi

@@ -0,0 +1,3 @@
+from typing import TypedDict as TypedDict, is_typeddict as is_typeddict
+
+__all__ = ['TypedDict', 'is_typeddict']

gllm_pipeline.pyi

@@ -22,6 +22,7 @@ import gllm_datastore.cache.cache
 import asyncio
 import uuid
 import copy
+import gllm_core.schema.tool
 import gllm_core.utils.imports
 import gllm_core.utils.logger_manager
 import gllm_inference
@@ -84,4 +85,5 @@ import langgraph.types
 import dataclasses
 import gllm_core.event.messenger
 import gllm_pipeline.steps.step_error_handler.RaiseStepErrorHandler
-import traceback
+import traceback
+import sys

{gllm_pipeline_binary-0.4.23.dist-info → gllm_pipeline_binary-0.4.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: gllm-pipeline-binary
-Version: 0.4.23
+Version: 0.4.24
 Summary: A library containing components related to Gen AI applications pipeline orchestration.
 Author-email: Dimitrij Ray <dimitrij.ray@gdplabs.id>, Henry Wicaksono <henry.wicaksono@gdplabs.id>, Kadek Denaya <kadek.d.r.diana@gdplabs.id>
 Requires-Python: <3.13,>=3.11
@@ -11,6 +11,7 @@ Requires-Dist: gllm-core-binary<0.4.0,>=0.3.0
 Requires-Dist: gllm-inference-binary<0.6.0,>=0.5.0
 Requires-Dist: aiohttp<3.13.0,>=3.12.14
 Requires-Dist: langgraph<0.7.0,>=0.6.0
+Requires-Dist: typing-extensions<5.0.0,>=4.5.0
 Provides-Extra: dev
 Requires-Dist: coverage<7.5.0,>=7.4.4; extra == "dev"
 Requires-Dist: mypy<1.16.0,>=1.15.0; extra == "dev"

{gllm_pipeline_binary-0.4.23.dist-info → gllm_pipeline_binary-0.4.24.dist-info}/RECORD

@@ -1,5 +1,5 @@
-gllm_pipeline.cp312-win_amd64.pyd,sha256=tzPFp3ZMu0Ek6EYDOwCAfzeEO6N-jxpuOPCntj91GHE,2177024
-gllm_pipeline.pyi,sha256=SCe8vHfVDlgMZ1I_dfYPa-qFjsj_IWl0lgTa7lRrQqQ,2306
+gllm_pipeline.cp312-win_amd64.pyd,sha256=qjhsaFUWKRHDq6mgIqfYy52U1uMsunzKOu0tQLcGRhk,2189312
+gllm_pipeline.pyi,sha256=KL3UbRkjsWRktt_UM70HNxLAvyYzPTQ0pvNfwTZt8SI,2346
 gllm_pipeline/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gllm_pipeline/alias.pyi,sha256=FbALRYZpDlmQMsKNUvgCi6ji11PrEtNo2kgzbt0iT7g,237
 gllm_pipeline/types.pyi,sha256=CV3cEAxlNsnVatYz5iCxqmEFPEqeKW5vv-qUD3FpF54,241
@@ -7,8 +7,8 @@ gllm_pipeline/exclusions/__init__.pyi,sha256=_LwIlqmH4Iiksn7p09d2vZG4Ek8CdKC8UcD
 gllm_pipeline/exclusions/exclusion_manager.pyi,sha256=DzoL-2KeTRmFgJEo8rzYViFYKbzZVTZGJmKvzaoTC0M,2960
 gllm_pipeline/exclusions/exclusion_set.pyi,sha256=11XTt6IfkHpzomcNybA78SfWlp752Z3AGhXfm2rL0Fk,1685
 gllm_pipeline/pipeline/__init__.pyi,sha256=1IKGdMvmLWEiOOmAKFNUPm-gdw13zrnU1gs7tDNzgEU,168
-gllm_pipeline/pipeline/pipeline.pyi,sha256=m8OStEgk9SsV6faKX6wEqGyygE_xF421wJWQJvBAEeY,14979
-gllm_pipeline/pipeline/states.pyi,sha256=EiyfBPwrVDZ336w5wyD1q8W4E6G1uZNzsP-bzrHDumo,6464
+gllm_pipeline/pipeline/pipeline.pyi,sha256=CgP179RgbwtIsz_U0cMi3jdwbVOBz-D4sIKjoLkV55g,16504
+gllm_pipeline/pipeline/states.pyi,sha256=WezH44ULqlim8CoEupQGHXs5xNzkqLvK8epwYQYRT4I,6523
 gllm_pipeline/pipeline/composer/__init__.pyi,sha256=-hcOUQgpTRt1QjQfRurTf-UApFnTrhilx6vN-gYd5J0,666
 gllm_pipeline/pipeline/composer/composer.pyi,sha256=7h7EhEA-hex6w36Is6uGTz9OBUbmq6C0SdkeBeLFcAI,28715
 gllm_pipeline/pipeline/composer/guard_composer.pyi,sha256=YfbXmzyU3CwAvGnCfM-6MVcTdxk53-j6Cv3IdzNr_-c,3335
@@ -40,7 +40,7 @@ gllm_pipeline/router/preset/aurelio/router_image_domain_specific.pyi,sha256=6pm2
 gllm_pipeline/router/preset/lm_based/__init__.pyi,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gllm_pipeline/router/preset/lm_based/router_image_domain_specific.pyi,sha256=UdiuoSXm2MVAL8AspAaSkyXYkE59bYj1y4xRRgKwavE,655
 gllm_pipeline/steps/__init__.pyi,sha256=5HtVA5CODr_9_7_OGEXFXU40edqhHa9YlCV5qVx3xbU,1989
-gllm_pipeline/steps/_func.pyi,sha256=FS-g9LiyHb8OHLM8bXrAMiqvk4-KgaxV8RacD0IQw-Q,64125
+gllm_pipeline/steps/_func.pyi,sha256=TDTSsBwiUMwlFXLcLVolcd_7g7bAcg4mDx_r_y2mDb4,64152
 gllm_pipeline/steps/branching_step.pyi,sha256=iNarrcZgWfiRdr8CJfGm8GzUlYq13Rx5cgYXnBsNWN4,1041
 gllm_pipeline/steps/component_step.pyi,sha256=VNBFZscK2Q4HgBt8ZrbE6B69oLTKcXo-KkqbTjmYYhM,5748
 gllm_pipeline/steps/composite_step.pyi,sha256=lvueTBQG_t0TtS5qRvUzZOIt3h6-uD26DJXW4ZSkuUc,3544
@@ -70,8 +70,9 @@ gllm_pipeline/utils/input_map.pyi,sha256=mPWU9_b3VGhszuTjB3yQggZWJCxjZth4_WQdKec
 gllm_pipeline/utils/mermaid.pyi,sha256=B096GTXxVAO--kw3UDsbysOsnjGOytYfozX39YaM21A,1174
 gllm_pipeline/utils/retry_converter.pyi,sha256=JPUuaGzKpVLshrbhX9rQHYl5XmC9GDa59rGU-FtOpWM,1128
 gllm_pipeline/utils/step_execution.pyi,sha256=3o28tiCHR8t-6Vk3Poz91V-CLdYrdhvJblPW9AoOK-c,996
+gllm_pipeline/utils/typing_compat.pyi,sha256=V4812i25ncSqZ0o_30lUX65tU07bWEs4LIpdLPt2ngg,116
 gllm_pipeline.build/.gitignore,sha256=aEiIwOuxfzdCmLZe4oB1JsBmCUxwG8x-u-HBCV9JT8E,1
-gllm_pipeline_binary-0.4.23.dist-info/METADATA,sha256=rro3zbAKhq4bHW2wFZZjHyFTFaAH_ZxbCCWFpJ9OKvY,4476
-gllm_pipeline_binary-0.4.23.dist-info/WHEEL,sha256=x5rgv--I0NI0IT1Lh9tN1VG2cI637p3deednwYLKnxc,96
-gllm_pipeline_binary-0.4.23.dist-info/top_level.txt,sha256=C3yeOtoE6ZhuOnBEq_FFc_Rp954IHJBlB6fBgSdAWYI,14
-gllm_pipeline_binary-0.4.23.dist-info/RECORD,,
+gllm_pipeline_binary-0.4.24.dist-info/METADATA,sha256=8mzbtoKhwpZLS3hpZ2rK6Rwu-eNZRs_fV_gD7rdh9eo,4524
+gllm_pipeline_binary-0.4.24.dist-info/WHEEL,sha256=x5rgv--I0NI0IT1Lh9tN1VG2cI637p3deednwYLKnxc,96
+gllm_pipeline_binary-0.4.24.dist-info/top_level.txt,sha256=C3yeOtoE6ZhuOnBEq_FFc_Rp954IHJBlB6fBgSdAWYI,14
+gllm_pipeline_binary-0.4.24.dist-info/RECORD,,