Add Memory Gateway agent plugin
This commit is contained in:
34
plugins/memory-gateway-agent/scripts/smoke_test.py
Normal file
34
plugins/memory-gateway-agent/scripts/smoke_test.py
Normal file
@ -0,0 +1,34 @@
|
||||
#!/usr/bin/env python3
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import sys
|
||||
import uuid
|
||||
from pathlib import Path
|
||||
|
||||
PLUGIN_ROOT = Path(__file__).resolve().parents[1]
|
||||
if str(PLUGIN_ROOT) not in sys.path:
|
||||
sys.path.insert(0, str(PLUGIN_ROOT))
|
||||
|
||||
from memory_gateway_plugin.output import dumps_safe, summarize_result
|
||||
from memory_gateway_plugin.tools import memory_append_episode, memory_commit_session, memory_search
|
||||
|
||||
|
||||
def main() -> None:
|
||||
user_id = "plugin_smoke_user"
|
||||
agent_id = "plugin_smoke_agent"
|
||||
session_id = f"plugin_smoke_{uuid.uuid4().hex[:8]}"
|
||||
episode = memory_append_episode(
|
||||
user_id=user_id,
|
||||
agent_id=agent_id,
|
||||
session_id=session_id,
|
||||
episode_summary="结论:Memory Gateway Agent Plugin smoke test 写入短期 episode。",
|
||||
tags=["smoke-test"],
|
||||
)
|
||||
commit = memory_commit_session(user_id=user_id, agent_id=agent_id, session_id=session_id)
|
||||
search = memory_search(query="Memory Gateway Agent Plugin smoke test", user_id=user_id, agent_id=agent_id, session_id=session_id)
|
||||
print(dumps_safe({"episode": summarize_result(episode), "commit": summarize_result(commit), "search": summarize_result(search)}))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user