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"

165
tests/test_smoke.py
View File

@@ -4,6 +4,9 @@ import tempfile
from datetime import date, timedelta
from pathlib import Path
from app.modules.ai_agent.constants import AIProviderName, AIResponseKey
from app.modules.business.constants import StatusValue
_db = tempfile.NamedTemporaryFile(delete=False, suffix=".db")
_db.close()
@@ -11,13 +14,21 @@ os.environ["DATABASE_URL"] = "sqlite:///" + _db.name.replace("\\", "/")
os.environ["API_KEY"] = "test-key"
os.environ["FEISHU_APP_ID"] = ""
os.environ["FEISHU_APP_SECRET"] = ""
os.environ["MODEL_PROVIDER"] = "noop"
os.environ["MODEL_PROVIDER"] = AIProviderName.NOOP
os.environ["SCHEDULER_ENABLED"] = "false"
from fastapi.testclient import TestClient
from app.core.database import Base, engine
from app.main import app
from app.modules.reports.constants import (
LifecycleAttentionKey,
LifecycleResponseKey,
LifecycleSection,
MetricKey,
ReportTitle,
ReportType,
)
Base.metadata.create_all(bind=engine)
@@ -203,10 +214,10 @@ def test_new_ledgers_reports_and_risk_events() -> None:
report_response = client.post(
"/api/v1/reports/work-reports/generate",
headers=headers,
json={"report_type": "daily", "reporter": "pytest", "actor": "pytest"},
json={"report_type": ReportType.DAILY, "reporter": "pytest", "actor": "pytest"},
)
assert report_response.status_code == 200
assert report_response.json()["data"]["report_type"] == "daily"
assert report_response.json()["data"]["report_type"] == ReportType.DAILY
risk_response = client.post(
"/api/v1/risks/events/generate?actor=pytest",
@@ -220,11 +231,155 @@ def test_new_ledgers_reports_and_risk_events() -> None:
assert any(item["risk_type"] == "overdue_task" for item in events_response.json()["items"])
def test_project_lifecycle_report_summarizes_progress_cost_and_risk() -> None:
today = date.today()
project_code = "P-LIFECYCLE-001"
project_response = client.post(
"/api/v1/business/projects",
headers=headers,
json={
"actor": "pytest",
"data": {
"code": project_code,
"name": "Lifecycle Project",
"owner": "lifecycle-owner",
"status": "执行中",
"progress_percent": 40,
"budget_amount": 1000,
"actual_amount": 1500,
"due_date": (today - timedelta(days=1)).isoformat(),
},
},
)
assert project_response.status_code == 200
task_response = client.post(
"/api/v1/business/tasks",
headers=headers,
json={
"actor": "pytest",
"data": {
"code": "TASK-LIFECYCLE-001",
"title": "Lifecycle overdue task",
"project_code": project_code,
"owner": "lifecycle-owner",
"status": "待办",
"due_date": (today - timedelta(days=1)).isoformat(),
"blocker": "waiting for decision",
},
},
)
assert task_response.status_code == 200
procurement_response = client.post(
"/api/v1/business/procurements",
headers=headers,
json={
"actor": "pytest",
"data": {
"code": "PROC-LIFECYCLE-001",
"name": "Lifecycle procurement",
"project_code": project_code,
"expected_amount": 300,
"actual_amount": 100,
"approval_status": StatusValue.PENDING_APPROVAL,
"delivery_status": StatusValue.UNDELIVERED,
"payment_status": StatusValue.UNPAID,
},
},
)
assert procurement_response.status_code == 200
expense_response = client.post(
"/api/v1/business/expenses",
headers=headers,
json={
"actor": "pytest",
"data": {
"code": "EXP-LIFECYCLE-001",
"expense_type": "差旅",
"amount": 80,
"project_code": project_code,
"approval_status": StatusValue.PENDING_APPROVAL,
"payment_status": StatusValue.UNPAID,
},
},
)
assert expense_response.status_code == 200
attendance_response = client.post(
"/api/v1/business/attendance-records",
headers=headers,
json={
"actor": "pytest",
"data": {
"code": "ATT-LIFECYCLE-001",
"employee_name": "Lifecycle Tester",
"project_code": project_code,
"work_date": today.isoformat(),
"status": StatusValue.MISSING_PUNCH,
},
},
)
assert attendance_response.status_code == 200
response = client.get(
f"/api/v1/reports/project-lifecycle?project_code={project_code}",
headers=headers,
)
assert response.status_code == 200
data = response.json()
assert data[LifecycleResponseKey.TITLE] == ReportTitle.PROJECT_LIFECYCLE
assert data[LifecycleResponseKey.METRICS][LifecycleSection.PROJECTS][MetricKey.TOTAL] == 1
assert data[LifecycleResponseKey.METRICS][LifecycleSection.PROJECTS][MetricKey.DELAYED] == 1
assert (
data[LifecycleResponseKey.METRICS][LifecycleSection.PROJECTS][MetricKey.OVER_BUDGET]
== 1
)
assert data[LifecycleResponseKey.METRICS][LifecycleSection.TASKS][MetricKey.OVERDUE] == 1
assert (
data[LifecycleResponseKey.METRICS][LifecycleSection.PROCUREMENTS][
MetricKey.PENDING_APPROVAL
]
== 1
)
assert (
data[LifecycleResponseKey.METRICS][LifecycleSection.EXPENSES][
MetricKey.PENDING_APPROVAL
]
== 1
)
assert data[LifecycleResponseKey.METRICS][LifecycleSection.ATTENDANCE][MetricKey.ABNORMAL] == 1
assert (
data[LifecycleResponseKey.ATTENTION][LifecycleAttentionKey.DELAYED_PROJECTS][0]["code"]
== project_code
)
assert "生命周期健康分" in data[LifecycleResponseKey.CONTENT]
assert data[LifecycleResponseKey.RECOMMENDATIONS]
ai_response = client.get(
f"/api/v1/reports/project-lifecycle?project_code={project_code}&include_ai=true",
headers=headers,
)
assert ai_response.status_code == 200
ai_data = ai_response.json()
assert ai_data[LifecycleResponseKey.AI_ANALYSIS][AIResponseKey.OK] is True
assert (
ai_data[LifecycleResponseKey.AI_ANALYSIS][AIResponseKey.PROVIDER]
== AIProviderName.NOOP
)
def test_ai_noop_provider() -> None:
response = client.post(
"/api/v1/ai/ask",
headers=headers,
json={"prompt": "生成项目摘要", "actor": "pytest", "context": {"project": "P-SMOKE-001"}},
json={
"prompt": "生成项目摘要",
"actor": "pytest",
"context": {"project": "P-SMOKE-001"},
},
)
assert response.status_code == 200
assert response.json()["provider"] == "noop"
assert response.json()[AIResponseKey.PROVIDER] == AIProviderName.NOOP