add replicate

conf/app.conf

@@ -5,19 +5,36 @@ copyrequestbody = true
 
 
 [dev]
+; aonet
+; replicate
+whoisApi = "replicate"
 apikey = "Rbhpcp0FKNrYNA1nZkrwrIbD0YSSRlVG"
 taskUrl = "https://api.aonet.ai/api/v1"
 imageUrl = "https://tmp-file.aigic.ai/api/v1/upload/persistence"
+replicateToken = "r8_9OCCea50go2Qkh0f0jhu3DbNjyzuyt61VNVI6"
+replicateTimeout = 10
+
 [test]
+whoisApi = "aonet"
 apikey = "Rbhpcp0FKNrYNA1nZkrwrIbD0YSSRlVG"
 taskUrl = "https://api.aonet.ai/api/v1"
 imageUrl = "https://tmp-file.aigic.ai/api/v1/upload/persistence"
+replicateToken = "r8_9OCCea50go2Qkh0f0jhu3DbNjyzuyt61VNVI6"
+replicateTimeout = 10
+
 [prod]
+whoisApi = "aonet"
 apikey = "Rbhpcp0FKNrYNA1nZkrwrIbD0YSSRlVG"
 taskUrl = "https://api.aonet.ai/api/v1"
 imageUrl = "https://tmp-file.aigic.ai/api/v1/upload/persistence"
+replicateToken = "r8_9OCCea50go2Qkh0f0jhu3DbNjyzuyt61VNVI6"
+replicateTimeout = 10
+
 
 
 
 include "mysql.conf"
-include "mongo.conf"
\ No newline at end of file
+include "mongo.conf"
+
+
+; include "replicate_models_version.conf"
\ No newline at end of file

conf/replicate_models_version.yaml

+pulid:
+  version: "43d309c37ab4e62361e5e29b8e9e867fb2dcbcec77ae91206a8d95ac5dd451a0"
+  url: "https://api.replicate.com/v1/predictions"
+  stream: false
+
+idm-vton:
+  version: "906425dbca90663ff5427624839572cc56ea7d380343d13e2a4c4b09d3f0c30f"
+  url: "https://api.replicate.com/v1/predictions"
+  stream: false
+
+face-swap:
+  version: "bc479d7d8ecc50eb83839af0c28210db75cac9c23837e2722028df4cddfafa22"
+  url: "https://api.replicate.com/v1/predictions"
+  stream: false
+
+lllama3:0.0.8:
+  version: ""
+  url: "https://api.replicate.com/v1/models/meta/meta-llama-3-8b/predictions"
+  stream: false
+
+meta-llama-3-8b-instruct:
+  version: ""
+  url: "https://api.replicate.com/v1/models/meta/meta-llama-3-8b-instruct/predictions"
+  stream: false
+
+stable-diffusion-3:
+  version: ""
+  url: "https://api.replicate.com/v1/models/stability-ai/stable-diffusion-3/predictions"
+  stream: false
+
+xtts-v2:
+  version: "684bc3855b37866c0c65add2ff39c78f3dea3f4ff103a436465326e0f438d55e"
+  url: "https://api.replicate.com/v1/predictions"
+  stream: false
+
+sadtalker:
+  version: "a519cc0cfebaaeade068b23899165a11ec76aaa1d2b313d40d214f204ec957a3"
+  url: "https://api.replicate.com/v1/predictions"
+  stream: false
+
+stable-diffusion:
+  version: "ac732df83cea7fff18b8472768c88ad041fa750ff7682a21affe81863cbe77e4"
+  url: "https://api.replicate.com/v1/predictions"
+  stream: false
\ No newline at end of file

controllers/log.go

