augagent 0.1.0__tar.gz

augagent-0.1.0/PKG-INFO

@@ -0,0 +1,139 @@
+Metadata-Version: 2.4
+Name: augagent
+Version: 0.1.0
+Summary: A multi-agent Python framework with intuitive DX, powered by Pydantic.
+Author: augagent contributors
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/augagent/augagent
+Project-URL: Documentation, https://augagent.readthedocs.io
+Project-URL: Repository, https://github.com/augagent/augagent
+Project-URL: Issues, https://github.com/augagent/augagent/issues
+Keywords: agents,multi-agent,ai,llm,pydantic,framework
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic>=2.0
+Requires-Dist: httpx>=0.24
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21; extra == "dev"
+Requires-Dist: ruff>=0.1; extra == "dev"
+Requires-Dist: mypy>=1.5; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: mkdocs>=1.5; extra == "docs"
+Requires-Dist: mkdocs-material>=9.0; extra == "docs"
+Requires-Dist: mkdocstrings[python]>=0.23; extra == "docs"
+
+# AugAgent 🤖
+
+**AugAgent** is a multi-agent framework designed in Python, providing a deeply intuitive Developer Experience (DX) heavily inspired by industry standards like CrewAI and AutoGen, but supercharged with **Pydantic** for rigid type safety and data validation.
+
+By utilizing strict Pydantic schemas under the hood, AugAgent ensures that Language Models produce predictable, structured outputs, effectively eliminating hallucinated parameters and malformed function arguments before they ever reach your runtime code.
+
+## 🚀 Why AugAgent?
+
+- **Familiar, Intuitive DX**: If you've used CrewAI or AutoGen, you'll feel right at home. The `AugAgent` (Role/Goal/Backstory), `AugTask`, and `AugTeam` abstractions are heavily aligned with the latest industry orchestration standards.
+- **Pydantic Native**: LLM configurations, chat schemas, and tool arguments are built on Pydantic v2.
+- **OpenAI Compatible**: Easily plug in any OpenAI-compatible API endpoint (e.g. OpenAI, Azure, vLLM, Ollama) directly into the `LLMConfig`.
+- **Automatic Tool Schemas**: The `@aug_tool` decorator dynamically generates strict OpenAI function-calling JSON schemas from standard Python type hints.
+- **Context Chaining**: `AugTeam` orchestrators automatically chain task outputs as context for subsequent tasks.
+
+---
+
+## 🛠️ Quick Start
+
+Install AugAgent:
+```bash
+pip install augagent
+```
+
+Below is an example of an orchestrated Research and Writing team using the framework:
+
+```python
+import asyncio
+from augagent import AugAgent, AugTask, AugTeam, aug_tool, LLMConfig
+from pydantic import Field
+
+# 1. Define tools
+
+@aug_tool
+def search_web(query: str = Field(description="Search query")) -> str:
+    """Search the web for information."""
+    print(f"  [Tool Executing] Searching the web for: {query}")
+    # Mock search results for demonstration
+    if "quantum" in query.lower():
+        return "Quantum computing uses qubits. It can solve complex problems faster than classical computers."
+    return f"Found generic results for {query}."
+
+@aug_tool
+def fetch_article(url: str = Field(description="URL to fetch")) -> str:
+    """Fetch the contents of an article."""
+    print(f"  [Tool Executing] Fetching article: {url}")
+    return "This is the content of the article regarding " + url
+
+# 2. Define agents
+
+llm_config = LLMConfig(model="gpt-4o-mini", temperature=0.2)
+
+researcher = AugAgent(
+    name="Researcher",
+    role="Senior Technology Researcher",
+    goal="Find comprehensive and accurate information on emerging technologies.",
+    backstory="You have 10 years of experience researching deep tech.",
+    llm_config=llm_config,
+    tools=[search_web, fetch_article],
+    verbose=True
+)
+
+writer = AugAgent(
+    name="Writer",
+    role="Technical Content Writer",
+    goal="Write engaging, accurate articles about technology.",
+    backstory="You write clear, accessible articles for a wide audience.",
+    llm_config=llm_config,
+    verbose=True
+)
+
+# 3. Define tasks
+
+research_task = AugTask(
+    description="Research the topic: {topic}. Find key facts and recent advancements.",
+    expected_output="A bulleted list of key facts about {topic}.",
+    agent=researcher
+)
+
+writing_task = AugTask(
+    description="Write a short, engaging article about {topic} based on the research provided.",
+    expected_output="A 2-paragraph article in Markdown format.",
+    agent=writer
+)
+
+# 4. Assemble the team and kickoff
+
+team = AugTeam(
+    agents=[researcher, writer],
+    tasks=[research_task, writing_task],
+    verbose=True
+)
+
+if __name__ == "__main__":
+    print(f"=== Kicking off AugTeam ===")
+    results = team.kickoff(inputs={"topic": "Quantum Computing"})
+    
+    print("\n=== Final Results ===")
+    for idx, result in enumerate(results):
+        print(f"\nTask {idx + 1} ({result.agent_name}):")
+        print("-" * 40)
+        print(result.output)
+```
+
+## 📚 Documentation
+- [Architecture & Design Choices (WHY)](docs/ARCHITECTURE_WHY.md)
+- [Antigravity Ledger (HOW)](docs/ANTIGRAVITY_LEDGER.md)

