如何在Go中扩展第三方包的New()函数以读取.env文件

完全可以不用修改第三方包的代码，核心思路是在调用第三方包的New()函数之前，把.env文件中的配置加载到系统环境变量中——因为第三方包的New()本质是读取系统环境变量，只要提前完成.env的加载，就能让它正常获取到令牌。

具体实现步骤 #

1. 创建.env配置文件
   在项目根目录新建.env文件，写入你的API令牌：

HUGGINGFACEHUB_API_TOKEN=你的实际令牌值

2. 在你的代码中先加载.env，再调用第三方包
   直接在自己的业务代码里先执行godotenv.Load()，再调用第三方包的New()即可，示例代码如下：

package main

import (
    "log"

    "github.com/joho/godotenv"
    // 替换成你实际使用的第三方包导入路径
    "your/thirdparty/llm/package/path"
)

func main() {
    // 加载.env文件到系统环境变量
    if err := godotenv.Load(); err != nil {
        log.Fatalf("加载.env文件失败: %v", err)
    }

    // 调用第三方包的New()，此时环境变量已就绪
    llmClient, err := llm.New()
    if err != nil {
        log.Fatalf("创建LLM客户端失败: %v", err)
    }

    // 后续正常使用llmClient进行业务操作
    log.Println("LLM客户端初始化成功")
}

额外注意事项 #

• 加载顺序：必须保证godotenv.Load()在llm.New()之前执行，否则第三方包会读取不到环境变量
• 自定义.env路径：如果你的.env文件不在项目根目录，可以指定路径加载，比如godotenv.Load("./config/.env")
• 环境变量优先级：如果系统环境变量中已经存在同名的HUGGINGFACEHUB_API_TOKEN，godotenv.Load()不会覆盖它；如果需要强制覆盖，可以使用godotenv.Overload()替代Load()

内容的提问来源于stack exchange，提问作者Hasan A Yousef