added oxy base bank.

Author: JDOxygen

CHANGELOG_zh.md

@@ -14,6 +14,7 @@
 ### Added
 - 新增Prompt管理界面，支持在线修改Prompt
 - 支持本地大模型 LocalLLM类
+- 新增oxy.BaseBank，用于标准化智能体输入信息
 
 ### Changed
 - 修改es包，由elasticsearch[async] 改为 elasticsearch

applications/bank_manager_by_api_router.py

@@ -0,0 +1,79 @@
+import uvicorn
+from fastapi import APIRouter, Body, FastAPI
+from fastapi.routing import APIRoute
+
+app = FastAPI()
+router = APIRouter(tags=["bank"])
+
+
+@router.get("/user_profile_retrieve", description="A tool for querying user profile")
+def user_profile_retrieve(
+    query: str = Body(description="query"),
+    user_pin: str = Body(description="SystemArg.agent_pin"),
+    agent_pin: str = Body(description="SystemArg.user_pin"),
+):
+    user_profile_dict = {
+        "001": "Arlen, a student, likes music",
+        "002": "Tom, a programmer, likes sports",
+    }
+    portrait = user_profile_dict.get(user_pin, "Nothing")
+    return f"The current user profile is: {portrait}"
+
+
+@router.post("/user_profile_deposit", description="A tool for updating user profile")
+def user_profile_deposit(
+    content: str = Body(description="content"),
+    user_pin: str = Body(description="SystemArg.agent_pin"),
+    agent_pin: str = Body(description="SystemArg.user_pin"),
+) -> str:
+    print(agent_pin, user_pin, content)
+    return "updated user_profile"
+
+
+app.include_router(router)
+
+
+@app.get("/list_banks")
+def list_banks():
+    return get_banks_from_router(router)
+
+
+def get_banks_from_router(router: APIRouter):
+    banks = []
+    for route in router.routes:
+        if isinstance(route, APIRoute) and "bank" in getattr(route, "tags", []):
+            description = route.description
+            input_schema = {"type": "object", "properties": {}, "required": []}
+            for param in route.dependant.query_params + route.dependant.body_params:
+                param_type = param.type_
+                # Type conversion (simple implementation)
+                if param_type is str:
+                    t = "string"
+                elif param_type is int:
+                    t = "integer"
+                elif param_type is float:
+                    t = "number"
+                elif param_type is bool:
+                    t = "boolean"
+                else:
+                    t = "string"
+                input_schema["properties"][param.name] = {
+                    "type": t,
+                    "description": param.field_info.description or "",
+                }
+                if param.required:
+                    input_schema["required"].append(param.name)
+            banks.append(
+                {
+                    "name": route.endpoint.__name__,
+                    "endpoint": route.path,
+                    "methods": route.methods,
+                    "description": description,
+                    "inputSchema": input_schema,
+                }
+            )
+    return banks
+
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="127.0.0.1", port=8090)

applications/bank_manager_by_bank_router.py

@@ -0,0 +1,70 @@
+import asyncio
+import os
+
+from fastapi import Body
+
+from oxygent import MAS, BankRouter, Config, oxy
+
+Config.set_server_port(8090)
+
+
+router = BankRouter()
+
+user_profile_dict = {
+    "001": "Arlen, a student, likes music",
+    "002": "Tom, a programmer, likes sports",
+}
+
+
+@router.post("/user_profile_retrieve", description="A tool for querying user profile")
+def user_profile_retrieve(
+    query: str = Body(description="query"),
+    user_pin: str = Body(description="SystemArg.agent_pin"),
+    agent_pin: str = Body(description="SystemArg.user_pin"),
+):
+    global user_profile_dict
+    portrait = user_profile_dict.get(user_pin, "Nothing")
+    return f"The current user profile is: {portrait}"
+
+
+@router.post("/user_profile_deposit", description="A tool for updating user profile")
+async def user_profile_deposit(
+    content: str = Body(description="content"),
+    user_pin: str = Body(description="SystemArg.agent_pin"),
+    agent_pin: str = Body(description="SystemArg.user_pin"),
+):
+    global user_profile_dict
+    output = await router.mas.call(
+        callee="bank_manager",
+        arguments={
+            "query": "Please update the user profile.",
+            "chat": content,
+            "profile": user_profile_dict.get(user_pin, "Nothing"),
+        },
+    )
+    user_profile_dict[user_pin] = output
+    return output
+
+
+oxy_space = [
+    oxy.HttpLLM(
+        name="default_llm",
+        api_key=os.getenv("DEFAULT_LLM_API_KEY"),
+        base_url=os.getenv("DEFAULT_LLM_BASE_URL"),
+        model_name=os.getenv("DEFAULT_LLM_MODEL_NAME"),
+    ),
+    oxy.ChatAgent(
+        name="bank_manager",
+        llm_model="default_llm",
+        prompt="You are an expert in user profiling. Please update and refine the current user profile by integrating our previous conversation:\n${chat}\nAnd the current user profile:\n${profile}",
+    ),
+]
+
+
+async def main():
+    async with MAS(oxy_space=oxy_space, routers=[router]) as mas:
+        await mas.start_web_service()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

