Merge: fix-docs - Add module docstrings
This commit is contained in:
@@ -1,144 +1,10 @@
|
|||||||
"""Opus Orchestrator AI.
|
"""Opus Orchestrator - AI Book Generation System.
|
||||||
|
|
||||||
Full-flow AI book generation using LangGraph, CrewAI, AutoGen, and PydanticAI.
|
Version: 0.2.0
|
||||||
Integrates Fiction Fortress and Nonfiction Fortress methodologies.
|
|
||||||
|
|
||||||
Usage:
|
Quick Start:
|
||||||
python -m opus_orchestrator generate --concept "Your story idea"
|
opus generate --concept "Your book idea"
|
||||||
opus generate --concept "Your story idea" # If installed
|
opus serve --port 8000
|
||||||
|
|
||||||
|
For full documentation, see README.md
|
||||||
"""
|
"""
|
||||||
|
|
||||||
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.agents.research import (
|
|
||||||
ResearchAgent,
|
|
||||||
VerifiedFactChecker,
|
|
||||||
create_research_agent,
|
|
||||||
)
|
|
||||||
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.utils.s3_ingest import S3Ingestor, create_s3_ingestor
|
|
||||||
from opus_orchestrator.utils.local_ingest import LocalIngestor, create_local_ingestor
|
|
||||||
from opus_orchestrator.frameworks import StoryFramework
|
|
||||||
from opus_orchestrator.nonfiction_frameworks import (
|
|
||||||
NonfictionFramework,
|
|
||||||
get_nonfiction_framework,
|
|
||||||
list_nonfiction_frameworks,
|
|
||||||
)
|
|
||||||
from opus_orchestrator.crews import (
|
|
||||||
OpusCrew,
|
|
||||||
FictionCrew,
|
|
||||||
NonfictionCrew,
|
|
||||||
create_fiction_crew,
|
|
||||||
create_nonfiction_crew,
|
|
||||||
)
|
|
||||||
from opus_orchestrator.pydanticai_agent import (
|
|
||||||
OpusPydanticAgent,
|
|
||||||
StorySeed,
|
|
||||||
CharacterProfile,
|
|
||||||
ChapterOutline,
|
|
||||||
ChapterDraft,
|
|
||||||
CritiqueResult,
|
|
||||||
StyleGuide,
|
|
||||||
create_story_seed_agent,
|
|
||||||
create_character_agent,
|
|
||||||
create_chapter_outline_agent,
|
|
||||||
create_chapter_draft_agent,
|
|
||||||
create_critique_agent,
|
|
||||||
create_style_guide_agent,
|
|
||||||
)
|
|
||||||
|
|
||||||
__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",
|
|
||||||
# Research Agent (NEW!)
|
|
||||||
"ResearchAgent",
|
|
||||||
"VerifiedFactChecker",
|
|
||||||
"create_research_agent",
|
|
||||||
# LangGraph
|
|
||||||
"OpusGraph",
|
|
||||||
"OpusGraphState",
|
|
||||||
"run_opus",
|
|
||||||
"StoryFramework",
|
|
||||||
# Nonfiction Frameworks (NEW!)
|
|
||||||
"NonfictionFramework",
|
|
||||||
"get_nonfiction_framework",
|
|
||||||
"list_nonfiction_frameworks",
|
|
||||||
# Crews
|
|
||||||
"OpusCrew",
|
|
||||||
"FictionCrew",
|
|
||||||
"NonfictionCrew",
|
|
||||||
"create_fiction_crew",
|
|
||||||
"create_nonfiction_crew",
|
|
||||||
# PydanticAI (NEW!)
|
|
||||||
"OpusPydanticAgent",
|
|
||||||
"StorySeed",
|
|
||||||
"CharacterProfile",
|
|
||||||
"ChapterOutline",
|
|
||||||
"ChapterDraft",
|
|
||||||
"CritiqueResult",
|
|
||||||
"StyleGuide",
|
|
||||||
"create_story_seed_agent",
|
|
||||||
"create_character_agent",
|
|
||||||
"create_chapter_outline_agent",
|
|
||||||
"create_chapter_draft_agent",
|
|
||||||
"create_critique_agent",
|
|
||||||
"create_style_guide_agent",
|
|
||||||
# Main
|
|
||||||
"OpusOrchestrator",
|
|
||||||
"CritiqueCrew",
|
|
||||||
"create_critique_crew",
|
|
||||||
"GitHubIngestor",
|
|
||||||
"create_github_ingestor",
|
|
||||||
# S3/MinIO
|
|
||||||
"S3Ingestor",
|
|
||||||
"create_s3_ingestor",
|
|
||||||
# Local Files
|
|
||||||
"LocalIngestor",
|
|
||||||
"create_local_ingestor",
|
|
||||||
]
|
|
||||||
|
|
||||||
# Import legacy orchestrator for backward compatibility
|
|
||||||
from opus_orchestrator.orchestrator import OpusOrchestrator
|
|
||||||
|
|||||||
Reference in New Issue
Block a user