-
Notifications
You must be signed in to change notification settings - Fork 18
/
chat.go
56 lines (54 loc) · 5.36 KB
/
chat.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package entity
import "github.com/iimeta/fastapi/internal/model/common"
type Chat struct {
Id string `bson:"_id,omitempty"` // ID
TraceId string `bson:"trace_id,omitempty"` // 日志ID
UserId int `bson:"user_id,omitempty"` // 用户ID
AppId int `bson:"app_id,omitempty"` // 应用ID
Corp string `bson:"corp,omitempty"` // 公司
ModelId string `bson:"model_id,omitempty"` // 模型ID
Name string `bson:"name,omitempty"` // 模型名称
Model string `bson:"model,omitempty"` // 模型
Type int `bson:"type,omitempty"` // 模型类型[1:文生文, 2:文生图, 3:图生文, 4:图生图, 5:文生语音, 6:语音生文, 100:多模态]
Key string `bson:"key,omitempty"` // 密钥
IsEnablePresetConfig bool `bson:"is_enable_preset_config,omitempty"` // 是否启用预设配置
PresetConfig common.PresetConfig `bson:"preset_config,omitempty"` // 预设配置
IsEnableModelAgent bool `bson:"is_enable_model_agent,omitempty"` // 是否启用模型代理
ModelAgentId string `bson:"model_agent_id,omitempty"` // 模型代理ID
ModelAgent *ModelAgent `bson:"model_agent,omitempty"` // 模型代理信息
IsEnableForward bool `bson:"is_enable_forward,omitempty"` // 是否启用模型转发
ForwardConfig *common.ForwardConfig `bson:"forward_config,omitempty"` // 模型转发配置
IsSmartMatch bool `bson:"is_smart_match,omitempty"` // 是否智能匹配
IsEnableFallback bool `bson:"is_enable_fallback,omitempty"` // 是否启用后备模型
FallbackConfig *common.FallbackConfig `bson:"fallback_config,omitempty"` // 后备模型配置
RealModelId string `bson:"real_model_id,omitempty"` // 真实模型ID
RealModelName string `bson:"real_model_name,omitempty"` // 真实模型名称
RealModel string `bson:"real_model,omitempty"` // 真实模型
Stream bool `bson:"stream,omitempty"` // 流式
Messages []common.Message `bson:"messages,omitempty"` // 完整提示(提问)
Prompt string `bson:"prompt,omitempty"` // 提示(提问)
Completion string `bson:"completion,omitempty"` // 补全(回答)
TextQuota common.TextQuota `bson:"text_quota,omitempty"` // 文本额度
ImageQuotas []common.ImageQuota `bson:"image_quotas,omitempty"` // 图像额度
MultimodalQuota common.MultimodalQuota `bson:"multimodal_quota,omitempty"` // 多模态额度
PromptTokens int `bson:"prompt_tokens,omitempty"` // 提示令牌数(提问令牌数)
CompletionTokens int `bson:"completion_tokens,omitempty"` // 补全令牌数(回答令牌数)
TotalTokens int `bson:"total_tokens,omitempty"` // 总令牌数
ConnTime int64 `bson:"conn_time,omitempty"` // 连接时间
Duration int64 `bson:"duration,omitempty"` // 持续时间
TotalTime int64 `bson:"total_time,omitempty"` // 总时间
InternalTime int64 `bson:"internal_time,omitempty"` // 内耗时间
ReqTime int64 `bson:"req_time,omitempty"` // 请求时间
ReqDate string `bson:"req_date,omitempty"` // 请求日期
ClientIp string `bson:"client_ip,omitempty"` // 客户端IP
RemoteIp string `bson:"remote_ip,omitempty"` // 远程IP
LocalIp string `bson:"local_ip,omitempty"` // 本地IP
ErrMsg string `bson:"err_msg,omitempty"` // 错误信息
IsRetry bool `bson:"is_retry,omitempty"` // 是否重试
Retry *common.Retry `bson:"retry,omitempty"` // 重试
Status int `bson:"status,omitempty"` // 状态[1:成功, -1:失败, 2:中止, 3:重试]
Creator string `bson:"creator,omitempty"` // 创建人
Updater string `bson:"updater,omitempty"` // 更新人
CreatedAt int64 `bson:"created_at,omitempty"` // 创建时间
UpdatedAt int64 `bson:"updated_at,omitempty"` // 更新时间
}