applications/bank_manager_by_manual_api.py

@@ -0,0 +1,75 @@
+import uvicorn
+from fastapi import APIRouter, FastAPI
+from pydantic import BaseModel
+
+app = FastAPI()
+
+router = APIRouter()
+
+
+class RetrievalRequest(BaseModel):
+    query: str
+    agent_pin: str
+    user_pin: str
+
+
+@router.post("/user_profile_retrieve")
+def user_profile_retrieve(request: RetrievalRequest):
+    user_profile_dict = {
+        "001": "Arlen, a student, likes music",
+        "002": "Tom, a programmer, likes sports",
+    }
+    portrait = user_profile_dict.get(request.user_pin, "Nothing")
+    return f"The current user profile is: {portrait}"
+
+
+class DepositRequest(BaseModel):
+    content: str
+    agent_pin: str
+    user_pin: str
+
+
+@router.post("/user_profile_deposit")
+def user_profile_deposit(request: DepositRequest):
+    print(request.content)
+    return "updated user_profile"
+
+
+@router.get("/list_banks")
+def list_banks():
+    return [
+        {
+            "name": "user_profile_retrieve",
+            "endpoint": "/user_profile_retrieve",
+            "type": "retrieve",
+            "description": "A tool for querying user profile",
+            "inputSchema": {
+                "properties": {
+                    "query": {"description": "query", "type": "str"},
+                    "agent_pin": {"description": "SystemArg.agent_pin", "type": "str"},
+                    "user_pin": {"description": "SystemArg.user_pin", "type": "str"},
+                },
+                "required": ["query", "agent_pin", "user_pin"],
+                "type": "object",
+            },
+        },
+        {
+            "name": "user_profile_deposit",
+            "endpoint": "/user_profile_deposit",
+            "type": "deposit",
+            "description": "A tool for updating user profile",
+            "inputSchema": {
+                "properties": {
+                    "content": {"description": "content", "type": "str"},
+                    "agent_pin": {"description": "SystemArg.agent_pin", "type": "str"},
+                    "user_pin": {"description": "SystemArg.user_pin", "type": "str"},
+                },
+                "required": ["content", "agent_pin", "user_pin"],
+                "type": "object",
+            },
+        },
+    ]
+
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="127.0.0.1", port=8090)

docs/development/api/agent/base_oxy.md

@@ -36,7 +36,7 @@ This class defines the core execution lifecycle, permission management,message h
 | `is_permission_required`         | `bool`               | `False`                                    | Whether execution needs permission |
 | `is_save_data`                   | `bool`               | `True`                                     | Persist execution data to store    |
 | `permitted_tool_name_list`       | `list`               | `[]`                                       | Tools the agent/tool may call      |
-| `extra_permitted_tool_name_list` | `list`               | `[]`                                       | Additional tool permissions        |
+| `permitted_oxy`                  | `list`               | `[]`                                       | Additional tool permissions        |
 | `is_send_tool_call`              | `bool`               | `Config.get_message_is_send_tool_call()`   | Send *tool\_call* messages         |
 | `is_send_observation`            | `bool`               | `Config.get_message_is_send_observation()` | Send *observation* messages        |
 | `is_send_answer`                 | `bool`               | `Config.get_message_is_send_answer()`      | Send *answer* messages             |

examples/banks/demo_bank_chat_agent_dump_memory.py

