feat: Support files delete, get, list, download/

PiperOrigin-RevId: 744834118

Author: qiaodev

api_client.go

@@ -25,6 +25,7 @@ import (
 	"log"
 	"math"
 	"net/http"
+	// "net/http/httputil"
 	"net/url"
 	"os"
 	"runtime"
@@ -70,6 +71,19 @@ func sendRequest(ctx context.Context, ac *apiClient, path string, method string,
 	return deserializeUnaryResponse(resp)
 }
 
+func downloadFile(ctx context.Context, ac *apiClient, path string, httpOptions *HTTPOptions) ([]byte, error) {
+	req, err := buildRequest(ctx, ac, path, nil, http.MethodGet, httpOptions)
+	if err != nil {
+		return nil, err
+	}
+
+	resp, err := doRequest(ac, req)
+	if err != nil {
+		return nil, err
+	}
+	return io.ReadAll(resp.Body)
+}
+
 func mapToStruct[R any](input map[string]any, output *R) error {
 	b := new(bytes.Buffer)
 	err := json.NewEncoder(b).Encode(input)

client.go

@@ -37,10 +37,10 @@ type Client struct {
 	Caches *Caches
 	// Chats provides util functions for creating a new chat session.
 	Chats *Chats
-	// Operations provides access to long-running operations.
-	Operations *Operations
 	// Files provides access to the Files service.
 	Files *Files
+	// Operations provides access to long-running operations.
+	Operations *Operations
 }
 
 // Backend is the GenAI backend to use for the client.

examples/files/README.md

@@ -22,5 +22,5 @@ example using the following commands.
 ```
 $ go get google.golang.org/genai
 $ cd `go list -f '{{.Dir}}' google.golang.org/genai/examples/files`
-$ go run files.go
+$ go run list_download.go
 ```

examples/files/list_download.go

@@ -0,0 +1,59 @@
+// Copyright 2025 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+//go:build ignore_vet
+
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"log"
+
+	"google.golang.org/genai"
+)
+
+var model = flag.String("model", "gemini-2.0-flash", "the model name, e.g. gemini-2.0-flash")
+
+func run(ctx context.Context) {
+	client, err := genai.NewClient(ctx, nil)
+	if err != nil {
+		log.Fatal(err)
+	}
+	if client.ClientConfig().Backend == genai.BackendVertexAI {
+		fmt.Println("Calling VertexAI Backend...")
+	} else {
+		fmt.Println("Calling GeminiAPI Backend...")
+	}
+
+	// Create a new Chat.
+	for file, err := range client.Files.All(ctx) {
+		if err != nil {
+			log.Fatal(err)
+		}
+		data, err := client.Files.Download(ctx, file, nil)
+		if err != nil {
+			log.Printf("Download %s failed: %w\n", file.Name, err)
+		} else {
+			fmt.Printf("Downloaded %s. Data size: %d\n", file.Name, len(data))
+		}
+	}
+}
+
+func main() {
+	ctx := context.Background()
+	flag.Parse()
+	run(ctx)
+}

examples/models/generate_videos/videos.go

@@ -39,12 +39,12 @@ func run(ctx context.Context) {
 	} else {
 		fmt.Println("Calling GeminiAI GenerateVideo API...")
 	}
-	// Pass in basic config
-	var config *genai.GenerateVideosConfig = &genai.GenerateVideosConfig{
-		OutputGCSURI: "gs://unified-genai-tests/tmp/genai/video/outputs",
+	var config genai.GenerateVideosConfig
+	if client.ClientConfig().Backend == genai.BackendVertexAI {
+		config.OutputGCSURI = "gs://unified-genai-tests/tmp/genai/video/outputs"
 	}
 	// Call the GenerateVideo method.
-	operation, err := client.Models.GenerateVideos(ctx, *model, "A neon hologram of a cat driving at top speed", nil, config)
+	operation, err := client.Models.GenerateVideos(ctx, *model, "A neon hologram of a cat driving at top speed", nil, &config)
 	if err != nil {
 		log.Fatal(err)
 	}
@@ -65,6 +65,18 @@ func run(ctx context.Context) {
 	}
 	// Log the output.
 	fmt.Println(string(response))
+
+	// Download the video file.
+	if client.ClientConfig().Backend != genai.BackendVertexAI {
+		for _, v := range operation.Response.GeneratedVideos {
+			data, err := client.Files.Download(ctx, genai.NewDownloadURIFromGeneratedVideo(v), nil)
+			if err != nil {
+				log.Println(err)
+				continue
+			}
+			fmt.Printf("Video file %s downloaded. Data size: %d. \n", v.Video.URI, len(data))
+		}
+	}
 }
 
 func main() {