t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
58 lines
2.1 KiB
Go
58 lines
2.1 KiB
Go
package repositories
|
|
|
|
import (
|
|
"time"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
var AgentToolInvocationRepository = newAgentToolInvocationRepository()
|
|
|
|
func newAgentToolInvocationRepository() *agentToolInvocationRepository {
|
|
return &agentToolInvocationRepository{}
|
|
}
|
|
|
|
type agentToolInvocationRepository struct{}
|
|
|
|
func (r *agentToolInvocationRepository) GetByIdempotencyKey(db *gorm.DB, conversationID int64, toolCode, idempotencyKey string) *models.AgentToolInvocation {
|
|
if conversationID <= 0 || toolCode == "" || idempotencyKey == "" {
|
|
return nil
|
|
}
|
|
var item models.AgentToolInvocation
|
|
if err := db.Where("conversation_id = ? AND tool_code = ? AND idempotency_key = ?", conversationID, toolCode, idempotencyKey).First(&item).Error; err != nil {
|
|
return nil
|
|
}
|
|
return &item
|
|
}
|
|
|
|
func (r *agentToolInvocationRepository) Create(db *gorm.DB, item *models.AgentToolInvocation) error {
|
|
return db.Create(item).Error
|
|
}
|
|
|
|
func (r *agentToolInvocationRepository) Updates(db *gorm.DB, id int64, values map[string]any) error {
|
|
return db.Model(&models.AgentToolInvocation{}).Where("id = ?", id).Updates(values).Error
|
|
}
|
|
|
|
func (r *agentToolInvocationRepository) TransitionStatus(db *gorm.DB, id int64, fromStatus string, values map[string]any) (bool, error) {
|
|
result := db.Model(&models.AgentToolInvocation{}).
|
|
Where("id = ? AND status = ?", id, fromStatus).
|
|
Updates(values)
|
|
return result.RowsAffected == 1, result.Error
|
|
}
|
|
|
|
func (r *agentToolInvocationRepository) TransitionLease(db *gorm.DB, id int64, leaseToken string, values map[string]any) (bool, error) {
|
|
result := db.Model(&models.AgentToolInvocation{}).
|
|
Where("id = ? AND status = ? AND result_data = ?", id, "running", leaseToken).
|
|
Updates(values)
|
|
return result.RowsAffected == 1, result.Error
|
|
}
|
|
|
|
func (r *agentToolInvocationRepository) RecoverStaleRunning(db *gorm.DB, id int64, staleBefore time.Time, values map[string]any) (bool, error) {
|
|
result := db.Model(&models.AgentToolInvocation{}).
|
|
Where("id = ? AND status = ? AND updated_at < ?", id, "running", staleBefore).
|
|
Updates(values)
|
|
return result.RowsAffected == 1, result.Error
|
|
}
|