+package controllers
+
+import (
+	"aon_app_server/models"
+	"aon_app_server/utils/mongo"
+	"encoding/json"
+	"github.com/beego/beego/v2/core/logs"
+	"net/http"
+)
+
+type LogController struct {
+	MainController
+}
+
+// Report @router /report/ [post]
+func (server *LogController) Report() {
+	body := server.Ctx.Input.RequestBody
+	log := models.Log{}
+	err := json.Unmarshal(body, &log) //解析body中数据
+	logs.Debug("appRequest", log, string(body))
+	if err != nil {
+		server.respond(models.NoRequestBody, err.Error())
+		return
+	}
+
+	header := server.Ctx.Request.Header
+	platform := header.Get("Sec-Ch-Ua-Platform")
+	if log.Platform == "" && platform != "" {
+		log.Platform = platform
+	}
+	userAgent := header.Get("User-Agent")
+	if log.UserAgent == "" && userAgent != "" {
+		log.UserAgent = userAgent
+	}
+	isMobile := header.Get("Sec-Ch-Ua-Mobile")
+	if log.IsMobile == "" && isMobile != "" {
+		log.IsMobile = isMobile
+	}
+	secUa := header.Get("Sec-Ch-Ua")
+	if log.SecUa == "" && secUa != "" {
+		log.SecUa = secUa
+	}
+	_, err = mongo.Insert(&log)
+	if err != nil {
+		server.respond(models.BusinessFailed, err.Error())
+		return
+	}
+	server.respond(http.StatusOK, "")
+}
+
+// BatchReport @router /batch_report/ [post]
+func (server *LogController) BatchReport() {
+	body := server.Ctx.Input.RequestBody
+	var data []models.Log
+	err := json.Unmarshal(body, &data) //解析body中数据
+	logs.Debug("appRequest", data)
+	if err != nil {
+		server.respond(models.NoRequestBody, err.Error())
+		return
+	}
+
+	header := server.Ctx.Request.Header
+	for _, log := range data {
+		platform := header.Get("Sec-Ch-Ua-Platform")
+		if log.Platform == "" && platform != "" {
+			log.Platform = platform
+		}
+		userAgent := header.Get("User-Agent")
+		if log.UserAgent == "" && userAgent != "" {
+			log.UserAgent = userAgent
+		}
+		isMobile := header.Get("Sec-Ch-Ua-Mobile")
+		if log.IsMobile == "" && isMobile != "" {
+			log.IsMobile = isMobile
+		}
+		secUa := header.Get("Sec-Ch-Ua")
+		if log.SecUa == "" && secUa != "" {
+			log.SecUa = secUa
+		}
+	}
+
+	var interfaceSlice []interface{} = make([]interface{}, len(data))
+	for i, v := range data {
+		interfaceSlice[i] = v
+	}
+
+	_, err = mongo.InsertMany(interfaceSlice)
+	if err != nil {
+		server.respond(models.BusinessFailed, err.Error())
+		return
+	}
+	server.respond(http.StatusOK, "")
+}

go.mod

@@ -43,5 +43,5 @@ require (
 	golang.org/x/text v0.14.0 // indirect
 	golang.org/x/tools v0.6.0 // indirect
 	google.golang.org/protobuf v1.23.0 // indirect
-	gopkg.in/yaml.v2 v2.2.8 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
 )

go.sum

@@ -324,6 +324,8 @@ gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.2.5/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.2.8 h1:obN1ZagJSUGI0Ek/LBmuj4SNLPfIny3KsKFopxRdj10=
 gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
 honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
 honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=
 honnef.co/go/tools v0.0.1-2019.2.3/go.mod h1:a3bituU0lyd329TUQxRnasdCoJDkEUEAqEt0JzvZhAg=

lastupdate.tmp

-{"/Users/brent/Documents/wubanWork/aon_app_server/controllers":1719017412063939595}
\ No newline at end of file
+{"/Users/brent/Documents/wubanWork/aon_app_server/controllers":1719995732323745955}
\ No newline at end of file

models/log.go

+package models
+
+import "time"
+
+type Log struct {
+	Id        interface{} `json:"id" bson:"_id,omitempty"`
+	Version   string      `json:"sdk_version,omitempty" bson:"sdk_version"`
+	Time      time.Time   `json:"time" bson:"time"`
+	UserId    string      `json:"user_id" bson:"user_id"`
+	Device    string      `json:"device" bson:"device"`
+	Platform  string      `json:"platform" bson:"platform"`
+	Error     string      `json:"error" bson:"error"`
+	Method    string      `json:"method" bson:"method"`
+	Input     interface{} `json:"input" bson:"input"`
+	UserAgent string      `json:"user_agent" bson:"user_agent"`
+	IsMobile  string      `json:"is_mobile" bson:"is_mobile"`
+	SecUa     string      `json:"sec_ua" bson:"sec_ua"`
+}

models/task.go

@@ -28,13 +28,37 @@ type Task struct {
 	Deleted     int         `json:"deleted" bson:"deleted"`
 }
 
