1 files changed, 6 insertions, 25 deletions
diff --git a/doConnect.go b/doConnect.go
index 2285f0b..b7abd40 100644
--- a/doConnect.go
+++ b/doConnect.go
@@ -10,42 +10,23 @@ import (
 )
 
 // doConnect initializes the Gemini client and handles the request flow.
-func doConnect() error {
+func doConnect() (*genai.Client, error) {
 	apiKey := os.Getenv("GEMINI_API_KEY")
 	if apiKey == "" {
-		return log.Errorf("GEMINI_API_KEY environment variable not set")
+		return nil, log.Errorf("GEMINI_API_KEY environment variable not set")
 	}
 
 	ctx := context.Background()
 	client, err := genai.NewClient(ctx, &genai.ClientConfig{APIKey: apiKey})
 	if err != nil {
-		return log.Errorf("failed to create new genai client: %w", err)
+		return nil, log.Errorf("failed to create new genai client: %w", err)
 	}
 
-	if argv.JsonFile != "" {
-		req, err := parseJSON(argv.JsonFile)
-		if err != nil {
-			return err
-		}
-		log.Info("parseJSON() ok. model =", req.Model)
-
-		genaiContent, err := convertToGenai(req)
-		if err != nil {
-			return log.Errorf("failed to convert to genai.Content: %w", err)
-		}
-		log.Info("Successfully converted JSON to genai.Content")
-		// Here you would now use the 'genaiContent' to send to the API
-		_ = genaiContent // Prevent unused variable error for now
-
-		return nil
-	}
-
-	log.Info("doing sampleHello()")
-	return sampleHello(client)
+	return client, err
 }
 
 // sampleHello sends a hardcoded prompt to the model and prints the response.
-func sampleHello(client *genai.Client) error {
+func simpleHello(client *genai.Client) error {
 	log.Info("Sending 'hello, how are you' to the Gemini API...")
 	ctx := context.Background()
 
@@ -56,7 +37,7 @@ func sampleHello(client *genai.Client) error {
 
 	content := []*genai.Content{{Parts: parts}}
 
-	resp, err := client.Models.GenerateContent(ctx, "gemini-1.5-flash-latest", content, nil)
+	resp, err := client.Models.GenerateContent(ctx, "gemini-2.5-flash", content, nil)
 	if err != nil {
 		return log.Errorf("error sending message: %v", err)
 	}