"Invalid constructor input for Tool" error when using function calling with gemini-pro

Question

I have the following code to enabling function calling with gemini-pro model (it is based on this example).

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


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

model._tools.to_proto()

For some reason, I received the following error:

TypeError: Invalid constructor input for Tool: <function getWordCount at 0x7a9baaf95c60>

The error repeats for all the following models:

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

I could not find any resources to resolve this issue. I appreciate any help.

Is this thread useful for your situation? stackoverflow.com/q/78090031 — Tanaike
– Tanaike, Commented Mar 22, 2024 at 1:38

Linda Lawton - DaImTo · Accepted Answer · 2024-03-21 21:02:31Z

1

This is how you declare a tool

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']
    )
  )
])

It's not just a python function

answered Mar 21, 2024 at 21:02

Linda Lawton - DaImTo

118k40 gold badges229 silver badges511 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

renakre Over a year ago

based on your example, so it will automatically understand that it needs to multiple a by b? there is no need to add the following return a * b?

Linda Lawton - DaImTo Over a year ago

No it will tell your application that it needs to run the code you have defined for the multiply function. I have a video on how to make a chatbot with functions if you're interested

Collectives™ on Stack Overflow

"Invalid constructor input for Tool" error when using function calling with gemini-pro

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related