+type ReplicateRequest struct {
+	Version string      `json:"version,omitempty"`
+	Input   interface{} `json:"input"`
+	Stream  bool        `json:"stream,omitempty"`
+}
+
+type ReplicateResponse struct {
+	Id          string      `json:"id"`
+	Model       string      `json:"model"`
+	Version     string      `json:"version"`
+	Input       interface{} `json:"input"`
+	Logs        string      `json:"logs"`
+	Output      interface{} `json:"output"`
+	DataRemoved bool        `json:"data_removed"`
+	Error       interface{} `json:"error"`
+	Status      string      `json:"status"`
+	CreatedAt   time.Time   `json:"created_at"`
+	Urls        struct {
+		Cancel string `json:"cancel"`
+		Get    string `json:"get"`
+		Stream string `json:"stream"`
+	} `json:"urls"`
+}
+
 type TaskReturn struct {
 	TaskId    string `json:"task_id"`
 	TaskUid   string `json:"task_uid"`
 	TaskFee   string `json:"task_fee"`
 	IsSuccess bool   `json:"is_success"`
 	TaskError string `json:"task_error"`
-	ExecCode  string `json:"exec_code"`
+	ExecCode  int    `json:"exec_code"`
 	ExecError string `json:"exec_error"`
 	ApiError  struct {
 		RequestId string `json:"request_id"`
@@ -43,8 +67,8 @@ type TaskReturn struct {
 }
 
 type TaskResponse struct {
-	Task   TaskReturn `json:"task"`
-	Output []string   `json:"output"`
+	Task   *TaskReturn `json:"task"`
+	Output []string    `json:"output"`
 }
 
 type TaskResult struct {

models/versions.go

+package models
+
+var Versions = map[string]string{
+	"pulid":     "43d309c37ab4e62361e5e29b8e9e867fb2dcbcec77ae91206a8d95ac5dd451a0",
+	"idm-vton":  "906425dbca90663ff5427624839572cc56ea7d380343d13e2a4c4b09d3f0c30f",
+	"face-swap": "bc479d7d8ecc50eb83839af0c28210db75cac9c23837e2722028df4cddfafa22",
+}
+
+type Model struct {
+	Version string `yaml:"version"`
+	Url     string `yaml:"url"`
+	Stream  bool   `yaml:"stream"`
+}

routers/router.go

@@ -8,6 +8,8 @@ import (
 func init() {
 	beego.Router("/", &controllers.MainController{})
 	beego.AutoPrefix("app/api/v1", &controllers.TaskController{})
+	beego.AutoPrefix("app/api/v1", &controllers.LogController{})
+
 	//ns := beego.NewNamespace("app",
 	//	beego.NSNamespace("api",
 	//		beego.NSNamespace("v1",

utils/mongo/mongo.go

@@ -91,6 +91,29 @@ func Insert(i interface{}) (interface{}, error) {
 	return res.InsertedID, err
 }
 
+func InsertMany(i []interface{}) ([]interface{}, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	collectionName := ""
+	for _, value := range i {
+		t := reflect.TypeOf(value)
+		if t.Kind() == reflect.Ptr {
+			t = t.Elem()
+		}
+		logs.Debug("结构体名称 = ", t.Name())
+		collectionName = t.Name()
+	}
+
+	collection := database.Collection(collectionName)
+
+	res, err := collection.InsertMany(ctx, i)
+	if err != nil {
+		return nil, err
+	}
+	return res.InsertedIDs, err
+}
+
 func Update(i interface{}) (interface{}, error) {
 	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
@@ -206,9 +229,31 @@ func Query(collectionName string, page int, size int, filter interface{}) (int64
 				fieldName := v.Type().Field(i).Name
 				fieldValue := v.Field(i).Interface()
 				if fieldName == "created_time" {
+					t := reflect.TypeOf(fieldValue)
+					logs.Debug("created_time type = ", t)
+					interfaceSlice, ok := fieldValue.([]interface{})
+					if !ok {
+						return 0, nil, errors.New("created_time must be a Array")
+					}
+					if len(interfaceSlice) < 2 {
+						return 0, nil, errors.New("created_time len must = 2")
+					}
+					var createdTime []time.Time
+					for _, v := range interfaceSlice {
+						if str, ok := v.(string); ok {
+							startTime, err := time.Parse(time.DateTime, str)
+							if err != nil {
+								logs.Debug(err)
+							}
+							createdTime = append(createdTime, startTime)
+						} else {
+							fmt.Printf("类型断言失败，遇到非 string 类型的值: %v\n", v)
+						}
+					}
+
 					finalFilter[fieldName] = bson.M{
-						"$gte": "",
-						"$lt":  "",
+						"$gte": createdTime[0],
+						"$lt":  createdTime[1],
 					}
 					continue
 				}
@@ -223,9 +268,31 @@ func Query(collectionName string, page int, size int, filter interface{}) (int64
 				keyStr := fmt.Sprintf("%v", key.Interface())
 				logs.Debug("keyStr: value\n", keyStr, value)
 				if keyStr == "created_time" {
+					t := reflect.TypeOf(value)
+					logs.Debug("created_time type = ", t)
+					interfaceSlice, ok := value.([]interface{})
+					if !ok {
+						return 0, nil, errors.New("created_time must be a Array")
+					}
+					if len(interfaceSlice) < 2 {
+						return 0, nil, errors.New("created_time len must = 2")
+					}
+					var createdTime []time.Time
+					for _, v := range interfaceSlice {
+						if str, ok := v.(string); ok {
+							startTime, err := time.Parse(time.DateTime, str)
+							if err != nil {
+								logs.Debug(err)
+							}
+							createdTime = append(createdTime, startTime)
+						} else {
+							fmt.Printf("类型断言失败，遇到非 string 类型的值: %v\n", v)
+						}
+					}
+
 					finalFilter[keyStr] = bson.M{
-						"$gte": "",
-						"$lt":  "",
+						"$gte": createdTime[0],
+						"$lt":  createdTime[1],
 					}
 					continue
 				}