Skip to content

Commit d9ff5da

Browse files
committed
feat(db): 完善字段可选类型
1 parent e2fbe70 commit d9ff5da

16 files changed

Lines changed: 58 additions & 56 deletions

src/db/models/agent_arena.rs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,11 @@ pub struct AgentArena {
44
pub chat_uuid: String,
55
pub round_uuid: String,
66
pub agent_id: i32,
7+
pub vote_num: i8,
78
pub vote_type: i8,
8-
pub comment: String,
9+
pub comment: Option<String>,
910
pub status: i8,
1011
pub created_at: DateTime<Utc>,
1112
pub updated_at: DateTime<Utc>,
12-
pub remark: String,
13+
pub remark: Option<String>,
1314
}

src/db/models/agent_info.rs

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,15 @@ pub struct AgentInfo {
33
pub id: i32,
44
pub llm_id: i32,
55
pub name: String,
6-
pub description: String,
6+
pub description: Option<String>,
77
pub prompt_id: i32,
88
pub avatar: String,
9-
pub params: String,
10-
pub call_count: i64,
11-
pub avg_score: f32,
12-
pub status: i8,
9+
pub call_count: i64,// default 0
10+
pub avg_score: f32,// default 0.0
11+
pub temperature: f32, // default 0.7
12+
pub top_p: f32, // default 0.95
13+
pub status: i8,// default 1
1314
pub created_at: DateTime<Utc>,
1415
pub updated_at: DateTime<Utc>,
15-
pub remark: String,
16+
pub remark: Option<String>,
1617
}

src/db/models/agent_mcp_rel.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@ pub struct AgentMCPRel {
33
pub id: i32,
44
pub agent_id: i32,
55
pub mcp_tool_id: i32,
6-
pub status: i8,
6+
pub status: i8,// default 1
77
pub created_at: DateTime<Utc>,
88
pub updated_at: DateTime<Utc>,
9-
pub remark: String,
9+
pub remark: Option<String>,
1010
}

src/db/models/ai_provider.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@ pub struct AiProvider {
44
pub name: String,
55
pub avatar: String,
66
pub api: String,
7-
pub key: String,
8-
pub timeout: i32,
9-
pub status: i8,
7+
pub key: Option<String>,
8+
pub timeout: i32,//ms default 5000
9+
pub status: i8,// default 1
1010
pub created_at: DateTime<Utc>,
1111
pub updated_at: DateTime<Utc>,
12-
pub remark: String,
12+
pub remark: Option<String>,
1313
}

src/db/models/chat_message.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@ pub struct ChatMessage {
66
pub round_uuid: String,
77
pub role: PromptRole,
88
pub content: String,
9-
pub input_tokens: i32,
10-
pub output_tokens: i32,
11-
pub duration: i32,
9+
pub input_tokens: i32, // default 0
10+
pub output_tokens: i32,// default 0
11+
pub duration: i32,// default 0
1212
pub model_config: String,
13-
pub status: i8,
13+
pub status: i8,// default 1
1414
pub created_at: DateTime<Utc>,
1515
pub updated_at: DateTime<Utc>,
16-
pub remark: String,
16+
pub remark: Option<String>,
1717
}

src/db/models/chat_session.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@ use chrono::{DateTime, Utc};
22
pub struct ChatSession {
33
pub id: i32,
44
pub chat_uuid: String,
5-
pub title: String,
5+
pub title: Option<String>,
66
pub llm_id: i32,
7-
pub status: i8,
7+
pub status: i8,// default 1
88
pub created_at: DateTime<Utc>,
99
pub updated_at: DateTime<Utc>,
10-
pub remark: String,
10+
pub remark: Option<String>,
1111
}

src/db/models/config.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@ pub struct Config {
33
pub key: String,
44
pub value: String,
55
pub description: String,
6-
pub type_: String,
6+
pub key_type: String,
77
pub tag: String,
8-
pub sort: i32,
8+
pub sort: i32,// default 0
99
pub created_at: DateTime<Utc>,
1010
pub updated_at: DateTime<Utc>,
11-
pub remark: String,
11+
pub remark: Option<String>,
1212
}

src/db/models/llm_model.rs

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,16 +2,15 @@ use chrono::{DateTime, Utc};
22
pub struct LLMModel {
33
pub id: i32,
44
pub provider_id: i32,
5-
pub name: String,
5+
pub name: Option<String>,
66
pub avatar: String,
77
pub model_code: String,
8-
pub max_tokens: i32,
9-
pub max_response_tokens: i32,
10-
pub temperature: f32,
11-
pub weight: i32,
12-
pub call_count: i64,
13-
pub status: i8,
8+
pub max_tokens: i32, // default 0
9+
pub max_response_tokens: i32, // default 0
10+
pub weight: i32, // default 0
11+
pub call_count: i64, // default 0
12+
pub status: i8,// default 1
1413
pub created_at: DateTime<Utc>,
1514
pub updated_at: DateTime<Utc>,
16-
pub remark: String,
15+
pub remark: Option<String>,
1716
}

src/db/models/mcp_provider.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,11 @@ pub struct MCPProvider {
33
pub id: i32,
44
pub name: String,
55
pub url: String,
6-
pub auth_key: String,
7-
pub timeout: i32,
8-
pub health_api: String,
9-
pub status: i8,
6+
pub auth_key: Option<String>,
7+
pub timeout: i32,// ms default 5000
8+
pub health_api: Option<String>,
9+
pub status: i8,// default 1
1010
pub created_at: DateTime<Utc>,
1111
pub updated_at: DateTime<Utc>,
12-
pub remark: String,
12+
pub remark: Option<String>,
1313
}

src/db/models/mcp_tool.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ pub struct MCPTool {
88
pub tag: String,
99
pub parameters: String,
1010
pub response_format: String,
11-
pub status: i8,
11+
pub status: i8,// default 1
1212
pub created_at: DateTime<Utc>,
1313
pub updated_at: DateTime<Utc>,
14-
pub remark: String,
14+
pub remark: Option<String>,
1515
}

0 commit comments

Comments
 (0)