mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-08 14:56:03 +00:00
0a3bafae4b
Add EventLog and Transcript persistence entities for storing auditable event facts and conversation history projection. Implement event-first AgentRunContext builder that produces Protocol v1 compliant context payloads with required fields: event, delivery, context (ContextAccess). Key changes: - EventLog ORM: auditable event records with indexes - Transcript ORM: conversation history projection with composite indexes - AgentRunContextBuilder: Protocol v1 payload with delivery, context, bootstrap - EventLogStore/TranscriptStore: async stores for fact sources - Host action handlers: HISTORY_PAGE, HISTORY_SEARCH, EVENT_GET, EVENT_PAGE - Context validation: build_context output validates via SDK AgentRunContext - Alembic migration for event_log and transcript tables - Alembic env.py imports all ORM models for autogenerate discovery Legacy compatibility: max-round messages go into bootstrap.messages and compatibility.legacy_messages, not top-level messages field.
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
"""Alembic environment for LangBot.
|
|
|
|
This env.py is designed to be called programmatically (not via CLI).
|
|
It supports both SQLite and PostgreSQL.
|
|
|
|
The sync connection is passed via config attributes by the runner.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from alembic import context
|
|
from sqlalchemy.engine import Connection
|
|
|
|
from langbot.pkg.entity.persistence.base import Base
|
|
|
|
# Import all ORM models so they are registered with Base.metadata
|
|
# This is required for autogenerate to detect model changes
|
|
from langbot.pkg.entity.persistence import (
|
|
apikey,
|
|
bot,
|
|
bstorage,
|
|
event_log,
|
|
mcp,
|
|
metadata,
|
|
model,
|
|
monitoring,
|
|
pipeline,
|
|
plugin,
|
|
rag,
|
|
transcript,
|
|
user,
|
|
vector,
|
|
webhook,
|
|
)
|
|
|
|
target_metadata = Base.metadata
|
|
|
|
|
|
def run_migrations_offline() -> None:
|
|
"""Run migrations in 'offline' mode — emit SQL without a live connection."""
|
|
url = context.config.get_main_option('sqlalchemy.url')
|
|
context.configure(
|
|
url=url,
|
|
target_metadata=target_metadata,
|
|
literal_binds=True,
|
|
dialect_opts={'paramstyle': 'named'},
|
|
)
|
|
with context.begin_transaction():
|
|
context.run_migrations()
|
|
|
|
|
|
def run_migrations_online() -> None:
|
|
"""Run migrations with a live sync connection passed via config attributes."""
|
|
connection: Connection = context.config.attributes.get('connection')
|
|
if connection is None:
|
|
raise RuntimeError('connection not provided in alembic config attributes')
|
|
|
|
context.configure(
|
|
connection=connection,
|
|
target_metadata=target_metadata,
|
|
# render_as_batch=True is critical for SQLite ALTER TABLE support
|
|
render_as_batch=True,
|
|
)
|
|
with context.begin_transaction():
|
|
context.run_migrations()
|
|
|
|
|
|
if context.is_offline_mode():
|
|
run_migrations_offline()
|
|
else:
|
|
run_migrations_online()
|