Feature/db

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,
+}

src/db/models/mcp_provider.rs

@@ -0,0 +1,13 @@
+use chrono::{DateTime, Utc};
+pub struct MCPProvider {
+    pub id: i32,
+    pub name: String,
+    pub url: String,
+    pub auth_key: String,
+    pub timeout: i32,
+    pub health_api: String,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/mcp_tool.rs

@@ -0,0 +1,15 @@
+use chrono::{DateTime, Utc};
+pub struct MCPTool {
+    pub id: i32,
+    pub provider_id: i32,
+    pub name: String,
+    pub tool_code: String,
+    pub description: String,
+    pub tag: String,
+    pub parameters: String,
+    pub response_format: String,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/memory_info.rs

@@ -0,0 +1,14 @@
+use chrono::{DateTime, Utc};
+pub struct MemoryInfo {
+    pub id: i32,
+    pub memory_uuid: String,
+    pub name: String,
+    pub memory_type: i8,
+    pub is_vector: i8,
+    pub vector_id: String,
+    pub keywords: String,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/mod.rs

@@ -1 +1,16 @@
-mod config;
+mod config;
+mod ai_provider;
+mod llm_model;
+mod agent_info;
+mod prompt_info;
+mod prompt_placeholder;
+mod agent_arena;
+mod chat_session;
+mod chat_message;
+mod memory_info;
+mod mcp_provider;
+mod mcp_tool;
+mod agent_mcp_rel;
+mod project_info;
+mod system_api_log;
+mod system_config;

src/db/models/project_info.rs

@@ -0,0 +1,10 @@
+use chrono::{DateTime, Utc};
+pub struct ProjectInfo {
+    pub id: i32,
+    pub name: String,
+    pub path: String,
+    pub is_deleted: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/prompt_info.rs

@@ -0,0 +1,15 @@
+use chrono::{DateTime, Utc};
+use crate::db::r#enum::PromptRole;
+pub struct PromptInfo {
+    pub id: i32,
+    pub name: String,
+    pub content: String,
+    pub category: String,
+    pub role: PromptRole,
+    pub version: String,
+    pub version_desc: String,
+    pub tag: String,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/prompt_placeholder.rs

@@ -0,0 +1,15 @@
+use chrono::{DateTime, Utc};
+pub struct PromptPlaceholder {
+    pub id: i32,
+    pub prompt_id: i32,
+    pub key: String,
+    pub label: String,
+    pub value: String,
+    pub default: String,
+    pub input_type: String,
+    pub required: i8,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/system_api_log.rs

@@ -0,0 +1,12 @@
+use chrono::{DateTime, Utc};
+pub struct SystemAPILog {
+    pub id: i32,
+    pub rel_uuid: String,
+    pub token_cost: i32,
+    pub duration: i32,
+    pub content: String,
+    pub status: i8, // 0: success, 1: failed
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}

src/db/models/system_config.rs

@@ -0,0 +1,13 @@
+use chrono::{DateTime, Utc};
+pub struct SystemConfig {
+    pub key: i32,
+    pub value: String,
+    pub key_type: String,
+    pub description: String,
+    pub tag: String,
+    pub sort: i32,
+    pub status: i8,
+    pub created_at: DateTime<Utc>,
+    pub updated_at: DateTime<Utc>,
+    pub remark: String,
+}