augagent-0.1.0/README.md

@@ -0,0 +1,106 @@
+# AugAgent 🤖
+
+**AugAgent** is a multi-agent framework designed in Python, providing a deeply intuitive Developer Experience (DX) heavily inspired by industry standards like CrewAI and AutoGen, but supercharged with **Pydantic** for rigid type safety and data validation.
+
+By utilizing strict Pydantic schemas under the hood, AugAgent ensures that Language Models produce predictable, structured outputs, effectively eliminating hallucinated parameters and malformed function arguments before they ever reach your runtime code.
+
+## 🚀 Why AugAgent?
+
+- **Familiar, Intuitive DX**: If you've used CrewAI or AutoGen, you'll feel right at home. The `AugAgent` (Role/Goal/Backstory), `AugTask`, and `AugTeam` abstractions are heavily aligned with the latest industry orchestration standards.
+- **Pydantic Native**: LLM configurations, chat schemas, and tool arguments are built on Pydantic v2.
+- **OpenAI Compatible**: Easily plug in any OpenAI-compatible API endpoint (e.g. OpenAI, Azure, vLLM, Ollama) directly into the `LLMConfig`.
+- **Automatic Tool Schemas**: The `@aug_tool` decorator dynamically generates strict OpenAI function-calling JSON schemas from standard Python type hints.
+- **Context Chaining**: `AugTeam` orchestrators automatically chain task outputs as context for subsequent tasks.
+
+---
+
+## 🛠️ Quick Start
+
+Install AugAgent:
+```bash
+pip install augagent
+```
+
+Below is an example of an orchestrated Research and Writing team using the framework:
+
+```python
+import asyncio
+from augagent import AugAgent, AugTask, AugTeam, aug_tool, LLMConfig
+from pydantic import Field
+
+# 1. Define tools
+
+@aug_tool
+def search_web(query: str = Field(description="Search query")) -> str:
+    """Search the web for information."""
+    print(f"  [Tool Executing] Searching the web for: {query}")
+    # Mock search results for demonstration
+    if "quantum" in query.lower():
+        return "Quantum computing uses qubits. It can solve complex problems faster than classical computers."
+    return f"Found generic results for {query}."
+
+@aug_tool
+def fetch_article(url: str = Field(description="URL to fetch")) -> str:
+    """Fetch the contents of an article."""
+    print(f"  [Tool Executing] Fetching article: {url}")
+    return "This is the content of the article regarding " + url
+
+# 2. Define agents
+
+llm_config = LLMConfig(model="gpt-4o-mini", temperature=0.2)
+
+researcher = AugAgent(
+    name="Researcher",
+    role="Senior Technology Researcher",
+    goal="Find comprehensive and accurate information on emerging technologies.",
+    backstory="You have 10 years of experience researching deep tech.",
+    llm_config=llm_config,
+    tools=[search_web, fetch_article],
+    verbose=True
+)
+
+writer = AugAgent(
+    name="Writer",
+    role="Technical Content Writer",
+    goal="Write engaging, accurate articles about technology.",
+    backstory="You write clear, accessible articles for a wide audience.",
+    llm_config=llm_config,
+    verbose=True
+)
+
+# 3. Define tasks
+
+research_task = AugTask(
+    description="Research the topic: {topic}. Find key facts and recent advancements.",
+    expected_output="A bulleted list of key facts about {topic}.",
+    agent=researcher
+)
+
+writing_task = AugTask(
+    description="Write a short, engaging article about {topic} based on the research provided.",
+    expected_output="A 2-paragraph article in Markdown format.",
+    agent=writer
+)
+
+# 4. Assemble the team and kickoff
+
+team = AugTeam(
+    agents=[researcher, writer],
+    tasks=[research_task, writing_task],
+    verbose=True
+)
+
+if __name__ == "__main__":
+    print(f"=== Kicking off AugTeam ===")
+    results = team.kickoff(inputs={"topic": "Quantum Computing"})
+    
+    print("\n=== Final Results ===")
+    for idx, result in enumerate(results):
+        print(f"\nTask {idx + 1} ({result.agent_name}):")
+        print("-" * 40)
+        print(result.output)
+```
+
+## 📚 Documentation
+- [Architecture & Design Choices (WHY)](docs/ARCHITECTURE_WHY.md)
+- [Antigravity Ledger (HOW)](docs/ANTIGRAVITY_LEDGER.md)

