mrhavens
fe1e001878
- Stage 1: One sentence summary - Stage 2: One paragraph outline - Stage 3: Character sheets - Stage 4: Four-page outline - Stage 5: Detailed character charts - Stage 6: Scene list - Stage 7: Scene descriptions - Then: Style guide → Write chapters → Critique → Compile Full pre-writing workflow now wired up.
131 lines
4.1 KiB
Python
131 lines
4.1 KiB
Python
"""Opus Orchestrator AI - Configuration."""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
def _load_env(key: str, default: Optional[str] = None) -> Optional[str]:
|
|
"""Load from environment variable."""
|
|
return os.environ.get(key, default)
|
|
|
|
|
|
class FortressConfig(BaseModel):
|
|
"""Configuration for Fortress integration."""
|
|
|
|
fiction_repo: str = "mrhavens/fiction-fortress"
|
|
nonfiction_repo: str = "mrhavens/nonfiction-fortress"
|
|
crewai_repo: str = "mrhavens/crewai-fortress"
|
|
autogen_repo: str = "mrhavens/autogen-fortress"
|
|
langgraph_repo: str = "mrhavens/langgraph-fortress"
|
|
|
|
|
|
class AgentConfig(BaseModel):
|
|
"""Configuration for AI agents."""
|
|
|
|
model: str = Field(default="MiniMax/MiniMax-M2.1", description="Default model for agents")
|
|
temperature: float = Field(default=0.7, ge=0.0, le=2.0)
|
|
max_tokens: Optional[int] = Field(default=None, description="Max tokens per response")
|
|
max_iterations: int = Field(default=10, description="Max iterations per agent task")
|
|
|
|
# Provider configuration
|
|
provider: str = Field(default="minimax", description="LLM provider: minimax, openai, anthropic")
|
|
api_key: Optional[str] = Field(default=None, description="API key for LLM provider")
|
|
|
|
|
|
class IterationConfig(BaseModel):
|
|
"""Configuration for iteration loops."""
|
|
|
|
min_critic_rounds: int = Field(default=2, description="Minimum critic review rounds")
|
|
max_critic_rounds: int = Field(default=5, description="Maximum critic review rounds")
|
|
approval_threshold: float = Field(default=0.8, description="Score threshold to proceed")
|
|
auto_proceed_threshold: float = Field(default=0.9, description="Score to auto-approve")
|
|
|
|
|
|
class OutputConfig(BaseModel):
|
|
"""Configuration for output generation."""
|
|
|
|
format: str = Field(default="markdown", description="Output format: markdown, epub, pdf")
|
|
include_frontmatter: bool = True
|
|
include_toc: bool = True
|
|
chapter_separator: str = "\n\n---\n\n"
|
|
output_dir: Path = Field(default=Path("./output"))
|
|
|
|
|
|
class OpusConfig(BaseModel):
|
|
"""Main configuration for Opus Orchestrator."""
|
|
|
|
fortress: FortressConfig = Field(default_factory=FortressConfig)
|
|
agent: AgentConfig = Field(default_factory=AgentConfig)
|
|
iteration: IterationConfig = Field(default_factory=IterationConfig)
|
|
output: OutputConfig = Field(default_factory=OutputConfig)
|
|
|
|
github_token: Optional[str] = Field(default=None, description="GitHub token for private repos")
|
|
|
|
class Config:
|
|
frozen = False
|
|
|
|
|
|
def load_config_from_env() -> OpusConfig:
|
|
"""Load configuration from environment variables.
|
|
|
|
Reads:
|
|
- MINIMAX_API_KEY or OPENAI_API_KEY for LLM
|
|
- GITHUB_TOKEN for GitHub operations
|
|
|
|
Prefers OPENAI_API_KEY if available (more reliable).
|
|
"""
|
|
# Load API keys - prefer OpenAI as MiniMax key may be invalid
|
|
openai_key = _load_env("OPENAI_API_KEY")
|
|
minimax_key = _load_env("MINIMAX_API_KEY")
|
|
|
|
# Use OpenAI by default if available, otherwise try MiniMax
|
|
if openai_key:
|
|
provider = "openai"
|
|
default_model = "gpt-4o"
|
|
api_key = openai_key
|
|
elif minimax_key:
|
|
provider = "minimax"
|
|
default_model = "MiniMax/MiniMax-M2.1"
|
|
api_key = minimax_key
|
|
else:
|
|
provider = "openai" # default
|
|
default_model = "gpt-4o"
|
|
api_key = None
|
|
|
|
github_token = _load_env("GITHUB_TOKEN")
|
|
|
|
agent_config = AgentConfig(
|
|
model=default_model,
|
|
provider=provider,
|
|
api_key=api_key,
|
|
)
|
|
|
|
return OpusConfig(
|
|
agent=agent_config,
|
|
github_token=github_token,
|
|
)
|
|
|
|
|
|
# Global config instance
|
|
_config: Optional[OpusConfig] = None
|
|
|
|
|
|
def get_config() -> OpusConfig:
|
|
"""Get the global configuration instance."""
|
|
global _config
|
|
if _config is None:
|
|
# Try to load from environment
|
|
try:
|
|
_config = load_config_from_env()
|
|
except Exception:
|
|
_config = OpusConfig()
|
|
return _config
|
|
|
|
|
|
def set_config(config: OpusConfig) -> None:
|
|
"""Set the global configuration instance."""
|
|
global _config
|
|
_config = config
|