feat(db): 新增数据库相关结构体和枚举类型

Author: bao-cn

src/db/enum/mod.rs

@@ -0,0 +1,6 @@
+pub enum PromptRole {
+    User,
+    Assistant,
+    System,
+    Tool,
+}

src/db/mod.rs

@@ -1,2 +1,3 @@
 mod models;
-mod store;
+mod store;
+mod r#enum;

src/db/models/agent_arena.rs

@@ -0,0 +1,13 @@
+use chrono::{DateTime, Utc};
+pub struct AgentArena {
+    pub id: i32,
+    pub chat_uuid: String,
+    pub round_uuid: String,
+    pub agent_id: i32,
+    pub vote_type: i8,
+    pub comment: String,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/agent_info.rs

@@ -0,0 +1,16 @@
+use chrono::{DateTime, Utc};
+pub struct AgentInfo {
+    pub id: i32,
+    pub llm_id: i32,
+    pub name: String,
+    pub description: String,
+    pub prompt_id: i32,
+    pub avatar: String,
+    pub params: String,
+    pub call_count: i64,
+    pub avg_score: f32,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/agent_mcp_rel.rs

@@ -0,0 +1,10 @@
+use chrono::{DateTime, Utc};
+pub struct AgentMCPRel {
+    pub id: i32,
+    pub agent_id: i32,
+    pub mcp_tool_id: i32,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/ai_provider.rs

@@ -0,0 +1,13 @@
+use chrono::{DateTime, Utc};
+pub struct AiProvider {
+    pub id: i32,
+    pub name: String,
+    pub avatar: String,
+    pub api: String,
+    pub key: String,
+    pub timeout: i32,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/chat_message.rs

@@ -0,0 +1,17 @@
+use chrono::{DateTime, Utc};
+use crate::db::r#enum::PromptRole;
+pub struct ChatMessage {
+    pub id: i32,
+    pub chat_uuid: String,
+    pub round_uuid: String,
+    pub role: PromptRole,
+    pub content: String,
+    pub input_tokens: i32,
+    pub output_tokens: i32,
+    pub duration: i32,
+    pub model_config: String,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/chat_session.rs

@@ -0,0 +1,11 @@
+use chrono::{DateTime, Utc};
+pub struct ChatSession {
+    pub id: i32,
+    pub chat_uuid: String,
+    pub title: String,
+    pub llm_id: i32,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/config.rs

@@ -1,8 +1,12 @@
+use chrono::{DateTime, Utc};
 pub struct Config {
     pub key: String,
     pub value: String,
     pub description: String,
     pub type_: String,
     pub tag: String,
     pub sort: i32,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
 }

src/db/models/llm_model.rs

@@ -0,0 +1,17 @@
+use chrono::{DateTime, Utc};
+pub struct LLMModel {
+    pub id: i32,
+    pub provider_id: i32,
+    pub name: String,
+    pub avatar: String,
+    pub model_code: String,
+    pub max_tokens: i32,
+    pub max_response_tokens: i32,
+    pub temperature: f32,
+    pub weight: i32,
+    pub call_count: i64,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}