134 lines
3.3 KiB
Go
134 lines
3.3 KiB
Go
package client
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"fmt"
|
||
"io"
|
||
"log"
|
||
"net/http"
|
||
"time"
|
||
|
||
"jimeng/signer"
|
||
)
|
||
|
||
// Client 即梦API客户端
|
||
type Client struct {
|
||
signer *signer.Config
|
||
hc *http.Client
|
||
}
|
||
|
||
// NewClient 创建客户端
|
||
func NewClient(ak, sk, host string) *Client {
|
||
return &Client{
|
||
signer: &signer.Config{
|
||
AccessKey: ak,
|
||
SecretKey: sk,
|
||
Host: host,
|
||
},
|
||
hc: &http.Client{Timeout: 30 * time.Second},
|
||
}
|
||
}
|
||
|
||
// post 统一发送带签名的 POST 请求
|
||
func (c *Client) post(path string, body any, out any) error {
|
||
b, err := json.Marshal(body)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
log.Printf("[HTTP] POST %s Body=%s", path, string(b))
|
||
hdrs, err := c.signer.Headers(&signer.Request{
|
||
Method: "POST",
|
||
Path: path,
|
||
Body: b,
|
||
ContentType: "application/json",
|
||
})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
req, err := http.NewRequest("POST", c.signer.Host+path, bytes.NewReader(b))
|
||
if err != nil {
|
||
return err
|
||
}
|
||
req.Header.Set("Content-Type", "application/json")
|
||
for k, v := range hdrs {
|
||
req.Header.Set(k, v)
|
||
}
|
||
resp, err := c.hc.Do(req)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer resp.Body.Close()
|
||
rb, _ := io.ReadAll(resp.Body)
|
||
log.Printf("[HTTP] Response status=%d body=%s", resp.StatusCode, string(rb))
|
||
if err := json.Unmarshal(rb, out); err != nil {
|
||
return fmt.Errorf("decode resp: %w, body=%s", err, string(rb))
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// SubmitNovelTask 提交短剧任务
|
||
func (c *Client) SubmitNovelTask(input map[string]interface{}) (map[string]interface{}, error) {
|
||
req := map[string]interface{}{
|
||
"capability_key": "pippit_novel_agent",
|
||
"action_key": input["action_key"],
|
||
"run_id": fmt.Sprintf("biz-%d", time.Now().UnixNano()),
|
||
"input": input,
|
||
}
|
||
var out map[string]interface{}
|
||
if err := c.post("/agent_openapi/v1/novel/submit", req, &out); err != nil {
|
||
return nil, err
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// QueryNovelTask 查询短剧任务
|
||
func (c *Client) QueryNovelTask(taskID, runID string) (map[string]interface{}, error) {
|
||
body := map[string]string{}
|
||
if taskID != "" {
|
||
body["task_id"] = taskID
|
||
} else {
|
||
body["run_id"] = runID
|
||
}
|
||
var out map[string]interface{}
|
||
if err := c.post("/agent_openapi/v1/novel/query", body, &out); err != nil {
|
||
return nil, err
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// SubmitVideoTask 提交视频任务（带货营销/剧情营销/短片创作）
|
||
func (c *Client) SubmitVideoTask(capabilityKey string, input map[string]interface{}) (map[string]interface{}, error) {
|
||
req := map[string]interface{}{
|
||
"capability_key": capabilityKey,
|
||
"action_key": "generate_video",
|
||
"run_id": fmt.Sprintf("biz-%d", time.Now().UnixNano()),
|
||
}
|
||
// 带货营销和剧情营销使用 avatar_marketing_input，短片创作使用 input
|
||
if capabilityKey == "pippit_video_part_agent" {
|
||
req["input"] = input
|
||
} else {
|
||
req["avatar_marketing_input"] = input
|
||
}
|
||
var out map[string]interface{}
|
||
if err := c.post("/agent_openapi/v1/video/submit", req, &out); err != nil {
|
||
return nil, err
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// QueryVideoTask 查询视频任务
|
||
func (c *Client) QueryVideoTask(taskID, runID string) (map[string]interface{}, error) {
|
||
body := map[string]string{}
|
||
if taskID != "" {
|
||
body["task_id"] = taskID
|
||
} else {
|
||
body["run_id"] = runID
|
||
}
|
||
var out map[string]interface{}
|
||
if err := c.post("/agent_openapi/v1/video/query", body, &out); err != nil {
|
||
return nil, err
|
||
}
|
||
return out, nil
|
||
}
|