mlogclub
3439d20537
- Added TitleKey and DescriptionKey fields to ToolSpec for dynamic localization. - Updated tool specifications to use i18nx for titles, descriptions, and appendices. - Refactored notification messages to utilize i18nx for localization in conversation and ticket assignment events. - Improved localization in the debug dialog component for skill definitions. - Added new translation keys in English and Chinese for ticket and conversation assignment notifications. - Ensured that fallback messages are properly localized based on user locale.
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
|
|
"agent-desk/internal/ai/runtime/graphs"
|
|
"agent-desk/internal/ai/runtime/registry"
|
|
"agent-desk/internal/models"
|
|
"agent-desk/internal/pkg/i18nx"
|
|
"agent-desk/internal/pkg/toolx"
|
|
|
|
einotool "github.com/cloudwego/eino/components/tool"
|
|
"github.com/cloudwego/eino/schema"
|
|
einojsonschema "github.com/eino-contrib/jsonschema"
|
|
orderedmap "github.com/wk8/go-ordered-map/v2"
|
|
)
|
|
|
|
type PrepareTicketDraftTool struct {
|
|
conversation models.Conversation
|
|
}
|
|
|
|
func NewPrepareTicketDraftTool() *PrepareTicketDraftTool {
|
|
return &PrepareTicketDraftTool{}
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) Spec() toolx.ToolSpec {
|
|
return toolx.GraphPrepareTicketDraft
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) Name() string {
|
|
return toolx.GraphPrepareTicketDraft.Name
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) Code() string {
|
|
return toolx.GraphPrepareTicketDraft.Code
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) Enabled(ctx registry.Context) bool {
|
|
return true
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) Build(ctx registry.Context) (einotool.BaseTool, error) {
|
|
if !t.Enabled(ctx) {
|
|
return nil, nil
|
|
}
|
|
return &PrepareTicketDraftTool{conversation: ctx.Conversation}, nil
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) Info(ctx context.Context) (*schema.ToolInfo, error) {
|
|
return &schema.ToolInfo{
|
|
Name: toolx.GraphPrepareTicketDraft.Name,
|
|
Desc: i18nx.Get("tool.graph.prepareTicketDraft.info"),
|
|
ParamsOneOf: schema.NewParamsOneOfByJSONSchema(&einojsonschema.Schema{
|
|
Version: einojsonschema.Version,
|
|
Type: "object",
|
|
Properties: orderedmap.New[string, *einojsonschema.Schema](orderedmap.WithInitialData(
|
|
orderedmap.Pair[string, *einojsonschema.Schema]{
|
|
Key: "title",
|
|
Value: &einojsonschema.Schema{
|
|
Type: "string",
|
|
Description: i18nx.Get("tool.graph.prepareTicketDraft.param.title"),
|
|
},
|
|
},
|
|
orderedmap.Pair[string, *einojsonschema.Schema]{
|
|
Key: "description",
|
|
Value: &einojsonschema.Schema{
|
|
Type: "string",
|
|
Description: i18nx.Get("tool.graph.prepareTicketDraft.param.description"),
|
|
},
|
|
},
|
|
orderedmap.Pair[string, *einojsonschema.Schema]{
|
|
Key: "issue",
|
|
Value: &einojsonschema.Schema{
|
|
Type: "string",
|
|
Description: i18nx.Get("tool.graph.prepareTicketDraft.param.issue"),
|
|
},
|
|
},
|
|
orderedmap.Pair[string, *einojsonschema.Schema]{
|
|
Key: "impact",
|
|
Value: &einojsonschema.Schema{
|
|
Type: "string",
|
|
Description: i18nx.Get("tool.graph.prepareTicketDraft.param.impact"),
|
|
},
|
|
},
|
|
orderedmap.Pair[string, *einojsonschema.Schema]{
|
|
Key: "expectedOutcome",
|
|
Value: &einojsonschema.Schema{
|
|
Type: "string",
|
|
Description: i18nx.Get("tool.graph.prepareTicketDraft.param.expectedOutcome"),
|
|
},
|
|
},
|
|
orderedmap.Pair[string, *einojsonschema.Schema]{
|
|
Key: "currentAttempt",
|
|
Value: &einojsonschema.Schema{
|
|
Type: "string",
|
|
Description: i18nx.Get("tool.graph.prepareTicketDraft.param.currentAttempt"),
|
|
},
|
|
},
|
|
)),
|
|
}),
|
|
Extra: map[string]any{
|
|
"toolCode": toolx.GraphPrepareTicketDraft.Code,
|
|
"sourceType": "graph",
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func (t *PrepareTicketDraftTool) InvokableRun(ctx context.Context, argumentsInJSON string, opts ...einotool.Option) (string, error) {
|
|
return graphs.NewPrepareTicketDraftGraph(t.conversation).Run(ctx, argumentsInJSON)
|
|
}
|