cf-mini-chat 0.1.25

Mini-chat module: multi-tenant AI chat
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use std::sync::Arc;

use crate::domain::models::{ChatPatch, NewChat};
use axum::Extension;
use axum::extract::Path;
use modkit::api::odata::OData;
use modkit::api::prelude::*;
use modkit_security::SecurityContext;
use uuid::Uuid;

use crate::api::rest::dto::{ChatDetailDto, CreateChatReq, UpdateChatReq};
use crate::module::AppServices;

/// POST /mini-chat/v1/chats
#[tracing::instrument(skip(svc, ctx, uri, req_body))]
pub(crate) async fn create_chat(
    uri: axum::http::Uri,
    Extension(ctx): Extension<SecurityContext>,
    Extension(svc): Extension<Arc<AppServices>>,
    Json(req_body): Json<CreateChatReq>,
) -> ApiResult<impl IntoResponse> {
    let new = NewChat {
        model: req_body.model,
        title: req_body.title,
        is_temporary: false,
    };

    let detail = svc.chats.create_chat(&ctx, new).await?;
    let id_str = detail.id.to_string();
    Ok(created_json(ChatDetailDto::from(detail), &uri, &id_str).into_response())
}

/// GET /mini-chat/v1/chats
#[tracing::instrument(skip(svc, ctx, query))]
pub(crate) async fn list_chats(
    Extension(ctx): Extension<SecurityContext>,
    Extension(svc): Extension<Arc<AppServices>>,
    OData(query): OData,
) -> ApiResult<JsonPage<ChatDetailDto>> {
    let page = svc.chats.list_chats(&ctx, &query).await?;
    let page = page.map_items(ChatDetailDto::from);
    Ok(Json(page))
}

/// GET /mini-chat/v1/chats/{id}
#[tracing::instrument(skip(svc, ctx), fields(chat_id = %id))]
pub(crate) async fn get_chat(
    Extension(ctx): Extension<SecurityContext>,
    Extension(svc): Extension<Arc<AppServices>>,
    Path(id): Path<Uuid>,
) -> ApiResult<JsonBody<ChatDetailDto>> {
    let detail = svc.chats.get_chat(&ctx, id).await?;
    Ok(Json(ChatDetailDto::from(detail)))
}

/// PATCH /mini-chat/v1/chats/{id}
#[tracing::instrument(skip(svc, ctx, req_body), fields(chat_id = %id))]
pub(crate) async fn update_chat(
    Extension(ctx): Extension<SecurityContext>,
    Extension(svc): Extension<Arc<AppServices>>,
    Path(id): Path<Uuid>,
    Json(req_body): Json<UpdateChatReq>,
) -> ApiResult<JsonBody<ChatDetailDto>> {
    let patch = ChatPatch {
        title: Some(Some(req_body.title)),
    };
    let detail = svc.chats.update_chat(&ctx, id, patch).await?;
    Ok(Json(ChatDetailDto::from(detail)))
}

/// DELETE /mini-chat/v1/chats/{id}
#[tracing::instrument(skip(svc, ctx), fields(chat_id = %id))]
pub(crate) async fn delete_chat(
    Extension(ctx): Extension<SecurityContext>,
    Extension(svc): Extension<Arc<AppServices>>,
    Path(id): Path<Uuid>,
) -> ApiResult<impl IntoResponse> {
    svc.chats.delete_chat(&ctx, id).await?;
    Ok(no_content().into_response())
}