@@ -0,0 +1,61 @@
+import asyncio
+import json
+import os
+
+from oxygent import MAS, OxyResponse, oxy
+
+
+async def dump_memory(oxy_response: OxyResponse) -> OxyResponse:
+    oxy_request = oxy_response.oxy_request
+    history = {
+        "query": oxy_request.get_query(),
+        "answer": oxy_response.output,
+    }
+
+    await oxy_request.call_async(
+        callee="user_profile_deposit",
+        arguments={"content": json.dumps(history, ensure_ascii=False)},
+        is_send_message=False,
+    )
+    return oxy_response
+
+
+oxy_space = [
+    oxy.HttpLLM(
+        name="default_llm",
+        api_key=os.getenv("DEFAULT_LLM_API_KEY"),
+        base_url=os.getenv("DEFAULT_LLM_BASE_URL"),
+        model_name=os.getenv("DEFAULT_LLM_MODEL_NAME"),
+        llm_params={"temperature": 0.01},
+        semaphore=4,
+    ),
+    oxy.ChatAgent(
+        name="qa_agent",
+        llm_model="default_llm",
+        prompt="You can refer to the following information to answer the question:\n${preceding_text}",
+        banks=["remote_user_profile_banks"],
+        preceding_oxy=["user_profile_retrieve"],
+        preceding_placeholder="preceding_text",
+        func_process_output=dump_memory,
+    ),
+    oxy.BankClient(
+        name="remote_user_profile_banks",
+        server_url="http://127.0.0.1:8090",
+    ),
+]
+
+
+def func_filter(payload):
+    payload["group_data"] = {"user_pin": "002"}
+    return payload
+
+
+async def main():
+    async with MAS(
+        oxy_space=oxy_space, func_filter=func_filter, name="temp_app"
+    ) as mas:
+        await mas.start_web_service(first_query="Who I am")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

examples/banks/demo_bank_react_agent_autonomy.py

@@ -0,0 +1,46 @@
+import asyncio
+import os
+
+from oxygent import MAS, oxy
+
+oxy_space = [
+    oxy.HttpLLM(
+        name="default_llm",
+        api_key=os.getenv("DEFAULT_LLM_API_KEY"),
+        base_url=os.getenv("DEFAULT_LLM_BASE_URL"),
+        model_name=os.getenv("DEFAULT_LLM_MODEL_NAME"),
+        llm_params={"temperature": 0.01},
+        semaphore=4,
+    ),
+    oxy.StdioMCPClient(
+        name="time_tools",
+        params={
+            "command": "uvx",
+            "args": ["mcp-server-time", "--local-timezone=Asia/Shanghai"],
+        },
+    ),
+    oxy.ReActAgent(
+        name="qa_agent",
+        llm_model="default_llm",
+        tools=["time_tools"],
+        banks=["remote_user_profile_banks"],
+    ),
+    oxy.BankClient(
+        name="remote_user_profile_banks",
+        server_url="http://127.0.0.1:8090",
+    ),
+]
+
+
+def func_filter(payload):
+    payload["group_data"] = {"user_pin": "002"}
+    return payload
+
+
+async def main():
+    async with MAS(oxy_space=oxy_space, func_filter=func_filter) as mas:
+        await mas.start_web_service(first_query="Who I am")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

examples/banks/demo_bank_react_agent_autonomy_by_mcp.py

@@ -0,0 +1,45 @@
+import asyncio
+import os
+
+from oxygent import MAS, oxy
+
+oxy_space = [
+    oxy.HttpLLM(
+        name="default_llm",
+        api_key=os.getenv("DEFAULT_LLM_API_KEY"),
+        base_url=os.getenv("DEFAULT_LLM_BASE_URL"),
+        model_name=os.getenv("DEFAULT_LLM_MODEL_NAME"),
+        llm_params={"temperature": 0.01},
+        semaphore=4,
+    ),
+    oxy.StdioMCPClient(
+        name="time_tools",
+        params={
+            "command": "uvx",
+            "args": ["mcp-server-time", "--local-timezone=Asia/Shanghai"],
+        },
+    ),
+    oxy.ReActAgent(
+        name="qa_agent",
+        llm_model="default_llm",
+        tools=["time_tools", "remote_user_profile_banks"],
+    ),
+    oxy.SSEMCPClient(
+        name="remote_user_profile_banks",
+        sse_url="http://127.0.0.1:8000/sse",
+    ),
+]
+
+
+def func_filter(payload):
+    payload["group_data"] = {"user_pin": "002"}
+    return payload
+
+
+async def main():
+    async with MAS(oxy_space=oxy_space, func_filter=func_filter) as mas:
+        await mas.start_web_service(first_query="Who I am")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())