2026-04-16 22:01:51 +08:00
|
|
|
use crate::app::AppState;
|
|
|
|
|
use crate::context::RequestContext;
|
2026-04-17 00:25:28 +08:00
|
|
|
use crate::error::WebError;
|
2026-05-11 13:16:34 +08:00
|
|
|
use axum::body::Body;
|
2026-04-16 22:01:51 +08:00
|
|
|
use axum::extract::{Extension, State};
|
2026-05-06 21:44:20 +08:00
|
|
|
use axum::http::{header, HeaderMap, HeaderName, HeaderValue, Request, StatusCode};
|
|
|
|
|
use axum::response::{IntoResponse, Response};
|
2026-04-29 12:24:44 +08:00
|
|
|
use axum::Json;
|
|
|
|
|
use serde_json::{json, Value};
|
2026-05-06 21:44:20 +08:00
|
|
|
use std::env;
|
|
|
|
|
use std::path::PathBuf;
|
|
|
|
|
use std::process::Command;
|
2026-04-17 00:25:28 +08:00
|
|
|
|
2026-04-16 22:01:51 +08:00
|
|
|
pub async fn next_ai_agent_run(
|
|
|
|
|
State(state): State<AppState>,
|
|
|
|
|
Extension(context): Extension<RequestContext>,
|
2026-05-06 21:44:20 +08:00
|
|
|
request: Request<Body>,
|
|
|
|
|
) -> Result<Response, WebError> {
|
2026-05-11 13:16:34 +08:00
|
|
|
let (_parts, body) = request.into_parts();
|
2026-05-06 21:44:20 +08:00
|
|
|
let body = axum::body::to_bytes(body, 10 * 1024 * 1024)
|
|
|
|
|
.await
|
|
|
|
|
.map_err(|error| WebError::internal(format!("读取 AI 请求体失败: {error}")))?;
|
|
|
|
|
let payload: Value = serde_json::from_slice(&body).map_err(|error| {
|
|
|
|
|
WebError::bad_request_code(
|
|
|
|
|
"ai_agent_bad_request",
|
|
|
|
|
format!("AI 请求体不是合法 JSON: {error}"),
|
|
|
|
|
)
|
|
|
|
|
.with_context(&context)
|
|
|
|
|
.with_header("x-mnote-web-owner", "mnote-web")
|
|
|
|
|
})?;
|
|
|
|
|
|
2026-05-08 00:41:03 +08:00
|
|
|
if let Some(provider) = explicit_agent_provider(&payload) {
|
|
|
|
|
return Err(WebError::bad_gateway_code(
|
|
|
|
|
"ai_provider_bridge_unavailable",
|
|
|
|
|
format!(
|
2026-05-11 13:16:34 +08:00
|
|
|
"{provider} provider 直连链路已退场；当前仅保留 mnote-cli host 主路径，不再静默降级。"
|
2026-05-08 00:41:03 +08:00
|
|
|
),
|
|
|
|
|
)
|
|
|
|
|
.with_context(&context)
|
|
|
|
|
.with_header("x-mnote-web-owner", "mnote-web")
|
|
|
|
|
.with_header("x-mnote-ai-execution-owner", "provider-bridge-unavailable"));
|
|
|
|
|
}
|
|
|
|
|
|
2026-05-11 13:16:34 +08:00
|
|
|
run_local_mnote_cli_ai_host(&state, &context, &payload).await
|
2026-05-06 21:44:20 +08:00
|
|
|
}
|
|
|
|
|
|
2026-05-08 00:41:03 +08:00
|
|
|
fn explicit_agent_provider(payload: &Value) -> Option<&'static str> {
|
|
|
|
|
let provider = payload
|
|
|
|
|
.get("options")
|
|
|
|
|
.and_then(|value| value.get("ai"))
|
|
|
|
|
.and_then(|value| value.get("provider"))
|
|
|
|
|
.and_then(Value::as_str)?
|
|
|
|
|
.trim()
|
|
|
|
|
.to_ascii_lowercase();
|
|
|
|
|
match provider.as_str() {
|
|
|
|
|
"hermes" => Some("hermes"),
|
|
|
|
|
"codex" => Some("codex"),
|
|
|
|
|
"claudecode" => Some("claudecode"),
|
|
|
|
|
_ => None,
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-05-06 21:44:20 +08:00
|
|
|
fn resolve_repo_root() -> PathBuf {
|
|
|
|
|
PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../../..")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn build_mnote_cli_args(context: &RequestContext, payload: &Value) -> Vec<String> {
|
|
|
|
|
let ai = payload
|
|
|
|
|
.get("options")
|
|
|
|
|
.and_then(|value| value.get("ai"))
|
|
|
|
|
.cloned()
|
|
|
|
|
.unwrap_or(Value::Null);
|
|
|
|
|
let runtime_context = payload.get("context").cloned().unwrap_or(Value::Null);
|
|
|
|
|
let document_id = runtime_context
|
|
|
|
|
.get("documentId")
|
|
|
|
|
.and_then(Value::as_str)
|
|
|
|
|
.unwrap_or("current");
|
|
|
|
|
let workspace_id = runtime_context
|
|
|
|
|
.get("workspaceId")
|
|
|
|
|
.and_then(Value::as_str)
|
|
|
|
|
.or(context.workspace.workspace_id.as_deref());
|
|
|
|
|
let session_id = ai
|
|
|
|
|
.get("sessionId")
|
|
|
|
|
.and_then(Value::as_str)
|
|
|
|
|
.filter(|value| !value.trim().is_empty())
|
|
|
|
|
.map(str::to_string)
|
|
|
|
|
.unwrap_or_else(|| format!("ai-{}", context.trace.request_id));
|
|
|
|
|
let args_json = json!({
|
|
|
|
|
"pageId": document_id,
|
|
|
|
|
"documentId": document_id,
|
|
|
|
|
"workspaceId": workspace_id,
|
|
|
|
|
"provider": ai.get("provider").cloned().unwrap_or(Value::Null),
|
|
|
|
|
"modelKey": ai.get("modelKey").cloned().unwrap_or(Value::Null),
|
|
|
|
|
"profileId": ai.get("profileId").cloned().unwrap_or(Value::Null),
|
|
|
|
|
"selectedUids": runtime_context.get("selectedUids").cloned().unwrap_or(Value::Null),
|
|
|
|
|
"pageOptions": runtime_context.get("pageOptions").cloned().unwrap_or(Value::Null),
|
2026-04-16 22:01:51 +08:00
|
|
|
})
|
2026-05-06 21:44:20 +08:00
|
|
|
.to_string();
|
|
|
|
|
|
|
|
|
|
vec![
|
|
|
|
|
"run".into(),
|
|
|
|
|
"--quiet".into(),
|
|
|
|
|
"--manifest-path".into(),
|
|
|
|
|
resolve_repo_root()
|
|
|
|
|
.join("rust")
|
|
|
|
|
.join("Cargo.toml")
|
|
|
|
|
.to_string_lossy()
|
|
|
|
|
.to_string(),
|
|
|
|
|
"-p".into(),
|
|
|
|
|
"mnote-cli".into(),
|
|
|
|
|
"--".into(),
|
|
|
|
|
"--json".into(),
|
|
|
|
|
"--validate-only".into(),
|
|
|
|
|
"--dry-run".into(),
|
|
|
|
|
"--actor-id".into(),
|
|
|
|
|
context.auth.actor_id.clone(),
|
|
|
|
|
"--actor-type".into(),
|
|
|
|
|
context.auth.actor_type.clone(),
|
|
|
|
|
"--session-id".into(),
|
|
|
|
|
session_id,
|
|
|
|
|
"--reason".into(),
|
|
|
|
|
"ai-agent-run:mnote-web-rust-host".into(),
|
|
|
|
|
"tool".into(),
|
|
|
|
|
"run".into(),
|
|
|
|
|
"--tool-name".into(),
|
|
|
|
|
"doc_get".into(),
|
|
|
|
|
"--kind".into(),
|
|
|
|
|
"query".into(),
|
|
|
|
|
"--mode".into(),
|
|
|
|
|
"explain-plan".into(),
|
|
|
|
|
"--args-json".into(),
|
|
|
|
|
args_json,
|
|
|
|
|
]
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
async fn run_local_mnote_cli_ai_host(
|
|
|
|
|
state: &AppState,
|
|
|
|
|
context: &RequestContext,
|
|
|
|
|
payload: &Value,
|
|
|
|
|
) -> Result<Response, WebError> {
|
|
|
|
|
let stream = payload
|
|
|
|
|
.get("stream")
|
|
|
|
|
.and_then(Value::as_bool)
|
|
|
|
|
.unwrap_or(true);
|
|
|
|
|
let args = build_mnote_cli_args(context, payload);
|
|
|
|
|
let repo_root = resolve_repo_root();
|
|
|
|
|
let actor_id =
|
|
|
|
|
if context.auth.actor_id.trim().is_empty() || context.auth.actor_id == "anonymous" {
|
|
|
|
|
state.config().dev_user_id.clone()
|
|
|
|
|
} else {
|
|
|
|
|
context.auth.actor_id.clone()
|
|
|
|
|
};
|
|
|
|
|
let actor_type =
|
|
|
|
|
if context.auth.actor_type.trim().is_empty() || context.auth.actor_type == "anonymous" {
|
|
|
|
|
"user".to_string()
|
|
|
|
|
} else {
|
|
|
|
|
context.auth.actor_type.clone()
|
|
|
|
|
};
|
|
|
|
|
let dev_email = state.config().dev_user_email.clone();
|
|
|
|
|
let dev_name = state.config().dev_user_name.clone();
|
|
|
|
|
|
|
|
|
|
let output = tokio::task::spawn_blocking(move || {
|
|
|
|
|
Command::new("cargo")
|
|
|
|
|
.args(args)
|
|
|
|
|
.current_dir(repo_root)
|
|
|
|
|
.env("CARGO_TERM_COLOR", "never")
|
|
|
|
|
.env(
|
|
|
|
|
"RUSTUP_TOOLCHAIN",
|
|
|
|
|
env::var("RUSTUP_TOOLCHAIN")
|
|
|
|
|
.ok()
|
|
|
|
|
.filter(|value| !value.trim().is_empty())
|
|
|
|
|
.unwrap_or_else(|| "1.89.0".into()),
|
|
|
|
|
)
|
|
|
|
|
.env("DEV_USER_ID", actor_id)
|
|
|
|
|
.env("DEV_USER_EMAIL", dev_email)
|
|
|
|
|
.env("DEV_USER_NAME", dev_name)
|
|
|
|
|
.env("MNOTE_CLI_ALLOW_CREATE_PAGE", "1")
|
|
|
|
|
.env("MNOTE_CLI_ALLOW_EDIT", "1")
|
|
|
|
|
.env("MNOTE_ACTOR_TYPE", actor_type)
|
|
|
|
|
.output()
|
|
|
|
|
})
|
|
|
|
|
.await
|
|
|
|
|
.map_err(|error| {
|
|
|
|
|
WebError::bad_gateway_code(
|
|
|
|
|
"mnote_cli_host_join_error",
|
|
|
|
|
format!("mnote-cli host join 失败: {error}"),
|
|
|
|
|
)
|
|
|
|
|
.with_context(context)
|
|
|
|
|
.with_header("x-mnote-web-owner", "mnote-web")
|
|
|
|
|
})?
|
|
|
|
|
.map_err(|error| {
|
|
|
|
|
WebError::bad_gateway_code(
|
|
|
|
|
"mnote_cli_host_spawn_error",
|
|
|
|
|
format!("mnote-cli host 启动失败: {error}"),
|
|
|
|
|
)
|
|
|
|
|
.with_context(context)
|
|
|
|
|
.with_header("x-mnote-web-owner", "mnote-web")
|
|
|
|
|
})?;
|
|
|
|
|
|
|
|
|
|
let stdout = String::from_utf8_lossy(&output.stdout).trim().to_string();
|
|
|
|
|
let stderr = String::from_utf8_lossy(&output.stderr).trim().to_string();
|
|
|
|
|
if !stream {
|
|
|
|
|
if !output.status.success() {
|
|
|
|
|
return Err(WebError::bad_gateway_code(
|
|
|
|
|
"mnote_cli_host_failed",
|
|
|
|
|
if stderr.is_empty() {
|
|
|
|
|
"mnote-cli 执行失败".into()
|
|
|
|
|
} else {
|
|
|
|
|
stderr
|
|
|
|
|
},
|
|
|
|
|
)
|
|
|
|
|
.with_context(context)
|
|
|
|
|
.with_header("x-mnote-web-owner", "mnote-web"));
|
|
|
|
|
}
|
|
|
|
|
let mut response = Json(json!({
|
|
|
|
|
"ok": true,
|
|
|
|
|
"bridgeOwner": "mnote-cli",
|
|
|
|
|
"text": if stdout.is_empty() { "mnote-cli 无输出" } else { &stdout },
|
|
|
|
|
}))
|
|
|
|
|
.into_response();
|
|
|
|
|
stamp_owner_header(response.headers_mut());
|
|
|
|
|
response.headers_mut().insert(
|
|
|
|
|
HeaderName::from_static("x-mnote-ai-execution-owner"),
|
|
|
|
|
HeaderValue::from_static("mnote-cli"),
|
|
|
|
|
);
|
|
|
|
|
return Ok(response);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
let body = if output.status.success() {
|
|
|
|
|
format!(
|
|
|
|
|
"event: ready\ndata: {}\n\nevent: assistant_message\ndata: {}\n\nevent: completion\ndata: {}\n\n",
|
|
|
|
|
json!({"ok": true, "bridgeOwner": "mnote-cli"}).to_string(),
|
|
|
|
|
json!({"text": if stdout.is_empty() { "mnote-cli 无输出" } else { &stdout }}).to_string(),
|
|
|
|
|
json!({"ok": true, "text": if stdout.is_empty() { "mnote-cli 无输出" } else { &stdout }, "steps": 1}).to_string(),
|
|
|
|
|
)
|
|
|
|
|
} else {
|
|
|
|
|
format!(
|
|
|
|
|
"event: ready\ndata: {}\n\nevent: error\ndata: {}\n\n",
|
|
|
|
|
json!({"ok": true, "bridgeOwner": "mnote-cli"}).to_string(),
|
|
|
|
|
json!({"ok": false, "message": if stderr.is_empty() { "mnote-cli 执行失败" } else { &stderr }}).to_string(),
|
|
|
|
|
)
|
|
|
|
|
};
|
|
|
|
|
let mut response = Response::builder()
|
|
|
|
|
.status(StatusCode::OK)
|
|
|
|
|
.header(header::CONTENT_TYPE, "text/event-stream; charset=utf-8")
|
|
|
|
|
.header(header::CACHE_CONTROL, "no-cache, no-transform")
|
|
|
|
|
.header(header::CONNECTION, "keep-alive")
|
|
|
|
|
.header("x-accel-buffering", "no")
|
|
|
|
|
.header("x-mnote-ai-execution-owner", "mnote-cli")
|
|
|
|
|
.body(Body::from(body))
|
|
|
|
|
.map_err(|error| WebError::internal(format!("mnote-cli SSE 响应构造失败: {error}")))?;
|
|
|
|
|
stamp_owner_header(response.headers_mut());
|
|
|
|
|
Ok(response)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn stamp_owner_header(headers: &mut HeaderMap) {
|
|
|
|
|
if let Ok(name) = HeaderName::from_lowercase(b"x-mnote-web-owner") {
|
|
|
|
|
headers.insert(name, HeaderValue::from_static("mnote-web"));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2026-04-17 00:25:28 +08:00
|
|
|
#[cfg(test)]
|
|
|
|
|
mod tests {
|
2026-04-29 12:24:44 +08:00
|
|
|
use crate::app::{build_app, AppConfig, AppState};
|
2026-04-17 00:25:28 +08:00
|
|
|
use axum::body::Body;
|
2026-05-11 13:16:34 +08:00
|
|
|
use axum::http::{Request, StatusCode};
|
2026-05-06 21:44:20 +08:00
|
|
|
use axum::response::IntoResponse;
|
|
|
|
|
use tokio::net::TcpListener;
|
2026-04-17 00:25:28 +08:00
|
|
|
use tower::util::ServiceExt;
|
|
|
|
|
|
|
|
|
|
fn app() -> axum::Router {
|
|
|
|
|
build_app(AppState::new(AppConfig {
|
|
|
|
|
service_name: "mnote-web".into(),
|
|
|
|
|
service_version: "0.1.0".into(),
|
|
|
|
|
bind_addr: "127.0.0.1:0".into(),
|
2026-04-29 12:24:44 +08:00
|
|
|
public_bind_addr: "127.0.0.1:3000".into(),
|
|
|
|
|
legacy_next_base_url: Some("http://127.0.0.1:3100".into()),
|
|
|
|
|
enable_legacy_next_compat: true,
|
2026-04-23 07:38:34 +08:00
|
|
|
enable_debug_shell_routes: false,
|
2026-04-17 00:25:28 +08:00
|
|
|
hermes_base_path: "/api/hermes".into(),
|
|
|
|
|
compat_next_base_path: "/api/compat/next".into(),
|
|
|
|
|
convex_url: None,
|
|
|
|
|
convex_admin_key: None,
|
|
|
|
|
allow_dev_fixtures: true,
|
2026-04-17 23:36:24 +08:00
|
|
|
query_fixtures_json: Some(r#"{"sidebar:datasetList":{"active_workspace_id":"ws_demo","workspaces":[],"documents":[{"id":"page_root","workspace_id":"ws_demo","title":"工作区首页","parent_id":null,"sort_order":0,"is_starred":true,"is_template":false,"created_at":"2026-04-16T00:00:00Z","updated_at":"2026-04-16T00:00:00Z"},{"id":"page_child","workspace_id":"ws_demo","title":"子页面","parent_id":"page_root","sort_order":1,"is_starred":false,"is_template":false,"created_at":"2026-04-16T00:00:00Z","updated_at":"2026-04-16T00:00:00Z"}],"trashed_documents":[],"media_assets":[],"trashed_media_assets":[],"mindmap_assets":[],"trashed_mindmap_assets":[],"table_assets":[],"trashed_table_assets":[],"mindmap_docs":[],"mindmap_asset_children":{}},"bridgeLogs:listWorkspaceOverview":{"workspace_id":"ws_demo","command_logs":[],"domain_events":[],"next_cursor":null,"has_more":false,"filters":{"command_status":null,"event_status":null,"target_page_id":null,"target_block_id":null,"aggregate_type":null,"aggregate_id":null},"generated_at":"2026-04-16T00:00:00Z"}}"#.into()),
|
|
|
|
|
mutation_fixtures_json: None,
|
2026-04-17 00:25:28 +08:00
|
|
|
dev_user_id: "dev-user".into(),
|
|
|
|
|
dev_user_name: "开发用户".into(),
|
|
|
|
|
dev_user_email: "dev@mnote.local".into(),
|
|
|
|
|
}))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[tokio::test]
|
2026-05-11 13:16:34 +08:00
|
|
|
async fn compat_next_sidebar_route_is_not_registered_by_default() {
|
2026-04-17 00:25:28 +08:00
|
|
|
let response = app()
|
|
|
|
|
.oneshot(
|
|
|
|
|
Request::builder()
|
|
|
|
|
.uri("/api/compat/next/sidebar?workspaceId=ws_demo")
|
|
|
|
|
.body(Body::empty())
|
|
|
|
|
.expect("request"),
|
|
|
|
|
)
|
|
|
|
|
.await
|
|
|
|
|
.expect("response");
|
|
|
|
|
|
2026-05-11 13:16:34 +08:00
|
|
|
assert_eq!(response.status(), StatusCode::NOT_FOUND);
|
2026-04-17 00:25:28 +08:00
|
|
|
}
|
2026-05-06 21:44:20 +08:00
|
|
|
|
|
|
|
|
#[tokio::test]
|
|
|
|
|
async fn direct_ai_agent_run_is_owned_by_rust_web_when_next_compat_disabled() {
|
|
|
|
|
let response = build_app(AppState::new(AppConfig {
|
|
|
|
|
service_name: "mnote-web".into(),
|
|
|
|
|
service_version: "0.1.0".into(),
|
|
|
|
|
bind_addr: "127.0.0.1:0".into(),
|
|
|
|
|
public_bind_addr: "127.0.0.1:3000".into(),
|
|
|
|
|
legacy_next_base_url: Some("http://127.0.0.1:3100".into()),
|
|
|
|
|
enable_legacy_next_compat: false,
|
|
|
|
|
enable_debug_shell_routes: false,
|
|
|
|
|
hermes_base_path: "/api/hermes".into(),
|
|
|
|
|
compat_next_base_path: "/api/compat/next".into(),
|
|
|
|
|
convex_url: None,
|
|
|
|
|
convex_admin_key: None,
|
|
|
|
|
allow_dev_fixtures: true,
|
|
|
|
|
query_fixtures_json: None,
|
|
|
|
|
mutation_fixtures_json: None,
|
|
|
|
|
dev_user_id: "dev-user".into(),
|
|
|
|
|
dev_user_name: "开发用户".into(),
|
|
|
|
|
dev_user_email: "dev@mnote.local".into(),
|
|
|
|
|
}))
|
|
|
|
|
.oneshot(
|
|
|
|
|
Request::builder()
|
|
|
|
|
.method("POST")
|
|
|
|
|
.uri("/api/ai-agent/run")
|
|
|
|
|
.header("content-type", "application/json")
|
|
|
|
|
.body(Body::from(r#"{"stream":true,"scope":"document","messages":[{"role":"user","content":"ping"}],"context":{"documentId":"doc-1"},"options":{"ai":{"provider":"online"}}}"#))
|
|
|
|
|
.expect("request"),
|
|
|
|
|
)
|
|
|
|
|
.await
|
|
|
|
|
.expect("response");
|
|
|
|
|
|
|
|
|
|
assert_ne!(response.status(), StatusCode::SERVICE_UNAVAILABLE);
|
|
|
|
|
assert_eq!(
|
|
|
|
|
response
|
|
|
|
|
.headers()
|
|
|
|
|
.get("x-mnote-web-owner")
|
|
|
|
|
.and_then(|value| value.to_str().ok()),
|
|
|
|
|
Some("mnote-web")
|
|
|
|
|
);
|
|
|
|
|
let body = axum::body::to_bytes(response.into_body(), usize::MAX)
|
|
|
|
|
.await
|
|
|
|
|
.expect("body");
|
|
|
|
|
let text = String::from_utf8(body.to_vec()).expect("utf8");
|
|
|
|
|
assert!(!text.contains("legacy_next_compat_disabled"));
|
|
|
|
|
}
|
|
|
|
|
|
2026-05-08 00:41:03 +08:00
|
|
|
#[tokio::test]
|
|
|
|
|
async fn explicit_agent_provider_does_not_fallback_to_mnote_cli_when_next_unavailable() {
|
|
|
|
|
let response = build_app(AppState::new(AppConfig {
|
|
|
|
|
service_name: "mnote-web".into(),
|
|
|
|
|
service_version: "0.1.0".into(),
|
|
|
|
|
bind_addr: "127.0.0.1:0".into(),
|
|
|
|
|
public_bind_addr: "127.0.0.1:3000".into(),
|
|
|
|
|
legacy_next_base_url: Some("http://127.0.0.1:9".into()),
|
|
|
|
|
enable_legacy_next_compat: false,
|
|
|
|
|
enable_debug_shell_routes: false,
|
|
|
|
|
hermes_base_path: "/api/hermes".into(),
|
|
|
|
|
compat_next_base_path: "/api/compat/next".into(),
|
|
|
|
|
convex_url: None,
|
|
|
|
|
convex_admin_key: None,
|
|
|
|
|
allow_dev_fixtures: true,
|
|
|
|
|
query_fixtures_json: None,
|
|
|
|
|
mutation_fixtures_json: None,
|
|
|
|
|
dev_user_id: "dev-user".into(),
|
|
|
|
|
dev_user_name: "开发用户".into(),
|
|
|
|
|
dev_user_email: "dev@mnote.local".into(),
|
|
|
|
|
}))
|
|
|
|
|
.oneshot(
|
|
|
|
|
Request::builder()
|
|
|
|
|
.method("POST")
|
|
|
|
|
.uri("/api/ai-agent/run")
|
|
|
|
|
.header("content-type", "application/json")
|
|
|
|
|
.body(Body::from(r#"{"stream":true,"scope":"document","messages":[{"role":"user","content":"ping"}],"context":{"documentId":"doc-1"},"options":{"ai":{"provider":"codex"}}}"#))
|
|
|
|
|
.expect("request"),
|
|
|
|
|
)
|
|
|
|
|
.await
|
|
|
|
|
.expect("response");
|
|
|
|
|
|
|
|
|
|
assert_eq!(response.status(), StatusCode::BAD_GATEWAY);
|
|
|
|
|
assert_eq!(
|
|
|
|
|
response
|
|
|
|
|
.headers()
|
|
|
|
|
.get("x-mnote-ai-execution-owner")
|
|
|
|
|
.and_then(|value| value.to_str().ok()),
|
|
|
|
|
Some("provider-bridge-unavailable")
|
|
|
|
|
);
|
|
|
|
|
let body = axum::body::to_bytes(response.into_body(), usize::MAX)
|
|
|
|
|
.await
|
|
|
|
|
.expect("body");
|
|
|
|
|
let text = String::from_utf8(body.to_vec()).expect("utf8");
|
|
|
|
|
assert!(text.contains("codex"));
|
2026-05-11 13:16:34 +08:00
|
|
|
assert!(text.contains("provider 直连链路已退场"));
|
2026-05-08 00:41:03 +08:00
|
|
|
}
|
|
|
|
|
|
2026-05-06 21:44:20 +08:00
|
|
|
#[tokio::test]
|
2026-05-11 13:16:34 +08:00
|
|
|
async fn explicit_hermes_provider_fails_instead_of_using_legacy_next_or_direct_bridge() {
|
2026-05-06 21:44:20 +08:00
|
|
|
let next_app = axum::Router::new().route(
|
|
|
|
|
"/api/ai-agent/run",
|
|
|
|
|
axum::routing::post(|| async move {
|
|
|
|
|
(
|
|
|
|
|
[(
|
|
|
|
|
axum::http::header::CONTENT_TYPE,
|
|
|
|
|
"text/event-stream; charset=utf-8",
|
|
|
|
|
)],
|
|
|
|
|
"event: assistant_message\ndata: {\"text\":\"hello from next ai\"}\n\n",
|
|
|
|
|
)
|
|
|
|
|
.into_response()
|
|
|
|
|
}),
|
|
|
|
|
);
|
|
|
|
|
let listener = TcpListener::bind("127.0.0.1:0").await.expect("bind next");
|
|
|
|
|
let addr = listener.local_addr().expect("local addr");
|
|
|
|
|
let server = tokio::spawn(async move {
|
|
|
|
|
axum::serve(listener, next_app).await.expect("serve next");
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
let response = build_app(AppState::new(AppConfig {
|
|
|
|
|
service_name: "mnote-web".into(),
|
|
|
|
|
service_version: "0.1.0".into(),
|
|
|
|
|
bind_addr: "127.0.0.1:0".into(),
|
|
|
|
|
public_bind_addr: "127.0.0.1:3000".into(),
|
|
|
|
|
legacy_next_base_url: Some(format!("http://{}", addr)),
|
|
|
|
|
enable_legacy_next_compat: true,
|
|
|
|
|
enable_debug_shell_routes: false,
|
|
|
|
|
hermes_base_path: "/api/hermes".into(),
|
|
|
|
|
compat_next_base_path: "/api/compat/next".into(),
|
|
|
|
|
convex_url: None,
|
|
|
|
|
convex_admin_key: None,
|
|
|
|
|
allow_dev_fixtures: true,
|
|
|
|
|
query_fixtures_json: None,
|
|
|
|
|
mutation_fixtures_json: None,
|
|
|
|
|
dev_user_id: "dev-user".into(),
|
|
|
|
|
dev_user_name: "开发用户".into(),
|
|
|
|
|
dev_user_email: "dev@mnote.local".into(),
|
|
|
|
|
}))
|
|
|
|
|
.oneshot(
|
|
|
|
|
Request::builder()
|
|
|
|
|
.method("POST")
|
|
|
|
|
.uri("/api/ai-agent/run")
|
|
|
|
|
.header("content-type", "application/json")
|
|
|
|
|
.body(Body::from(r#"{"stream":true,"scope":"document","messages":[{"role":"user","content":"ping"}],"context":{"documentId":"doc-1"},"options":{"ai":{"provider":"hermes"}}}"#))
|
|
|
|
|
.expect("request"),
|
|
|
|
|
)
|
|
|
|
|
.await
|
|
|
|
|
.expect("response");
|
|
|
|
|
|
2026-05-11 13:16:34 +08:00
|
|
|
assert_eq!(response.status(), StatusCode::BAD_GATEWAY);
|
2026-05-06 21:44:20 +08:00
|
|
|
assert_eq!(
|
|
|
|
|
response
|
|
|
|
|
.headers()
|
2026-05-11 13:16:34 +08:00
|
|
|
.get("x-mnote-ai-execution-owner")
|
2026-05-06 21:44:20 +08:00
|
|
|
.and_then(|value| value.to_str().ok()),
|
2026-05-11 13:16:34 +08:00
|
|
|
Some("provider-bridge-unavailable")
|
2026-05-06 21:44:20 +08:00
|
|
|
);
|
|
|
|
|
let body = axum::body::to_bytes(response.into_body(), usize::MAX)
|
|
|
|
|
.await
|
|
|
|
|
.expect("body");
|
|
|
|
|
let text = String::from_utf8(body.to_vec()).expect("utf8");
|
2026-05-11 13:16:34 +08:00
|
|
|
assert!(text.contains("hermes"));
|
|
|
|
|
assert!(!text.contains("hello from next ai"));
|
2026-05-06 21:44:20 +08:00
|
|
|
|
|
|
|
|
server.abort();
|
|
|
|
|
}
|
2026-04-17 00:25:28 +08:00
|
|
|
}
|
