d96a9f384a
## Merged Features from sub2apipro - Sora video generation integration (OpenAI Sora API) - Group management enhancements - Usage log improvements - Security headers middleware ## Chinese Model Pricing Updates - GLM-5, GLM-5-Turbo, GLM-5.1, GLM-4.7, GLM-4.5-Air - Baichuan4, Baichuan4-Turbo, Baichuan4-Air, Baichuan-M3-Plus - DeepSeek-V3, DeepSeek-V3.2, DeepSeek-R1 - Qwen3-8B (free), Qwen2.5-72B-Instruct ## URL Whitelist Additions - api.baichuan-ai.com (百川智能) - api.siliconflow.cn (硅基流动) - api.z.ai (智谱国际) - api.groq.com (Groq加速推理) ## Documentation - Added merge guide (docs/MERGE_GUIDE.md) - Added quick reference (docs/MERGE_QUICKREF.md) - Added review reports (docs/reviews/)
118 lines
3.6 KiB
Go
118 lines
3.6 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
// SoraClient 定义直连 Sora 的任务操作接口。
|
|
type SoraClient interface {
|
|
Enabled() bool
|
|
UploadImage(ctx context.Context, account *Account, data []byte, filename string) (string, error)
|
|
CreateImageTask(ctx context.Context, account *Account, req SoraImageRequest) (string, error)
|
|
CreateVideoTask(ctx context.Context, account *Account, req SoraVideoRequest) (string, error)
|
|
CreateStoryboardTask(ctx context.Context, account *Account, req SoraStoryboardRequest) (string, error)
|
|
UploadCharacterVideo(ctx context.Context, account *Account, data []byte) (string, error)
|
|
GetCameoStatus(ctx context.Context, account *Account, cameoID string) (*SoraCameoStatus, error)
|
|
DownloadCharacterImage(ctx context.Context, account *Account, imageURL string) ([]byte, error)
|
|
UploadCharacterImage(ctx context.Context, account *Account, data []byte) (string, error)
|
|
FinalizeCharacter(ctx context.Context, account *Account, req SoraCharacterFinalizeRequest) (string, error)
|
|
SetCharacterPublic(ctx context.Context, account *Account, cameoID string) error
|
|
DeleteCharacter(ctx context.Context, account *Account, characterID string) error
|
|
PostVideoForWatermarkFree(ctx context.Context, account *Account, generationID string) (string, error)
|
|
DeletePost(ctx context.Context, account *Account, postID string) error
|
|
GetWatermarkFreeURLCustom(ctx context.Context, account *Account, parseURL, parseToken, postID string) (string, error)
|
|
EnhancePrompt(ctx context.Context, account *Account, prompt, expansionLevel string, durationS int) (string, error)
|
|
GetImageTask(ctx context.Context, account *Account, taskID string) (*SoraImageTaskStatus, error)
|
|
GetVideoTask(ctx context.Context, account *Account, taskID string) (*SoraVideoTaskStatus, error)
|
|
}
|
|
|
|
// SoraImageRequest 图片生成请求参数
|
|
type SoraImageRequest struct {
|
|
Prompt string
|
|
Width int
|
|
Height int
|
|
MediaID string
|
|
}
|
|
|
|
// SoraVideoRequest 视频生成请求参数
|
|
type SoraVideoRequest struct {
|
|
Prompt string
|
|
Orientation string
|
|
Frames int
|
|
Model string
|
|
Size string
|
|
VideoCount int
|
|
MediaID string
|
|
RemixTargetID string
|
|
CameoIDs []string
|
|
}
|
|
|
|
// SoraStoryboardRequest 分镜视频生成请求参数
|
|
type SoraStoryboardRequest struct {
|
|
Prompt string
|
|
Orientation string
|
|
Frames int
|
|
Model string
|
|
Size string
|
|
MediaID string
|
|
}
|
|
|
|
// SoraImageTaskStatus 图片任务状态
|
|
type SoraImageTaskStatus struct {
|
|
ID string
|
|
Status string
|
|
ProgressPct float64
|
|
URLs []string
|
|
ErrorMsg string
|
|
}
|
|
|
|
// SoraVideoTaskStatus 视频任务状态
|
|
type SoraVideoTaskStatus struct {
|
|
ID string
|
|
Status string
|
|
ProgressPct int
|
|
URLs []string
|
|
GenerationID string
|
|
ErrorMsg string
|
|
}
|
|
|
|
// SoraCameoStatus 角色处理中间态
|
|
type SoraCameoStatus struct {
|
|
Status string
|
|
StatusMessage string
|
|
DisplayNameHint string
|
|
UsernameHint string
|
|
ProfileAssetURL string
|
|
InstructionSetHint any
|
|
InstructionSet any
|
|
}
|
|
|
|
// SoraCharacterFinalizeRequest 角色定稿请求参数
|
|
type SoraCharacterFinalizeRequest struct {
|
|
CameoID string
|
|
Username string
|
|
DisplayName string
|
|
ProfileAssetPointer string
|
|
InstructionSet any
|
|
}
|
|
|
|
// SoraUpstreamError 上游错误
|
|
type SoraUpstreamError struct {
|
|
StatusCode int
|
|
Message string
|
|
Headers http.Header
|
|
Body []byte
|
|
}
|
|
|
|
func (e *SoraUpstreamError) Error() string {
|
|
if e == nil {
|
|
return "sora upstream error"
|
|
}
|
|
if e.Message != "" {
|
|
return fmt.Sprintf("sora upstream error: %d %s", e.StatusCode, e.Message)
|
|
}
|
|
return fmt.Sprintf("sora upstream error: %d", e.StatusCode)
|
|
}
|