JiuContinent/company-ai-platform
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

```

Browse Source 
feat(ai_agent): 完善AI适配器和服务功能

- 添加OpenClaw和Hermes健康检查接口
- 实现OpenClaw工具调用功能
- 重构AI适配器使用常量定义
- 增加AI技能系统支持
- 更新配置文件中的默认模型提供者设置

refactor(scheduler): 使用常量替换硬编码值

- 将硬编码的actor值替换为ActorValue常量
- 将receive_id_type替换为FeishuReceiveIdType枚举

refactor(audit): 统一审计日志常量使用

- 将硬编码的actor、source、risk_level等值替换为对应常量
- 更新审核服务中的状态和操作常量引用

refactor(approvals): 标准化审批模块常量使用

- 将applicant默认值替换为ActorValue.API常量
- 使用ApprovalStatus常量替代硬编码状态值
- 更新审核操作常量引用
```
This commit is contained in:
JiuContinent
2026-07-06 00:02:03 +08:00
parent d82116d637
commit aa81fc5321
36 changed files with 2328 additions and 337 deletions
Download Patch File Download Diff File Expand all files Collapse all files

163
tests/test_ai_adapters.py
View File

@@ -2,6 +2,16 @@ from typing import Any
from app.core.config import Settings
from app.modules.ai_agent import adapters
from app.modules.ai_agent.constants import (
OPENCLAW_HERMES_PIPELINE,
AIChatRole,
AIContextKey,
AIHttpHeader,
AIHttpPath,
AIHttpPayloadKey,
AIMemoryMode,
AIResponseKey,
)
class DummyResponse:
@@ -14,11 +24,26 @@ class DummyResponse:
return self._data
def chat_response(content: str) -> DummyResponse:
return DummyResponse(
{
AIHttpPayloadKey.CHOICES: [
{
AIHttpPayloadKey.MESSAGE: {
AIHttpPayloadKey.CONTENT: content,
}
}
]
}
)
class DummyClient:
calls: list[dict[str, Any]] = []
def __init__(self, timeout: int):
def __init__(self, timeout: int, trust_env: bool = True):
self.timeout = timeout
self.trust_env = trust_env
def __enter__(self) -> "DummyClient":
return self
@@ -32,16 +57,98 @@ class DummyClient:
json: dict[str, Any],
headers: dict[str, str],
) -> DummyResponse:
self.calls.append({"url": url, "json": json, "headers": headers})
if url.endswith("/api/v1/agent/ask"):
assert json["context"]["hermes_memory"] == "remembered project preference"
return DummyResponse({"answer": "openclaw final answer", "tool_calls": 1})
mode = json["context"]["mode"]
if mode == "memory_recall":
return DummyResponse({"answer": "remembered project preference"})
if mode == "memory_write":
return DummyResponse({"answer": "stored"})
raise AssertionError(f"unexpected Hermes mode: {mode}")
self.calls.append({"method": "POST", "url": url, "json": json, "headers": headers})
if url.endswith(AIHttpPath.CHAT_COMPLETIONS):
user_content = json[AIHttpPayloadKey.MESSAGES][-1][AIHttpPayloadKey.CONTENT]
if AIMemoryMode.RECALL in user_content:
return chat_response("remembered project preference")
if AIMemoryMode.WRITE in user_content:
return chat_response("stored")
if AIContextKey.AGENT_PIPELINE in user_content:
return chat_response("hermes final answer")
return chat_response("hermes direct answer")
if url.endswith(AIHttpPath.TOOLS_INVOKE):
return DummyResponse(
{
AIResponseKey.OK: True,
AIHttpPayloadKey.TOOL: json[AIHttpPayloadKey.TOOL],
AIHttpPayloadKey.ACTION: json[AIHttpPayloadKey.ACTION],
AIHttpPayloadKey.ARGS: json[AIHttpPayloadKey.ARGS],
AIHttpPayloadKey.SESSION_KEY: json[AIHttpPayloadKey.SESSION_KEY],
}
)
raise AssertionError(f"unexpected URL: {url}")
def get(self, url: str, headers: dict[str, str]) -> DummyResponse:
self.calls.append({"method": "GET", "url": url, "headers": headers})
if url.endswith(AIHttpPath.HEALTHZ):
return DummyResponse({"status": "ok"})
if url.endswith(AIHttpPath.READYZ):
return DummyResponse({"status": "ready"})
if url.endswith(AIHttpPath.HEALTH):
return DummyResponse({"status": "ok", "platform": "hermes-agent"})
raise AssertionError(f"unexpected URL: {url}")
def test_hermes_adapter_uses_openai_chat_completions(monkeypatch) -> None:
DummyClient.calls = []
monkeypatch.setattr(adapters.httpx, "Client", DummyClient)
settings = Settings(
hermes_base_url="http://hermes.local/v1",
hermes_api_key="hermes-key",
hermes_model="hermes-agent",
hermes_session_id="company-ai-platform-test",
)
result = adapters.HermesAdapter(settings).ask(
"summarize project risk",
{"project_code": "P-001"},
)
assert result[AIResponseKey.ANSWER] == "hermes direct answer"
assert DummyClient.calls[0]["url"] == "http://hermes.local/v1/chat/completions"
assert DummyClient.calls[0]["headers"][AIHttpHeader.AUTHORIZATION] == "Bearer hermes-key"
assert (
DummyClient.calls[0]["headers"][AIHttpHeader.HERMES_SESSION_ID]
== "company-ai-platform-test"
)
assert DummyClient.calls[0]["json"][AIHttpPayloadKey.MODEL] == "hermes-agent"
assert (
DummyClient.calls[0]["json"][AIHttpPayloadKey.MESSAGES][0][AIHttpPayloadKey.ROLE]
== AIChatRole.SYSTEM
)
assert "P-001" in DummyClient.calls[0]["json"][AIHttpPayloadKey.MESSAGES][1][
AIHttpPayloadKey.CONTENT
]
def test_openclaw_adapter_uses_gateway_health_and_tool_invoke(monkeypatch) -> None:
DummyClient.calls = []
monkeypatch.setattr(adapters.httpx, "Client", DummyClient)
settings = Settings(
openclaw_http_url="http://openclaw.local",
openclaw_gateway_token="gateway-token",
)
health = adapters.OpenClawAdapter(settings).health()
result = adapters.OpenClawAdapter(settings).ask(
"list sessions",
{
AIContextKey.OPENCLAW_TOOL: "sessions_list",
AIContextKey.OPENCLAW_ARGS: {},
AIContextKey.OPENCLAW_SESSION_KEY: "main",
},
)
assert health[AIResponseKey.OK] is True
assert result[AIResponseKey.RAW][AIResponseKey.DATA][AIHttpPayloadKey.TOOL] == "sessions_list"
assert [call["url"] for call in DummyClient.calls] == [
"http://openclaw.local/healthz",
"http://openclaw.local/readyz",
"http://openclaw.local/tools/invoke",
]
assert DummyClient.calls[0]["headers"][AIHttpHeader.AUTHORIZATION] == "Bearer gateway-token"
assert DummyClient.calls[2]["json"][AIHttpPayloadKey.SESSION_KEY] == "main"
def test_openclaw_hermes_adapter_runs_recall_answer_and_remember(monkeypatch) -> None:
@@ -49,24 +156,34 @@ def test_openclaw_hermes_adapter_runs_recall_answer_and_remember(monkeypatch) ->
monkeypatch.setattr(adapters.httpx, "Client", DummyClient)
settings = Settings(
model_provider="openclaw_hermes",
openclaw_base_url="http://openclaw.local",
openclaw_api_key="openclaw-key",
hermes_base_url="http://hermes.local",
openclaw_http_url="http://openclaw.local",
openclaw_gateway_token="openclaw-key",
hermes_base_url="http://hermes.local/v1",
hermes_api_key="hermes-key",
)
result = adapters.OpenClawHermesAdapter(settings).ask(
"summarize project risk",
{"project_code": "P-001"},
{
"project_code": "P-001",
AIContextKey.OPENCLAW_TOOL: "sessions_list",
AIContextKey.OPENCLAW_ARGS: {},
},
)
assert result["answer"] == "openclaw final answer"
assert result["raw"]["pipeline"] == "hermes_recall -> openclaw_answer -> hermes_remember"
assert result[AIResponseKey.ANSWER] == "hermes final answer"
assert result[AIResponseKey.RAW][AIResponseKey.PIPELINE] == OPENCLAW_HERMES_PIPELINE
assert [call["url"] for call in DummyClient.calls] == [
"http://hermes.local/api/v1/ask",
"http://openclaw.local/api/v1/agent/ask",
"http://hermes.local/api/v1/ask",
"http://hermes.local/v1/chat/completions",
"http://openclaw.local/healthz",
"http://openclaw.local/readyz",
"http://openclaw.local/tools/invoke",
"http://hermes.local/v1/chat/completions",
"http://hermes.local/v1/chat/completions",
]
assert DummyClient.calls[0]["headers"][AIHttpHeader.AUTHORIZATION] == "Bearer hermes-key"
assert DummyClient.calls[1]["headers"][AIHttpHeader.AUTHORIZATION] == "Bearer openclaw-key"
assert DummyClient.calls[3]["json"][AIHttpPayloadKey.TOOL] == "sessions_list"
assert AIContextKey.OPENCLAW in DummyClient.calls[4]["json"][AIHttpPayloadKey.MESSAGES][1][
AIHttpPayloadKey.CONTENT
]
assert DummyClient.calls[0]["headers"]["Authorization"] == "Bearer hermes-key"
assert DummyClient.calls[1]["headers"]["Authorization"] == "Bearer openclaw-key"
assert DummyClient.calls[1]["json"]["context"]["agent_pipeline"] == "openclaw_hermes"