velaclaw 0.3.0

Protocol-driven autonomous AI agent runtime with intelligent model selection and multi-model negotiation.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//! POST `/api/chat` — non-streaming agent-loop chat (VL-UI-002).
//! POST `/api/chat` — 非流式 agent 循环对话（VL-UI-002）。

use super::local_control::auth::check_pairing_auth;
use super::local_control::runner::{persist_chat_turn, run_agent_chat};
use super::local_control::types::ChatApiRequest;
use super::{client_key_from_request, AppState, RATE_LIMIT_WINDOW_SECS};
use axum::extract::{ConnectInfo, State};
use axum::http::{HeaderMap, StatusCode};
use axum::response::{IntoResponse, Json};
use std::net::SocketAddr;

/// POST /api/chat — run one agent turn and return the full assistant reply.
pub async fn handle_post_chat(
    State(state): State<AppState>,
    ConnectInfo(peer_addr): ConnectInfo<SocketAddr>,
    headers: HeaderMap,
    body: Result<Json<ChatApiRequest>, axum::extract::rejection::JsonRejection>,
) -> impl IntoResponse {
    let rate_key =
        client_key_from_request(Some(peer_addr), &headers, state.trust_forwarded_headers);
    if !state.rate_limiter.allow_webhook(&rate_key) {
        let err = serde_json::json!({
            "error": "Too many chat requests. Please retry later.",
            "retry_after": RATE_LIMIT_WINDOW_SECS,
        });
        return (StatusCode::TOO_MANY_REQUESTS, Json(err)).into_response();
    }

    if let Err(response) = check_pairing_auth(&state.pairing, &headers, None) {
        return response.into_response();
    }

    let Json(req) = match body {
        Ok(b) => b,
        Err(e) => {
            let err = serde_json::json!({
                "error": format!("Invalid JSON body: {e}"),
            });
            return (StatusCode::BAD_REQUEST, Json(err)).into_response();
        }
    };

    if req.messages.is_empty() {
        let err = serde_json::json!({
            "error": "messages must not be empty",
        });
        return (StatusCode::BAD_REQUEST, Json(err)).into_response();
    }

    let config = state.config.lock().clone();
    match run_agent_chat(&config, &req, Some(&state.approval_hub)).await {
        Ok(resp) => {
            if let Err(e) = persist_chat_turn(
                &config.workspace_dir,
                req.session_id.as_deref(),
                &req,
                &resp.content,
            )
            .await
            {
                tracing::warn!("session persist failed: {e:#}");
            }
            (StatusCode::OK, Json(resp)).into_response()
        }
        Err(e) => {
            tracing::warn!("POST /api/chat failed: {e:#}");
            let err = serde_json::json!({ "error": e.to_string() });
            (StatusCode::INTERNAL_SERVER_ERROR, Json(err)).into_response()
        }
    }
}