fix: proactive examples (#273)

Author: evan-ak

examples/proactive/memory/config.py

@@ -9,7 +9,7 @@
                 "prompt": "# Task Objective\nYou will be given a conversation between a user and an coding agent. Your goal is to extract detailed records for what are planed to do, and what have been done.",
             },
             "workflow": {
-                "ordinal": 10,
+                "ordinal": 20,
                 "prompt": "# Workflow\nRead through the conversation and extract records. You should expecially focus on:\n- What the user ask the agent to do\n- What plan does the agent suggest\n- What the agent has done",
             },
             "rules": {
@@ -37,14 +37,30 @@
                     "prompt": "# Workflow\nRead through the existing markdown file and the new records. Then update the markdown file to reflect:\n- What existing tasks are completed\n- What new tasks are added\n- What tasks are still in progress",
                 },
                 "rules": {
-                    "ordinal": -1,
-                    "prompt": None,
+                    "ordinal": 30,
+                    "prompt": "# Rules\nFor each action-like record, explictly mark it as [Done] or [Todo].",
                 },
                 "examples": {
                     "ordinal": 50,
-                    "prompt": "# Example\n## Output\n```markdown\n# Task\n## Task Objective\nThe user ask the agent to generate a code example for fastapi\n## Workflow\nThe agent suggest to use the code example from the document\nThe agent ask the user to specify the response type\n```",
+                    "prompt": "# Example\n## Output\n```markdown\n# Task\n## Task Objective\nThe user ask the agent to generate a code example for fastapi\n## Breakdown\n- [Done] The agent suggest to use the code example from the document\n- [Todo] The agent ask the user to specify the response type\n```",
                 },
             },
         }
     ],
 }
+
+retrieve_config = {
+    "method": "rag",
+    "route_intention": False,
+    "sufficiency_check": False,
+    "category": {
+        "enabled": False,
+    },
+    "item": {
+        "enabled": True,
+        "top_k": 10,
+    },
+    "resource": {
+        "enabled": False,
+    },
+}

examples/proactive/memory/local/common.py

@@ -0,0 +1,31 @@
+import os
+
+from memu.app import MemoryService
+
+from ..config import memorize_config, retrieve_config
+
+USER_ID = "claude_user"
+SHARED_MEMORY_SERVICE = None
+
+
+def get_memory_service() -> MemoryService:
+    global SHARED_MEMORY_SERVICE
+    if SHARED_MEMORY_SERVICE is not None:
+        return SHARED_MEMORY_SERVICE
+
+    api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        msg = "Please set OPENAI_API_KEY environment variable"
+        raise ValueError(msg)
+
+    SHARED_MEMORY_SERVICE = MemoryService(
+        llm_profiles={
+            "default": {
+                "api_key": api_key,
+                "chat_model": "gpt-4o-mini",
+            },
+        },
+        memorize_config=memorize_config,
+        retrieve_config=retrieve_config,
+    )
+    return SHARED_MEMORY_SERVICE

examples/proactive/memory/local/memorize.py

@@ -1,14 +1,11 @@
 import json
-import os
 from collections.abc import Awaitable
 from pathlib import Path
 from typing import Any
 
 import pendulum
 
-from memu.app import MemoryService
-
-from ..config import memorize_config
+from .common import get_memory_service
 
 USER_ID = "claude_user"
 
@@ -28,26 +25,14 @@ def dump_conversation_resource(
     }
     time_string = pendulum.now().format("YYYYMMDD_HHmmss")
     resource_url = Path(__file__).parent / "data" / f"conv_{time_string}.json"
+    resource_url.parent.mkdir(parents=True, exist_ok=True)
     with open(resource_url, "w") as f:
         json.dump(resource_data, f, indent=4, ensure_ascii=False)
     return resource_url.as_posix()
 
 
 def memorize(conversation_messages: list[dict[str, Any]]) -> Awaitable[dict[str, Any]]:
-    api_key = os.getenv("OPENAI_API_KEY")
-    if not api_key:
-        msg = "Please set OPENAI_API_KEY environment variable"
-        raise ValueError(msg)
-
-    memory_service = MemoryService(
-        llm_profiles={
-            "default": {
-                "api_key": api_key,
-                "chat_model": "gpt-4o-mini",
-            },
-        },
-        memorize_config=memorize_config,
-    )
+    memory_service = get_memory_service()
 
     resource_url = dump_conversation_resource(conversation_messages)
     return memory_service.memorize(resource_url=resource_url, modality="conversation", user={"user_id": USER_ID})

examples/proactive/memory/local/tools.py

@@ -1,9 +1,8 @@
-import os
 from typing import Any
 
 from claude_agent_sdk import create_sdk_mcp_server, tool
 
-from memu.app import MemoryService
+from .common import get_memory_service
 
 USER_ID = "claude_user"
 
@@ -13,42 +12,15 @@ async def get_memory(args: dict[str, Any]) -> dict[str, Any]:
     """Retrieve memory from the memory API based on the provided query."""
     query = {"role": "user", "content": args["query"]}
 
-    api_key = os.getenv("OPENAI_API_KEY")
-    if not api_key:
-        msg = "Please set OPENAI_API_KEY environment variable"
-        raise ValueError(msg)
-
-    memory_service = MemoryService(
-        llm_profiles={
-            "default": {
-                "api_key": api_key,
-                "chat_model": "gpt-4o-mini",
-            },
-        },
-        retrieve_config={
-            "method": "rag",
-            "route_intention": False,
-            "sufficiency_check": False,
-            "category": {
-                "enabled": False,
-            },
-            "item": {
-                "enabled": True,
-                "top_k": 10,
-            },
-            "resource": {
-                "enabled": False,
-            },
-        },
-    )
+    memory_service = get_memory_service()
 
     result = await memory_service.retrieve(query, where={"user_id": USER_ID})
 
     return {"content": [{"type": "text", "text": str(result)}]}
 
 
 async def _get_todos() -> str:
-    memory_service = MemoryService()
+    memory_service = get_memory_service()
 
     result = await memory_service.list_memory_categories(where={"user_id": USER_ID})
 

examples/proactive/proactive.py

@@ -47,7 +47,11 @@ async def get_next_input(iteration: int) -> tuple[str | None, bool]:
         return await get_user_input()
 
     todos = await _get_todos()
-    if todos:
+
+    print(f">>> Todos:\n{todos}\n")
+    print("-" * 40)
+
+    if todos and "[todo]" in todos.lower():
         return f"Please continue with the following todos:\n{todos}", False
 
     return await get_user_input()