Dave
26 lines
650 B
Rust
26 lines
650 B
Rust
use crate::http::context::{AppContext, OpenApiResult, bad_request};
|
|
use crate::llm::chat;
|
|
use poem_openapi::{OpenApi, Tags, payload::Json};
|
|
use std::sync::Arc;
|
|
|
|
#[derive(Tags)]
|
|
enum ChatTags {
|
|
Chat,
|
|
}
|
|
|
|
pub struct ChatApi {
|
|
pub ctx: Arc<AppContext>,
|
|
}
|
|
|
|
#[OpenApi(tag = "ChatTags::Chat")]
|
|
impl ChatApi {
|
|
/// Cancel the currently running chat stream, if any.
|
|
///
|
|
/// Returns `true` once the cancellation signal is issued.
|
|
#[oai(path = "/chat/cancel", method = "post")]
|
|
async fn cancel_chat(&self) -> OpenApiResult<Json<bool>> {
|
|
chat::cancel_chat(&self.ctx.state).map_err(bad_request)?;
|
|
Ok(Json(true))
|
|
}
|
|
}
|