如何使用google的gemini AI设置型号名称？

Question

我在设置型号名称时犯了一个错误，并且无法弄清楚。 我正在使用node.js。

我的代码：

const { TextServiceClient } = require("@google-ai/generativelanguage").v1beta2;
const { GoogleAuth } = require("google-auth-library");

  const aiClient = new TextServiceClient({
    authClient: new GoogleAuth().fromAPIKey(PALM_API_KEY),
  });

  const model = "models/gemini-1.0-pro";

  aiClient
    .generateText({
      model,
      prompt: {
        text: 'somePrompt',
      },
    })
    .then((res) => {
      //do stuff with the response
    })

输出

  Error: Error: 3 INVALID_ARGUMENT: * GenerateTextRequest.model: unexpected model name format


  code: 3,
  details: '* GenerateTextRequest.model: unexpected model name format\n',
  metadata: Metadata { internalRepr: Map(0) {}, options: {} },
  note: 'Exception occurred in retry method that was not classified as transient'

文档 https://www.npmjs.com/package/@google-ai/generativelanguage https://ai.google.dev/gemini-api/docs/models/gemini#gemini-1.0-pro

我已经尝试了文档中列出的每个版本，但最终出现了相同的错误。我做错了什么？

Answer 1

您没有使用正确的nodejs模型。

您可以按照此 Gemini API nodejs 快速入门

中的说明进行操作

但基本上：

1. 正确的SDK包是

   @google/generative-ai

2. 然后在您的代码中您将有以下声明，您可以在其中选择所需的模型：

const model = genAI.getGenerativeModel({ model: "the-model-name-here" });

希望有帮助。