voxagent 0.1.0__py3-none-any.whl

voxagent/types/messages.py

@@ -0,0 +1,134 @@
+"""Message types for voxagent.
+
+This module defines the core message types used in agent conversations:
+- ContentBlock types (TextBlock, ImageBlock, ToolUseBlock, ToolResultBlock)
+- ToolCall and ToolResult for tool interactions
+- Message for conversation messages
+"""
+
+from typing import Annotated, Any, Literal, Union
+
+from pydantic import BaseModel, Field
+
+
+class TextBlock(BaseModel):
+    """A text content block.
+
+    Attributes:
+        type: Discriminator field, always "text".
+        text: The text content.
+    """
+
+    type: Literal["text"] = "text"
+    text: str
+
+
+class ImageBlock(BaseModel):
+    """An image content block with base64-encoded data.
+
+    Attributes:
+        type: Discriminator field, always "image".
+        media_type: The MIME type of the image (e.g., "image/png", "image/jpeg").
+        data: Base64-encoded image data.
+    """
+
+    type: Literal["image"] = "image"
+    media_type: str
+    data: str
+
+
+class ToolUseBlock(BaseModel):
+    """A tool use block representing a tool call in assistant content.
+
+    Attributes:
+        type: Discriminator field, always "tool_use".
+        id: Unique identifier for this tool use.
+        name: Name of the tool to call.
+        input: Dictionary of input parameters for the tool.
+    """
+
+    type: Literal["tool_use"] = "tool_use"
+    id: str
+    name: str
+    input: dict[str, Any] = Field(default_factory=dict)
+
+
+class ToolResultBlock(BaseModel):
+    """A tool result block representing a tool's output in user content.
+
+    Attributes:
+        type: Discriminator field, always "tool_result".
+        tool_use_id: The ID of the corresponding ToolUseBlock.
+        tool_name: Optional name of the tool (for providers that need it).
+        content: The result content from the tool.
+        is_error: Whether the result represents an error.
+    """
+
+    type: Literal["tool_result"] = "tool_result"
+    tool_use_id: str
+    tool_name: str | None = None
+    content: str
+    is_error: bool = False
+
+
+# ContentBlock is a discriminated union of all content block types
+# Using Pydantic's discriminator for robust JSON parsing
+ContentBlock = Annotated[
+    Union[TextBlock, ImageBlock, ToolUseBlock, ToolResultBlock],
+    Field(discriminator="type"),
+]
+
+
+class ToolCall(BaseModel):
+    """A tool call request.
+
+    Attributes:
+        id: Unique identifier for this tool call.
+        name: Name of the tool to call.
+        params: Dictionary of parameters for the tool.
+    """
+
+    id: str
+    name: str
+    params: dict[str, Any] = Field(default_factory=dict)
+
+
+class ToolResult(BaseModel):
+    """Result from a tool execution.
+
+    Attributes:
+        tool_use_id: The ID of the corresponding ToolCall.
+        content: The result content from the tool.
+        is_error: Whether the result represents an error.
+    """
+
+    tool_use_id: str
+    content: str
+    is_error: bool = False
+
+
+class Message(BaseModel):
+    """A message in a conversation.
+
+    Attributes:
+        role: The role of the message sender ("user", "assistant", or "system").
+        content: The message content, either a string or a list of ContentBlocks.
+        tool_calls: Optional list of tool calls (only for assistant messages).
+    """
+
+    role: Literal["user", "assistant", "system"]
+    content: str | list[ContentBlock]
+    tool_calls: list[ToolCall] | None = None
+
+
+__all__ = [
+    "ContentBlock",
+    "ImageBlock",
+    "Message",
+    "TextBlock",
+    "ToolCall",
+    "ToolResult",
+    "ToolResultBlock",
+    "ToolUseBlock",
+]
+

voxagent/types/run.py

