systemprompt-api 0.2.0

HTTP API server and gateway for systemprompt.io OS
Documentation 
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

use axum::extract::State;
use axum::http::StatusCode;
use axum::response::{IntoResponse, Response};
use axum::{Extension, Json};
use systemprompt_content::{ContentError, SearchRequest, SearchService};
use systemprompt_models::RequestContext;
use systemprompt_runtime::AppContext;

pub async fn query_handler(
    Extension(_req_ctx): Extension<RequestContext>,
    State(ctx): State<AppContext>,
    Json(request): Json<SearchRequest>,
) -> Response {
    log_search_start(&request.query);

    let search_service = match SearchService::new(ctx.db_pool()) {
        Ok(service) => service,
        Err(e) => return handle_service_error(&e),
    };

    execute_search(&search_service, &request).await
}

fn log_search_start(query: &str) {
    tracing::info!(query = %query, "Searching");
}

fn handle_service_error(e: &ContentError) -> Response {
    tracing::error!(error = %e, "Failed to create search service");
    internal_error(&e.to_string())
}

async fn execute_search(service: &SearchService, request: &SearchRequest) -> Response {
    match service.search(request).await {
        Ok(response) => {
            tracing::info!(total = response.total, "Search completed");
            Json(response).into_response()
        },
        Err(e) => {
            tracing::error!(error = %e, "Search error");
            internal_error(&e.to_string())
        },
    }
}

fn internal_error(message: &str) -> Response {
    (
        StatusCode::INTERNAL_SERVER_ERROR,
        Json(serde_json::json!({"error": message})),
    )
        .into_response()
}