gitdataai/libs/service/agent/model_capability.rs
ZhenYi 881fbdb6ea refactor(service): clean up agent modules, use agent crate types
- service now delegates model/provider/pricing logic to agent crate
- ChatService built at startup with EmbedService (graceful degradation)
- RoomService wired with EmbedService for Qdrant embedding
- Add error types for embedding service
2026-04-25 20:09:45 +08:00

54 lines
1.8 KiB
Rust

//! Capability management — delegates to agent crate.
use crate::AppService;
use crate::error::AppError;
use session::Session;
pub use agent::model::capability::{CreateModelCapabilityRequest, ModelCapabilityResponse, UpdateModelCapabilityRequest};
impl AppService {
pub async fn agent_model_capability_list(
&self,
model_version_id: i64,
_ctx: &Session,
) -> Result<Vec<agent::model::capability::ModelCapabilityResponse>, AppError> {
Ok(agent::model::capability::list_capabilities(&self.db, model_version_id).await?)
}
pub async fn agent_model_capability_get(
&self,
id: i64,
_ctx: &Session,
) -> Result<agent::model::capability::ModelCapabilityResponse, AppError> {
Ok(agent::model::capability::get_capability(&self.db, id).await?)
}
pub async fn agent_model_capability_create(
&self,
request: agent::model::capability::CreateModelCapabilityRequest,
ctx: &Session,
) -> Result<agent::model::capability::ModelCapabilityResponse, AppError> {
super::provider::require_system_caller(ctx)?;
Ok(agent::model::capability::create_capability(&self.db, request).await?)
}
pub async fn agent_model_capability_update(
&self,
id: i64,
request: agent::model::capability::UpdateModelCapabilityRequest,
ctx: &Session,
) -> Result<agent::model::capability::ModelCapabilityResponse, AppError> {
super::provider::require_system_caller(ctx)?;
Ok(agent::model::capability::update_capability(&self.db, id, request).await?)
}
pub async fn agent_model_capability_delete(
&self,
id: i64,
ctx: &Session,
) -> Result<(), AppError> {
super::provider::require_system_caller(ctx)?;
Ok(agent::model::capability::delete_capability(&self.db, id).await?)
}
}