agentforge-chat-history-postgres 0.2.3__tar.gz → 0.2.4__tar.gz

{agentforge_chat_history_postgres-0.2.3 → agentforge_chat_history_postgres-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentforge-chat-history-postgres
-Version: 0.2.3
+Version: 0.2.4
 Summary: Postgres-backed ChatHistoryStore for AgentForge
 Project-URL: Homepage, https://github.com/Scaffoldic/agentforge-py
 Project-URL: Repository, https://github.com/Scaffoldic/agentforge-py
@@ -19,7 +19,7 @@ Classifier: Topic :: Database
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Typing :: Typed
 Requires-Python: >=3.13
-Requires-Dist: agentforge-core~=0.2.3
+Requires-Dist: agentforge-core~=0.2.4
 Requires-Dist: asyncpg>=0.30
 Description-Content-Type: text/markdown
 

{agentforge_chat_history_postgres-0.2.3 → agentforge_chat_history_postgres-0.2.4}/pyproject.toml

@@ -8,7 +8,7 @@
 
 [project]
 name = "agentforge-chat-history-postgres"
-version = "0.2.3"
+version = "0.2.4"
 description = "Postgres-backed ChatHistoryStore for AgentForge"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -30,7 +30,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "agentforge-core ~= 0.2.3",
+    "agentforge-core ~= 0.2.4",
     "asyncpg>=0.30",
 ]
 

{agentforge_chat_history_postgres-0.2.3 → agentforge_chat_history_postgres-0.2.4}/src/agentforge_chat_history_postgres/_inmem_runner.py

@@ -79,7 +79,22 @@ class PostgresFakeRunner:
             self._insert_turn(params)
             return
         if sql_n.startswith("insert into chat_sessions"):
-            self._upsert_session(params[0], params[1])
+            # DO NOTHING (bug-018 create-if-missing) must not bump
+            # last_active_at on an existing row; DO UPDATE (append path)
+            # does. Mirror the real ON CONFLICT clause.
+            if "do nothing" in sql_n:
+                self._sessions.setdefault(
+                    params[0],
+                    {
+                        "id": params[0],
+                        "owner": None,
+                        "created_at": params[1],
+                        "last_active_at": params[1],
+                        "metadata": {},
+                    },
+                )
+            else:
+                self._upsert_session(params[0], params[1])
             return
         if sql_n.startswith("update chat_sessions set metadata"):
             metadata_json, owner, session_id = params

{agentforge_chat_history_postgres-0.2.3 → agentforge_chat_history_postgres-0.2.4}/src/agentforge_chat_history_postgres/history.py

@@ -13,7 +13,7 @@ from __future__ import annotations
 
 import json
 from collections.abc import Mapping
-from datetime import datetime
+from datetime import UTC, datetime
 from typing import Any
 
 from agentforge_core.contracts.chat import ChatHistoryStore
@@ -199,11 +199,22 @@ class PostgresChatHistory(ChatHistoryStore):
         return [await self._row_to_info(row) for row in rows]
 
     async def update_session_metadata(self, session_id: str, metadata: Mapping[str, Any]) -> None:
+        # Create the session row if it doesn't exist yet (bug-018):
+        # ChatServer records owner/metadata before the first turn. DO
+        # NOTHING leaves an existing row (and last_active_at) untouched.
+        await self._r.execute(
+            """INSERT INTO chat_sessions
+                 (id, owner, created_at, last_active_at, metadata)
+               VALUES ($1, NULL, $2, $2, '{}'::jsonb)
+               ON CONFLICT (id) DO NOTHING""",
+            session_id,
+            datetime.now(UTC),
+        )
         row = await self._r.fetchrow(
             "SELECT owner, metadata FROM chat_sessions WHERE id = $1",
             session_id,
         )
-        if row is None:
+        if row is None:  # pragma: no cover — the INSERT above guarantees a row
             raise ModuleError(f"Cannot update metadata for unknown session {session_id!r}")
         existing: dict[str, Any] = dict(_coerce_jsonb(row["metadata"]))
         existing.update(dict(metadata))

{agentforge_chat_history_postgres-0.2.3 → agentforge_chat_history_postgres-0.2.4}/tests/unit/test_postgres_history_branches.py

@@ -9,7 +9,6 @@ from datetime import UTC, datetime, timedelta
 import pytest
 from agentforge_chat_history_postgres import PostgresChatHistory
 from agentforge_chat_history_postgres._inmem_runner import PostgresFakeRunner
-from agentforge_core.production.exceptions import ModuleError
 from agentforge_core.values.chat import ChatTurn
 from agentforge_core.values.messages import ToolCall
 
@@ -63,11 +62,14 @@ async def test_list_sessions_with_before_filter() -> None:
 
 
 @pytest.mark.asyncio
-async def test_update_metadata_unknown_session_raises() -> None:
+async def test_update_metadata_creates_unknown_session() -> None:
+    """bug-018: metadata on a not-yet-existing session upserts the row
+    rather than raising, so ChatServer can record owner before turn 1."""
     runner = PostgresFakeRunner()
     store = await PostgresChatHistory.from_dsn("p", runner=runner)
-    with pytest.raises(ModuleError, match="unknown session"):
-        await store.update_session_metadata("nope", {"k": "v"})
+    await store.update_session_metadata("nope", {"k": "v", "owner": "u1"})
+    sessions = await store.list_sessions(owner="u1")
+    assert [s.id for s in sessions] == ["nope"]
 
 
 @pytest.mark.asyncio