962bf0312d
OTLP tracing: - libs/observability/otlp.rs: SdkTracerProvider via HTTP/proto OTLP exporter - libs/observability/tracing_middleware.rs: Actix-web span with trace_id propagation - libs/observability/tracing_fmt.rs: JSON fmt + registry.try_init for layered init - libs/rpc: gRPC method spans via info_span - libs/agent, libs/room, libs/service, libs/api: structured tracing throughout Prometheus metrics: - libs/observability/prometheus_exporter.rs: /metrics HTTP handler + metrics crate - libs/observability/metrics_middleware.rs: HttpMetrics middleware + AtomicU64 - libs/observability/redis_metrics.rs: Redis counter poller via RedisMetrics - libs/room/metrics.rs: RoomMetrics (connections, messages, presence counters) Config env vars: APP_OTEL_ENABLED, APP_OTEL_ENDPOINT, APP_OTEL_SERVICE_NAME
58 lines
1.6 KiB
Rust
58 lines
1.6 KiB
Rust
use crate::{DateTimeUtc, ModelId, ModelVersionId, RoomId};
|
|
use sea_orm::entity::prelude::*;
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Serialize, Deserialize)]
|
|
#[sea_orm(table_name = "room_ai")]
|
|
pub struct Model {
|
|
#[sea_orm(primary_key)]
|
|
pub room: RoomId,
|
|
#[sea_orm(primary_key)]
|
|
pub model: ModelId,
|
|
pub version: Option<ModelVersionId>,
|
|
pub call_count: i64,
|
|
pub last_call_at: Option<DateTimeUtc>,
|
|
pub history_limit: Option<i64>,
|
|
pub system_prompt: Option<String>,
|
|
pub temperature: Option<f64>,
|
|
pub max_tokens: Option<i64>,
|
|
pub use_exact: bool,
|
|
pub think: bool,
|
|
pub stream: bool,
|
|
pub min_score: Option<f32>,
|
|
/// Agent type: "chat" (default) or "react" for ReAct reasoning agent.
|
|
pub agent_type: Option<String>,
|
|
pub created_at: DateTimeUtc,
|
|
pub updated_at: DateTimeUtc,
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
pub enum Relation {
|
|
#[sea_orm(
|
|
belongs_to = "super::room::Entity",
|
|
from = "Column::Room",
|
|
to = "super::room::Column::Id"
|
|
)]
|
|
Room,
|
|
#[sea_orm(
|
|
belongs_to = "super::super::agents::model::Entity",
|
|
from = "Column::Model",
|
|
to = "super::super::agents::model::Column::Id"
|
|
)]
|
|
AiModel,
|
|
}
|
|
|
|
impl Related<super::room::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::Room.def()
|
|
}
|
|
}
|
|
|
|
impl Related<super::super::agents::model::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::AiModel.def()
|
|
}
|
|
}
|
|
|
|
impl ActiveModelBehavior for ActiveModel {}
|