环境变量使用说明添加

Author: wangle201210

README.md

@@ -19,6 +19,10 @@ go get github.com/wangle201210/text2sql
 
 ### 基本用法
 
+export OPENAI_API_KEY="sk-******"
+export OPENAI_MODEL_NAME="gpt-4o-mini"
+export OPENAI_BASE_URL="https://api.openai.com/v1"
+
 ```go
 package main
 

eino/model.go

@@ -10,6 +10,14 @@ import (
 )
 
 func createOpenAIChatModel(ctx context.Context) model.ChatModel {
+	m := os.Getenv("OPENAI_MODEL_NAME")
+	if m == "" {
+		log.Fatalf("请在环境变量中设置你的 OPENAI_MODEL_NAME")
+	}
+	apikey := os.Getenv("OPENAI_API_KEY")
+	if apikey == "" {
+		log.Fatalf("请在环境变量中设置你的 OPENAI_API_KEY")
+	}
 	chatModel, err := openai.NewChatModel(ctx, &openai.ChatModelConfig{
 		Model:   os.Getenv("OPENAI_MODEL_NAME"),
 		APIKey:  os.Getenv("OPENAI_API_KEY"),