mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
5a3d8ab74d
- Add summary module with simple summary and personal summary tasks - Implement new vector store operations for updating and managing memory - Refactor retrieve module to improve query building and memory rewriting - Update schema and utils modules for better data handling and PDF processing
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from fastmcp import Client
|
|
from mcp.types import CallToolResult
|
|
|
|
|
|
async def main():
|
|
async with Client("http://0.0.0.0:8002/sse/") as client:
|
|
tools = await client.list_tools()
|
|
for tool in tools:
|
|
print(tool.model_dump_json())
|
|
|
|
workspace_id = "default"
|
|
|
|
result: CallToolResult = await client.call_tool("retrieve_task_memory_simple",
|
|
arguments={
|
|
"query": "茅台怎么样？",
|
|
"workspace_id": workspace_id,
|
|
})
|
|
print(result.content)
|
|
|
|
trajectories = [
|
|
{
|
|
"task_id": "t1",
|
|
"messages": [
|
|
{"role": "user", "content": "今天天气不错"}
|
|
],
|
|
"score": 0.9,
|
|
}
|
|
]
|
|
|
|
|
|
result: CallToolResult = await client.call_tool("summary_task_memory_simple",
|
|
arguments={
|
|
"trajectories": trajectories,
|
|
"workspace_id": workspace_id,
|
|
})
|
|
print(result.content)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import asyncio
|
|
|
|
asyncio.run(main())
|