diff options
Diffstat (limited to 'vendor/github.com')
| -rw-r--r-- | vendor/github.com/google/generative-ai-go/genai/files.go | 14 | ||||
| -rw-r--r-- | vendor/github.com/google/generative-ai-go/genai/internal/version.go | 2 |
2 files changed, 15 insertions, 1 deletions
diff --git a/vendor/github.com/google/generative-ai-go/genai/files.go b/vendor/github.com/google/generative-ai-go/genai/files.go index 12de62457..7fe57db15 100644 --- a/vendor/github.com/google/generative-ai-go/genai/files.go +++ b/vendor/github.com/google/generative-ai-go/genai/files.go @@ -19,6 +19,7 @@ package genai import ( "context" "io" + "os" "strings" gl "cloud.google.com/go/ai/generativelanguage/apiv1beta" @@ -83,6 +84,19 @@ func (c *Client) UploadFile(ctx context.Context, name string, r io.Reader, opts return c.GetFile(ctx, res.File.Name) } +// UploadFileFromPath is a convenience method wrapping [UploadFile]. It takes +// a path to read the file from, and uses a default auto-generated ID for the +// uploaded file. +func (c *Client) UploadFileFromPath(ctx context.Context, path string, opts *UploadFileOptions) (*File, error) { + osf, err := os.Open(path) + if err != nil { + return nil, err + } + defer osf.Close() + + return c.UploadFile(ctx, "", osf, opts) +} + // GetFile returns the named file. func (c *Client) GetFile(ctx context.Context, name string) (*File, error) { req := &pb.GetFileRequest{Name: userNameToServiceName(name)} diff --git a/vendor/github.com/google/generative-ai-go/genai/internal/version.go b/vendor/github.com/google/generative-ai-go/genai/internal/version.go index c71bf93f9..e6f17360d 100644 --- a/vendor/github.com/google/generative-ai-go/genai/internal/version.go +++ b/vendor/github.com/google/generative-ai-go/genai/internal/version.go @@ -5,4 +5,4 @@ package internal // Version is the current tagged release of the library. -const Version = "0.17.0" +const Version = "0.18.0" |