Merge pull request baidubce#904 from zhangairku/go-support-CustomMetadata

Go support custom metadata

Author: baowei-fw

go/appbuilder/app_builder_client_data.go

@@ -69,6 +69,8 @@ type AppBuilderClientRunRequest struct {
 	ToolChoice       *ToolChoice               `json:"tool_choice"`
 	Action           *Action                   `json:"action"`
 	McpAuthorization *[]map[string]interface{} `json:"mcp_authorization,omitempty"`
+	Parameters       map[string]any            `json:"parameters,omitempty"`
+	CustomMetadata   *CustomMetadata           `json:"custom_metadata,omitempty"`
 }
 
 type AppBuilderClientUploadFileRequest struct {
@@ -127,6 +129,10 @@ type ActionInterruptEvent struct {
 	Type string `json:"type"`
 }
 
+type CustomMetadata struct {
+	OverrideRoleInstruction string `json:"override_role_instruction"`
+}
+
 func NewResumeAction(eventId string) *Action {
 	return NewAction("resume", eventId, "chat")
 }

go/appbuilder/app_builder_client_test.go

@@ -923,3 +923,94 @@ func TestAppBuilderClientUploadFile(t *testing.T) {
 		t.Logf("%s========== OK:  %s ==========%s", "\033[32m", t.Name(), "\033[0m")
 	}
 }
+
+func TestAppBuilderClientRunParameters(t *testing.T) {
+	t.Parallel() // 并发运行
+	// 测试逻辑
+	config, err := NewSDKConfig("", "")
+	if err != nil {
+		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+		t.Fatalf("new http client config failed: %v", err)
+	}
+
+	appID := "2313e282-baa6-4db6-92dd-a21e99cfd59e"
+	client, err := NewAppBuilderClient(appID, config)
+	if err != nil {
+		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+		t.Fatalf("new AppBuilderClient instance failed")
+	}
+
+	conversationID, err := client.CreateConversation()
+	if err != nil {
+		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+		t.Fatalf("create conversation failed: %v", err)
+	}
+
+	res, err := client.Run(AppBuilderClientRunRequest{
+		AppID:          appID,
+		Query:          "国庆长假",
+		Stream:         false,
+		ConversationID: conversationID,
+		Parameters: map[string]any{
+			"city": "上海",
+		},
+	})
+	if err != nil {
+		t.Errorf("run err: %v", err)
+	}
+
+	for answer, err := res.Next(); err == nil; answer, err = res.Next() {
+		t.Log(answer.Answer)
+	}
+
+}
+func TestAppBuilderClientRunCustomMetadata(t *testing.T) {
+	t.Parallel() // 并发运行
+	// 测试逻辑
+	config, err := NewSDKConfig("", "")
+	if err != nil {
+		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+		t.Fatalf("new http client config failed: %v", err)
+	}
+
+	appID := "a3654cd9-378a-4b46-a33b-2259ca3b304e"
+	client, err := NewAppBuilderClient(appID, config)
+	if err != nil {
+		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+		t.Fatalf("new AppBuilderClient instance failed")
+	}
+
+	conversationID, err := client.CreateConversation()
+	if err != nil {
+		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+		t.Fatalf("create conversation failed: %v", err)
+	}
+
+	res, err := client.Run(AppBuilderClientRunRequest{
+		AppID:          appID,
+		Query:          "我要回老家相亲",
+		Stream:         false,
+		ConversationID: conversationID,
+		CustomMetadata: &CustomMetadata{
+			OverrideRoleInstruction: `# 角色任务\n" +
+                "作为高情商大师，你的主要任务是根据提问，做出最佳的建议。\n" +
+                "\n" +
+                "# 工具能力\n" +
+                "\n" +
+                "无工具集提供\n" +
+                "\n" +
+                "# 要求与限制\n" +
+                "\n" +
+                "1. 输出内容的风格为幽默\n" +
+                "2.输出的字数限制为100字以内"`,
+		},
+	})
+	if err != nil {
+		t.Errorf("run err: %v", err)
+	}
+
+	for answer, err := res.Next(); err == nil; answer, err = res.Next() {
+		t.Log(answer.Answer)
+	}
+
+}