pybotchi 3.4.0__tar.gz → 3.4.2__tar.gz

{pybotchi-3.4.0 → pybotchi-3.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybotchi
-Version: 3.4.0
+Version: 3.4.2
 Summary: A deterministic, intent-based AI agent builder.
 License-File: LICENSE
 Author: Alexie (Boyong) Madolid
@@ -129,11 +129,16 @@ LLM.add(base=ChatOpenAI(
 ```python
 from pybotchi import Action, ActionReturn
 
+from pydantic import Field
+
 class Translation(Action):
-    """Translate to specified language."""
+    """Translate to specific language."""
+
+    message: str = Field(description="The text content to be translated.")
+    language: str = Field(description="The ISO code or name of the target language.")
 
     async def pre(self, context):
-        message = await context.llm.ainvoke(context.prompts)
+        message = await context.llm.ainvoke(f"Reply only with translation of `{self.message}` to {self.language}.")
         await context.add_response(self, message.text)
         return ActionReturn.GO
 ```
@@ -181,22 +186,20 @@ async def test():
     context = Context(
         prompts=[
             {"role": "system", "content": "You're an AI that can solve math problems and translate requests."},
-            {"role": "user", "content": "4 x 4 and explain in Filipino"}
+            {"role": "user", "content": "4 x 4 then explain in Filipino"},
         ],
     )
     await context.start(MultiAgent)
-    print(context.prompts[-1]["content"])
+    print(f"MathProblem: {context.prompts[-3]['content']}")
+    print(f"Translate: {context.prompts[-1]['content']}")
 
 asyncio.run(test())
 ```
 
 **Result:**
 ```
-Ang sagot sa 4 x 4 ay 16.
-
-Paliwanag: Kapag sinabi nating 4 x 4, ibig sabihin ay apat na grupo ng apat. Kung bibilangin natin ito, makakakuha tayo ng kabuuang labing-anim (16).
-
-Ibig sabihin, 4 + 4 + 4 + 4 = 16.
+MathProblem: Four multiplied by four is sixteen. Imagine you have four groups, and each group has four candies. If you count all the candies together, you will have sixteen candies. That's what 4 x 4 means!
+Translate: Ang apat na pinarami sa apat ay labing-anim. Ipagpalagay mong may apat na grupo, at bawat grupo ay may apat na kendi. Kung bibilangin mo lahat ng kendi, magkakaroon ka ng labing-anim na kendi. Iyan ang ibig sabihin ng 4 x 4!
 ```
 
 ### Visualize Your Graph
@@ -210,7 +213,7 @@ async def print_mermaid_graph():
     multi_agent_graph = await graph(MultiAgent)
     print(multi_agent_graph.flowchart())
 
-run(print_mermaid_graph())
+asyncio.run(print_mermaid_graph())
 ```
 **Result:**
 ```
@@ -393,14 +396,12 @@ pybotchi-grpc server.py
 **Result**
 ```bash
 #-------------------------------------------------------#
-# Agent ID: agent_b6c9ada82c7444818356a6338e975c09
-# Agent Path: server.py
+# Agent ID: agent_8b3c5685c84b4602966d1b3252916aa7
 # Agent Path: server.py
 # Starting None worker(s) on 0.0.0.0:50051
 #-------------------------------------------------------#
-# Agent Path: server.py
+# Agent Process: Process-1 [173012]
 # Agent Handler: PyBotchiGRPC
-# gRPC server running on 0.0.0.0:50051
 #-------------------------------------------------------#
 ```
 gRPC client print graph:
@@ -409,9 +410,9 @@ python3 client.py
 ```
 ```bash
 flowchart TD
+grpc.agent_8b3c5685c84b4602966d1b3252916aa7.MathProblem[MathProblem]
 __main__.MultiAgent[MultiAgent]
-grpc.agent_b6c9ada82c7444818356a6338e975c09.MathProblem[MathProblem]
-__main__.MultiAgent --**GRPC** : remote--> grpc.agent_b6c9ada82c7444818356a6338e975c09.MathProblem
+__main__.MultiAgent --"`**GRPC** : remote`"--> grpc.agent_8b3c5685c84b4602966d1b3252916aa7.MathProblem
 style __main__.MultiAgent fill:#4CAF50,color:#000000
 ```
 ![gRPC MultiAgent Graph](docs/mermaid2.png)

{pybotchi-3.4.0 → pybotchi-3.4.2}/README.md

@@ -103,11 +103,16 @@ LLM.add(base=ChatOpenAI(
 ```python
 from pybotchi import Action, ActionReturn
 
+from pydantic import Field
+
 class Translation(Action):
-    """Translate to specified language."""
+    """Translate to specific language."""
+
+    message: str = Field(description="The text content to be translated.")
+    language: str = Field(description="The ISO code or name of the target language.")
 
     async def pre(self, context):
-        message = await context.llm.ainvoke(context.prompts)
+        message = await context.llm.ainvoke(f"Reply only with translation of `{self.message}` to {self.language}.")
         await context.add_response(self, message.text)
         return ActionReturn.GO
 ```
@@ -155,22 +160,20 @@ async def test():
     context = Context(
         prompts=[
             {"role": "system", "content": "You're an AI that can solve math problems and translate requests."},
-            {"role": "user", "content": "4 x 4 and explain in Filipino"}
+            {"role": "user", "content": "4 x 4 then explain in Filipino"},
         ],
     )
     await context.start(MultiAgent)
-    print(context.prompts[-1]["content"])
+    print(f"MathProblem: {context.prompts[-3]['content']}")
+    print(f"Translate: {context.prompts[-1]['content']}")
 
 asyncio.run(test())
 ```
 
 **Result:**
 ```
-Ang sagot sa 4 x 4 ay 16.
-
-Paliwanag: Kapag sinabi nating 4 x 4, ibig sabihin ay apat na grupo ng apat. Kung bibilangin natin ito, makakakuha tayo ng kabuuang labing-anim (16).
-
-Ibig sabihin, 4 + 4 + 4 + 4 = 16.
+MathProblem: Four multiplied by four is sixteen. Imagine you have four groups, and each group has four candies. If you count all the candies together, you will have sixteen candies. That's what 4 x 4 means!
+Translate: Ang apat na pinarami sa apat ay labing-anim. Ipagpalagay mong may apat na grupo, at bawat grupo ay may apat na kendi. Kung bibilangin mo lahat ng kendi, magkakaroon ka ng labing-anim na kendi. Iyan ang ibig sabihin ng 4 x 4!
 ```
 
 ### Visualize Your Graph
@@ -184,7 +187,7 @@ async def print_mermaid_graph():
     multi_agent_graph = await graph(MultiAgent)
     print(multi_agent_graph.flowchart())
 
-run(print_mermaid_graph())
+asyncio.run(print_mermaid_graph())
 ```
 **Result:**
 ```
@@ -367,14 +370,12 @@ pybotchi-grpc server.py
 **Result**
 ```bash
 #-------------------------------------------------------#
-# Agent ID: agent_b6c9ada82c7444818356a6338e975c09
-# Agent Path: server.py
+# Agent ID: agent_8b3c5685c84b4602966d1b3252916aa7
 # Agent Path: server.py
 # Starting None worker(s) on 0.0.0.0:50051
 #-------------------------------------------------------#
-# Agent Path: server.py
+# Agent Process: Process-1 [173012]
 # Agent Handler: PyBotchiGRPC
-# gRPC server running on 0.0.0.0:50051
 #-------------------------------------------------------#
 ```
 gRPC client print graph:
@@ -383,9 +384,9 @@ python3 client.py
 ```
 ```bash
 flowchart TD
+grpc.agent_8b3c5685c84b4602966d1b3252916aa7.MathProblem[MathProblem]
 __main__.MultiAgent[MultiAgent]
-grpc.agent_b6c9ada82c7444818356a6338e975c09.MathProblem[MathProblem]
-__main__.MultiAgent --**GRPC** : remote--> grpc.agent_b6c9ada82c7444818356a6338e975c09.MathProblem
+__main__.MultiAgent --"`**GRPC** : remote`"--> grpc.agent_8b3c5685c84b4602966d1b3252916aa7.MathProblem
 style __main__.MultiAgent fill:#4CAF50,color:#000000
 ```
 ![gRPC MultiAgent Graph](docs/mermaid2.png)

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/action.pyi

@@ -38,10 +38,18 @@ class Action(BaseModel, Generic[TContext]):
     __display_name__: str
     __groups__: Groups | set[str] | None
     __to_commit__: bool
+    _usage: list[UsageData]
+    _actions: list[Action | ActionEntry]
+    _parent: Action | None
+    _children: list['Action']
     @classmethod
     def __pydantic_init_subclass__(cls, **kwargs: Any) -> None: ...
     @classmethod
     def __init_child_actions__(cls) -> None: ...
+    @property
+    def _tool_call(self) -> ToolCall: ...
+    @classmethod
+    async def _as_tool(cls, context: TContext) -> dict[str, Any] | type[BaseModel]: ...
     async def pre(self, context: TContext) -> ActionReturn: ...
     async def fallback(self, context: TContext, content: str) -> ActionReturn: ...
     async def on_child_init_error(self, context: TContext, next_actions: list['Action'], child_cls: type[Action], child_args: dict[str, Any], exception: Exception) -> str | None: ...

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/common.py

@@ -110,11 +110,13 @@ class Graph(BaseModel):
 
             if concurrent:
                 connection = (
-                    f"ed{con}@--**{base}** : {alias}<br>*[concurrent]*-->" if alias else f"ed{con}@--*[concurrent]*-->"
+                    f'ed{con}@--"`**{base}** : {alias}<br>*[concurrent]*`"-->'
+                    if alias
+                    else f'ed{con}@--"`*[concurrent]*`"-->'
                 )
                 con += 1
             else:
-                connection = f"--**{base}** : {alias}-->" if alias else "-->"
+                connection = f'--"`**{base}** : {alias}`"-->' if alias else "-->"
             content += f"{source} {connection} {target}\n"
 
         constraints = (

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/context.pyi

@@ -22,6 +22,7 @@ class Context(BaseModel, Generic[TLLM]):
     streaming: bool
     max_self_loop: int | None
     parent: Self | None
+    _action_call: dict[str, int]
     @cached_property
     def llm(self) -> TLLM: ...
     @cached_property

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/grpc/action.pyi

@@ -36,6 +36,7 @@ class GRPCAction(Action[TContext], Generic[TContext]):
     @classmethod
     def __init_child_actions__(cls) -> None: ...
     async def pre_grpc(self, context: TContext) -> ActionReturn: ...
+    _parent: Incomplete
     __to_commit__: bool
     async def execute(self, context: TContext, parent: Action | None = None, append: bool = True) -> ActionReturn: ...
     async def get_child_actions(self, context: TContext) -> ChildActions: ...

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/grpc/cli.py

@@ -4,7 +4,7 @@ from asyncio import run
 from importlib.resources import files
 from importlib.util import module_from_spec, spec_from_file_location
 from inspect import getmembers, isclass
-from multiprocessing import Process, cpu_count
+from multiprocessing import Process, cpu_count, current_process
 from os import getenv
 from pathlib import Path
 from signal import SIGINT, SIGTERM, signal
@@ -96,10 +96,12 @@ async def serve(
         server.add_insecure_port(address)
     await server.start()
 
-    echo(f"# Agent Path: {path}")
-    echo(f"# Agent Handler: {grpc_handler.__name__}")
-    echo(f"# gRPC server running on {address}")
-    echo("#-------------------------------------------------------#")
+    process = current_process()
+    echo(
+        f"# Agent Process: {process.name} [{process.pid}]\n"
+        f"# Agent Handler: {grpc_handler.__name__}\n"
+        "#-------------------------------------------------------#"
+    )
     await server.wait_for_termination()
 
 
@@ -191,7 +193,6 @@ def main(
     else:
         _certificate_chain = None
 
-    echo(f"# Agent Path: {path}")
     echo(f"# Starting {workers} worker(s) on {host}:{port}")
     echo("#-------------------------------------------------------#")
 

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/grpc/context.pyi

@@ -3,6 +3,7 @@ from ..context import Action as Action, ChatRole as ChatRole, Context as Context
 from ..utils import uuid as uuid
 from .common import GRPCIntegration as GRPCIntegration
 from .pybotchi_pb2 import Event as Event
+from asyncio import Queue
 from typing import Any, Generic, TypeVar
 
 TContext = TypeVar('TContext', bound='GRPCContext')
@@ -11,6 +12,8 @@ class GRPCContext(Context[TLLM], Generic[TLLM]):
     integrations: dict[str, GRPCIntegration]
     source_id: str | None
     context_id: str
+    _response_queue: Queue[Event] | None
+    _request_queues: dict[str, Queue]
     def grpc_dump(self) -> dict[str, Any]: ...
     def grpc_sharing_dump(self) -> dict[str, Any]: ...
     async def grpc_send_up(self, source_id: str | None, name: str, data: dict[str, Any]) -> None: ...

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/grpc/exception.pyi

@@ -7,3 +7,4 @@ class GRPCRemoteError(Exception):
     message: Incomplete
     tracebacks: Incomplete
     def __init__(self, cls: str, alias: str, type: str, message: str, tracebacks: list[str]) -> None: ...
+    def __str__(self) -> str: ...

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/grpc/pybotchi_pb2.pyi

@@ -1,3 +1,5 @@
 from _typeshed import Incomplete
 
+_sym_db: Incomplete
 DESCRIPTOR: Incomplete
+_globals: Incomplete

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/grpc/pybotchi_pb2_grpc.pyi

@@ -2,6 +2,7 @@ from _typeshed import Incomplete
 
 GRPC_GENERATED_VERSION: str
 GRPC_VERSION: Incomplete
+_version_not_supported: bool
 
 class PyBotchiGRPCStub:
     connect: Incomplete

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/mcp/action.py

@@ -402,7 +402,13 @@ async def multi_mcp_clients(
                 )
             else:
                 async_client = await stack.enter_async_context(
-                    AsyncClient(**overrided_config["async_client_args"], follow_redirects=True)
+                    AsyncClient(
+                        base_url=overrided_config["url"],
+                        headers=overrided_config["headers"],
+                        timeout=overrided_config["timeout"],
+                        **overrided_config["async_client_args"],
+                        follow_redirects=True,
+                    )
                 )
 
                 streams = await stack.enter_async_context(

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/mcp/action.pyi

@@ -38,6 +38,7 @@ class MCPAction(Action[TContext], Generic[TContext]):
     @classmethod
     def __init_child_actions__(cls) -> None: ...
     async def pre_mcp(self, context: TContext) -> ActionReturn: ...
+    _parent: Incomplete
     __to_commit__: bool
     async def execute(self, context: TContext, parent: Action | None = None, append: bool = True) -> ActionReturn: ...
     async def get_child_actions(self, context: TContext) -> ChildActions: ...

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/mcp/common.py

@@ -35,18 +35,13 @@ class AsyncClientArgs(TypedDict, total=False):
         | bytes
         | None
     )
-    headers: (
-        Mapping[str, str] | Mapping[bytes, bytes] | Sequence[tuple[str, str]] | Sequence[tuple[bytes, bytes]] | None
-    )
     cookies: dict[str, str] | list[tuple[str, str]] | None
     verify: str | bool
     cert: CertTypes | None
     http1: bool
     http2: bool
     proxy: str | None
-    timeout: float | None | tuple[float | None, float | None, float | None, float | None] | None
     max_redirects: int
-    base_url: str
     trust_env: bool
     default_encoding: str
 

{pybotchi-3.4.0 → pybotchi-3.4.2}/pybotchi/mcp/common.pyi

@@ -14,16 +14,13 @@ class MCPMode(StrEnum):
 class AsyncClientArgs(TypedDict, total=False):
     auth: tuple[str | bytes, str | bytes] | None
     params: Mapping[str, PrimitiveData | Sequence[PrimitiveData]] | list[tuple[str, PrimitiveData]] | tuple[tuple[str, PrimitiveData], ...] | str | bytes | None
-    headers: Mapping[str, str] | Mapping[bytes, bytes] | Sequence[tuple[str, str]] | Sequence[tuple[bytes, bytes]] | None
     cookies: dict[str, str] | list[tuple[str, str]] | None
     verify: str | bool
     cert: CertTypes | None
     http1: bool
     http2: bool
     proxy: str | None
-    timeout: float | None | tuple[float | None, float | None, float | None, float | None] | None
     max_redirects: int
-    base_url: str
     trust_env: bool
     default_encoding: str
 

{pybotchi-3.4.0 → pybotchi-3.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pybotchi"
-version = "3.4.0"
+version = "3.4.2"
 description = "A deterministic, intent-based AI agent builder."
 authors = ["Alexie (Boyong) Madolid <madolid.alexie@gmail.com>"]
 readme = "README.md"