cc078b953a
- Added a `hashUserID` function to generate a collision-safe hash for user IDs, improving metadata handling. - Updated the environment setup in `buildEnv` to override `metadata.user_id` with a sanitized value, ensuring compatibility with third-party APIs. - Enhanced logging for the environment variables to aid in debugging and monitoring.
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
package image
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/gou/process"
|
|
agentLLM "github.com/yaoapp/yao/agent/llm"
|
|
"github.com/yaoapp/yao/openapi/oauth/authorized"
|
|
)
|
|
|
|
//go:embed generate_schema.json
|
|
var GenerateSchemaJSON []byte
|
|
|
|
// GenerateHandler is the tools.image_generate process handler.
|
|
func GenerateHandler(proc *process.Process) interface{} {
|
|
prompt := proc.ArgsString(0)
|
|
if prompt == "" {
|
|
return map[string]interface{}{"error": "prompt is required"}
|
|
}
|
|
|
|
provider := proc.ArgsString(1)
|
|
size := proc.ArgsString(2, "1024x1024")
|
|
model := proc.ArgsString(3)
|
|
|
|
authInfo := authorized.ProcessAuthInfo(proc)
|
|
if authInfo == nil {
|
|
return map[string]interface{}{"error": "unauthorized: no auth info in request"}
|
|
}
|
|
|
|
connectorID := provider
|
|
if connectorID == "" {
|
|
connectorID = findFirstImageGenConnector(authInfo)
|
|
if connectorID == "" {
|
|
return map[string]interface{}{"error": "no image generation provider available; configure one or specify a provider"}
|
|
}
|
|
}
|
|
|
|
conn, _, err := agentLLM.ResolveConnector(connectorID, authInfo)
|
|
if err != nil {
|
|
return map[string]interface{}{"error": fmt.Sprintf("resolve connector: %v", err)}
|
|
}
|
|
|
|
options := map[string]interface{}{"size": size}
|
|
if model != "" {
|
|
options["model"] = model
|
|
}
|
|
resp, err := agentLLM.GenerateImage(conn, prompt, options)
|
|
if err != nil {
|
|
return map[string]interface{}{"error": fmt.Sprintf("image generation failed: %v", err)}
|
|
}
|
|
|
|
return map[string]interface{}{
|
|
"image": resp.Image,
|
|
"format": resp.Format,
|
|
"size": size,
|
|
}
|
|
}
|