vibe-aigc 0.1.0__tar.gz → 0.2.0__tar.gz

vibe_aigc-0.2.0/PKG-INFO

@@ -0,0 +1,179 @@
+Metadata-Version: 2.4
+Name: vibe-aigc
+Version: 0.2.0
+Summary: A New Paradigm for Content Generation via Agentic Orchestration
+Author: Vibe AIGC Contributors
+License-Expression: MIT
+Project-URL: Homepage, https://jmanhype.github.io/vibe-aigc
+Project-URL: Repository, https://github.com/jmanhype/vibe-aigc
+Project-URL: Documentation, https://jmanhype.github.io/vibe-aigc
+Project-URL: Changelog, https://jmanhype.github.io/vibe-aigc/changelog
+Project-URL: Issues, https://github.com/jmanhype/vibe-aigc/issues
+Keywords: ai,aigc,agents,orchestration,content-generation,llm,workflow
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: openai>=1.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: mkdocs-material>=9.0.0; extra == "docs"
+Requires-Dist: mkdocstrings[python]>=0.24.0; extra == "docs"
+Requires-Dist: pymdown-extensions>=10.0.0; extra == "docs"
+Dynamic: license-file
+
+# Vibe AIGC
+
+[![CI](https://github.com/jmanhype/vibe-aigc/actions/workflows/ci.yml/badge.svg)](https://github.com/jmanhype/vibe-aigc/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/gh/jmanhype/vibe-aigc/branch/main/graph/badge.svg)](https://codecov.io/gh/jmanhype/vibe-aigc)
+[![PyPI](https://img.shields.io/pypi/v/vibe-aigc.svg)](https://pypi.org/project/vibe-aigc/)
+[![Python 3.12+](https://img.shields.io/badge/python-3.12+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Docs](https://img.shields.io/badge/docs-GitHub%20Pages-blue)](https://jmanhype.github.io/vibe-aigc)
+
+**A New Paradigm for Content Generation via Agentic Orchestration**
+
+Based on [arXiv:2602.04575](https://arxiv.org/abs/2602.04575)
+
+📚 **[Documentation](https://jmanhype.github.io/vibe-aigc)** | 🚀 **[Quick Start](https://jmanhype.github.io/vibe-aigc/getting-started/quickstart/)** | 📖 **[API Reference](https://jmanhype.github.io/vibe-aigc/api/models/)**
+
+---
+
+## What is Vibe AIGC?
+
+Vibe AIGC bridges the **Intent-Execution Gap** in AI content generation. Instead of prompt engineering, you provide a **Vibe** — a high-level representation of your creative intent — and the system automatically decomposes it into executable workflows.
+
+```python
+from vibe_aigc import MetaPlanner, Vibe
+
+# Express your intent
+vibe = Vibe(
+    description="Create a cinematic sci-fi trailer",
+    style="dark, atmospheric, Blade Runner aesthetic",
+    constraints=["under 60 seconds", "no dialogue"]
+)
+
+# Let the Meta-Planner handle the rest
+planner = MetaPlanner()
+result = await planner.execute(vibe)
+```
+
+## Architecture (Paper Section 5)
+
+The implementation follows the paper's three-part architecture:
+
+| Component | Purpose | Module |
+|-----------|---------|--------|
+| **MetaPlanner** | Decomposes Vibes into workflows | `vibe_aigc.planner` |
+| **KnowledgeBase** | Domain expertise for intent understanding | `vibe_aigc.knowledge` |
+| **ToolRegistry** | Atomic tools for content generation | `vibe_aigc.tools` |
+
+```python
+from vibe_aigc import MetaPlanner, Vibe, create_knowledge_base, create_default_registry
+
+# The full architecture
+kb = create_knowledge_base()  # Film, writing, design, music knowledge
+tools = create_default_registry()  # LLM, templates, combine tools
+
+planner = MetaPlanner(knowledge_base=kb, tool_registry=tools)
+
+# Query knowledge for "Hitchcockian suspense" → technical specs
+result = kb.query("Hitchcockian suspense")
+# Returns: camera techniques, lighting specs, editing patterns
+```
+
+## Features
+
+- 🎯 **Vibe-based Planning** — High-level intent → executable workflows
+- 🧠 **Domain Knowledge** — Built-in expertise for film, writing, design, music
+- 🔧 **Tool Library** — Pluggable tools for actual content generation
+- ⚡ **Parallel Execution** — Independent nodes run concurrently
+- 🔄 **Adaptive Replanning** — Automatic recovery from failures
+- 💾 **Checkpoint/Resume** — Save and restore workflow state
+- 📊 **Progress Tracking** — Real-time callbacks and visualization
+- 🎨 **Workflow Visualization** — ASCII and Mermaid diagrams
+
+## Installation
+
+```bash
+pip install vibe-aigc
+```
+
+## CLI Usage
+
+```bash
+# Generate a workflow plan
+vibe-aigc plan "Create a blog post about AI" --style "informative" --format ascii
+
+# Execute a vibe
+vibe-aigc execute "Design a landing page" --visualize --checkpoint
+
+# Manage checkpoints
+vibe-aigc checkpoints --list
+```
+
+## Quick Example
+
+```python
+import asyncio
+from vibe_aigc import Vibe, MetaPlanner
+
+async def main():
+    vibe = Vibe(
+        description="Write a blog post about AI agents",
+        style="informative, engaging",
+        constraints=["under 1000 words"]
+    )
+    
+    planner = MetaPlanner()
+    result = await planner.execute_with_visualization(vibe)
+    
+    print(f"Status: {result.get_summary()['status']}")
+
+asyncio.run(main())
+```
+
+## Architecture
+
+```
+User Vibe → MetaPlanner → Agentic Pipeline → Execution → Result
+     ↑                           ↓
+     └──── Feedback Loop ────────┘
+```
+
+## Documentation
+
+- [Installation Guide](https://jmanhype.github.io/vibe-aigc/getting-started/installation/)
+- [Quick Start Tutorial](https://jmanhype.github.io/vibe-aigc/getting-started/quickstart/)
+- [Core Concepts](https://jmanhype.github.io/vibe-aigc/guide/concepts/)
+- [API Reference](https://jmanhype.github.io/vibe-aigc/api/models/)
+- [Examples](https://jmanhype.github.io/vibe-aigc/examples/)
+
+## Contributing
+
+Contributions welcome! See [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines.
+
+## License
+
+MIT — see [LICENSE](LICENSE) for details.
+
+## Citation
+
+```bibtex
+@article{vibe-aigc-2025,
+  title={Vibe AIGC: A New Paradigm for Content Generation via Agentic Orchestration},
+  journal={arXiv preprint arXiv:2602.04575},
+  year={2025}
+}
+```

vibe_aigc-0.2.0/README.md

@@ -0,0 +1,144 @@
+# Vibe AIGC
+
+[![CI](https://github.com/jmanhype/vibe-aigc/actions/workflows/ci.yml/badge.svg)](https://github.com/jmanhype/vibe-aigc/actions/workflows/ci.yml)
+[![codecov](https://codecov.io/gh/jmanhype/vibe-aigc/branch/main/graph/badge.svg)](https://codecov.io/gh/jmanhype/vibe-aigc)
+[![PyPI](https://img.shields.io/pypi/v/vibe-aigc.svg)](https://pypi.org/project/vibe-aigc/)
+[![Python 3.12+](https://img.shields.io/badge/python-3.12+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Docs](https://img.shields.io/badge/docs-GitHub%20Pages-blue)](https://jmanhype.github.io/vibe-aigc)
+
+**A New Paradigm for Content Generation via Agentic Orchestration**
+
+Based on [arXiv:2602.04575](https://arxiv.org/abs/2602.04575)
+
+📚 **[Documentation](https://jmanhype.github.io/vibe-aigc)** | 🚀 **[Quick Start](https://jmanhype.github.io/vibe-aigc/getting-started/quickstart/)** | 📖 **[API Reference](https://jmanhype.github.io/vibe-aigc/api/models/)**
+
+---
+
+## What is Vibe AIGC?
+
+Vibe AIGC bridges the **Intent-Execution Gap** in AI content generation. Instead of prompt engineering, you provide a **Vibe** — a high-level representation of your creative intent — and the system automatically decomposes it into executable workflows.
+
+```python
+from vibe_aigc import MetaPlanner, Vibe
+
+# Express your intent
+vibe = Vibe(
+    description="Create a cinematic sci-fi trailer",
+    style="dark, atmospheric, Blade Runner aesthetic",
+    constraints=["under 60 seconds", "no dialogue"]
+)
+
+# Let the Meta-Planner handle the rest
+planner = MetaPlanner()
+result = await planner.execute(vibe)
+```
+
+## Architecture (Paper Section 5)
+
+The implementation follows the paper's three-part architecture:
+
+| Component | Purpose | Module |
+|-----------|---------|--------|
+| **MetaPlanner** | Decomposes Vibes into workflows | `vibe_aigc.planner` |
+| **KnowledgeBase** | Domain expertise for intent understanding | `vibe_aigc.knowledge` |
+| **ToolRegistry** | Atomic tools for content generation | `vibe_aigc.tools` |
+
+```python
+from vibe_aigc import MetaPlanner, Vibe, create_knowledge_base, create_default_registry
+
+# The full architecture
+kb = create_knowledge_base()  # Film, writing, design, music knowledge
+tools = create_default_registry()  # LLM, templates, combine tools
+
+planner = MetaPlanner(knowledge_base=kb, tool_registry=tools)
+
+# Query knowledge for "Hitchcockian suspense" → technical specs
+result = kb.query("Hitchcockian suspense")
+# Returns: camera techniques, lighting specs, editing patterns
+```
+
+## Features
+
+- 🎯 **Vibe-based Planning** — High-level intent → executable workflows
+- 🧠 **Domain Knowledge** — Built-in expertise for film, writing, design, music
+- 🔧 **Tool Library** — Pluggable tools for actual content generation
+- ⚡ **Parallel Execution** — Independent nodes run concurrently
+- 🔄 **Adaptive Replanning** — Automatic recovery from failures
+- 💾 **Checkpoint/Resume** — Save and restore workflow state
+- 📊 **Progress Tracking** — Real-time callbacks and visualization
+- 🎨 **Workflow Visualization** — ASCII and Mermaid diagrams
+
+## Installation
+
+```bash
+pip install vibe-aigc
+```
+
+## CLI Usage
+
+```bash
+# Generate a workflow plan
+vibe-aigc plan "Create a blog post about AI" --style "informative" --format ascii
+
+# Execute a vibe
+vibe-aigc execute "Design a landing page" --visualize --checkpoint
+
+# Manage checkpoints
+vibe-aigc checkpoints --list
+```
+
+## Quick Example
+
+```python
+import asyncio
+from vibe_aigc import Vibe, MetaPlanner
+
+async def main():
+    vibe = Vibe(
+        description="Write a blog post about AI agents",
+        style="informative, engaging",
+        constraints=["under 1000 words"]
+    )
+    
+    planner = MetaPlanner()
+    result = await planner.execute_with_visualization(vibe)
+    
+    print(f"Status: {result.get_summary()['status']}")
+
+asyncio.run(main())
+```
+
+## Architecture
+
+```
+User Vibe → MetaPlanner → Agentic Pipeline → Execution → Result
+     ↑                           ↓
+     └──── Feedback Loop ────────┘
+```
+
+## Documentation
+
+- [Installation Guide](https://jmanhype.github.io/vibe-aigc/getting-started/installation/)
+- [Quick Start Tutorial](https://jmanhype.github.io/vibe-aigc/getting-started/quickstart/)
+- [Core Concepts](https://jmanhype.github.io/vibe-aigc/guide/concepts/)
+- [API Reference](https://jmanhype.github.io/vibe-aigc/api/models/)
+- [Examples](https://jmanhype.github.io/vibe-aigc/examples/)
+
+## Contributing
+
+Contributions welcome! See [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines.
+
+## License
+
+MIT — see [LICENSE](LICENSE) for details.
+
+## Citation
+
+```bibtex
+@article{vibe-aigc-2025,
+  title={Vibe AIGC: A New Paradigm for Content Generation via Agentic Orchestration},
+  journal={arXiv preprint arXiv:2602.04575},
+  year={2025}
+}
+```

{vibe_aigc-0.1.0 → vibe_aigc-0.2.0}/pyproject.toml

@@ -2,9 +2,13 @@
 requires = ["setuptools>=61.0", "wheel"]
 build-backend = "setuptools.build_meta"
 
+[tool.setuptools.packages.find]
+include = ["vibe_aigc*"]
+exclude = ["tests*", "docs*", "examples*", "landing*"]
+
 [project]
 name = "vibe-aigc"
-version = "0.1.0"
+version = "0.2.0"
 description = "A New Paradigm for Content Generation via Agentic Orchestration"
 authors = [{name = "Vibe AIGC Contributors"}]
 license = "MIT"
@@ -32,11 +36,20 @@ dev = [
     "ruff>=0.1.0",
     "mypy>=1.0.0",
 ]
+docs = [
+    "mkdocs-material>=9.0.0",
+    "mkdocstrings[python]>=0.24.0",
+    "pymdown-extensions>=10.0.0",
+]
+
+[project.scripts]
+vibe-aigc = "vibe_aigc.cli:main"
 
 [project.urls]
-Homepage = "https://github.com/jmanhype/vibe-aigc"
+Homepage = "https://jmanhype.github.io/vibe-aigc"
 Repository = "https://github.com/jmanhype/vibe-aigc"
-Documentation = "https://github.com/jmanhype/vibe-aigc#readme"
+Documentation = "https://jmanhype.github.io/vibe-aigc"
+Changelog = "https://jmanhype.github.io/vibe-aigc/changelog"
 Issues = "https://github.com/jmanhype/vibe-aigc/issues"
 
 [tool.pytest.ini_options]

{vibe_aigc-0.1.0 → vibe_aigc-0.2.0}/tests/test_adaptive_replanning.py

@@ -282,7 +282,7 @@ class TestAdaptiveReplanning:
         ]
 
         plan_call_count = 0
-        async def plan_side_effect(vibe):
+        async def plan_side_effect(vibe, knowledge_context=None, tools_context=None):
             nonlocal plan_call_count
             result = plans[min(plan_call_count, len(plans) - 1)]
             plan_call_count += 1

vibe_aigc-0.2.0/tests/test_agents.py

@@ -0,0 +1,231 @@
+"""Tests for Specialized Agent Framework."""
+
+import pytest
+from unittest.mock import AsyncMock, MagicMock
+
+from vibe_aigc.agents import (
+    BaseAgent,
+    AgentRole,
+    AgentContext,
+    AgentResult,
+    AgentRegistry,
+    WriterAgent,
+    ResearcherAgent,
+    EditorAgent,
+    DirectorAgent,
+    DesignerAgent,
+    ScreenwriterAgent,
+    ComposerAgent,
+    create_default_agents
+)
+from vibe_aigc.tools import ToolRegistry, ToolResult
+
+
+class TestAgentContext:
+    """Test AgentContext class."""
+    
+    def test_create_context(self):
+        ctx = AgentContext(
+            task="Write a blog post",
+            vibe_description="Technical but accessible",
+            style="informative",
+            constraints=["under 1000 words"]
+        )
+        
+        assert ctx.task == "Write a blog post"
+        assert ctx.style == "informative"
+        assert len(ctx.constraints) == 1
+
+
+class TestAgentResult:
+    """Test AgentResult class."""
+    
+    def test_success_result(self):
+        result = AgentResult(
+            success=True,
+            output="Generated content",
+            artifacts={"image": "url"},
+            messages=["Completed"]
+        )
+        
+        assert result.success
+        assert result.output == "Generated content"
+        assert "image" in result.artifacts
+    
+    def test_to_dict(self):
+        result = AgentResult(success=True, output="test")
+        d = result.to_dict()
+        
+        assert d["success"] is True
+        assert d["output"] == "test"
+
+
+class TestWriterAgent:
+    """Test WriterAgent."""
+    
+    def test_creation(self):
+        agent = WriterAgent()
+        
+        assert agent.name == "Writer"
+        assert agent.role == AgentRole.WRITER
+        assert "llm_generate" in agent._capabilities
+    
+    @pytest.mark.asyncio
+    async def test_execute_without_registry(self):
+        agent = WriterAgent()
+        ctx = AgentContext(task="Write something", vibe_description="Test")
+        
+        result = await agent.execute(ctx)
+        
+        assert not result.success
+        assert "No tool registry" in result.messages[0]
+    
+    @pytest.mark.asyncio
+    async def test_execute_with_mock_registry(self):
+        # Create mock registry
+        mock_registry = MagicMock(spec=ToolRegistry)
+        mock_tool = AsyncMock()
+        mock_tool.execute.return_value = ToolResult(
+            success=True,
+            output={"text": "Generated text", "tokens_used": 100}
+        )
+        mock_registry.get.return_value = mock_tool
+        
+        agent = WriterAgent(tool_registry=mock_registry)
+        ctx = AgentContext(task="Write a haiku", vibe_description="Peaceful")
+        
+        result = await agent.execute(ctx)
+        
+        assert result.success
+        assert "Generated text" in result.output
+
+
+class TestResearcherAgent:
+    """Test ResearcherAgent."""
+    
+    def test_creation(self):
+        agent = ResearcherAgent()
+        
+        assert agent.role == AgentRole.RESEARCHER
+        assert "search" in agent._capabilities
+
+
+class TestEditorAgent:
+    """Test EditorAgent."""
+    
+    def test_creation(self):
+        agent = EditorAgent()
+        
+        assert agent.role == AgentRole.EDITOR
+    
+    @pytest.mark.asyncio
+    async def test_execute_without_content(self):
+        mock_registry = MagicMock(spec=ToolRegistry)
+        mock_registry.get.return_value = None
+        
+        agent = EditorAgent(tool_registry=mock_registry)
+        ctx = AgentContext(task="Edit", vibe_description="Test")
+        
+        result = await agent.execute(ctx)
+        
+        assert not result.success
+        assert "No content" in result.messages[0]
+
+
+class TestDirectorAgent:
+    """Test DirectorAgent."""
+    
+    def test_add_managed_agent(self):
+        director = DirectorAgent()
+        writer = WriterAgent()
+        
+        director.add_agent(writer)
+        
+        assert "Writer" in director._managed_agents
+
+
+class TestDesignerAgent:
+    """Test DesignerAgent."""
+    
+    def test_creation(self):
+        agent = DesignerAgent()
+        
+        assert agent.role == AgentRole.DESIGNER
+        assert "image_generate" in agent._capabilities
+
+
+class TestScreenwriterAgent:
+    """Test ScreenwriterAgent."""
+    
+    def test_creation(self):
+        agent = ScreenwriterAgent()
+        
+        assert agent.role == AgentRole.SCREENWRITER
+
+
+class TestComposerAgent:
+    """Test ComposerAgent."""
+    
+    def test_creation(self):
+        agent = ComposerAgent()
+        
+        assert agent.role == AgentRole.COMPOSER
+        assert "audio_generate" in agent._capabilities
+
+
+class TestAgentRegistry:
+    """Test AgentRegistry."""
+    
+    def test_register_and_get(self):
+        registry = AgentRegistry()
+        agent = WriterAgent()
+        
+        registry.register(agent)
+        
+        assert registry.get("Writer") == agent
+    
+    def test_find_by_role(self):
+        registry = AgentRegistry()
+        registry.register(WriterAgent())
+        registry.register(EditorAgent())
+        
+        writers = registry.find_by_role(AgentRole.WRITER)
+        
+        assert len(writers) == 1
+        assert writers[0].role == AgentRole.WRITER
+    
+    def test_list_agents(self):
+        registry = AgentRegistry()
+        registry.register(WriterAgent())
+        registry.register(EditorAgent())
+        
+        names = registry.list_agents()
+        
+        assert "Writer" in names
+        assert "Editor" in names
+    
+    def test_create_team(self):
+        registry = AgentRegistry()
+        registry.register(WriterAgent())
+        registry.register(EditorAgent())
+        registry.register(DirectorAgent())
+        
+        team = registry.create_team([AgentRole.WRITER, AgentRole.EDITOR])
+        
+        assert AgentRole.WRITER in team
+        assert AgentRole.EDITOR in team
+
+
+class TestCreateDefaultAgents:
+    """Test create_default_agents factory."""
+    
+    def test_creates_all_agents(self):
+        registry = create_default_agents()
+        
+        assert registry.get("Writer") is not None
+        assert registry.get("Researcher") is not None
+        assert registry.get("Editor") is not None
+        assert registry.get("Director") is not None
+        assert registry.get("Designer") is not None
+        assert registry.get("Screenwriter") is not None
+        assert registry.get("Composer") is not None