WangLiZhao
1c1db7e30c
- 在 handleCallbackSuccess 函数中新增获取协议配置逻辑 - 实现历史消息获取并在 rounds 中注入历史消息 - 添加 InjectHistory 函数实现按协议顺序合并历史消息 - 在 GetPromptText 接口中集成历史消息注入测试 - 更新 ProviderProtocol 实体中的 MergeOrder 类型为 []string - 新增 Capabilities 字段支持最大 token 配置 - 修改 renderTemplate 函数接收协议对象参数 - 优化会话历史存储逻辑，提取用户消息内容进行记录 - 移除无用的注释代码 handleCallbackSuccess 处理回调成功
50 lines
1.8 KiB
Go
50 lines
1.8 KiB
Go
package entity
|
|
|
|
import "gitea.com/red-future/common/beans"
|
|
|
|
// ProviderProtocol 模型协议映射配置
|
|
type ProviderProtocol struct {
|
|
beans.SQLBaseDO `orm:",inherit"`
|
|
// 业务字段
|
|
ProviderName string `orm:"provider_name" json:"providerName"`
|
|
TargetField string `orm:"target_field" json:"targetField"`
|
|
MergeOrder []string `orm:"merge_order" json:"mergeOrder"`
|
|
RoleMapping map[string]any `orm:"role_mapping" json:"roleMapping"`
|
|
ContentMapping map[string]any `orm:"content_mapping" json:"contentMapping"`
|
|
Capabilities map[string]any `orm:"capabilities" json:"capabilities"`
|
|
RequestTemplate map[string]any `orm:"request_template" json:"requestTemplate"`
|
|
SystemPromptTemplate string `orm:"system_prompt_template" json:"systemPromptTemplate"`
|
|
Status int `orm:"status" json:"status"`
|
|
Remark string `orm:"remark" json:"remark"`
|
|
}
|
|
|
|
// providerProtocolCol 列名
|
|
type providerProtocolCol struct {
|
|
beans.SQLBaseCol
|
|
ProviderName string
|
|
TargetField string
|
|
MergeOrder string
|
|
RoleMapping string
|
|
ContentMapping string
|
|
Capabilities string
|
|
RequestTemplate string
|
|
SystemPromptTemplate string
|
|
Status string
|
|
Remark string
|
|
}
|
|
|
|
// ProviderProtocolCol 列名常量
|
|
var ProviderProtocolCol = providerProtocolCol{
|
|
SQLBaseCol: beans.DefSQLBaseCol,
|
|
ProviderName: "provider_name",
|
|
TargetField: "target_field",
|
|
MergeOrder: "merge_order",
|
|
RoleMapping: "role_mapping",
|
|
ContentMapping: "content_mapping",
|
|
Capabilities: "capabilities",
|
|
RequestTemplate: "request_template",
|
|
SystemPromptTemplate: "system_prompt_template",
|
|
Status: "status",
|
|
Remark: "remark",
|
|
}
|