feat(genai): Add generate content with routing option

Author: nardosalemu

genai/content_cache/contentcache_update.go

@@ -38,26 +38,26 @@ func updateContentCache(w io.Writer, cacheName string) error {
 
 	// Update expire time using TTL
 	resp, err := client.Caches.Update(ctx, cacheName, &genai.UpdateCachedContentConfig{
-		TTL: "36000s",
+		TTL: int64(36000),
 	})
 	if err != nil {
 		return fmt.Errorf("failed to update content cache exp. time with TTL: %w", err)
 	}
 
-	fmt.Fprintf(w, "Cache expires in: %s\n", time.Until(*resp.ExpireTime))
+	fmt.Fprintf(w, "Cache expires in: %s\n", time.Until(resp.ExpireTime))
 	// Example response:
 	// Cache expires in: 10h0m0.005875s
 
 	// Update expire time using specific time stamp
 	inSevenDays := time.Now().Add(7 * 24 * time.Hour)
 	resp, err = client.Caches.Update(ctx, cacheName, &genai.UpdateCachedContentConfig{
-		ExpireTime: &inSevenDays,
+		ExpireTime: inSevenDays,
 	})
 	if err != nil {
 		return fmt.Errorf("failed to update content cache expire time: %w", err)
 	}
 
-	fmt.Fprintf(w, "Cache expires in: %s\n", time.Until(*resp.ExpireTime))
+	fmt.Fprintf(w, "Cache expires in: %s\n", time.Until(resp.ExpireTime))
 	// Example response:
 	// Cache expires in: 167h59m59.80327s
 

genai/content_cache/contentcache_use_with_txt.go

@@ -46,10 +46,8 @@ func useContentCacheWithTxt(w io.Writer, cacheName string) error {
 		return fmt.Errorf("failed to use content cache to generate content: %w", err)
 	}
 
-	respText, err := resp.Text()
-	if err != nil {
-		return fmt.Errorf("failed to convert model response to text: %w", err)
-	}
+	respText := resp.Text()
+
 	fmt.Fprintln(w, respText)
 
 	// Example response:

genai/controlled_generation/ctrlgen_with_enum_schema.go

@@ -54,10 +54,8 @@ func generateWithEnumSchema(w io.Writer) error {
 		return fmt.Errorf("failed to generate content: %w", err)
 	}
 
-	respText, err := resp.Text()
-	if err != nil {
-		return fmt.Errorf("failed to convert model response to text: %w", err)
-	}
+	respText := resp.Text()
+
 	fmt.Fprintln(w, respText)
 
 	// Example response:

genai/controlled_generation/ctrlgen_with_nullable_schema.go

@@ -63,11 +63,11 @@ Finally, Saturday rounds off the week with sunny skies, a temperature of 80°F,
 					Items: &genai.Schema{
 						Type: "object",
 						Properties: map[string]*genai.Schema{
-							"Day":         {Type: "string", Nullable: true},
-							"Forecast":    {Type: "string", Nullable: true},
-							"Temperature": {Type: "integer", Nullable: true},
-							"Humidity":    {Type: "string", Nullable: true},
-							"Wind Speed":  {Type: "integer", Nullable: true},
+							"Day":         {Type: "string", Nullable: genai.Ptr(true)},
+							"Forecast":    {Type: "string", Nullable: genai.Ptr(true)},
+							"Temperature": {Type: "integer", Nullable: genai.Ptr(true)},
+							"Humidity":    {Type: "string", Nullable: genai.Ptr(true)},
+							"Wind Speed":  {Type: "integer", Nullable: genai.Ptr(true)},
 						},
 						Required: []string{"Day", "Temperature", "Forecast", "Wind Speed"},
 					},
@@ -81,10 +81,8 @@ Finally, Saturday rounds off the week with sunny skies, a temperature of 80°F,
 		return fmt.Errorf("failed to generate content: %w", err)
 	}
 
-	respText, err := resp.Text()
-	if err != nil {
-		return fmt.Errorf("failed to convert model response to text: %w", err)
-	}
+	respText := resp.Text()
+
 	fmt.Fprintln(w, respText)
 
 	// Example response:

genai/controlled_generation/ctrlgen_with_resp_schema.go

@@ -66,10 +66,8 @@ func generateWithRespSchema(w io.Writer) error {
 		return fmt.Errorf("failed to generate content: %w", err)
 	}
 
-	respText, err := resp.Text()
-	if err != nil {
-		return fmt.Errorf("failed to convert model response to text: %w", err)
-	}
+	respText := resp.Text()
+
 	fmt.Fprintln(w, respText)
 
 	// Example response: