kevin/ops2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
kevin
2026-06-10 19:50:14 +08:00
parent dc39d28894
commit 2182d0312b
6 changed files with 122 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/my_work/agents/function_tool.go
+9 -1
View File
@@ -15,7 +15,11 @@ type FunctionToolSchema struct {
}
type FunctionToolRuntime struct {
UserID uint
UserID uint
UserName string
UserEmail string
UserType string
UserInfo *CurrentUserInfo
}
func FunctionToolSchemas(configs []ToolConfig) []FunctionToolSchema {
@@ -29,6 +33,8 @@ func FunctionToolSchemas(configs []ToolConfig) []FunctionToolSchema {
tools = append(tools, timeFunctionToolSchema())
case "ops_ai_assistant_schedule_query", "ops_ai_assistant":
tools = append(tools, opsAIAssistantScheduleQuerySchema())
case "ops_ai_assistant_current_user":
tools = append(tools, opsAIAssistantCurrentUserSchema())
}
}
return tools
@@ -56,6 +62,8 @@ func ExecuteFunctionTool(ctx context.Context, runtime FunctionToolRuntime, name
return json.Marshal(result)
case "ops_ai_assistant_schedule_query", "ops_ai_assistant":
return executeOpsAIAssistantScheduleQuery(ctx, runtime, rawArgs)
case "ops_ai_assistant_current_user":
return executeOpsAIAssistantCurrentUser(ctx, runtime, rawArgs)
default:
return nil, fmt.Errorf("unknown tool: %s", name)
}
backend/my_work/agents/ops_ai_assistant.go
+84 -2
View File
@@ -8,12 +8,29 @@ import (
)
type ScheduleQueryArgs struct {
Action string `json:"action,omitempty"`
StartDate string `json:"start_date"`
EndDate string `json:"end_date"`
CalendarID uint `json:"calendar_id,omitempty"`
Limit int `json:"limit,omitempty"`
}
type CurrentUserArgs struct {
Action string `json:"action,omitempty"`
}
type CurrentUserInfo struct {
ID uint `json:"id,omitempty"`
UserID uint `json:"user_id,omitempty"`
FirstName string `json:"first_name,omitempty"`
Username string `json:"username,omitempty"`
Birthdate string `json:"birthdate,omitempty"`
Gender string `json:"gender,omitempty"`
AvatarPath string `json:"avatar_path,omitempty"`
Region string `json:"region,omitempty"`
Language string `json:"language,omitempty"`
}
type ScheduleCalendar struct {
ID uint `json:"id"`
Name string `json:"name"`
@@ -29,6 +46,7 @@ type ScheduleEvent struct {
ScheduleType string `json:"schedule_type"`
IsPublic bool `json:"is_public"`
Remark string `json:"remark,omitempty"`
AccessNote string `json:"access_note,omitempty"`
CanEdit bool `json:"canEdit"`
}
@@ -53,26 +71,90 @@ func RegisterScheduleProvider(provider ScheduleProvider) {
func opsAIAssistantScheduleQuerySchema() FunctionToolSchema {
return FunctionToolSchema{
Name: "ops_ai_assistant_schedule_query",
Description: "按明确日期范围查询当前用户可见的 OPS 日历/日程。相对日期需先调用 time 获取 start_date/end_date。",
Description: "只读工具：按明确日期范围查询当前用户可见的 OPS 日历/日程。禁止新增、修改、删除数据，禁止批量添加、批量导入或执行任何写操作。相对日期需先调用 time 获取 start_date/end_date。",
Parameters: map[string]interface{}{
"type": "object",
"properties": map[string]interface{}{
"action": map[string]interface{}{
"type": "string",
"enum": []string{"query"},
"description": "固定为 query。本工具只允许查询，禁止 create/update/delete/batch_add/import 等写操作。",
},
"start_date": map[string]interface{}{"type": "string", "description": "开始日期，格式 YYYY-MM-DD。"},
"end_date": map[string]interface{}{"type": "string", "description": "结束日期，格式 YYYY-MM-DD。"},
"calendar_id": map[string]interface{}{"type": "integer", "description": "可选日历 ID；不传则查询全部可见日程。"},
"limit": map[string]interface{}{"type": "integer", "description": "可选返回上限，默认 100，最大 200。"},
},
"required": []string{"start_date", "end_date"},
"required": []string{"action", "start_date", "end_date"},
},
// ScheduleQueryResult is the return type packed as JSON
}
}
func opsAIAssistantCurrentUserSchema() FunctionToolSchema {
return FunctionToolSchema{
Name: "ops_ai_assistant_current_user",
Description: "只读工具：当用户询问“我是谁”“当前登录用户是谁”“我的用户信息”等当前身份问题时调用。先判断是否已登录；已登录则返回当前登录用户信息，未登录则返回 unknown 并提示需要登录才能获取信息。",
Parameters: map[string]interface{}{
"type": "object",
"properties": map[string]interface{}{
"action": map[string]interface{}{
"type": "string",
"enum": []string{"get"},
"description": "固定为 get。本工具只读取当前登录用户信息。",
},
},
"required": []string{"action"},
},
}
}
func executeOpsAIAssistantCurrentUser(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) {
var args CurrentUserArgs
if len(rawArgs) > 0 {
if err := json.Unmarshal(rawArgs, &args); err != nil {
return nil, err
}
}
if args.Action != "" && args.Action != "get" {
return json.Marshal(map[string]interface{}{
"ok": false,
"error": "ops_ai_assistant_current_user 是只读工具，仅允许 get 读取当前用户信息",
})
}
if runtime.UserID <= 0 {
return json.Marshal(map[string]interface{}{
"ok": true,
"loggedIn": false,
"unknown": true,
"message": "不知道你是谁，需要登录才能获取当前用户信息。",
})
}
return json.Marshal(map[string]interface{}{
"ok": true,
"loggedIn": true,
"user": map[string]interface{}{
"id": runtime.UserID,
"name": runtime.UserName,
"email": runtime.UserEmail,
"type": runtime.UserType,
"info": runtime.UserInfo,
},
})
}
func executeOpsAIAssistantScheduleQuery(ctx context.Context, runtime FunctionToolRuntime, rawArgs []byte) ([]byte, error) {
var args ScheduleQueryArgs
if err := json.Unmarshal(rawArgs, &args); err != nil {
return nil, err
}
if args.Action != "" && args.Action != "query" {
return json.Marshal(map[string]interface{}{
"ok": false,
"error": "ops_ai_assistant 是只读工具，仅允许 query 查询操作，禁止新增、修改、删除、批量添加或导入数据",
})
}
startDate, err := time.Parse("2006-01-02", args.StartDate)
if err != nil {
return nil, fmt.Errorf("invalid start_date: %w", err)