Set temperature

hupe1980 · Aug 1, 2023 · a019a32 · a019a32
1 parent b9497c7
commit a019a32
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docs/content/en/docs/chains/structured_output.md b/docs/content/en/docs/chains/structured_output.md
@@ -26,7 +26,9 @@ type Person struct {
 }
 
 func main() {
-	chatModel, err := chatmodel.NewOpenAI(os.Getenv("OPENAI_API_KEY"))
+	chatModel, err := chatmodel.NewOpenAI(os.Getenv("OPENAI_API_KEY"), func(o *chatmodel.OpenAIOptions) {
+		o.Temperature = 0
+	})
 	if err != nil {
 		log.Fatal(err)
 	}

diff --git a/examples/structured_output_chain/main.go b/examples/structured_output_chain/main.go
@@ -20,7 +20,9 @@ type Person struct {
 }
 
 func main() {
-	chatModel, err := chatmodel.NewOpenAI(os.Getenv("OPENAI_API_KEY"))
+	chatModel, err := chatmodel.NewOpenAI(os.Getenv("OPENAI_API_KEY"), func(o *chatmodel.OpenAIOptions) {
+		o.Temperature = 0
+	})
 	if err != nil {
 		log.Fatal(err)
 	}