initial

mindforge.cronjob/internal/llm/gemini.go

@@ -0,0 +1,86 @@
+package llm
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+)
+
+func (s *llmService) SendGeminiRequest(systemPrompt string, userPrompt string, model string) (string, error) {
+	apiKey := getEnvConfig("GEMINI_API_KEY")
+	if apiKey == "" {
+		return "", errors.New("GEMINI_API_KEY not found in .env or environment")
+	}
+
+	apiBase := "https://generativelanguage.googleapis.com/v1beta"
+
+	url := fmt.Sprintf("%s/models/%s:generateContent?key=%s", strings.TrimRight(apiBase, "/"), model, apiKey)
+
+	reqBody := map[string]interface{}{}
+	if systemPrompt != "" {
+		reqBody["system_instruction"] = map[string]interface{}{
+			"parts": []map[string]string{
+				{"text": systemPrompt},
+			},
+		}
+	}
+	reqBody["contents"] = []map[string]interface{}{
+		{
+			"role": "user",
+			"parts": []map[string]string{
+				{"text": userPrompt},
+			},
+		},
+	}
+
+	jsonBody, err := json.Marshal(reqBody)
+	if err != nil {
+		return "", err
+	}
+
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBody))
+	if err != nil {
+		return "", err
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	client := &http.Client{Timeout: 120 * time.Second}
+	resp, err := client.Do(req)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	bodyBytes, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return "", err
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return "", fmt.Errorf("Gemini API error status %d: %s", resp.StatusCode, string(bodyBytes))
+	}
+
+	var result struct {
+		Candidates []struct {
+			Content struct {
+				Parts []struct {
+					Text string `json:"text"`
+				} `json:"parts"`
+			} `json:"content"`
+		} `json:"candidates"`
+	}
+	if err := json.Unmarshal(bodyBytes, &result); err != nil {
+		return "", err
+	}
+
+	if len(result.Candidates) > 0 && len(result.Candidates[0].Content.Parts) > 0 {
+		return result.Candidates[0].Content.Parts[0].Text, nil
+	}
+
+	return "", errors.New("empty response from Gemini API")
+}

mindforge.cronjob/internal/llm/llm.go

@@ -0,0 +1,25 @@
+package llm
+
+import (
+	"os"
+)
+
+// Service defines the interface for connecting to LLMs
+type Service interface {
+	SendOpenAIRequest(systemPrompt string, userPrompt string, model string) (string, error)
+	SendGeminiRequest(systemPrompt string, userPrompt string, model string) (string, error)
+}
+
+type llmService struct{}
+
+// NewLLMService creates a new LLM service instance
+func NewLLMService() Service {
+	return &llmService{}
+}
+
+func getEnvConfig(key string) string {
+	if val := os.Getenv(key); val != "" {
+		return val
+	}
+	return ""
+}

mindforge.cronjob/internal/llm/openai.go

@@ -0,0 +1,88 @@
+package llm
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+)
+
+func (s *llmService) SendOpenAIRequest(systemPrompt string, userPrompt string, model string) (string, error) {
+	apiKey := getEnvConfig("OPENAI_API_KEY")
+	if apiKey == "" {
+		return "", errors.New("OPENAI_API_KEY not found in .env or environment")
+	}
+
+	apiBase := "https://api.openai.com/v1"
+
+	url := fmt.Sprintf("%s/chat/completions", strings.TrimRight(apiBase, "/"))
+
+	reqBody := map[string]interface{}{
+		"model": model,
+		"messages": []map[string]string{
+			{"role": "system", "content": systemPrompt},
+			{"role": "user", "content": userPrompt},
+		},
+	}
+
+	jsonBody, err := json.Marshal(reqBody)
+	if err != nil {
+		return "", err
+	}
+
+	var lastErr error
+	// Retry up to 5 times (total 5 attempts)
+	for i := 0; i < 5; i++ {
+		req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBody))
+		if err != nil {
+			return "", err
+		}
+		req.Header.Set("Content-Type", "application/json")
+		req.Header.Set("Authorization", "Bearer "+apiKey)
+
+		client := &http.Client{Timeout: 120 * time.Second}
+		resp, err := client.Do(req)
+
+		if err != nil {
+			lastErr = err
+			time.Sleep(time.Second * time.Duration(1<<i))
+			continue
+		}
+
+		bodyBytes, err := io.ReadAll(resp.Body)
+		resp.Body.Close()
+		if err != nil {
+			lastErr = err
+			time.Sleep(time.Second * time.Duration(1<<i))
+			continue
+		}
+
+		if resp.StatusCode != http.StatusOK {
+			lastErr = fmt.Errorf("OpenAI API error status %d: %s", resp.StatusCode, string(bodyBytes))
+			time.Sleep(time.Second * time.Duration(1<<i))
+			continue
+		}
+
+		var result struct {
+			Choices []struct {
+				Message struct {
+					Content string `json:"content"`
+				} `json:"message"`
+			} `json:"choices"`
+		}
+		if err := json.Unmarshal(bodyBytes, &result); err != nil {
+			return "", err
+		}
+
+		if len(result.Choices) > 0 {
+			return result.Choices[0].Message.Content, nil
+		}
+		return "", errors.New("empty response from OpenAI API")
+	}
+
+	return "", fmt.Errorf("failed to get OpenAI response after 5 attempts. Last error: %v", lastErr)
+}