mrhavens
4b8ae306e6
- OpusCrew base class with CrewAI LLM integration
- FictionCrew: Writer, Editor, Proofreader agents
- NonfictionCrew: Researcher, Writer, Fact-Checker, Editor agents
- CLI entry point: python -m opus_orchestrator
- Commands: generate, frameworks, config
- Test: generated 282-word story with CrewAI crews
Usage:
python -m opus_orchestrator generate --concept 'Your idea' --use-crewai
python -m opus_orchestrator frameworks
python -m opus_orchestrator config
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
"""Opus Orchestrator AI.
|
|
|
|
Full-flow AI book generation using LangGraph, CrewAI, AutoGen, and PydanticAI.
|
|
Integrates Fiction Fortress and Nonfiction Fortress methodologies.
|
|
|
|
Usage:
|
|
python -m opus_orchestrator generate --concept "Your story idea"
|
|
opus generate --concept "Your story idea" # If installed
|
|
"""
|
|
|
|
from opus_orchestrator.agents.fiction import (
|
|
ArchitectAgent,
|
|
CharacterLeadAgent,
|
|
EditorAgent,
|
|
VoiceAgent,
|
|
WorldsmithAgent,
|
|
)
|
|
from opus_orchestrator.agents.nonfiction import (
|
|
AnalystAgent,
|
|
FactCheckerAgent,
|
|
NonfictionEditorAgent,
|
|
NonfictionWriterAgent,
|
|
ResearcherAgent,
|
|
)
|
|
from opus_orchestrator.config import OpusConfig, get_config
|
|
from opus_orchestrator.schemas import (
|
|
BookIntent,
|
|
BookType,
|
|
Manuscript,
|
|
RawContent,
|
|
)
|
|
from opus_orchestrator.state import OpusState, create_initial_state
|
|
from opus_orchestrator.langgraph_workflow import OpusGraph, run_opus, OpusGraphState
|
|
from opus_orchestrator.autogen_critique import CritiqueCrew, create_critique_crew
|
|
from opus_orchestrator.utils.github_ingest import GitHubIngestor, create_github_ingestor
|
|
from opus_orchestrator.frameworks import StoryFramework
|
|
from opus_orchestrator.crews import (
|
|
OpusCrew,
|
|
FictionCrew,
|
|
NonfictionCrew,
|
|
create_fiction_crew,
|
|
create_nonfiction_crew,
|
|
)
|
|
|
|
__all__ = [
|
|
# Config
|
|
"OpusConfig",
|
|
"get_config",
|
|
# State
|
|
"OpusState",
|
|
"create_initial_state",
|
|
# Schemas
|
|
"BookIntent",
|
|
"BookType",
|
|
"Manuscript",
|
|
"RawContent",
|
|
# Fiction Agents
|
|
"ArchitectAgent",
|
|
"CharacterLeadAgent",
|
|
"EditorAgent",
|
|
"VoiceAgent",
|
|
"WorldsmithAgent",
|
|
# Nonfiction Agents
|
|
"ResearcherAgent",
|
|
"AnalystAgent",
|
|
"NonfictionWriterAgent",
|
|
"FactCheckerAgent",
|
|
"NonfictionEditorAgent",
|
|
# LangGraph
|
|
"OpusGraph",
|
|
"OpusGraphState",
|
|
"run_opus",
|
|
"StoryFramework",
|
|
# Crews (NEW!)
|
|
"OpusCrew",
|
|
"FictionCrew",
|
|
"NonfictionCrew",
|
|
"create_fiction_crew",
|
|
"create_nonfiction_crew",
|
|
# Main
|
|
"OpusOrchestrator",
|
|
"CritiqueCrew",
|
|
"create_critique_crew",
|
|
"GitHubIngestor",
|
|
"create_github_ingestor",
|
|
]
|
|
|
|
# Import legacy orchestrator for backward compatibility
|
|
from opus_orchestrator.orchestrator import OpusOrchestrator
|