vibe-aigc 0.5.0__tar.gz → 0.6.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {vibe_aigc-0.5.0/vibe_aigc.egg-info → vibe_aigc-0.6.0}/PKG-INFO +1 -1
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/pyproject.toml +5 -5
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/__init__.py +116 -110
- vibe_aigc-0.6.0/vibe_aigc/composer_general.py +453 -0
- vibe_aigc-0.6.0/vibe_aigc/discovery.py +416 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/knowledge.py +368 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/llm.py +38 -5
- vibe_aigc-0.6.0/vibe_aigc/model_registry.py +790 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/mv_pipeline.py +650 -448
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/planner.py +169 -1
- vibe_aigc-0.6.0/vibe_aigc/vibe_backend.py +381 -0
- vibe_aigc-0.6.0/vibe_aigc/vlm_feedback.py +289 -0
- vibe_aigc-0.6.0/vibe_aigc/workflow_backend.py +318 -0
- vibe_aigc-0.6.0/vibe_aigc/workflow_composer.py +661 -0
- vibe_aigc-0.6.0/vibe_aigc/workflow_executor.py +530 -0
- vibe_aigc-0.6.0/vibe_aigc/workflow_registry.py +609 -0
- vibe_aigc-0.6.0/vibe_aigc/workflow_strategies.py +778 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0/vibe_aigc.egg-info}/PKG-INFO +1 -1
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc.egg-info/SOURCES.txt +10 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/LICENSE +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/README.md +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/setup.cfg +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_adaptive_replanning.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_agents.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_assets.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_auto_checkpoint.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_automatic_checkpoints.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_checkpoint_serialization.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_error_handling.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_executor.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_feedback_system.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_integration.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_knowledge_base.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_metaplanner_resume.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_metaplanner_visualization.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_models.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_parallel_execution.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_planner.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_progress_callbacks.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_tools.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_visualization.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/tests/test_workflow_resume.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/agents.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/assets.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/audio.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/character.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/cli.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/comfyui.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/executor.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/models.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/persistence.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/tools.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/tools_multimodal.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/video.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/visualization.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc/workflows.py +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc.egg-info/dependency_links.txt +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc.egg-info/entry_points.txt +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc.egg-info/requires.txt +0 -0
- {vibe_aigc-0.5.0 → vibe_aigc-0.6.0}/vibe_aigc.egg-info/top_level.txt +0 -0
|
@@ -8,7 +8,7 @@ exclude = ["tests*", "docs*", "examples*", "landing*"]
|
|
|
8
8
|
|
|
9
9
|
[project]
|
|
10
10
|
name = "vibe-aigc"
|
|
11
|
-
version = "0.
|
|
11
|
+
version = "0.6.0"
|
|
12
12
|
description = "A New Paradigm for Content Generation via Agentic Orchestration"
|
|
13
13
|
authors = [{name = "Vibe AIGC Contributors"}]
|
|
14
14
|
license = "MIT"
|
|
@@ -66,7 +66,7 @@ python_version = "3.12"
|
|
|
66
66
|
warn_return_any = true
|
|
67
67
|
warn_unused_configs = true
|
|
68
68
|
ignore_missing_imports = true
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
69
|
+
|
|
70
|
+
|
|
71
|
+
|
|
72
|
+
|
|
@@ -1,112 +1,118 @@
|
|
|
1
|
-
"""Vibe AIGC: A New Paradigm for Content Generation via Agentic Orchestration.
|
|
2
|
-
|
|
3
|
-
This package implements the Vibe AIGC paradigm from the paper:
|
|
4
|
-
"Vibe AIGC: A New Paradigm for Content Generation via Agentic Orchestration"
|
|
5
|
-
|
|
6
|
-
Architecture (Paper Section 5):
|
|
7
|
-
- MetaPlanner: Decomposes Vibes into workflows (Section 5.2)
|
|
8
|
-
- KnowledgeBase: Domain-specific expert knowledge (Section 5.3)
|
|
9
|
-
- ToolRegistry: Atomic tool library for content generation (Section 5.4)
|
|
10
|
-
- Agents: Specialized role-based agents (Section 4 examples)
|
|
11
|
-
- AssetBank: Character and style consistency management
|
|
12
|
-
"""
|
|
13
|
-
|
|
14
|
-
from .models import Vibe, WorkflowPlan, WorkflowNode, WorkflowNodeType
|
|
15
|
-
from .planner import MetaPlanner
|
|
16
|
-
from .llm import LLMClient, LLMConfig
|
|
17
|
-
from .executor import WorkflowExecutor, ExecutionStatus, ExecutionResult
|
|
18
|
-
|
|
19
|
-
# Paper Section 5.3: Domain-Specific Expert Knowledge Base
|
|
20
|
-
from .knowledge import (
|
|
21
|
-
KnowledgeBase,
|
|
22
|
-
DomainKnowledge,
|
|
23
|
-
create_knowledge_base
|
|
24
|
-
)
|
|
25
|
-
|
|
26
|
-
# Paper Section 5.4: Atomic Tool Library
|
|
27
|
-
from .tools import (
|
|
28
|
-
ToolRegistry,
|
|
29
|
-
BaseTool,
|
|
30
|
-
ToolResult,
|
|
31
|
-
ToolSpec,
|
|
32
|
-
ToolCategory,
|
|
33
|
-
LLMTool,
|
|
34
|
-
TemplateTool,
|
|
35
|
-
CombineTool,
|
|
36
|
-
create_default_registry
|
|
37
|
-
)
|
|
38
|
-
|
|
39
|
-
# Multi-Modal Tools (Image, Video, Audio, Search)
|
|
40
|
-
from .tools_multimodal import (
|
|
41
|
-
ImageGenerationTool,
|
|
42
|
-
VideoGenerationTool,
|
|
43
|
-
AudioGenerationTool,
|
|
44
|
-
TTSTool,
|
|
45
|
-
SearchTool,
|
|
46
|
-
ScrapeTool,
|
|
47
|
-
register_multimodal_tools,
|
|
48
|
-
create_full_registry
|
|
49
|
-
)
|
|
50
|
-
|
|
51
|
-
# Paper Section 4: Specialized Agents
|
|
52
|
-
from .agents import (
|
|
53
|
-
BaseAgent,
|
|
54
|
-
AgentRole,
|
|
55
|
-
AgentContext,
|
|
56
|
-
AgentResult,
|
|
57
|
-
AgentRegistry,
|
|
58
|
-
WriterAgent,
|
|
59
|
-
ResearcherAgent,
|
|
60
|
-
EditorAgent,
|
|
61
|
-
DirectorAgent,
|
|
62
|
-
DesignerAgent,
|
|
63
|
-
ScreenwriterAgent,
|
|
64
|
-
ComposerAgent,
|
|
65
|
-
create_default_agents
|
|
66
|
-
)
|
|
67
|
-
|
|
68
|
-
# Asset Bank for Consistency
|
|
69
|
-
from .assets import (
|
|
70
|
-
AssetBank,
|
|
71
|
-
Character,
|
|
72
|
-
StyleGuide,
|
|
73
|
-
Artifact,
|
|
74
|
-
create_asset_bank
|
|
75
|
-
)
|
|
76
|
-
|
|
77
|
-
__version__ = "0.2.0"
|
|
78
|
-
__all__ = [
|
|
79
|
-
# Core models
|
|
80
|
-
"Vibe", "WorkflowPlan", "WorkflowNode", "WorkflowNodeType",
|
|
81
|
-
# MetaPlanner (Section 5.2)
|
|
82
|
-
"MetaPlanner", "LLMClient", "LLMConfig",
|
|
83
|
-
# Executor
|
|
84
|
-
"WorkflowExecutor", "ExecutionStatus", "ExecutionResult",
|
|
85
|
-
# Knowledge Base (Section 5.3)
|
|
86
|
-
"KnowledgeBase", "DomainKnowledge", "create_knowledge_base",
|
|
87
|
-
# Tool Registry (Section 5.4)
|
|
88
|
-
"ToolRegistry", "BaseTool", "ToolResult", "ToolSpec", "ToolCategory",
|
|
89
|
-
"LLMTool", "TemplateTool", "CombineTool", "create_default_registry",
|
|
90
|
-
# Multi-Modal Tools
|
|
91
|
-
"ImageGenerationTool", "VideoGenerationTool", "AudioGenerationTool",
|
|
92
|
-
"TTSTool", "SearchTool", "ScrapeTool",
|
|
93
|
-
"register_multimodal_tools", "create_full_registry",
|
|
94
|
-
# Agents (Section 4 examples)
|
|
95
|
-
"BaseAgent", "AgentRole", "AgentContext", "AgentResult", "AgentRegistry",
|
|
96
|
-
"WriterAgent", "ResearcherAgent", "EditorAgent", "DirectorAgent",
|
|
97
|
-
"DesignerAgent", "ScreenwriterAgent", "ComposerAgent",
|
|
98
|
-
"create_default_agents",
|
|
99
|
-
# Asset Bank
|
|
100
|
-
"AssetBank", "Character", "StyleGuide", "Artifact", "create_asset_bank"
|
|
101
|
-
]
|
|
102
|
-
#
|
|
1
|
+
"""Vibe AIGC: A New Paradigm for Content Generation via Agentic Orchestration.
|
|
2
|
+
|
|
3
|
+
This package implements the Vibe AIGC paradigm from the paper:
|
|
4
|
+
"Vibe AIGC: A New Paradigm for Content Generation via Agentic Orchestration"
|
|
5
|
+
|
|
6
|
+
Architecture (Paper Section 5):
|
|
7
|
+
- MetaPlanner: Decomposes Vibes into workflows (Section 5.2)
|
|
8
|
+
- KnowledgeBase: Domain-specific expert knowledge (Section 5.3)
|
|
9
|
+
- ToolRegistry: Atomic tool library for content generation (Section 5.4)
|
|
10
|
+
- Agents: Specialized role-based agents (Section 4 examples)
|
|
11
|
+
- AssetBank: Character and style consistency management
|
|
12
|
+
"""
|
|
13
|
+
|
|
14
|
+
from .models import Vibe, WorkflowPlan, WorkflowNode, WorkflowNodeType
|
|
15
|
+
from .planner import MetaPlanner
|
|
16
|
+
from .llm import LLMClient, LLMConfig
|
|
17
|
+
from .executor import WorkflowExecutor, ExecutionStatus, ExecutionResult
|
|
18
|
+
|
|
19
|
+
# Paper Section 5.3: Domain-Specific Expert Knowledge Base
|
|
20
|
+
from .knowledge import (
|
|
21
|
+
KnowledgeBase,
|
|
22
|
+
DomainKnowledge,
|
|
23
|
+
create_knowledge_base
|
|
24
|
+
)
|
|
25
|
+
|
|
26
|
+
# Paper Section 5.4: Atomic Tool Library
|
|
27
|
+
from .tools import (
|
|
28
|
+
ToolRegistry,
|
|
29
|
+
BaseTool,
|
|
30
|
+
ToolResult,
|
|
31
|
+
ToolSpec,
|
|
32
|
+
ToolCategory,
|
|
33
|
+
LLMTool,
|
|
34
|
+
TemplateTool,
|
|
35
|
+
CombineTool,
|
|
36
|
+
create_default_registry
|
|
37
|
+
)
|
|
38
|
+
|
|
39
|
+
# Multi-Modal Tools (Image, Video, Audio, Search)
|
|
40
|
+
from .tools_multimodal import (
|
|
41
|
+
ImageGenerationTool,
|
|
42
|
+
VideoGenerationTool,
|
|
43
|
+
AudioGenerationTool,
|
|
44
|
+
TTSTool,
|
|
45
|
+
SearchTool,
|
|
46
|
+
ScrapeTool,
|
|
47
|
+
register_multimodal_tools,
|
|
48
|
+
create_full_registry
|
|
49
|
+
)
|
|
50
|
+
|
|
51
|
+
# Paper Section 4: Specialized Agents
|
|
52
|
+
from .agents import (
|
|
53
|
+
BaseAgent,
|
|
54
|
+
AgentRole,
|
|
55
|
+
AgentContext,
|
|
56
|
+
AgentResult,
|
|
57
|
+
AgentRegistry,
|
|
58
|
+
WriterAgent,
|
|
59
|
+
ResearcherAgent,
|
|
60
|
+
EditorAgent,
|
|
61
|
+
DirectorAgent,
|
|
62
|
+
DesignerAgent,
|
|
63
|
+
ScreenwriterAgent,
|
|
64
|
+
ComposerAgent,
|
|
65
|
+
create_default_agents
|
|
66
|
+
)
|
|
67
|
+
|
|
68
|
+
# Asset Bank for Consistency
|
|
69
|
+
from .assets import (
|
|
70
|
+
AssetBank,
|
|
71
|
+
Character,
|
|
72
|
+
StyleGuide,
|
|
73
|
+
Artifact,
|
|
74
|
+
create_asset_bank
|
|
75
|
+
)
|
|
76
|
+
|
|
77
|
+
__version__ = "0.2.0"
|
|
78
|
+
__all__ = [
|
|
79
|
+
# Core models
|
|
80
|
+
"Vibe", "WorkflowPlan", "WorkflowNode", "WorkflowNodeType",
|
|
81
|
+
# MetaPlanner (Section 5.2)
|
|
82
|
+
"MetaPlanner", "LLMClient", "LLMConfig",
|
|
83
|
+
# Executor
|
|
84
|
+
"WorkflowExecutor", "ExecutionStatus", "ExecutionResult",
|
|
85
|
+
# Knowledge Base (Section 5.3)
|
|
86
|
+
"KnowledgeBase", "DomainKnowledge", "create_knowledge_base",
|
|
87
|
+
# Tool Registry (Section 5.4)
|
|
88
|
+
"ToolRegistry", "BaseTool", "ToolResult", "ToolSpec", "ToolCategory",
|
|
89
|
+
"LLMTool", "TemplateTool", "CombineTool", "create_default_registry",
|
|
90
|
+
# Multi-Modal Tools
|
|
91
|
+
"ImageGenerationTool", "VideoGenerationTool", "AudioGenerationTool",
|
|
92
|
+
"TTSTool", "SearchTool", "ScrapeTool",
|
|
93
|
+
"register_multimodal_tools", "create_full_registry",
|
|
94
|
+
# Agents (Section 4 examples)
|
|
95
|
+
"BaseAgent", "AgentRole", "AgentContext", "AgentResult", "AgentRegistry",
|
|
96
|
+
"WriterAgent", "ResearcherAgent", "EditorAgent", "DirectorAgent",
|
|
97
|
+
"DesignerAgent", "ScreenwriterAgent", "ComposerAgent",
|
|
98
|
+
"create_default_agents",
|
|
99
|
+
# Asset Bank
|
|
100
|
+
"AssetBank", "Character", "StyleGuide", "Artifact", "create_asset_bank"
|
|
101
|
+
]
|
|
102
|
+
# Model Registry - Auto-detect available models
|
|
103
|
+
from .model_registry import ModelRegistry, ModelCapability, ModelFamily, ModelSpec
|
|
104
|
+
|
|
105
|
+
# VLM Feedback - Visual quality assessment
|
|
106
|
+
from .vlm_feedback import VLMFeedback, FeedbackResult, MediaType, create_vlm_feedback
|
|
107
|
+
|
|
108
|
+
# ComfyUI backend for actual image generation
|
|
103
109
|
from .comfyui import ComfyUIBackend, ComfyUIConfig, ComfyUIImageTool, create_comfyui_registry
|
|
104
|
-
|
|
105
|
-
# Workflow templates
|
|
106
|
-
from .workflows import WorkflowLibrary, WorkflowTemplate, create_workflow_library
|
|
107
|
-
|
|
108
|
-
# Audio generation
|
|
109
|
-
from .audio import MusicGenBackend, RiffusionBackend, ElevenLabsBackend, MusicGenerationTool, TTSTool
|
|
110
|
-
|
|
111
|
-
# MV Pipeline
|
|
110
|
+
|
|
111
|
+
# Workflow templates
|
|
112
|
+
from .workflows import WorkflowLibrary, WorkflowTemplate, create_workflow_library
|
|
113
|
+
|
|
114
|
+
# Audio generation
|
|
115
|
+
from .audio import MusicGenBackend, RiffusionBackend, ElevenLabsBackend, MusicGenerationTool, TTSTool
|
|
116
|
+
|
|
117
|
+
# MV Pipeline
|
|
112
118
|
from .mv_pipeline import MVPipeline, Shot, Storyboard, create_mv
|