adds support for gemini format (native and openai)

Author: akshaydeo

go.mod

@@ -3,3 +3,5 @@ module github.com/maximhq/maxim-go
 go 1.21
 
 require github.com/google/uuid v1.6.0
+
+require github.com/joho/godotenv v1.5.1

go.sum

@@ -1,2 +1,4 @@
 github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
 github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=

logging/base.go

@@ -11,6 +11,7 @@ const (
 	ProviderAzure     = "azure"
 	ProviderAnthropic = "anthropic"
 	ProviderBedrock   = "aws"
+	ProviderGemini    = "gemini"
 )
 
 // baseConfig is the configuration for a base entity.

logging/generation.go

@@ -2,6 +2,7 @@ package logging
 
 import (
 	"encoding/json"
+	"fmt"
 	"log"
 	"time"
 )
@@ -165,6 +166,11 @@ type Generation struct {
 }
 
 func newGeneration(c *GenerationConfig, w *writer) *Generation {
+	// Validating provider
+	if c.Provider != ProviderOpenAI && c.Provider != ProviderAzure && c.Provider != ProviderBedrock && c.Provider != ProviderAnthropic && c.Provider != ProviderGemini {
+		fmt.Printf("[MaximSDK] Invalid provider %s, allowed providers are %s, %s, %s, %s, %s\n", c.Provider, ProviderOpenAI, ProviderAzure, ProviderBedrock, ProviderAnthropic, ProviderGemini)
+		return nil
+	}
 	return &Generation{
 		base: newBase(EntityGeneration, c.Id, &baseConfig{
 			SpanId: c.SpanId,
@@ -219,6 +225,11 @@ func (g *Generation) handleAnthropicResult(jsonData []byte) (*MaximLLMResult, er
 	return ParseAnthropicResult(jsonData)
 }
 
+// handleGeminiResult extracts and logs data from a Gemini completion
+func (g *Generation) handleGeminiResult(jsonData []byte) (*MaximLLMResult, error) {
+	return ParseGeminiResult(jsonData)
+}
+
 // handleAzure extracts and logs data from an Azure OpenAI completion
 func (g *Generation) handleAzure(jsonData []byte, _ time.Duration) (*MaximLLMResult, error) {
 	// Azure OpenAI has the same response format as OpenAI
@@ -255,6 +266,8 @@ func (g *Generation) SetResult(r interface{}) {
 		finalResult, err = g.handleBedrockConverseResult(jsonData)
 	case ProviderAnthropic:
 		finalResult, err = g.handleAnthropicResult(jsonData)
+	case ProviderGemini:
+		finalResult, err = g.handleGeminiResult(jsonData)
 	}
 	if err != nil {
 		log.Println("[MaximSDK] Failed to parse result", err)

logging/parser.go

@@ -24,6 +24,8 @@ func ParseResult(provider, model string, r interface{}) (*MaximLLMResult, error)
 		finalResult, err = ParseBedrockResult(model, jsonData)
 	case ProviderAnthropic:
 		finalResult, err = ParseAnthropicResult(jsonData)
+	case ProviderGemini:
+		finalResult, err = ParseGeminiResult(jsonData)
 	}
 	return finalResult, err
 }