133 lines
3.9 KiB
Rust
133 lines
3.9 KiB
Rust
use crate::{ApiResponse, error::ApiError};
|
|
use actix_web::{HttpResponse, Result, web};
|
|
use service::AppService;
|
|
use service::agent::model_version::{CreateModelVersionRequest, UpdateModelVersionRequest};
|
|
use session::Session;
|
|
use uuid::Uuid;
|
|
|
|
#[derive(serde::Deserialize, utoipa::IntoParams)]
|
|
pub struct ListQuery {
|
|
pub model_id: Option<String>,
|
|
}
|
|
|
|
#[utoipa::path(
|
|
get,
|
|
path = "/api/agents/versions",
|
|
params(ListQuery),
|
|
responses(
|
|
(status = 200, body = Vec<service::agent::model_version::ModelVersionResponse>),
|
|
(status = 401, description = "Unauthorized"),
|
|
),
|
|
tag = "Agent"
|
|
)]
|
|
pub async fn model_version_list(
|
|
service: web::Data<AppService>,
|
|
session: Session,
|
|
query: web::Query<ListQuery>,
|
|
) -> Result<HttpResponse, ApiError> {
|
|
let model_id =
|
|
if let Some(ref s) = query.model_id {
|
|
Some(Uuid::parse_str(s).map_err(|_| {
|
|
service::error::AppError::BadRequest("Invalid model UUID".to_string())
|
|
})?)
|
|
} else {
|
|
None
|
|
};
|
|
let resp = service.agent_model_version_list(model_id, &session).await?;
|
|
Ok(ApiResponse::ok(resp).to_response())
|
|
}
|
|
|
|
#[utoipa::path(
|
|
get,
|
|
path = "/api/agents/versions/{id}",
|
|
params(("id" = String, Path)),
|
|
responses(
|
|
(status = 200, body = service::agent::model_version::ModelVersionResponse),
|
|
(status = 401, description = "Unauthorized"),
|
|
(status = 404),
|
|
),
|
|
tag = "Agent"
|
|
)]
|
|
pub async fn model_version_get(
|
|
service: web::Data<AppService>,
|
|
session: Session,
|
|
path: web::Path<String>,
|
|
) -> Result<HttpResponse, ApiError> {
|
|
let id = Uuid::parse_str(&path.into_inner())
|
|
.map_err(|_| service::error::AppError::BadRequest("Invalid UUID".to_string()))?;
|
|
let resp = service.agent_model_version_get(id, &session).await?;
|
|
Ok(ApiResponse::ok(resp).to_response())
|
|
}
|
|
|
|
#[utoipa::path(
|
|
post,
|
|
path = "/api/agents/versions",
|
|
request_body = CreateModelVersionRequest,
|
|
responses(
|
|
(status = 200, body = service::agent::model_version::ModelVersionResponse),
|
|
(status = 401),
|
|
(status = 403),
|
|
),
|
|
tag = "Agent"
|
|
)]
|
|
pub async fn model_version_create(
|
|
service: web::Data<AppService>,
|
|
session: Session,
|
|
body: web::Json<CreateModelVersionRequest>,
|
|
) -> Result<HttpResponse, ApiError> {
|
|
let resp = service
|
|
.agent_model_version_create(body.into_inner(), &session)
|
|
.await?;
|
|
Ok(ApiResponse::ok(resp).to_response())
|
|
}
|
|
|
|
#[utoipa::path(
|
|
patch,
|
|
path = "/api/agents/versions/{id}",
|
|
params(("id" = String, Path)),
|
|
request_body = UpdateModelVersionRequest,
|
|
responses(
|
|
(status = 200, body = service::agent::model_version::ModelVersionResponse),
|
|
(status = 401),
|
|
(status = 403),
|
|
(status = 404),
|
|
),
|
|
tag = "Agent"
|
|
)]
|
|
pub async fn model_version_update(
|
|
service: web::Data<AppService>,
|
|
session: Session,
|
|
path: web::Path<String>,
|
|
body: web::Json<UpdateModelVersionRequest>,
|
|
) -> Result<HttpResponse, ApiError> {
|
|
let id = Uuid::parse_str(&path.into_inner())
|
|
.map_err(|_| service::error::AppError::BadRequest("Invalid UUID".to_string()))?;
|
|
let resp = service
|
|
.agent_model_version_update(id, body.into_inner(), &session)
|
|
.await?;
|
|
Ok(ApiResponse::ok(resp).to_response())
|
|
}
|
|
|
|
#[utoipa::path(
|
|
delete,
|
|
path = "/api/agents/versions/{id}",
|
|
params(("id" = String, Path)),
|
|
responses(
|
|
(status = 200),
|
|
(status = 401),
|
|
(status = 403),
|
|
(status = 404),
|
|
),
|
|
tag = "Agent"
|
|
)]
|
|
pub async fn model_version_delete(
|
|
service: web::Data<AppService>,
|
|
session: Session,
|
|
path: web::Path<String>,
|
|
) -> Result<HttpResponse, ApiError> {
|
|
let id = Uuid::parse_str(&path.into_inner())
|
|
.map_err(|_| service::error::AppError::BadRequest("Invalid UUID".to_string()))?;
|
|
service.agent_model_version_delete(id, &session).await?;
|
|
Ok(HttpResponse::Ok().json(serde_json::json!({ "success": true })))
|
|
}
|