steven_li
ebfa242862
添加完整的Outlook MCP集成，包括邮件和日历功能，通过AuthZ模式进行认证和权限管理， 支持邮箱连接、断开、状态检查和数据同步等功能。 fix(config): 统一配置文件路径从.nanobot到.beaver 将配置文件路径从/root/.nanobot统一更改为/root/.beaver，更新Dockerfile中的环境变量定义， 确保所有组件使用一致的配置目录结构。 feat(agent): 添加代理删除功能和助手身份提示 为代理注册表添加delete_agent方法，实现代理的动态删除功能；同时添加海狸助手身份提示， 确保AI助手在交互中保持一致的身份认知。 feat(engine): 增强引擎循环并添加意图决策快照 扩展AgentLoop类，添加intent_agent_decision参数用于意图驱动的代理决策，并在会话中记录 决策快照，便于后续分析和调试。 feat(authz): 扩展认证客户端功能 为AuthzClient添加设置权限、用户注册、后端注册和Outlook设置管理等新方法，增强系统 的认证和授权能力。
136 lines
4.1 KiB
Python
136 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
|
|
from beaver.engine.providers.base import LLMProvider, LLMResponse
|
|
from beaver.tasks import MainAgentRouter, TaskRecord
|
|
|
|
|
|
class RouterProvider(LLMProvider):
|
|
def __init__(self, response: str | Exception) -> None:
|
|
super().__init__()
|
|
self.response = response
|
|
self.calls: list[dict] = []
|
|
|
|
async def chat(
|
|
self,
|
|
messages: list[dict],
|
|
tools: list[dict] | None = None,
|
|
model: str | None = None,
|
|
max_tokens: int = 4096,
|
|
temperature: float = 0.7,
|
|
thinking_enabled: bool | None = None,
|
|
) -> LLMResponse:
|
|
self.calls.append(
|
|
{
|
|
"messages": messages,
|
|
"max_tokens": max_tokens,
|
|
"temperature": temperature,
|
|
"model": model,
|
|
"thinking_enabled": thinking_enabled,
|
|
}
|
|
)
|
|
if isinstance(self.response, Exception):
|
|
raise self.response
|
|
return LLMResponse(content=self.response, finish_reason="stop", provider_name="stub", model="stub-model")
|
|
|
|
def get_default_model(self) -> str:
|
|
return "stub-model"
|
|
|
|
|
|
def _task() -> TaskRecord:
|
|
return TaskRecord(
|
|
task_id="task-1",
|
|
session_id="web:task",
|
|
description="实现任务连续性",
|
|
goal="实现任务连续性",
|
|
constraints=[],
|
|
priority=0,
|
|
status="awaiting_feedback",
|
|
creator="test",
|
|
created_at="now",
|
|
updated_at="now",
|
|
metadata={"short_title": "任务连续性"},
|
|
)
|
|
|
|
|
|
def test_router_continues_active_task_from_llm_decision() -> None:
|
|
provider = RouterProvider('{"action":"continue_task","reason":"related","short_title":"任务连续性"}')
|
|
decision = asyncio.run(
|
|
MainAgentRouter().classify(
|
|
"再把输入框标识也补上",
|
|
active_task=_task(),
|
|
provider=provider,
|
|
)
|
|
)
|
|
|
|
assert decision.is_task
|
|
assert decision.starts_new_task is False
|
|
assert decision.short_title == "任务连续性"
|
|
assert provider.calls[0]["max_tokens"] == 256
|
|
|
|
|
|
def test_router_receives_thinking_mode() -> None:
|
|
provider = RouterProvider('{"action":"simple_chat","reason":"simple"}')
|
|
decision = asyncio.run(
|
|
MainAgentRouter().classify(
|
|
"你好",
|
|
provider=provider,
|
|
thinking_enabled=False,
|
|
)
|
|
)
|
|
|
|
assert not decision.is_task
|
|
assert provider.calls[0]["thinking_enabled"] is False
|
|
|
|
|
|
def test_router_injects_intent_skill_guidance() -> None:
|
|
provider = RouterProvider('{"action":"new_task","reason":"needs weather tool","short_title":"珠海天气"}')
|
|
decision = asyncio.run(
|
|
MainAgentRouter().classify(
|
|
"帮我查一下今天珠海天气",
|
|
provider=provider,
|
|
intent_skill="Weather and current external data must be routed to new_task.",
|
|
)
|
|
)
|
|
|
|
assert decision.is_task
|
|
assert decision.starts_new_task is True
|
|
assert decision.action == "create_task"
|
|
prompt = provider.calls[0]["messages"][1]["content"]
|
|
assert "Intent Agent skill guidance" in prompt
|
|
assert "Weather and current external data" in prompt
|
|
|
|
|
|
def test_router_closes_active_task_from_llm_decision() -> None:
|
|
decision = asyncio.run(
|
|
MainAgentRouter().classify(
|
|
"这个任务结束了",
|
|
active_task=_task(),
|
|
provider=RouterProvider('{"action":"close_task","reason":"user said done"}'),
|
|
)
|
|
)
|
|
|
|
assert not decision.is_task
|
|
assert decision.closes_task is True
|
|
|
|
|
|
def test_router_fallback_keeps_active_task_but_not_new_task() -> None:
|
|
active = asyncio.run(
|
|
MainAgentRouter().classify(
|
|
"继续",
|
|
active_task=_task(),
|
|
provider=RouterProvider(RuntimeError("provider down")),
|
|
)
|
|
)
|
|
inactive = asyncio.run(
|
|
MainAgentRouter().classify(
|
|
"implement something",
|
|
active_task=None,
|
|
provider=RouterProvider(RuntimeError("provider down")),
|
|
)
|
|
)
|
|
|
|
assert active.is_task
|
|
assert not inactive.is_task
|