40 lines
1.3 KiB
Go
40 lines
1.3 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/cloudwego/eino/components/tool"
|
|
"github.com/cloudwego/eino/components/tool/utils"
|
|
)
|
|
|
|
type OrderLog struct {
|
|
OrderID string `json:"order_id"`
|
|
Timestamp time.Time `json:"timestamp"`
|
|
Level string `json:"level"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type OrderLogQueryInput struct {
|
|
OrderID string `json:"order_id" jsonschema:"description=订单ID"`
|
|
}
|
|
|
|
func NewOrderLogQueryTool() tool.InvokableTool {
|
|
t, err := utils.InferTool("query_order_logs", "查询订单处理日志，返回日志数组。", orderLogQuery)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func orderLogQuery(ctx context.Context, in *OrderLogQueryInput) ([]*OrderLog, error) {
|
|
logs := []*OrderLog{
|
|
{OrderID: in.OrderID, Timestamp: time.Now().Add(-5 * time.Minute), Level: "INFO", Message: "订单已创建"},
|
|
{OrderID: in.OrderID, Timestamp: time.Now().Add(-4 * time.Minute), Level: "INFO", Message: "支付完成"},
|
|
{OrderID: in.OrderID, Timestamp: time.Now().Add(-3 * time.Minute), Level: "INFO", Message: "仓库打包"},
|
|
{OrderID: in.OrderID, Timestamp: time.Now().Add(-2 * time.Minute), Level: "WARN", Message: "快递延迟，重新调度"},
|
|
{OrderID: in.OrderID, Timestamp: time.Now().Add(-1 * time.Minute), Level: "INFO", Message: "发货完成"},
|
|
}
|
|
return logs, nil
|
|
}
|