Add initial implementation of Gema CLI with environment configuration and command execution

Author: 4nkitd

.env

@@ -0,0 +1,2 @@
+GENAI_API_KEY=
+GENAI_DEFAULT_MODEL=

Readme.md

@@ -0,0 +1,58 @@
+# Gema CLI
+
+Gema CLI is an AI assistant for your terminal that can run commands based on your queries. It leverages the power of AI to understand your requests and suggest appropriate commands to execute.
+
+## Installation
+
+To install Gema CLI, clone the repository and build the executable:
+
+```sh
+git clone https://github.com/4nkitd/gemini-cli.git
+cd gemini-cli
+go build -o ?
+```
+
+## Usage
+
+Before using Gema CLI, you need to set the following environment variables:
+
+- `GENAI_DEFAULT_MODEL`: The default AI model to use.
+- `GENAI_API_KEY`: Your API key for accessing the AI service.
+
+You can set these environment variables in your terminal:
+
+```sh
+export GENAI_DEFAULT_MODEL="models/gemini-2.0-flash-exp" 
+export GENAI_API_KEY=your_api_key
+```
+
+To use Gema CLI, simply run the executable with your query:
+
+```sh
+./? "your query here"
+```
+
+Gema CLI will process your query, provide a response, and suggest a command to run. You will be prompted to confirm whether you want to execute the suggested command.
+
+## Example
+
+```sh
+./? "list all files in the current directory"
+```
+
+Output:
+
+```
+Response:
+Here is the command to list all files in the current directory:
+
+Suggested Command to RUN: ls -la
+
+Run command (y for yes, n for no):
+```
+
+If you type `y`, the command will be executed, and the output will be displayed.
+
+## Contributing
+
+Contributions are welcome! Please open an issue or submit a pull request on GitHub.

gemini.go

@@ -0,0 +1,103 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/google/generative-ai-go/genai"
+	"google.golang.org/api/option"
+)
+
+type AiResponse struct {
+	Response string `json:"response"`
+	Command  string `json:"command"`
+}
+
+func AskQuery(query string) AiResponse {
+
+	apiKey := os.Getenv("GENAI_API_KEY")
+	if apiKey == "" {
+		log.Fatal("GENAI_API_KEY environment variable is not set")
+	}
+
+	defaultModel := os.Getenv("GENAI_DEFAULT_MODEL")
+	if defaultModel == "" {
+		log.Fatal("GENAI_DEFAULT_MODEL environment variable is not set")
+	}
+	var aiTemp float32 = 1
+
+	ctx := context.Background()
+	client, err := genai.NewClient(ctx, option.WithAPIKey(apiKey))
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer client.Close()
+	model := client.GenerativeModel(defaultModel)
+	model.Temperature = &aiTemp
+
+	model.SystemInstruction = &genai.Content{Parts: []genai.Part{genai.Text(SystemInstruction)}}
+	model.ResponseMIMEType = "application/json"
+	model.ResponseSchema = &genai.Schema{
+		Type: genai.TypeObject,
+		Properties: map[string]*genai.Schema{
+			"response": {
+				Type: genai.TypeString,
+			},
+			"command": {
+				Type: genai.TypeString,
+			},
+		},
+		Required: []string{"response"},
+	}
+
+	resp, err := model.GenerateContent(ctx, genai.Text(query))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	jsonResponse := responseString(resp)
+
+	if !json.Valid([]byte(jsonResponse)) {
+		log.Fatal("Invalid JSON response")
+	}
+
+	var result AiResponse
+	if err := json.Unmarshal([]byte(jsonResponse), &result); err != nil {
+		log.Fatal(err)
+	}
+
+	if result.Response == "" {
+		log.Fatal("Response field is missing or not a string")
+	}
+
+	return result
+}
+
+func responseString(resp *genai.GenerateContentResponse) string {
+	var b strings.Builder
+	for i, cand := range resp.Candidates {
+		if len(resp.Candidates) > 1 {
+			fmt.Fprintf(&b, "%d:", i+1)
+		}
+		b.WriteString(contentString(cand.Content))
+	}
+	return b.String()
+}
+
+func contentString(c *genai.Content) string {
+	var b strings.Builder
+	if c == nil || c.Parts == nil {
+		return ""
+	}
+	for i, part := range c.Parts {
+		if i > 0 {
+			fmt.Fprintf(&b, ";")
+		}
+		fmt.Fprintf(&b, "%v", part)
+	}
+	return b.String()
+}

go.mod

@@ -0,0 +1,60 @@
+module github.com/4nkitd/gemini-cli
+
+go 1.23.4
+
+require (
+	github.com/google/generative-ai-go v0.19.0
+	github.com/spf13/cobra v1.8.1
+	google.golang.org/api v0.186.0
+)
+
+require (
+	cloud.google.com/go v0.115.0 // indirect
+	cloud.google.com/go/ai v0.8.0 // indirect
+	cloud.google.com/go/auth v0.6.0 // indirect
+	cloud.google.com/go/auth/oauth2adapt v0.2.2 // indirect
+	cloud.google.com/go/compute/metadata v0.3.0 // indirect
+	cloud.google.com/go/longrunning v0.5.7 // indirect
+	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
+	github.com/charmbracelet/bubbletea v1.2.4 // indirect
+	github.com/charmbracelet/lipgloss v1.0.0 // indirect
+	github.com/charmbracelet/x/ansi v0.4.5 // indirect
+	github.com/charmbracelet/x/term v0.2.1 // indirect
+	github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
+	github.com/felixge/httpsnoop v1.0.4 // indirect
+	github.com/go-logr/logr v1.4.1 // indirect
+	github.com/go-logr/stdr v1.2.2 // indirect
+	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
+	github.com/golang/protobuf v1.5.4 // indirect
+	github.com/google/s2a-go v0.1.7 // indirect
+	github.com/google/uuid v1.6.0 // indirect
+	github.com/googleapis/enterprise-certificate-proxy v0.3.2 // indirect
+	github.com/googleapis/gax-go/v2 v2.12.5 // indirect
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
+	github.com/lucasb-eyer/go-colorful v1.2.0 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/mattn/go-localereader v0.0.1 // indirect
+	github.com/mattn/go-runewidth v0.0.15 // indirect
+	github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 // indirect
+	github.com/muesli/cancelreader v0.2.2 // indirect
+	github.com/muesli/termenv v0.15.2 // indirect
+	github.com/rivo/uniseg v0.4.7 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	go.opencensus.io v0.24.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.51.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.51.0 // indirect
+	go.opentelemetry.io/otel v1.26.0 // indirect
+	go.opentelemetry.io/otel/metric v1.26.0 // indirect
+	go.opentelemetry.io/otel/trace v1.26.0 // indirect
+	golang.org/x/crypto v0.24.0 // indirect
+	golang.org/x/net v0.26.0 // indirect
+	golang.org/x/oauth2 v0.21.0 // indirect
+	golang.org/x/sync v0.9.0 // indirect
+	golang.org/x/sys v0.27.0 // indirect
+	golang.org/x/text v0.16.0 // indirect
+	golang.org/x/time v0.5.0 // indirect
+	google.golang.org/genproto/googleapis/api v0.0.0-20240617180043-68d350f18fd4 // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20240617180043-68d350f18fd4 // indirect
+	google.golang.org/grpc v1.64.1 // indirect
+	google.golang.org/protobuf v1.34.2 // indirect
+)