OpenAI

OpenAI 提供了一系列不同级别的模型，适用于不同的任务。

本示例介绍了如何使用 LangChain 与 OpenAI 模型进行交互。

设置 OpenAI 密钥有两种方法：

1. 可以通过将环境变量 OPENAI_API_KEY 设置为 API 密钥来实现。
2. 或者在初始化包装器时与其他参数一起指定：

model, err := openai.New(openai.WithToken(apiToken))

示例

package main

import (
	"context"
	"fmt"
	"log"

	"github.com/tmc/langchaingo/llms"
	"github.com/tmc/langchaingo/llms/openai"
)

func main() {
	llm, err := openai.New()
	if err != nil {
		log.Fatal(err)
	}
	ctx := context.Background()
	completion, err := llms.GenerateFromSinglePrompt(ctx,
		llm,
		"The first man to walk on the moon",
		llms.WithTemperature(0.8),
		llms.WithStopWords([]string{"Armstrong"}),
	)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("The first man to walk on the moon:")
	fmt.Println(completion)
}