googlai: align option types with langchain's default

tmc · Jan 18, 2024 · 470ee50 · 470ee50
1 parent 232daa6
commit 470ee50
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/llms/googleai/googleai_llm.go b/llms/googleai/googleai_llm.go
@@ -77,10 +77,10 @@ func (g *GoogleAI) GenerateContent(ctx context.Context, messages []llms.MessageC
 
 	opts := llms.CallOptions{
 		Model:       g.opts.defaultModel,
-		MaxTokens:   int(g.opts.defaultMaxTokens),
-		Temperature: float64(g.opts.defaultTemperature),
-		TopP:        float64(g.opts.defaultTopP),
-		TopK:        int(g.opts.defaultTopK),
+		MaxTokens:   g.opts.defaultMaxTokens,
+		Temperature: g.opts.defaultTemperature,
+		TopP:        g.opts.defaultTopP,
+		TopK:        g.opts.defaultTopK,
 	}
 	for _, opt := range options {
 		opt(&opts)

diff --git a/llms/googleai/googleai_option.go b/llms/googleai/googleai_option.go
@@ -6,10 +6,10 @@ type options struct {
 	apiKey                string
 	defaultModel          string
 	defaultEmbeddingModel string
-	defaultMaxTokens      int32
-	defaultTemperature    float32
-	defaultTopK           int32
-	defaultTopP           float32
+	defaultMaxTokens      int
+	defaultTemperature    float64
+	defaultTopK           int
+	defaultTopP           float64
 }
 
 func defaultOptions() options {