pybotchi 3.4.1__tar.gz → 3.4.3__tar.gz

{pybotchi-3.4.1 → pybotchi-3.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pybotchi
-Version: 3.4.1
+Version: 3.4.3
 Summary: A deterministic, intent-based AI agent builder.
 License-File: LICENSE
 Author: Alexie (Boyong) Madolid

{pybotchi-3.4.1 → pybotchi-3.4.3}/pybotchi/action.py

@@ -7,7 +7,7 @@ from collections import deque
 from collections.abc import Generator
 from inspect import getmembers
 from os import getenv
-from typing import Any, Generic, TYPE_CHECKING, TypeVar
+from typing import Any, Generic, Literal, TYPE_CHECKING, TypeVar
 
 from pydantic import BaseModel, PrivateAttr
 
@@ -503,13 +503,13 @@ class Action(BaseModel, Generic[TContext]):
 
         return ActionReturn.GO
 
-    def serialize(self) -> ActionEntry:
+    def serialize(self, mode: str | Literal["python", "json"] = "python") -> ActionEntry:
         """Serialize Action."""
         return {
             "name": self.__class__.__name__,
-            "args": self.model_dump(),
+            "args": self.model_dump(mode=mode),
             "usages": self._usage,
-            "actions": [a.serialize() if isinstance(a, Action) else a for a in self._actions],
+            "actions": [a.serialize(mode) if isinstance(a, Action) else a for a in self._actions],
         }
 
     ####################################################################################################

{pybotchi-3.4.1 → pybotchi-3.4.3}/pybotchi/action.pyi

@@ -4,7 +4,7 @@ from .utils import apply_placeholders as apply_placeholders, unwrap_exceptions a
 from _typeshed import Incomplete
 from collections.abc import Generator
 from pydantic import BaseModel
-from typing import Any, Generic, TypeVar
+from typing import Any, Generic, Literal, TypeVar
 
 DEFAULT_ACTION: Incomplete
 DEFAULT_TOOL_CALL_PROMPT: Incomplete
@@ -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: ...
@@ -58,7 +66,7 @@ class Action(BaseModel, Generic[TContext]):
     async def execution(self, context: TContext) -> ActionReturn: ...
     async def concurrent_children_execution(self, context: TContext, next_actions: list[Action]) -> ActionReturn: ...
     async def sequential_children_execution(self, context: TContext, next_actions: list[Action]) -> ActionReturn: ...
-    def serialize(self) -> ActionEntry: ...
+    def serialize(self, mode: str | Literal['python', 'json'] = 'python') -> ActionEntry: ...
     @classmethod
     def add_child(cls, action: type['Action'], name: str | None = None, override: bool = False, extended: bool = True) -> None: ...
     @classmethod

{pybotchi-3.4.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/pybotchi/grpc/pybotchi_pb2.pyi

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

{pybotchi-3.4.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/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.1 → pybotchi-3.4.3}/pyproject.toml

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