Merge: fix-docs - Add module docstrings

This commit is contained in:
2026-03-14 05:37:34 +00:00
+7 -141
View File
@@ -1,144 +1,10 @@
"""Opus Orchestrator AI.
"""Opus Orchestrator - AI Book Generation System.
Full-flow AI book generation using LangGraph, CrewAI, AutoGen, and PydanticAI.
Integrates Fiction Fortress and Nonfiction Fortress methodologies.
Version: 0.2.0
Usage:
python -m opus_orchestrator generate --concept "Your story idea"
opus generate --concept "Your story idea" # If installed
Quick Start:
opus generate --concept "Your book idea"
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