attp-client 0.0.1__tar.gz

attp_client-0.0.1/PKG-INFO

@@ -0,0 +1,278 @@
+Metadata-Version: 2.3
+Name: attp-client
+Version: 0.0.1
+Summary: A python-sdk client for interacting with AgentHub's ATTP protocol (Agent Tool Transport Protocol)
+License: MIT
+Author: Ascender Team
+Requires-Python: >=3.11,<3.14
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: ascender-framework (>=2.0rc7,<3.0)
+Requires-Dist: attp-core (==0.1.10)
+Requires-Dist: msgpack (>=1.1.1,<2.0.0)
+Requires-Dist: pydantic (>=2.11.7,<3.0.0)
+Description-Content-Type: text/markdown
+
+# ATTP Client
+
+A Python SDK client for interacting with AgentHub's ATTP (Agent Tool Transport Protocol).
+
+## Overview
+
+The ATTP Client provides a comprehensive Python interface for connecting to and interacting with AgentHub's ATTP protocol. It enables real-time communication with AI agents, tool management, and inference operations through a high-performance async/await interface.
+
+## Features
+
+- **Async/Await Support**: Built with modern Python async patterns for optimal performance
+- **Agent Communication**: Direct communication with AI agents through the ATTP protocol
+- **Tool Management**: Register, unregister, and manage tools in organized catalogs
+- **Inference API**: Invoke AI agent inference with configurable parameters
+- **Real-time Messaging**: Stream responses and handle real-time agent interactions
+- **Event Handling**: Register custom event handlers for connect/disconnect and custom routes
+- **Authentication**: Secure token-based authentication with organization support
+
+## Installation
+
+```bash
+pip install attp-client
+```
+
+Or with Poetry:
+
+```bash
+poetry add attp-client
+```
+
+## Quick Start
+
+### Basic Connection
+
+```python
+import asyncio
+from attp_client.client import ATTPClient
+
+async def main():
+    # Initialize the client
+    client = ATTPClient(
+        agt_token="your_agent_token_here",
+        organization_id=1,
+        connection_url="attp://localhost:6563"  # Optional, defaults to localhost
+    )
+    
+    # Connect to AgentHub
+    await client.connect()
+    
+    # Your code here...
+    
+    # Close the connection
+    await client.close()
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Agent Inference
+
+```python
+from attp_client.interfaces.inference.message import IMessageDTOV2
+from attp_client.interfaces.inference.enums.message_type import MessageTypeEnum
+from uuid import UUID
+
+# Invoke inference by agent ID
+response = await client.inference.invoke_inference(
+    agent_id=17,
+    input_configuration={},
+    messages=[
+        IMessageDTOV2(
+            content="Hello, how can you help me?",
+            message_type=MessageTypeEnum.USER_MESSAGE,
+            chat_id=UUID("your-chat-id-here")
+        )
+    ],
+    stream=False,
+    timeout=30
+)
+
+print("Agent response:", response)
+```
+
+### Tool Management
+
+```python
+# Register a tool
+tool_id = await client.tools.register(
+    catalog_name="my_catalog",
+    name="example_tool",
+    description="An example tool for demonstration",
+    schema_id="tool_schema_v1",
+    return_direct=False,
+    timeout_ms=20000
+)
+
+# Access a specific catalog
+catalog = await client.catalog("my_catalog")
+
+# Unregister a tool
+await client.tools.unregister("my_catalog", str(tool_id))
+```
+
+### Event Handling
+
+```python
+def on_connect():
+    print("Connected to AgentHub!")
+
+def on_disconnect():
+    print("Disconnected from AgentHub!")
+
+def handle_custom_event(data):
+    print(f"Received custom event: {data}")
+
+# Register event handlers
+client.add_event_handler("", "connect", on_connect)
+client.add_event_handler("", "disconnect", on_disconnect)
+client.add_event_handler("custom:pattern", "custom", handle_custom_event)
+```
+
+## Configuration
+
+### Client Parameters
+
+- `agt_token` (str): Your AgentHub authentication token
+- `organization_id` (int): Your organization ID
+- `connection_url` (str, optional): ATTP server URL (default: "attp://localhost:6563")
+- `max_retries` (int, optional): Maximum connection retry attempts (default: 20)
+- `limits` (Limits, optional): Connection limits configuration
+- `logger` (Logger, optional): Custom logger instance
+
+### Connection Limits
+
+```python
+from attp_core.rs_api import Limits
+
+limits = Limits(max_payload_size=50000)
+client = ATTPClient(
+    agt_token="...",
+    organization_id=1,
+    limits=limits
+)
+```
+
+## API Reference
+
+### ATTPClient
+
+The main client class for ATTP communication.
+
+#### Methods
+
+- `connect()`: Establish connection to AgentHub
+- `close()`: Close the connection
+- `catalog(catalog_name: str)`: Access a specific tool catalog
+- `add_event_handler(pattern: str, route_type: RouteType, callback: Callable)`: Register event handlers
+
+#### Properties
+
+- `inference`: Access to the inference API
+- `tools`: Access to the tools manager
+- `router`: Access to the low-level router
+
+### AttpInferenceAPI
+
+Handles AI agent inference operations.
+
+#### Methods
+
+- `invoke_inference()`: Invoke inference for a specific agent
+- `invoke_chat_inference()`: Invoke inference for a chat session
+
+### ToolsManager
+
+Manages tool registration and organization.
+
+#### Methods
+
+- `register()`: Register a new tool
+- `unregister()`: Remove tool registration
+
+## Error Handling
+
+The client includes comprehensive error handling:
+
+```python
+from attp_client.errors import (
+    AttpException,
+    DeadSessionError,
+    NotFoundError,
+    SerializationError,
+    UnauthenticatedError
+)
+
+try:
+    await client.connect()
+except UnauthenticatedError:
+    print("Authentication failed - check your token")
+except DeadSessionError:
+    print("Session has died - reconnection required")
+except AttpException as e:
+    print(f"ATTP error: {e}")
+```
+
+## Requirements
+
+- Python 3.11 - 3.13
+- Dependencies:
+  - `pydantic` (>=2.11.7,<3.0.0)
+  - `attp-core` (==0.1.10)
+  - `msgpack` (>=1.1.1,<2.0.0)
+  - `ascender-framework` (>=2.0rc7,<3.0)
+
+## Development
+
+### Setup
+
+```bash
+# Clone the repository
+git clone https://github.com/AscenderTeam/attp-client.git
+cd attp-client
+
+# Install dependencies
+poetry install
+
+# Run tests
+python -m pytest tests/
+```
+
+### Project Structure
+
+```
+src/
+├── attp_client/           # Main package
+│   ├── client.py         # Main client class
+│   ├── inference.py      # AI inference API
+│   ├── tools.py          # Tool management
+│   ├── catalog.py        # Catalog operations
+│   ├── router.py         # Message routing
+│   ├── session.py        # Session management
+│   ├── errors/           # Exception classes
+│   ├── interfaces/       # Protocol interfaces
+│   ├── misc/             # Utilities
+│   ├── types/            # Type definitions
+│   └── utils/            # Helper utilities
+└── tests/                # Test suite
+```
+
+## License
+
+MIT License - see LICENSE file for details.
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## Support
+
+For issues and questions, please use the GitHub Issues page or contact the Ascender Team.

attp_client-0.0.1/README.md

@@ -0,0 +1,260 @@
+# ATTP Client
+
+A Python SDK client for interacting with AgentHub's ATTP (Agent Tool Transport Protocol).
+
+## Overview
+
+The ATTP Client provides a comprehensive Python interface for connecting to and interacting with AgentHub's ATTP protocol. It enables real-time communication with AI agents, tool management, and inference operations through a high-performance async/await interface.
+
+## Features
+
+- **Async/Await Support**: Built with modern Python async patterns for optimal performance
+- **Agent Communication**: Direct communication with AI agents through the ATTP protocol
+- **Tool Management**: Register, unregister, and manage tools in organized catalogs
+- **Inference API**: Invoke AI agent inference with configurable parameters
+- **Real-time Messaging**: Stream responses and handle real-time agent interactions
+- **Event Handling**: Register custom event handlers for connect/disconnect and custom routes
+- **Authentication**: Secure token-based authentication with organization support
+
+## Installation
+
+```bash
+pip install attp-client
+```
+
+Or with Poetry:
+
+```bash
+poetry add attp-client
+```
+
+## Quick Start
+
+### Basic Connection
+
+```python
+import asyncio
+from attp_client.client import ATTPClient
+
+async def main():
+    # Initialize the client
+    client = ATTPClient(
+        agt_token="your_agent_token_here",
+        organization_id=1,
+        connection_url="attp://localhost:6563"  # Optional, defaults to localhost
+    )
+    
+    # Connect to AgentHub
+    await client.connect()
+    
+    # Your code here...
+    
+    # Close the connection
+    await client.close()
+
+if __name__ == "__main__":
+    asyncio.run(main())
+```
+
+### Agent Inference
+
+```python
+from attp_client.interfaces.inference.message import IMessageDTOV2
+from attp_client.interfaces.inference.enums.message_type import MessageTypeEnum
+from uuid import UUID
+
+# Invoke inference by agent ID
+response = await client.inference.invoke_inference(
+    agent_id=17,
+    input_configuration={},
+    messages=[
+        IMessageDTOV2(
+            content="Hello, how can you help me?",
+            message_type=MessageTypeEnum.USER_MESSAGE,
+            chat_id=UUID("your-chat-id-here")
+        )
+    ],
+    stream=False,
+    timeout=30
+)
+
+print("Agent response:", response)
+```
+
+### Tool Management
+
+```python
+# Register a tool
+tool_id = await client.tools.register(
+    catalog_name="my_catalog",
+    name="example_tool",
+    description="An example tool for demonstration",
+    schema_id="tool_schema_v1",
+    return_direct=False,
+    timeout_ms=20000
+)
+
+# Access a specific catalog
+catalog = await client.catalog("my_catalog")
+
+# Unregister a tool
+await client.tools.unregister("my_catalog", str(tool_id))
+```
+
+### Event Handling
+
+```python
+def on_connect():
+    print("Connected to AgentHub!")
+
+def on_disconnect():
+    print("Disconnected from AgentHub!")
+
+def handle_custom_event(data):
+    print(f"Received custom event: {data}")
+
+# Register event handlers
+client.add_event_handler("", "connect", on_connect)
+client.add_event_handler("", "disconnect", on_disconnect)
+client.add_event_handler("custom:pattern", "custom", handle_custom_event)
+```
+
+## Configuration
+
+### Client Parameters
+
+- `agt_token` (str): Your AgentHub authentication token
+- `organization_id` (int): Your organization ID
+- `connection_url` (str, optional): ATTP server URL (default: "attp://localhost:6563")
+- `max_retries` (int, optional): Maximum connection retry attempts (default: 20)
+- `limits` (Limits, optional): Connection limits configuration
+- `logger` (Logger, optional): Custom logger instance
+
+### Connection Limits
+
+```python
+from attp_core.rs_api import Limits
+
+limits = Limits(max_payload_size=50000)
+client = ATTPClient(
+    agt_token="...",
+    organization_id=1,
+    limits=limits
+)
+```
+
+## API Reference
+
+### ATTPClient
+
+The main client class for ATTP communication.
+
+#### Methods
+
+- `connect()`: Establish connection to AgentHub
+- `close()`: Close the connection
+- `catalog(catalog_name: str)`: Access a specific tool catalog
+- `add_event_handler(pattern: str, route_type: RouteType, callback: Callable)`: Register event handlers
+
+#### Properties
+
+- `inference`: Access to the inference API
+- `tools`: Access to the tools manager
+- `router`: Access to the low-level router
+
+### AttpInferenceAPI
+
+Handles AI agent inference operations.
+
+#### Methods
+
+- `invoke_inference()`: Invoke inference for a specific agent
+- `invoke_chat_inference()`: Invoke inference for a chat session
+
+### ToolsManager
+
+Manages tool registration and organization.
+
+#### Methods
+
+- `register()`: Register a new tool
+- `unregister()`: Remove tool registration
+
+## Error Handling
+
+The client includes comprehensive error handling:
+
+```python
+from attp_client.errors import (
+    AttpException,
+    DeadSessionError,
+    NotFoundError,
+    SerializationError,
+    UnauthenticatedError
+)
+
+try:
+    await client.connect()
+except UnauthenticatedError:
+    print("Authentication failed - check your token")
+except DeadSessionError:
+    print("Session has died - reconnection required")
+except AttpException as e:
+    print(f"ATTP error: {e}")
+```
+
+## Requirements
+
+- Python 3.11 - 3.13
+- Dependencies:
+  - `pydantic` (>=2.11.7,<3.0.0)
+  - `attp-core` (==0.1.10)
+  - `msgpack` (>=1.1.1,<2.0.0)
+  - `ascender-framework` (>=2.0rc7,<3.0)
+
+## Development
+
+### Setup
+
+```bash
+# Clone the repository
+git clone https://github.com/AscenderTeam/attp-client.git
+cd attp-client
+
+# Install dependencies
+poetry install
+
+# Run tests
+python -m pytest tests/
+```
+
+### Project Structure
+
+```
+src/
+├── attp_client/           # Main package
+│   ├── client.py         # Main client class
+│   ├── inference.py      # AI inference API
+│   ├── tools.py          # Tool management
+│   ├── catalog.py        # Catalog operations
+│   ├── router.py         # Message routing
+│   ├── session.py        # Session management
+│   ├── errors/           # Exception classes
+│   ├── interfaces/       # Protocol interfaces
+│   ├── misc/             # Utilities
+│   ├── types/            # Type definitions
+│   └── utils/            # Helper utilities
+└── tests/                # Test suite
+```
+
+## License
+
+MIT License - see LICENSE file for details.
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## Support
+
+For issues and questions, please use the GitHub Issues page or contact the Ascender Team.

attp_client-0.0.1/pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "attp-client"
+version = "0.0.1"
+description = "A python-sdk client for interacting with AgentHub's ATTP protocol (Agent Tool Transport Protocol)"
+authors = [
+    {name = "Ascender Team"}
+]
+license = {text = "MIT"}
+readme = "README.md"
+requires-python = ">=3.11,<3.14"
+dependencies = [
+    "pydantic (>=2.11.7,<3.0.0)",
+    "attp-core (==0.1.10)",
+    "msgpack (>=1.1.1,<2.0.0)",
+    "ascender-framework (>=2.0rc7,<3.0)"
+]
+
+[tool.poetry]
+package-mode = true
+packages = [
+    { include = "attp_client", from = "src" }
+]
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"

attp_client-0.0.1/src/attp_client/catalog.py

@@ -0,0 +1,59 @@
+from typing import Any, Callable, MutableMapping
+from attp_client.errors.not_found import NotFoundError
+from attp_client.tools import ToolsManager
+
+
+class AttpCatalog:
+    attached_tools: MutableMapping[str, Callable[..., Any]]
+    tool_name_to_id_symlink: MutableMapping[str, str]
+    
+    def __init__(
+        self,
+        id: int,
+        catalog_name: str,
+        manager: ToolsManager
+    ) -> None:
+        self.id = id
+        self.catalog_name = catalog_name
+        self.tool_manager = manager
+        self.attached_tools = {}
+        self.tool_name_to_id_symlink = {}
+    
+    async def attach_tool(
+        self,
+        callback: Callable[..., Any],
+        name: str, 
+        description: str | None = None,
+        schema_id: str | None = None,
+        *,
+        return_direct: bool = False,
+        schema_ver: str = "1.0",
+        timeout_ms: float = 20000,
+        idempotent: bool = False
+    ):
+        assigned_id = await self.tool_manager.register(
+            self.catalog_name,
+            name=name,
+            description=description,
+            schema_id=schema_id,
+            return_direct=return_direct,
+            schema_ver=schema_ver,
+            timeout_ms=timeout_ms,
+            idempotent=idempotent
+        )
+        
+        self.attached_tools[str(assigned_id)] = callback
+        self.tool_name_to_id_symlink[name] = str(assigned_id)
+        return assigned_id
+    
+    async def detatch_tool(
+        self,
+        name: str
+    ):
+        tool_id = self.tool_name_to_id_symlink.get(name)
+        
+        if not tool_id:
+            raise NotFoundError(f"Tool {name} not marked as registered and wasn't found in the catalog {self.catalog_name}.")
+        
+        await self.tool_manager.unregister(self.catalog_name, tool_id)
+        return tool_id