steve/go-llm
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Refactor Toolbox handling in Google LLM integration.

Browse Source 
Implemented a nil check for Toolbox to prevent potential nil pointer dereferences. Cleaned up and reorganized code for better readability and maintainability while keeping placeholder functionality intact.
This commit is contained in:
Steve Dudenhoeffer 2025-01-22 19:49:37 -05:00
parent 0d70ec46de
commit e7b7aab62e
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
google.go
View File

@ -4,6 +4,7 @@ import (
"context" "context"
"encoding/json" "encoding/json"
"fmt" "fmt"
"github.com/google/generative-ai-go/genai" "github.com/google/generative-ai-go/genai"
"google.golang.org/api/option" "google.golang.org/api/option"
) )
@ -31,6 +32,7 @@ func (g google) requestToGoogleRequest(in Request, model *genai.GenerativeModel)
res = append(res, genai.Text(c.Text)) res = append(res, genai.Text(c.Text))
} }
if in.Toolbox != nil {
for _, tool := range in.Toolbox.funcs { for _, tool := range in.Toolbox.funcs {
panic("google ToolBox is todo" + tool.Name) panic("google ToolBox is todo" + tool.Name)
@ -43,6 +45,7 @@ func (g google) requestToGoogleRequest(in Request, model *genai.GenerativeModel)
}) })
*/ */
} }
}
return res return res
} }