@@ -0,0 +1,176 @@
+"""Run parameters and result types for voxagent.
+
+This module defines the types used for agent run configuration and results:
+- ModelConfig for model provider and settings
+- AgentConfig for agent behavior settings
+- ToolPolicy for tool access control
+- RunParams for run configuration
+- ToolMeta for tool execution metadata
+- RunResult for run outcome
+"""
+
+from pathlib import Path
+from typing import Any
+
+from pydantic import BaseModel, Field, field_validator
+
+from .messages import Message
+
+
+class ModelConfig(BaseModel):
+    """Configuration for the model to use.
+
+    Attributes:
+        provider: The model provider (e.g., "anthropic", "openai", "ollama").
+        model: The model name (e.g., "claude-3-5-sonnet", "gpt-4o").
+        temperature: Optional sampling temperature.
+        max_tokens: Optional maximum tokens for generation.
+    """
+
+    provider: str
+    model: str
+    temperature: float | None = None
+    max_tokens: int | None = None
+
+    @classmethod
+    def from_string(cls, model_string: str) -> "ModelConfig":
+        """Parse a model string in 'provider:model' format.
+
+        Args:
+            model_string: String in format "provider:model"
+
+        Returns:
+            ModelConfig instance
+
+        Raises:
+            ValueError: If the string doesn't contain a colon separator
+        """
+        if ":" not in model_string:
+            raise ValueError(
+                f"Invalid model string format: '{model_string}'. Expected 'provider:model'"
+            )
+        provider, model = model_string.split(":", 1)
+        return cls(provider=provider, model=model)
+
+    def to_string(self) -> str:
+        """Convert to 'provider:model' string format.
+
+        Returns:
+            String in format "provider:model"
+        """
+        return f"{self.provider}:{self.model}"
+
+
+class AgentConfig(BaseModel):
+    """Configuration for agent behavior.
+
+    Attributes:
+        max_turns: Maximum number of conversation turns. Default 50.
+        context_limit: Maximum context size in tokens. Default 100000.
+    """
+
+    max_turns: int = 50
+    context_limit: int = 100000
+
+
+class ToolPolicy(BaseModel):
+    """Policy for controlling tool access.
+
+    Attributes:
+        allow_list: List of allowed tool names. None means allow all.
+        deny_list: List of denied tool names. Defaults to empty.
+    """
+
+    allow_list: list[str] | None = None
+    deny_list: list[str] = Field(default_factory=list)
+
+
+class RunParams(BaseModel):
+    """Parameters for an agent run.
+
+    Attributes:
+        session_key: Unique session identifier (required, non-empty).
+        prompt: The user prompt to process (required).
+        model: Model configuration (required).
+        config: Agent configuration (required).
+        timeout_ms: Timeout in milliseconds (required, must be > 0).
+        workspace_dir: Working directory for the agent (required).
+        abort_signal: Optional signal for aborting the run.
+        channel: Optional channel identifier.
+        policies: List of tool policies to apply. Defaults to empty.
+    """
+
+    session_key: str
+    prompt: str
+    model: ModelConfig
+    config: AgentConfig
+    timeout_ms: int
+    workspace_dir: Path
+    abort_signal: Any | None = None
+    channel: str | None = None
+    policies: list[ToolPolicy] = Field(default_factory=list)
+
+    @field_validator("session_key")
+    @classmethod
+    def session_key_must_not_be_empty(cls, v: str) -> str:
+        """Validate that session_key is not empty."""
+        if not v:
+            raise ValueError("session_key must not be empty")
+        return v
+
+    @field_validator("timeout_ms")
+    @classmethod
+    def timeout_ms_must_be_positive(cls, v: int) -> int:
+        """Validate that timeout_ms is positive."""
+        if v <= 0:
+            raise ValueError("timeout_ms must be positive")
+        return v
+
+
+class ToolMeta(BaseModel):
+    """Metadata about a tool execution.
+
+    Attributes:
+        tool_name: Name of the tool that was called.
+        tool_call_id: Unique identifier for the tool call.
+        execution_time_ms: Time taken to execute in milliseconds.
+        success: Whether the execution was successful.
+        error: Error message if execution failed.
+    """
+
+    tool_name: str
+    tool_call_id: str
+    execution_time_ms: int = 0
+    success: bool = True
+    error: str | None = None
+
+
+class RunResult(BaseModel):
+    """Result of an agent run.
+
+    Attributes:
+        messages: List of messages from the conversation.
+        assistant_texts: List of text responses from the assistant.
+        tool_metas: List of tool execution metadata.
+        aborted: Whether the run was aborted.
+        timed_out: Whether the run timed out.
+        error: Error message if the run failed.
+    """
+
+    messages: list[Message]
+    assistant_texts: list[str]
+    tool_metas: list[ToolMeta] = Field(default_factory=list)
+    aborted: bool = False
+    timed_out: bool = False
+    error: str | None = None
+
+
+__all__ = [
+    "AgentConfig",
+    "ModelConfig",
+    "RunParams",
+    "RunResult",
+    "ToolMeta",
+    "ToolPolicy",
+]
+

voxagent-0.1.0.dist-info/METADATA

