巧用 Python/Go 调用 Gemini CLI：提效 AI 开发的实用指南

在AI应用开发中，我们经常需要与各种模型进行交互。Google 的 Gemini 模型也提供了 CLI 工具，方便我们在命令行中进行快速测试和原型验证。本文将介绍如何使用 Python 和 Go 两种语言调用 Gemini CLI，提升开发效率。

问题背景与解决方案

在实际项目中，直接使用 Gemini CLI 命令字符串拼接的方式进行调用，存在可维护性差、容易出错等问题。我们需要一种更优雅、更高效的方式，将 Gemini CLI 集成到我们的 Python 或 Go 代码中。本文将分别介绍使用 Python 的 subprocess 模块和 Go 的 os/exec 包，封装 Gemini CLI 调用，实现更简洁的代码。

Python 调用 Gemini CLI

Python 作为数据科学和 AI 领域的热门语言，拥有丰富的库和工具。我们可以使用 subprocess 模块来调用 Gemini CLI。

import subprocess
import json

def run_gemini_cli(prompt):
    command = ["gemini", "--query", prompt, "--output", "json"]
    result = subprocess.run(command, capture_output=True, text=True)

    if result.returncode != 0:
        print(f"Error: {result.stderr}")
        return None

    try:
        return json.loads(result.stdout)
    except json.JSONDecodeError:
        print(f"Error decoding JSON: {result.stdout}")
        return None


# 示例用法
prompt = "介绍一下北京烤鸭的特点。"
response = run_gemini_cli(prompt)
if response:
    print(response)

这段代码首先定义了一个 run_gemini_cli 函数，该函数接受一个 prompt 作为输入，并构造 Gemini CLI 命令。然后，使用 subprocess.run 函数执行该命令，并捕获输出。如果命令执行成功，则将输出解析为 JSON 格式，并返回。如果命令执行失败，则打印错误信息，并返回 None。

实战避坑：

• 确保 Gemini CLI 已经正确安装并配置到环境变量中。
• subprocess.run 函数默认不会输出到控制台，需要设置 capture_output=True 来捕获输出。
• text=True 参数可以确保输出以文本格式返回，方便后续处理。
• 异常处理非常重要，确保能够处理 Gemini CLI 返回的错误信息。

Go 调用 Gemini CLI

Go 语言以其高性能和并发性，在后端开发中占据重要地位。我们可以使用 os/exec 包来调用 Gemini CLI。

package main

import (
    "fmt"
    "os/exec"
    "encoding/json"
)

func runGeminiCLI(prompt string) (map[string]interface{}, error) {
    cmd := exec.Command("gemini", "--query", prompt, "--output", "json")
    output, err := cmd.CombinedOutput()
    if err != nil {
        return nil, fmt.Errorf("error running Gemini CLI: %w, output: %s", err, string(output))
    }

    var result map[string]interface{}
    err = json.Unmarshal(output, &result)
    if err != nil {
        return nil, fmt.Errorf("error unmarshaling JSON: %w, output: %s", err, string(output))
    }

    return result, nil
}

func main() {
    prompt := "介绍一下上海的豫园。"
    response, err := runGeminiCLI(prompt)
    if err != nil {
        fmt.Println("Error:", err)
        return
    }
    fmt.Println(response)
}

这段代码定义了一个 runGeminiCLI 函数，它接受一个 prompt 作为输入，并使用 exec.Command 函数创建一个 Gemini CLI 命令。然后，使用 cmd.CombinedOutput 函数执行该命令，并捕获标准输出和标准错误。如果命令执行成功，则将输出解析为 JSON 格式，并返回。如果命令执行失败，则返回错误信息。

实战避坑：

• 同样需要确保 Gemini CLI 已经正确安装并配置到环境变量中。
• cmd.CombinedOutput 函数会同时捕获标准输出和标准错误，方便排查问题。
• 在处理 JSON 输出时，需要使用 json.Unmarshal 函数将字节数组转换为 Go 中的数据结构。
• Go 语言的错误处理机制非常重要，需要认真处理每个可能出错的地方。

总结

本文介绍了如何使用 Python 和 Go 两种语言调用 Gemini CLI。通过封装 Gemini CLI 调用，我们可以编写更简洁、更易于维护的代码。在实际项目中，我们可以根据具体的需求选择合适的语言和方法。同时，也需要注意异常处理和错误排查，确保程序的稳定性和可靠性。另外，可以使用 Nginx 作为反向代理，对服务进行负载均衡，提高并发连接数，保障服务稳定性。也可以使用宝塔面板等工具来简化服务器管理。