q792602257
a5aeb8889c
Tests / Test failed: 2, passed: 849
- ToolCallDecision captures thinking blocks and pre-tool text output
- AnthropicToolCallingClient supports optional extended thinking (budget_tokens + beta header)
- PlannedStep carries rationale and thinking from each LLM decision
- WorldEvent replaces scene_summary with rationale/thinking/page fields (backward-compatible)
- AI planner system prompt instructs reflection before each tool call
- _history_summary() emits compact {page, rationale, action, success} dicts
- Cloud DB migration 0011 adds nullable rationale/thinking columns to planner_decision_log
- OpenAI client extracts reasoning_content into thinking field
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from alembic import command
|
|
from alembic.config import Config
|
|
from alembic.runtime.migration import MigrationContext
|
|
|
|
from cloud.database import create_database_engine, normalize_database_url
|
|
|
|
|
|
HEAD_REVISION = "0011_planner_decision_log_reflection"
|
|
|
|
|
|
class SchemaVersionError(RuntimeError):
|
|
"""Raised when the database schema is not at the required revision."""
|
|
|
|
|
|
def upgrade_database(database_url: str, revision: str = "head") -> None:
|
|
command.upgrade(_alembic_config(database_url), revision)
|
|
|
|
|
|
def downgrade_database(database_url: str, revision: str = "base") -> None:
|
|
command.downgrade(_alembic_config(database_url), revision)
|
|
|
|
|
|
def current_revision(database_url: str) -> str | None:
|
|
engine = create_database_engine(database_url)
|
|
try:
|
|
with engine.connect() as connection:
|
|
context = MigrationContext.configure(connection)
|
|
return context.get_current_revision()
|
|
finally:
|
|
engine.dispose()
|
|
|
|
|
|
def is_schema_current(database_url: str) -> bool:
|
|
return current_revision(database_url) == HEAD_REVISION
|
|
|
|
|
|
def require_current_schema(database_url: str) -> None:
|
|
revision = current_revision(database_url)
|
|
if revision != HEAD_REVISION:
|
|
raise SchemaVersionError(
|
|
f"cloud database schema is {revision or 'unversioned'}; "
|
|
f"required revision is {HEAD_REVISION}"
|
|
)
|
|
|
|
|
|
def _alembic_config(database_url: str) -> Config:
|
|
migrations_dir = Path(__file__).resolve().parent / "migrations"
|
|
config = Config(str(migrations_dir / "alembic.ini"))
|
|
config.set_main_option("script_location", str(migrations_dir))
|
|
config.set_main_option(
|
|
"sqlalchemy.url",
|
|
normalize_database_url(database_url).replace("%", "%%"),
|
|
)
|
|
return config
|