使用gemini-pro调用函数时出现“工具的构造函数输入无效”错误

Question

我有以下代码来启用gemini-pro模型的函数调用（它基于这个示例）。

def getWordCount(sentence:str):
    return len(sentence.split(' '))


model = genai.GenerativeModel(model_name='models/gemini-pro', tools=[getWordCount])

model._tools.to_proto()

由于某种原因，我收到以下错误：

TypeError：工具的构造函数输入无效：

以下所有型号都会重复出现此错误：

models/gemini-1.0-pro
models/gemini-1.0-pro-001
models/gemini-1.0-pro-latest
models/gemini-1.0-pro-vision-latest
models/gemini-pro
models/gemini-pro-vision

我找不到任何资源来解决此问题。我很感激任何帮助。

Answer 1

这就是声明工具的方式

calculator = glm.Tool(
function_declarations=[
  glm.FunctionDeclaration(
    name='multiply',
    description="Returns the product of two numbers.",
    parameters=glm.Schema(
        type=glm.Type.OBJECT,
        properties={
            'a':glm.Schema(type=glm.Type.NUMBER),
            'b':glm.Schema(type=glm.Type.NUMBER)
        },
        required=['a','b']
    )
  )
])

它不仅仅是一个Python函数