added support for tools on VertexAI

[lkuligin]

• Description: added support for tools on VertexAI
• Twitter handle: lkuligin

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

1 Ignored Deployment

Name	Status	Preview	Comments	Updated (UTC)
langchain	⬜️ Ignored (Inspect)	Visit Preview		Dec 18, 2023 4:46pm

[hwchase17]

i tried this out and it errors, i think you need to remove title from the json schema that is generated

[lkuligin]

@hwchase17 which error do you get?

I've just tried this again

raw_tools = [
    Tool(
        name="Search",
        func=search.run,
        description="useful for when you need to answer questions about current events. You should ask targeted questions",
    ),
    Tool(
        name="Calculator",
        func=llm_math_chain.run,
        description="useful for when you need to answer questions about math",
    )
]
tools = [format_tool_to_vertex_tool(tool) for tool in raw_tools]

and it seems to be working fine.

and there's also a working integration test:

langchain/libs/community/tests/integration_tests/chat_models/test_vertexai.py

Line 299 in 45f2fd5

def test_tools() -> None:

[holtskinner]

I like creating a BaseFunctionsAgentOutputParser that both of these fit into, but I think this class and the VertexAIFunctionsAgentOutputParser should be in a separate file outside of the OpenAI parser

[efriis]

Adding to vertex partner package in #15822