alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Standardize action results

Browse Source
This commit is contained in:
mudler
2025-03-01 17:27:07 +01:00
parent 8492c95cb6
commit 5b4f618ca3
22 changed files with 108 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
services/actions/genimage.go
View File

@@ -24,18 +24,18 @@ type GenImageAction struct {
imageModel string
}
func (a *GenImageAction) Run(ctx context.Context, params action.ActionParams) (string, error) {
func (a *GenImageAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct {
Prompt string `json:"prompt"`
Size string `json:"size"`
}{}
err := params.Unmarshal(&result)
if err != nil {
return "", err
return action.ActionResult{}, err
}
if result.Prompt == "" {
return "", fmt.Errorf("prompt is required")
return action.ActionResult{}, fmt.Errorf("prompt is required")
}
req := openai.ImageRequest{
@@ -56,14 +56,17 @@ func (a *GenImageAction) Run(ctx context.Context, params action.ActionParams) (s
resp, err := a.client.CreateImage(ctx, req)
if err != nil {
return "Failed to generate image " + err.Error(), err
return action.ActionResult{Result: "Failed to generate image " + err.Error()}, err
}
if len(resp.Data) == 0 {
return "Failed to generate image", nil
return action.ActionResult{Result: "Failed to generate image"}, nil
}
return fmt.Sprintf("The image was generated and available at: %s", resp.Data[0].URL), nil
return action.ActionResult{
Result: fmt.Sprintf("The image was generated and available at: %s", resp.Data[0].URL), Metadata: map[string]interface{}{
"url": resp.Data[0].URL,
}}, nil
}
func (a *GenImageAction) Definition() action.ActionDefinition {