@@ -0,0 +1,186 @@
+Metadata-Version: 2.4
+Name: voxagent
+Version: 0.1.0
+Summary: A lightweight, model-agnostic LLM provider abstraction with streaming and tool support
+Project-URL: Homepage, https://github.com/lensator/voxagent
+Project-URL: Documentation, https://github.com/lensator/voxagent#readme
+Project-URL: Repository, https://github.com/lensator/voxagent
+Project-URL: Issues, https://github.com/lensator/voxagent/issues
+Project-URL: Changelog, https://github.com/lensator/voxagent/blob/main/CHANGELOG.md
+Author: voxDomus team
+License-Expression: MIT
+Keywords: agent,ai,anthropic,llm,mcp,openai,streaming,tools
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+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
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Requires-Python: >=3.11
+Requires-Dist: anyio>=4.0
+Requires-Dist: httpx>=0.25
+Requires-Dist: pydantic>=2.0
+Provides-Extra: all
+Requires-Dist: anthropic>=0.25; extra == 'all'
+Requires-Dist: google-generativeai>=0.5; extra == 'all'
+Requires-Dist: groq>=0.4; extra == 'all'
+Requires-Dist: mcp>=1.0; extra == 'all'
+Requires-Dist: ollama>=0.2; extra == 'all'
+Requires-Dist: openai>=1.0; extra == 'all'
+Requires-Dist: tiktoken>=0.5; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.25; extra == 'anthropic'
+Provides-Extra: dev
+Requires-Dist: mypy>=1.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
+Requires-Dist: pytest-cov>=4.0; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: ruff>=0.1.0; extra == 'dev'
+Provides-Extra: google
+Requires-Dist: google-generativeai>=0.5; extra == 'google'
+Provides-Extra: groq
+Requires-Dist: groq>=0.4; extra == 'groq'
+Provides-Extra: mcp
+Requires-Dist: mcp>=1.0; extra == 'mcp'
+Provides-Extra: ollama
+Requires-Dist: ollama>=0.2; extra == 'ollama'
+Provides-Extra: openai
+Requires-Dist: openai>=1.0; extra == 'openai'
+Requires-Dist: tiktoken>=0.5; extra == 'openai'
+Description-Content-Type: text/markdown
+
+# voxagent
+
+[![PyPI version](https://badge.fury.io/py/voxagent.svg)](https://badge.fury.io/py/voxagent)
+[![Python 3.11+](https://img.shields.io/badge/python-3.11+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Typed](https://img.shields.io/badge/typed-yes-green.svg)](https://peps.python.org/pep-0561/)
+
+A lightweight, model-agnostic LLM provider abstraction with streaming and tool support.
+
+## Features
+
+- **Multi-Provider**: Unified interface for OpenAI, Anthropic, Google, Groq, Ollama
+- **Streaming**: Typed `StreamChunk` union (TextDelta, ToolUse, MessageEnd, Error)
+- **Tool System**: `@tool` decorator for easy function-to-tool conversion
+- **MCP Integration**: First-class Model Context Protocol support
+- **Type Safe**: Full type hints with `py.typed` marker
+- **Minimal Dependencies**: Core requires only `pydantic`, `httpx`, `anyio`
+
+## Installation
+
+```bash
+# Core only (no provider SDKs)
+pip install voxagent
+
+# With specific providers
+pip install voxagent[openai]
+pip install voxagent[anthropic]
+pip install voxagent[google]
+pip install voxagent[ollama]
+
+# All providers
+pip install voxagent[all]
+```
+
+## Quick Start
+
+```python
+import asyncio
+from voxagent import Agent
+
+async def main():
+    agent = Agent(model="openai:gpt-4o")
+    result = await agent.run("Hello, world!")
+    print(result.output)
+
+asyncio.run(main())
+```
+
+## Streaming
+
+```python
+from voxagent import Agent
+from voxagent.providers import TextDeltaChunk
+
+agent = Agent(model="anthropic:claude-3-5-sonnet")
+
+async for chunk in agent.stream("Tell me a story"):
+    if isinstance(chunk, TextDeltaChunk):
+        print(chunk.delta, end="", flush=True)
+```
+
+## Tools
+
+```python
+from voxagent import Agent
+from voxagent.tools import tool
+
+@tool()
+def get_weather(city: str) -> str:
+    """Get the current weather for a city."""
+    return f"Sunny, 72°F in {city}"
+
+agent = Agent(
+    model="openai:gpt-4o",
+    tools=[get_weather],
+)
+
+result = await agent.run("What's the weather in Paris?")
+```
+
+## Supported Providers
+
+| Provider | Model Format | Example |
+|----------|--------------|---------|
+| OpenAI | `openai:model` | `openai:gpt-4o` |
+| Anthropic | `anthropic:model` | `anthropic:claude-3-5-sonnet` |
+| Google | `google:model` | `google:gemini-1.5-pro` |
+| Groq | `groq:model` | `groq:llama-3.1-70b` |
+| Ollama | `ollama:model` | `ollama:llama3.2` |
+
+## API Reference
+
+### Agent
+
+```python
+from voxagent import Agent
+
+agent = Agent(
+    model="provider:model",      # Required: provider:model string
+    system_prompt="...",         # Optional: system instructions
+    tools=[...],                 # Optional: list of tools
+    temperature=0.7,             # Optional: sampling temperature
+)
+
+# Single response
+result = await agent.run("prompt")
+
+# Streaming
+async for chunk in agent.stream("prompt"):
+    ...
+```
+
+### StreamChunk Types
+
+```python
+from voxagent.providers import (
+    TextDeltaChunk,    # Text content
+    ToolUseChunk,      # Tool invocation
+    MessageEndChunk,   # End of message
+    ErrorChunk,        # Error occurred
+)
+```
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md) for development setup and guidelines.
+
+## License
+
+MIT License - see [LICENSE](LICENSE) for details.
+

voxagent-0.1.0.dist-info/RECORD

@@ -0,0 +1,53 @@
+voxagent/__init__.py,sha256=YMYC95iwWXK26hicGYmd2erNOInrYtohwUVOuNmpTCs,3927
+voxagent/_version.py,sha256=UZNgW5i4fgEyRh_mbeD3SQyS6bLMfXkimp37vJd2Kso,129
+voxagent/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+voxagent/agent/__init__.py,sha256=eASoU7Zhvw8BtJ-iUqVN06S4fMLkHwDgUZbHeH2AUOM,755
+voxagent/agent/abort.py,sha256=2Wnnxq8Dcn7wQkKPHrba2o0OeOdzF4NNsl-usgE4CJw,5191
+voxagent/agent/core.py,sha256=zXDUubx6oWQCj1V997s2zJ0Xj0XCaWIPZNinTNMz2zY,30581
+voxagent/code/__init__.py,sha256=6-i-FjYxwNay7XvWt9kwwg2nk3wPNzRp54AtEBdQ13U,237
+voxagent/mcp/__init__.py,sha256=_3Rsn7nIuivdWLv0MzpyjRGsPuCgr4LrXCge6FCb3nE,470
+voxagent/mcp/manager.py,sha256=sECOhw-f6HB6NV-mBqcgJzsEt28acdQI_O2cT-41Rxw,6606
+voxagent/mcp/tool.py,sha256=j_p5Kmra_g69vucohQRr0xguy6lXPFzwSAdX4OFsNK4,5113
+voxagent/providers/__init__.py,sha256=lUmbIoiVFSX-CqfrVYiS5Rbriy_U061NR0vC4qChWR8,3202
+voxagent/providers/anthropic.py,sha256=A9aJUBc2nwqFdtvh5CyHp0GSQR82DaxwJQsgA79cffg,15889
+voxagent/providers/augment.py,sha256=K8ORZsOh88FBh4d7pYZUgMQHtQXnhJizpjwGkbWf53Y,8623
+voxagent/providers/auth.py,sha256=X3yDSHzj7gP-XxiYVpTSpDZs608gxnd-0UYX5aNNsEw,3650
+voxagent/providers/base.py,sha256=f_tPz4LFWX2nU9q1aa__Z3UpT35VpQnDVr40Z3VM6uk,6933
+voxagent/providers/chatgpt.py,sha256=DBjoc8UTONxwz2BGhsPVevSJBlLduai4gjzt62_lZrE,15620
+voxagent/providers/claudecode.py,sha256=4pUnWO9IJlXnagnqu3bl6SIDdyjpDohjOwJqRe96g6U,4331
+voxagent/providers/cli_base.py,sha256=0N4psjsH9ekgfq6ynspPQ48iB5BmN6qiOSDPMHjJdBA,7554
+voxagent/providers/codex.py,sha256=NUtCH1fqf1OTI2JEOlNClofm6spWq4ZvlaPaHbyotyE,5397
+voxagent/providers/failover.py,sha256=fBwzD_iJ1bJ3W7kQzT0qoogTXg0G9NMbAucJOIT3R5k,2719
+voxagent/providers/google.py,sha256=f8QdbZxqi6L5syzmYAOiCVnJwN0Bs1ym3nDNvCbQ8Fg,18239
+voxagent/providers/groq.py,sha256=KEwscwKBJdft6ZST3nTEcodh06AxoAzTsDvIr7O-pzQ,2765
+voxagent/providers/ollama.py,sha256=WKHgWMfo4cAFwKYxRe2m3sNOg91OlUL1Uo8MUH6hzuQ,13166
+voxagent/providers/openai.py,sha256=bs8iwU0qy35obV1cjv6gCzsOKBR-BSrq8p5Z2oO_hmE,14475
+voxagent/providers/registry.py,sha256=kcHBlbveiVufJTDxUQ65GBVm5qJdxycsXioWJ2SJeTI,5441
+voxagent/security/__init__.py,sha256=EF3Z30AMQNCOoyjsa8QNG6EA4uSyKv54exkM-5HYylE,361
+voxagent/security/events.py,sha256=9yb0PC9iX8MZC53I-nNcKE_4iQRdCB2FE4lvDxgJT18,2123
+voxagent/security/filter.py,sha256=d-ZbZi2MrwJB8ulY_lBXxFfS_YY-W-kcapRvvdLNEp4,4932
+voxagent/security/registry.py,sha256=Nt2s92o0UPTbgoxHzwfb5HU6U0R-nbf6oZdgGIC66FY,2657
+voxagent/session/__init__.py,sha256=OCwAh0UM-yv6BPFVIGKVkB_tuMW_Ixs5K8P7fIKx9u8,981
+voxagent/session/compaction.py,sha256=8ksRKPkYpF6S2jO3lsVqkrLpDmc2SxRGSQWmf2A_yhg,7560
+voxagent/session/lock.py,sha256=5T1Lymf4A9B1s9TvwFCjUgG6zaNzi7d5HC_rbdImc8Y,3294
+voxagent/session/model.py,sha256=MkdbwMmvT5PFiCJZl0arW2uC64SPcAOuWIDEIwf2M-k,3592
+voxagent/session/storage.py,sha256=ma0UWbOeVeIdYcfzMwGtw1QGSoosdY1NPo2lKd-x7ms,5829
+voxagent/streaming/__init__.py,sha256=gMiAnqVB24Y9KwkgolkMTtySyD8XnBe4Iva6szCPWXc,1092
+voxagent/streaming/emitter.py,sha256=GEAzkbFjG7KZcCYUY5YRdBb0sU5eVw3NevVcEC9W9Sw,9513
+voxagent/streaming/events.py,sha256=unWh38vxSiDoXU02x3e-TM0tclNsPR7JsH5YrkIFb_g,6996
+voxagent/subagent/__init__.py,sha256=sH7Z6wqWJiR1SSvC1z20XizwWKFWIkUs4KwJTvwr0rc,487
+voxagent/subagent/context.py,sha256=2JCzHxavVASkeTfS99Q6tzPamT3moLgHM9MFWfFs7m8,4117
+voxagent/subagent/definition.py,sha256=pbBvuUdUjfCeFbTASGhd_ljceADHy9hL2iEuMxDtHlk,5803
+voxagent/tools/__init__.py,sha256=JJ66f_fjwdcVmLwyp3B0sQA1RVPZN0PdS-SJPMU2nGU,867
+voxagent/tools/context.py,sha256=0DWC2QF-1vWMmst-4IoEtAovzE-g1f8hhmboQpXWH1w,1343
+voxagent/tools/decorator.py,sha256=fQj2kOcCeVRZ21XbrZ9TDBx6OOFZ-waxDoLNsGc7z-o,5030
+voxagent/tools/definition.py,sha256=OfYwO402gDGbbRjjcr0dJTWsEBOQLkhJmdIIiVSHA6A,4113
+voxagent/tools/executor.py,sha256=fK3DlS9c3ROVx2ixbAiAn-0e13GYBe6jOyHqze3AJig,2678
+voxagent/tools/policy.py,sha256=mCdYYnITrVTg9w5bwiCMOiXknrU1SQza_qZo02juQdg,2496
+voxagent/tools/registry.py,sha256=MNJzgcmKT0AoMWIky9TJY4WVhzn5dkmjIHsUiZ3mv3U,2593
+voxagent/types/__init__.py,sha256=3VunuprKKEpOR9Cg-UITHJXds_xQ-tfqQb4S7wD3nP4,933
+voxagent/types/messages.py,sha256=c6hNi9w6C8gbFoFm5fFge35vwJGywaoR_OiPQprfyVs,3494
+voxagent/types/run.py,sha256=4vYq0pCqH7_7SWbMb1SplWj4TLiE3DELDYMi0HefFmo,5071
+voxagent-0.1.0.dist-info/METADATA,sha256=2Rx_AsdBcjXYytjFNIR03hZ-53mN3Nz-rntojpKAVcc,5610
+voxagent-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+voxagent-0.1.0.dist-info/RECORD,,

voxagent-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any