调用大模型示例
package main
import (
"context"
"fmt"
"log"
"os"
"github.com/tmc/langchaingo/llms"
"github.com/tmc/langchaingo/llms/openai"
)
func main() {
token := os.Getenv("LLM_API_KEY")
log.Println("token:" + token)
ctx := context.Background()
llm, err := openai.New(
openai.WithBaseURL("https://api.siliconflow.cn/v1"),
openai.WithModel("deepseek-ai/DeepSeek-R1-0528-Qwen3-8B"),
openai.WithToken(token),
)
if err != nil {
log.Fatal(err)
}
prompt := "你是谁,版本是多少?"
completion, err := llms.GenerateFromSinglePrompt(ctx, llm, prompt)
if err != nil {
log.Fatal(err)
}
fmt.Println(completion)
}
输出

image.png