cmd/starlet: remove Gemini API support, fully replace by llm (#350)

Author: astrophena

cmd/starlet/doc.go

@@ -8,7 +8,7 @@ Starlet is a Telegram bot runner powered by Starlark.
 It fetches bot logic written in Starlark from a specified GitHub Gist,
 executes it within a Starlark interpreter, and handles incoming Telegram updates
 via webhooks. Starlet provides a sandboxed environment with pre-defined modules
-for interacting with Telegram, Gemini AI, key-value caching, and more.
+for interacting with Telegram, LLM, key-value caching, and more.
 
 When a Telegram update arrives, Starlet parses it, converts the JSON payload
 into a Starlark dictionary, and invokes the handle function defined in the
@@ -71,7 +71,6 @@ Required:
 Optional:
 
   - DATABASE_PATH: Path to a SQLite database file. If not provided, an in-memory store is used.
-  - GEMINI_KEY: API key for Google Gemini (required to use the gemini module).
   - LLM_API_KEY: API key for an OpenAI-compatible gateway (required to use llm).
   - LLM_API_URL: API root URL for an OpenAI-compatible gateway, for example
     https://api.openai.com/v1 (required to use llm).

cmd/starlet/internal/bot/bot.go

@@ -22,7 +22,6 @@ import (
 	"go.astrophena.name/base/request"
 	"go.astrophena.name/base/version"
 	"go.astrophena.name/base/web"
-	"go.astrophena.name/tools/internal/api/gemini"
 	"go.astrophena.name/tools/internal/api/llm"
 	"go.astrophena.name/tools/internal/starlark/go2star"
 	"go.astrophena.name/tools/internal/starlark/interpreter"
@@ -66,7 +65,6 @@ type Bot struct {
 	tgBotUsername string
 
 	httpc        *http.Client
-	geminic      *gemini.Client
 	llmc         *llm.Client
 	llmUsagePath string
 	kvCache      *starlarkstruct.Module
@@ -97,8 +95,6 @@ type Opts struct {
 	BotUsername string
 	// HTTPClient is the HTTP client to use for making requests.
 	HTTPClient *http.Client
-	// GeminiClient is the client for interacting with the Google Gemini API.
-	GeminiClient *gemini.Client
 	// LLMClient is the client for interacting with an OpenAI-compatible LLM API.
 	LLMClient *llm.Client
 	// LLMUsagePath is path to persistent usage stats JSON for llm module.
@@ -120,7 +116,6 @@ func New(opts Opts) *Bot {
 		tgBotID:       opts.BotID,
 		tgBotUsername: opts.BotUsername,
 		httpc:         opts.HTTPClient,
-		geminic:       opts.GeminiClient,
 		llmc:          opts.LLMClient,
 		llmUsagePath:  opts.LLMUsagePath,
 		kvCache:       opts.KVCache,

cmd/starlet/internal/bot/doc.md

@@ -46,49 +46,6 @@ A module for accessing files provided to the bot.
 
 Reads the content of a file.
 
-## `gemini`
-
-This module provides a single function, generate_content, which uses the
-Gemini API to generate text, optionally with an image as context.
-
-It accepts the following keyword arguments:
-
-  - model (str): The name of the model to use for generation (e.g., "gemini-1.5-flash").
-  - contents (list of (str, str) tuples): A list of (role, text) tuples representing
-    the conversation history. Valid roles are typically "user" and "model".
-  - image (bytes, optional): The raw bytes of an image to include. The image is
-    inserted as a new part just before the last part of the 'contents'.
-    This is useful for multimodal prompts (e.g., asking a question about an image).
-  - system_instructions (str, optional): System instructions to guide Gemini's response.
-  - unsafe (bool, optional): If set to true, disables all safety settings for the
-    content generation, allowing potentially harmful content. Use with caution.
-
-For example, for a text-only prompt:
-
-	responses = gemini.generate_content(
-	    model="gemini-1.5-flash",
-	    contents=[
-	        ("user", "Once upon a time,"),
-	        ("model", "there was a brave knight."),
-	        ("user", "What happened next?")
-	    ],
-	    system_instructions="You are a creative story writer. Write a short story based on the provided prompt."
-	)
-
-To ask a question about an image:
-
-	image_data = ... # read image file content as bytes
-	responses = gemini.generate_content(
-	    model="gemini-1.5-flash",
-	    contents=[
-	        ("user", "Describe this image in detail.")
-	    ],
-	    image=image_data
-	)
-
-The responses variable will contain a list of generated responses, where each response
-is a list of strings representing the parts of the generated content.
-
 ## `llm`
 
 LLM provider.

cmd/starlet/internal/bot/env.go

@@ -11,7 +11,6 @@ import (
 	"strings"
 
 	"go.astrophena.name/base/version"
-	"go.astrophena.name/tools/internal/starlark/gemini"
 	"go.astrophena.name/tools/internal/starlark/go2star"
 	"go.astrophena.name/tools/internal/starlark/kvcache"
 	"go.astrophena.name/tools/internal/starlark/llm"
@@ -163,11 +162,6 @@ func (b *Bot) environment() Environment {
 				},
 			},
 		},
-		{
-			Name:  "gemini",
-			Doc:   gemini.Documentation(),
-			Value: gemini.Module(b.geminic),
-		},
 		{
 			Name:  "llm",
 			Doc:   llm.Documentation(),

cmd/starlet/main.go

@@ -22,7 +22,6 @@ import (
 	"go.astrophena.name/base/web"
 	"go.astrophena.name/base/web/service"
 	"go.astrophena.name/tools/cmd/starlet/internal/bot"
-	"go.astrophena.name/tools/internal/api/gemini"
 	"go.astrophena.name/tools/internal/api/gist"
 	"go.astrophena.name/tools/internal/api/llm"
 	"go.astrophena.name/tools/internal/starlark/kvcache"
@@ -48,7 +47,6 @@ type engine struct {
 
 	// configuration, read-only after initialization
 	databasePath string
-	geminiKey    string
 	llmAPIKey    string
 	llmAPIURL    string
 	llmUsagePath string
@@ -93,7 +91,6 @@ func (e *engine) doInit(ctx context.Context) error {
 
 	// Load configuration from environment variables.
 	e.databasePath = cmp.Or(e.databasePath, env.Getenv("DATABASE_PATH"))
-	e.geminiKey = cmp.Or(e.geminiKey, env.Getenv("GEMINI_KEY"))
 	e.llmAPIKey = cmp.Or(e.llmAPIKey, env.Getenv("LLM_API_KEY"))
 	e.llmAPIURL = cmp.Or(e.llmAPIURL, env.Getenv("LLM_API_URL"))
 	e.llmUsagePath = cmp.Or(e.llmUsagePath, env.Getenv("LLM_USAGE_PATH"))
@@ -109,7 +106,7 @@ func (e *engine) doInit(ctx context.Context) error {
 
 	if e.httpc == nil {
 		e.httpc = &http.Client{
-			// Increase timeout to properly handle Gemini API response times.
+			// Increase timeout to properly handle LLM API response times.
 			Timeout: 60 * time.Second,
 		}
 	}
@@ -120,7 +117,6 @@ func (e *engine) doInit(ctx context.Context) error {
 		e.gistID,
 		e.tgSecret,
 		e.tgToken,
-		e.geminiKey,
 		e.llmAPIKey,
 	} {
 		if val != "" {
@@ -163,14 +159,6 @@ func (e *engine) doInit(ctx context.Context) error {
 		LLMUsagePath: e.llmUsagePath,
 	}
 
-	if e.geminiKey != "" {
-		opts.GeminiClient = &gemini.Client{
-			APIKey:     e.geminiKey,
-			HTTPClient: e.httpc,
-			Scrubber:   e.scrubber,
-		}
-	}
-
 	if e.llmAPIKey != "" && e.llmAPIURL != "" {
 		opts.LLMClient = &llm.Client{
 			APIKey:     e.llmAPIKey,