agentseek-api 0.1.0__tar.gz → 0.2.0__tar.gz

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/.env.example

@@ -7,6 +7,10 @@ AGENTSEEK_GRAPHS=
 # If not set, all requests pass through as default_user (noop).
 AUTH_MODULE_PATH=
 
+# Redis worker stream replay buffer.
+# REDIS_STREAM_MAXLEN=10000
+# REDIS_STREAM_TTL_SECONDS=3600
+
 # Embedded SeekDB — set to true for a zero-config local backend.
 # Data stored in SEEKDB_EMBED_DIR (default: ~/.agentseek/seekdb_data).
 # Requires: uv sync --dev --extra embedded

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/.github/workflows/ci.yml

@@ -37,6 +37,9 @@ jobs:
           cache-python: true
           python-version: "3.12"
 
+      - name: Check lockfile
+        run: uv lock --check
+
       - name: Sync dependencies
         run: uv sync --dev
 

agentseek_api-0.2.0/CHANGELOG.md

@@ -0,0 +1,22 @@
+# Changelog
+
+Notable changes to AgentSeek API are documented in this file.
+
+## 0.2.0 - 2026-07-12
+
+### Highlights
+
+- Expanded the Crons API to align its request, response, filtering, sorting,
+  selection, and lifecycle behavior with the LangGraph Platform contract.
+- Moved Redis-worker run and protocol event persistence from SQL to bounded
+  Redis Streams, removing metadata-database writes from the streaming hot path.
+- Preserved UTF-8 output across SSE, wait, and A2A JSON responses while keeping
+  active thread event retention bounded without dropping live events.
+- Upgraded checkpoint integration to `langchain-oceanbase` 0.6.0.
+
+### Upgrade notes
+
+- Drain active Redis-worker runs before upgrading. Stream-event rows written to
+  SQL by earlier versions are not imported into Redis Streams.
+- Redis stream replay is bounded by `REDIS_STREAM_MAXLEN` and
+  `REDIS_STREAM_TTL_SECONDS`; review these settings for long-running workloads.

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentseek-api
-Version: 0.1.0
+Version: 0.2.0
 Summary: AgentSeek API core runtime with OceanBase checkpoints.
 License-File: LICENSE
 Requires-Python: >=3.12
@@ -12,7 +12,7 @@ Requires-Dist: fastapi>=0.110.1
 Requires-Dist: greenlet>=3.1.0
 Requires-Dist: langchain-anthropic>=1.0.0
 Requires-Dist: langchain-core>=1.0.0
-Requires-Dist: langchain-oceanbase==0.5.2
+Requires-Dist: langchain-oceanbase==0.6.0
 Requires-Dist: langchain-openai>=1.0.0
 Requires-Dist: langchain>=0.3.9
 Requires-Dist: langgraph-sdk>=0.3.5
@@ -26,13 +26,15 @@ Requires-Dist: scalar-fastapi>=1.0.3
 Requires-Dist: sqlalchemy>=2.0.0
 Requires-Dist: uvicorn>=0.30.0
 Provides-Extra: embedded
-Requires-Dist: langchain-oceanbase[pyseekdb]==0.5.2; extra == 'embedded'
+Requires-Dist: langchain-oceanbase[pyseekdb]==0.6.0; extra == 'embedded'
 Description-Content-Type: text/markdown
 
 # AgentSeek API
 
 **English** | [中文](README.zh-CN.md)
 
+[![PyPI version](https://img.shields.io/pypi/v/agentseek-api.svg)](https://pypi.org/project/agentseek-api/) [![Python >=3.12](https://img.shields.io/badge/python-%3E%3D3.12-blue.svg)](https://pypi.org/project/agentseek-api/) [![CI](https://github.com/ob-labs/agentseek-api/actions/workflows/ci.yml/badge.svg?branch=main)](https://github.com/ob-labs/agentseek-api/actions/workflows/ci.yml?query=branch%3Amain)
+
 > [!WARNING]
 > This project is under active development and is **not production-ready**.
 > Pull requests for bug fixes and enhancements are warmly welcomed!
@@ -214,7 +216,7 @@ When the server starts it prints the banner and local URLs:
 ╠═╣│ ┬├┤ │││ │ ╚═╗├┤ ├┤ ├┴┐
 ╩ ╩└─┘└─┘┘└┘ ┴ ╚═╝└─┘└─┘┴ ┴
 
-     AgentSeek v0.1.0
+     AgentSeek v0.2.0
 
 - 🚀 API: http://localhost:2024
 - 📚 Docs: http://localhost:2024/docs
@@ -653,6 +655,8 @@ parent api build --config ./langgraph.json -t my-api:dev
   - `REDIS_RUN_PROCESSING_KEY=agentseek:runs:processing`
   - `REDIS_WORKER_LOCK_KEY=agentseek:worker:active`
   - `REDIS_WORKER_LOCK_TTL_SECONDS=30`
+  - `REDIS_STREAM_MAXLEN=10000`
+  - `REDIS_STREAM_TTL_SECONDS=3600`
 - `METADATA_DB_BACKEND=auto` normalizes drivers:
   - PostgreSQL: `postgresql+asyncpg://...`
   - OceanBase / MySQL: `mysql+aiomysql://...`
@@ -665,8 +669,12 @@ parent api build --config ./langgraph.json -t my-api:dev
 
 ### Durable execution
 
-- Redis mode persists run stream events and protocol stream events into the
-  metadata database so stream replay does not depend on API-process memory.
+- Redis mode stores run stream events and protocol stream events in bounded
+  Redis Streams, so replay does not depend on API-process memory and streaming
+  writes do not put the metadata database on the hot path.
+- When upgrading from a version that stored Redis-worker stream events in the
+  metadata database, drain active runs first. Existing SQL stream-event rows
+  are not imported into Redis.
 - Interrupted runs can be resumed after worker restart as long as Redis and the
   metadata database stay available.
 - The worker owns a renewable Redis lease and exits if that lease is lost,

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/README.md

@@ -2,6 +2,8 @@
 
 **English** | [中文](README.zh-CN.md)
 
+[![PyPI version](https://img.shields.io/pypi/v/agentseek-api.svg)](https://pypi.org/project/agentseek-api/) [![Python >=3.12](https://img.shields.io/badge/python-%3E%3D3.12-blue.svg)](https://pypi.org/project/agentseek-api/) [![CI](https://github.com/ob-labs/agentseek-api/actions/workflows/ci.yml/badge.svg?branch=main)](https://github.com/ob-labs/agentseek-api/actions/workflows/ci.yml?query=branch%3Amain)
+
 > [!WARNING]
 > This project is under active development and is **not production-ready**.
 > Pull requests for bug fixes and enhancements are warmly welcomed!
@@ -183,7 +185,7 @@ When the server starts it prints the banner and local URLs:
 ╠═╣│ ┬├┤ │││ │ ╚═╗├┤ ├┤ ├┴┐
 ╩ ╩└─┘└─┘┘└┘ ┴ ╚═╝└─┘└─┘┴ ┴
 
-     AgentSeek v0.1.0
+     AgentSeek v0.2.0
 
 - 🚀 API: http://localhost:2024
 - 📚 Docs: http://localhost:2024/docs
@@ -622,6 +624,8 @@ parent api build --config ./langgraph.json -t my-api:dev
   - `REDIS_RUN_PROCESSING_KEY=agentseek:runs:processing`
   - `REDIS_WORKER_LOCK_KEY=agentseek:worker:active`
   - `REDIS_WORKER_LOCK_TTL_SECONDS=30`
+  - `REDIS_STREAM_MAXLEN=10000`
+  - `REDIS_STREAM_TTL_SECONDS=3600`
 - `METADATA_DB_BACKEND=auto` normalizes drivers:
   - PostgreSQL: `postgresql+asyncpg://...`
   - OceanBase / MySQL: `mysql+aiomysql://...`
@@ -634,8 +638,12 @@ parent api build --config ./langgraph.json -t my-api:dev
 
 ### Durable execution
 
-- Redis mode persists run stream events and protocol stream events into the
-  metadata database so stream replay does not depend on API-process memory.
+- Redis mode stores run stream events and protocol stream events in bounded
+  Redis Streams, so replay does not depend on API-process memory and streaming
+  writes do not put the metadata database on the hot path.
+- When upgrading from a version that stored Redis-worker stream events in the
+  metadata database, drain active runs first. Existing SQL stream-event rows
+  are not imported into Redis.
 - Interrupted runs can be resumed after worker restart as long as Redis and the
   metadata database stay available.
 - The worker owns a renewable Redis lease and exits if that lease is lost,

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/README.zh-CN.md

@@ -2,6 +2,8 @@
 
 [English](README.md) | **中文**
 
+[![PyPI version](https://img.shields.io/pypi/v/agentseek-api.svg)](https://pypi.org/project/agentseek-api/) [![Python >=3.12](https://img.shields.io/badge/python-%3E%3D3.12-blue.svg)](https://pypi.org/project/agentseek-api/) [![CI](https://github.com/ob-labs/agentseek-api/actions/workflows/ci.yml/badge.svg?branch=main)](https://github.com/ob-labs/agentseek-api/actions/workflows/ci.yml?query=branch%3Amain)
+
 > [!WARNING]
 > 本项目正在积极开发中，**尚未达到生产可用状态**。
 > 欢迎提交 Pull Request 来修复 Bug 或贡献增强功能！
@@ -179,7 +181,7 @@ uv run agentseek-api dev --config ./langgraph.json
 ╠═╣│ ┬├┤ │││ │ ╚═╗├┤ ├┤ ├┴┐
 ╩ ╩└─┘└─┘┘└┘ ┴ ╚═╝└─┘└─┘┴ ┴
 
-     AgentSeek v0.1.0
+     AgentSeek v0.2.0
 
 - 🚀 API: http://localhost:2024
 - 📚 Docs: http://localhost:2024/docs
@@ -616,6 +618,8 @@ parent api build --config ./langgraph.json -t my-api:dev
   - `REDIS_RUN_PROCESSING_KEY=agentseek:runs:processing`
   - `REDIS_WORKER_LOCK_KEY=agentseek:worker:active`
   - `REDIS_WORKER_LOCK_TTL_SECONDS=30`
+  - `REDIS_STREAM_MAXLEN=10000`
+  - `REDIS_STREAM_TTL_SECONDS=3600`
 - `METADATA_DB_BACKEND=auto` 会对驱动进行归一化：
   - PostgreSQL：`postgresql+asyncpg://...`
   - OceanBase / MySQL：`mysql+aiomysql://...`
@@ -628,8 +632,11 @@ parent api build --config ./langgraph.json -t my-api:dev
 
 ### 持久化执行
 
-- Redis 模式会把 run 流事件与 protocol 流事件持久化到元数据数据库，
-  这样流回放就不再依赖 API 进程的内存。
+- Redis 模式会把 run 流事件与 protocol 流事件存入有界 Redis Stream，
+  这样流回放既不依赖 API 进程内存，也不会让流式写入占用元数据数据库
+  的关键路径。
+- 从使用元数据数据库保存 Redis worker 流事件的旧版本升级时，请先等待
+  活跃 run 执行完成；已有的 SQL 流事件不会导入 Redis。
 - 只要 Redis 与元数据数据库仍然可用，被中断的 run 可以在 worker 重启
   后恢复。
 - Worker 持有一个可续期的 Redis lease，在 lease 丢失时会主动退出，从而

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentseek-api"
-version = "0.1.0"
+version = "0.2.0"
 description = "AgentSeek API core runtime with OceanBase checkpoints."
 readme = "README.md"
 requires-python = ">=3.12"
@@ -20,7 +20,7 @@ dependencies = [
     "langchain-core>=1.0.0",
     "langchain-openai>=1.0.0",
     "langchain-anthropic>=1.0.0",
-    "langchain-oceanbase==0.5.2",
+    "langchain-oceanbase==0.6.0",
     "cryptography>=45.0.0",
     "pymysql>=1.1.0",
     "langchain>=0.3.9",
@@ -30,7 +30,7 @@ dependencies = [
 
 [project.optional-dependencies]
 embedded = [
-    "langchain-oceanbase[pyseekdb]==0.5.2",
+    "langchain-oceanbase[pyseekdb]==0.6.0",
 ]
 
 [project.scripts]

agentseek_api-0.2.0/src/agentseek_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.0"

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/a2a_server.py

@@ -282,7 +282,7 @@ def _message_content(message: BaseMessage) -> str:
     content = getattr(message, "content", "")
     if isinstance(content, str):
         return content
-    return json.dumps(content, default=str)
+    return json.dumps(content, ensure_ascii=False, default=str)
 
 
 def _extract_chunk_messages(chunk: Any) -> list[BaseMessage]:
@@ -366,7 +366,7 @@ def _extract_output_text(extracted: Any) -> str:
             message_texts = [_message_content(message) for message in messages if isinstance(message, BaseMessage)]
             if message_texts:
                 return message_texts[-1]
-    return json.dumps(extracted, default=str)
+    return json.dumps(extracted, ensure_ascii=False, default=str)
 
 
 async def _invoke_a2a_graph(
@@ -558,7 +558,7 @@ def _canonical_a2a_method(method: str) -> tuple[str, bool]:
 
 
 def _sse_jsonrpc_event(*, request_id: Any, result: dict[str, Any]) -> str:
-    return f"event: message\ndata: {json.dumps(_jsonrpc_result(request_id, result))}\n\n"
+    return f"event: message\ndata: {json.dumps(_jsonrpc_result(request_id, result), ensure_ascii=False)}\n\n"
 
 
 async def handle_a2a_request(

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/api/crons.py

@@ -1,9 +1,9 @@
 from sqlalchemy import select
 
 from fastapi import APIRouter, Depends, HTTPException
-from fastapi.responses import Response
+from fastapi.responses import JSONResponse, Response
 
-from agentseek_api.core.auth_deps import authorize, apply_metadata_filters, get_current_user
+from agentseek_api.core.auth_deps import authorize, get_current_user
 from agentseek_api.core.database import db_manager
 from agentseek_api.core.orm import Assistant, Thread
 from agentseek_api.models.api import (
@@ -14,6 +14,7 @@ from agentseek_api.models.api import (
     CronRead,
     CronSearchRequest,
     CronSearchResponse,
+    ThreadCronCreate,
 )
 from agentseek_api.models.auth import User
 from agentseek_api.services import cron_service
@@ -36,7 +37,7 @@ async def _create_cron(
     *,
     assistant_id: str,
     thread_id: str | None,
-    payload: CronCreate,
+    payload: CronCreate | ThreadCronCreate,
     user: User,
 ) -> CronRead:
     try:
@@ -45,7 +46,7 @@ async def _create_cron(
         raise HTTPException(status_code=400, detail=str(exc)) from exc
 
 
-@router.post("/runs/crons", response_model=CronRead)
+@router.post("/runs/crons", response_model=CronRead, response_model_exclude_none=True)
 async def create_stateless_cron(payload: CronCreate, user: User = Depends(get_current_user)) -> CronRead:
     value: dict = {
         "metadata": dict(payload.metadata) if payload.metadata else {},
@@ -57,8 +58,8 @@ async def create_stateless_cron(payload: CronCreate, user: User = Depends(get_cu
     return await _create_cron(assistant_id=resolved_id, thread_id=None, payload=payload, user=user)
 
 
-@router.post("/threads/{thread_id}/runs/crons", response_model=CronRead)
-async def create_thread_cron(thread_id: str, payload: CronCreate, user: User = Depends(get_current_user)) -> CronRead:
+@router.post("/threads/{thread_id}/runs/crons", response_model=CronRead, response_model_exclude_none=True)
+async def create_thread_cron(thread_id: str, payload: ThreadCronCreate, user: User = Depends(get_current_user)) -> CronRead:
     value: dict = {
         "metadata": dict(payload.metadata) if payload.metadata else {},
         "assistant_id": payload.assistant_id,
@@ -75,10 +76,18 @@ async def create_thread_cron(thread_id: str, payload: CronCreate, user: User = D
     return await _create_cron(assistant_id=resolved_id, thread_id=thread_id, payload=payload, user=user)
 
 
-@router.post("/runs/crons/search", response_model=CronSearchResponse)
-async def search_crons(payload: CronSearchRequest, user: User = Depends(get_current_user)) -> CronSearchResponse:
+@router.post("/runs/crons/search", response_model=CronSearchResponse, response_model_exclude_none=True)
+async def search_crons(payload: CronSearchRequest, user: User = Depends(get_current_user)) -> CronSearchResponse | JSONResponse:
     filters = await authorize(user, "crons", "search", {})
-    return await cron_service.search_crons(payload=payload, user=user, filters=filters)
+    result = await cron_service.search_crons(payload=payload, user=user, filters=filters)
+    if payload.select is not None:
+        fields = set(payload.select)
+        # exclude_none keeps the select path's null handling consistent with the
+        # default path (response_model_exclude_none) and the single-cron routes.
+        return JSONResponse(
+            content={"items": [item.model_dump(mode="json", include=fields, exclude_none=True) for item in result.items]}
+        )
+    return result
 
 
 @router.post("/runs/crons/count", response_model=CronCountResponse)
@@ -87,13 +96,13 @@ async def count_crons(payload: CronCountRequest, user: User = Depends(get_curren
     return await cron_service.count_crons(payload=payload, user=user, filters=filters)
 
 
-@router.get("/runs/crons/{cron_id}", response_model=CronRead)
+@router.get("/runs/crons/{cron_id}", response_model=CronRead, response_model_exclude_none=True)
 async def get_cron(cron_id: str, user: User = Depends(get_current_user)) -> CronRead:
     filters = await authorize(user, "crons", "read", {"cron_id": cron_id})
     return await cron_service.get_cron(cron_id=cron_id, user=user, filters=filters)
 
 
-@router.patch("/runs/crons/{cron_id}", response_model=CronRead)
+@router.patch("/runs/crons/{cron_id}", response_model=CronRead, response_model_exclude_none=True)
 async def patch_cron(cron_id: str, payload: CronPatch, user: User = Depends(get_current_user)) -> CronRead:
     filters = await authorize(user, "crons", "update", {"cron_id": cron_id})
     try:

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/api/runs.py

@@ -31,6 +31,10 @@ from agentseek_api.services.stream_persistence import (
     parse_last_event_id,
 )
 from agentseek_api.services.sse import iter_with_sse_keepalives, safe_json_dumps, sse_keepalive_comment
+from agentseek_api.services.stream_modes import (
+    SUPPORTED_RUN_STREAM_MODES,
+    normalize_stream_modes as _normalize_stream_modes_shared,
+)
 from agentseek_api.services.thread_protocol import thread_protocol_broker
 from agentseek_api.settings import settings
 
@@ -39,8 +43,6 @@ router = APIRouter(prefix="/threads/{thread_id}/runs", tags=["Thread Runs"])
 TERMINAL_RUN_STATUSES = ("success", "error", "interrupted")
 REDIS_STREAM_POLL_INTERVAL_SECONDS = 0.05
 REDIS_STREAM_TERMINAL_IDLE_POLLS = 2
-SUPPORTED_RUN_STREAM_MODES = {"values", "updates", "messages", "messages-tuple", "debug", "events", "tasks", "checkpoints", "custom"}
-RUN_STREAM_MODE_ALIASES: dict[str, str] = {}
 RUN_CHECKPOINT_ID_METADATA_KEY = "__agentseek_checkpoint_id"
 
 
@@ -107,35 +109,10 @@ def _uses_redis_executor() -> bool:
 
 
 def _normalize_stream_modes(stream_mode: str | list[str] | None) -> list[str]:
-    if stream_mode is None:
-        return ["values"]
-
-    raw_modes = [stream_mode] if isinstance(stream_mode, str) else list(stream_mode)
-    modes: list[str] = []
-    invalid_modes: list[str] = []
-
-    if not raw_modes:
-        invalid_modes.append("<empty>")
-
-    for mode in raw_modes:
-        raw_mode = str(mode).strip()
-        if not raw_mode:
-            invalid_modes.append("<empty>")
-            continue
-        normalized = RUN_STREAM_MODE_ALIASES.get(raw_mode, raw_mode)
-        if normalized not in modes:
-            modes.append(normalized)
-
-    unsupported = invalid_modes + [mode for mode in modes if mode not in SUPPORTED_RUN_STREAM_MODES]
-    if unsupported:
-        raise HTTPException(
-            status_code=422,
-            detail=(
-                "Unsupported stream_mode value(s): "
-                f"{', '.join(unsupported)}. Supported values: {', '.join(sorted(SUPPORTED_RUN_STREAM_MODES))}."
-            ),
-        )
-    return modes
+    try:
+        return _normalize_stream_modes_shared(stream_mode)
+    except ValueError as exc:
+        raise HTTPException(status_code=422, detail=str(exc)) from exc
 
 
 def _parse_stream_mode_query_param(stream_mode: list[str] | None) -> list[str] | None:
@@ -275,7 +252,7 @@ def _wait_json_stream_response(
                         raise
                     yield b"\n"
             payload = await _wait_response_payload(current_run, user=user)
-            yield json.dumps(jsonable_encoder(payload), separators=(",", ":")).encode()
+            yield safe_json_dumps(jsonable_encoder(payload), separators=(",", ":")).encode("utf-8")
         finally:
             if cancel_on_disconnect:
                 try:

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/api/threads.py

@@ -48,7 +48,11 @@ from agentseek_api.services.thread_checkpoint_store import (
     snapshot_to_payload,
 )
 from agentseek_api.services.thread_protocol import thread_protocol_broker
-from agentseek_api.services.thread_service import create_thread_for_user, to_read_model
+from agentseek_api.services.thread_service import (
+    create_thread_for_user,
+    delete_threads_cascade,
+    to_read_model,
+)
 from agentseek_api.settings import settings
 
 router = APIRouter(prefix="/threads", tags=["Threads"])
@@ -59,18 +63,6 @@ REDIS_STREAM_TERMINAL_IDLE_POLLS = 2
 THREAD_STREAM_CHANNELS = ["input", "lifecycle", "messages", "tools", "values"]
 
 
-async def _best_effort_checkpointer_call(method_name: str, *args: object, **kwargs: object) -> None:
-    method = getattr(db_manager.get_langgraph_checkpointer(), method_name, None)
-    if method is None:
-        return
-    try:
-        result = method(*args, **kwargs)
-        if hasattr(result, "__await__"):
-            await result
-    except NotImplementedError:
-        return
-
-
 logger = logging.getLogger(__name__)
 
 @functools.lru_cache(maxsize=64)
@@ -461,23 +453,11 @@ async def delete_thread(thread_id: str, user: User = Depends(get_current_user))
 
     session_factory = db_manager.get_session_factory()
     async with session_factory() as session:
-        stmt = select(Thread).where(Thread.thread_id == thread_id)
+        stmt = select(Thread.thread_id).where(Thread.thread_id == thread_id)
         stmt = apply_metadata_filters(stmt, Thread, filters)
-        row = await session.scalar(stmt)
-        if row is None:
+        if await session.scalar(stmt) is None:
             raise HTTPException(status_code=404, detail="Thread not found")
-        run_ids = (
-            await session.scalars(select(Run.run_id).where(Run.thread_id == thread_id))
-        ).all()
-        await session.execute(delete(Run).where(Run.thread_id == thread_id))
-        await session.delete(row)
-        await session.commit()
-    await _best_effort_checkpointer_call("adelete_thread", thread_id)
-    if run_ids:
-        await _best_effort_checkpointer_call("adelete_for_runs", list(run_ids))
-        await delete_run_stream_events(list(run_ids))
-    thread_protocol_broker.delete_thread(thread_id)
-    await delete_thread_stream_events(thread_id)
+    await delete_threads_cascade([thread_id])
     return Response(status_code=204)
 
 

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/core/database.py

@@ -172,6 +172,7 @@ class DatabaseManager:
             self.session_factory = async_sessionmaker(self.engine, expire_on_commit=False)
             async with self.engine.begin() as conn:
                 await conn.run_sync(Base.metadata.create_all)
+                await conn.run_sync(self._apply_additive_migrations)
             embed_conn_args = {"path": embed_dir, "db_name": settings.OCEANBASE_DB_NAME}
             self._checkpointer = OceanBaseCheckpointSaver(
                 connection_args=embed_conn_args
@@ -195,6 +196,7 @@ class DatabaseManager:
             self.session_factory = async_sessionmaker(self.engine, expire_on_commit=False)
             async with self.engine.begin() as conn:
                 await conn.run_sync(Base.metadata.create_all)
+                await conn.run_sync(self._apply_additive_migrations)
             conn_args = _resolve_connection_args()
             self._checkpointer = OceanBaseCheckpointSaver(
                 connection_args=conn_args
@@ -220,6 +222,51 @@ class DatabaseManager:
         await self._setup_langgraph_checkpointer_once()
         await self._setup_store_once()
 
+    @staticmethod
+    def _apply_additive_migrations(connection) -> None:
+        """Add columns introduced after a table was first created.
+
+        ``Base.metadata.create_all`` only creates *missing* tables; it never
+        alters an existing one. Deployments whose tables predate a new ORM
+        column would therefore be missing it, breaking queries that reference
+        it. This inspects each mapped table and issues ``ALTER TABLE ADD
+        COLUMN`` for any column present in the model but absent in the DB.
+        Idempotent and safe to run on every startup.
+
+        Invariant for new columns: they must be either nullable or carry a
+        ``server_default``. ``ADD COLUMN ... NOT NULL`` without a server-side
+        default fails on a populated table (there is no value for existing
+        rows). Such a column cannot be added automatically here — it requires a
+        deliberate data migration — so we skip it with a loud error rather than
+        crash startup or emit failing DDL.
+        """
+        from sqlalchemy import inspect
+        from sqlalchemy.schema import CreateColumn
+
+        inspector = inspect(connection)
+        existing_tables = set(inspector.get_table_names())
+        dialect = connection.dialect
+        preparer = dialect.identifier_preparer
+        for table in Base.metadata.sorted_tables:
+            if table.name not in existing_tables:
+                continue  # create_all already built it with all columns
+            db_columns = {col["name"] for col in inspector.get_columns(table.name)}
+            for column in table.columns:
+                if column.name in db_columns:
+                    continue
+                if not column.nullable and column.server_default is None:
+                    logger.error(
+                        "Cannot auto-add NOT NULL column %s.%s without a server_default; "
+                        "a manual data migration is required. Skipping.",
+                        table.name,
+                        column.name,
+                    )
+                    continue
+                column_spec = CreateColumn(column).compile(dialect=dialect)
+                table_name = preparer.format_table(table)
+                logger.info("Adding missing column %s.%s", table.name, column.name)
+                connection.exec_driver_sql(f"ALTER TABLE {table_name} ADD COLUMN {column_spec}")
+
     async def _setup_checkpointer_once(self) -> None:
         if self._checkpointer_setup_done:
             return
@@ -262,6 +309,14 @@ class DatabaseManager:
             self._store_setup_done = True
 
     async def close(self) -> None:
+        langgraph_checkpointer = self._langgraph_checkpointer
+        self._langgraph_checkpointer = None
+        close_checkpointer = getattr(langgraph_checkpointer, "close", None)
+        if callable(close_checkpointer):
+            try:
+                await asyncio.to_thread(close_checkpointer)
+            except Exception:  # noqa: BLE001
+                logger.warning("Failed to close LangGraph checkpointer", exc_info=True)
         if self.engine is not None:
             await self.engine.dispose()
         store_engine = getattr(getattr(self._store, "obvector", None), "engine", None)
@@ -271,7 +326,6 @@ class DatabaseManager:
         self.engine = None
         self.session_factory = None
         self._checkpointer = None
-        self._langgraph_checkpointer = None
         self._store = None
         self._checkpointer_setup_done = False
         self._langgraph_checkpointer_setup_done = False

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/core/oceanbase_checkpointer.py

@@ -3,7 +3,6 @@ from datetime import UTC, datetime
 from typing import Any
 
 import pymysql
-from pymysql.connections import Connection
 from sqlalchemy import text
 from sqlalchemy.engine import Engine
 

{agentseek_api-0.1.0 → agentseek_api-0.2.0}/src/agentseek_api/core/orm.py

@@ -3,7 +3,7 @@ from typing import Any
 from datetime import UTC, datetime
 from uuid import uuid4
 
-from sqlalchemy import JSON, Boolean, DateTime, Index, Integer, String, Text, UniqueConstraint
+from sqlalchemy import JSON, Boolean, DateTime, Index, Integer, String, Text, UniqueConstraint, text
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
 from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
 
@@ -74,6 +74,10 @@ class CronJob(Base):
     metadata_json: Mapped[dict] = mapped_column("metadata", JSON, default=dict, nullable=False)
     kwargs_json: Mapped[dict] = mapped_column("kwargs", JSON, default=dict, nullable=False)
     webhook: Mapped[str | None] = mapped_column(Text, nullable=True)
+    end_time: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)
+    on_run_completed: Mapped[str] = mapped_column(
+        String(16), nullable=False, default="delete", server_default=text("'delete'")
+    )
     max_webhook_attempts: Mapped[int] = mapped_column(nullable=False, default=3)
     next_run_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), nullable=False)
     last_run_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True), nullable=True)