augagent-0.1.0/pyproject.toml

@@ -0,0 +1,68 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "augagent"
+version = "0.1.0"
+description = "A multi-agent Python framework with intuitive DX, powered by Pydantic."
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.10"
+authors = [
+    { name = "augagent contributors" },
+]
+keywords = ["agents", "multi-agent", "ai", "llm", "pydantic", "framework"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Typing :: Typed",
+]
+dependencies = [
+    "pydantic>=2.0",
+    "httpx>=0.24",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=7.0",
+    "pytest-asyncio>=0.21",
+    "ruff>=0.1",
+    "mypy>=1.5",
+]
+docs = [
+    "mkdocs>=1.5",
+    "mkdocs-material>=9.0",
+    "mkdocstrings[python]>=0.23",
+]
+
+[project.urls]
+Homepage = "https://github.com/augagent/augagent"
+Documentation = "https://augagent.readthedocs.io"
+Repository = "https://github.com/augagent/augagent"
+Issues = "https://github.com/augagent/augagent/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.ruff]
+target-version = "py310"
+line-length = 100
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "N", "W", "UP", "ANN", "S", "B"]
+ignore = ["ANN101", "ANN102"]
+
+[tool.mypy]
+python_version = "3.10"
+strict = true
+plugins = ["pydantic.mypy"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+asyncio_mode = "auto"

augagent-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

augagent-0.1.0/src/augagent/__init__.py

@@ -0,0 +1,90 @@
+"""AugAgent — A multi-agent framework with intuitive DX, powered by Pydantic.
+
+Quick Start::
+
+    from augagent import AugAgent, AugTask, AugTeam, aug_tool, LLMConfig
+    from pydantic import Field
+
+    @aug_tool
+    def greet(name: str = Field(description="Person to greet")) -> str:
+        \"\"\"Greet someone by name.\"\"\"
+        return f"Hello, {name}!"
+
+    assistant = AugAgent(
+        name="Assistant",
+        role="Friendly Helper",
+        goal="Help users with their requests",
+        llm_config=LLMConfig(model="gpt-4o"),
+        tools=[greet],
+    )
+
+    task = AugTask(
+        description="Greet the user warmly",
+        expected_output="A friendly greeting message",
+        agent=assistant,
+    )
+
+    team = AugTeam(agents=[assistant], tasks=[task], verbose=True)
+    result = team.kickoff()
+"""
+
+from augagent.agent import Agent, AugAgent
+from augagent.models import (
+    AgentConfig,
+    ChatCompletion,
+    ChatMessage,
+    ChatToolCall,
+    FunctionCall,
+    LLMConfig,
+    Message,
+    Role,
+    TaskResult,
+    TaskStatus,
+    TokenUsage,
+    ToolCall,
+    ToolResponse,
+)
+from augagent.task import AugTask, Task
+from augagent.team import AugTeam, Process, Team
+from augagent.telemetry import AgentLogger, get_logger
+from augagent.tools import AugTool, Tool, aug_tool, tool
+
+__version__ = "0.1.0"
+
+__all__ = [
+    # Core orchestration
+    "AugAgent",
+    "AugTask",
+    "AugTeam",
+    "Process",
+    # Tools
+    "AugTool",
+    "aug_tool",
+    # Backward compatibility aliases
+    "Agent",
+    "Task",
+    "Team",
+    "Tool",
+    "tool",
+    # Models — LLM configuration
+    "LLMConfig",
+    # Models — chat completion response
+    "ChatCompletion",
+    "ChatMessage",
+    "ChatToolCall",
+    "FunctionCall",
+    "TokenUsage",
+    # Models — internal
+    "AgentConfig",
+    "Message",
+    "Role",
+    "TaskResult",
+    "TaskStatus",
+    "ToolCall",
+    "ToolResponse",
+    # Telemetry
+    "AgentLogger",
+    "get_logger",
+    # Metadata
+    "__version__",
+]