langchain/templates/gemini-functions-agent
Erick Friis 3a2eb6e12b
infra: add print rule to ruff (#16221)
Added noqa for existing prints. Can slowly remove / will prevent more
being intro'd
2024-02-09 16:13:30 -08:00
..
gemini_functions_agent templates: simplify tool in gemini-functions-agent 2 (#17283) 2024-02-08 19:39:29 -08:00
tests templates: add gemini functions agent (#17141) 2024-02-07 17:27:01 -08:00
LICENSE templates: add gemini functions agent (#17141) 2024-02-07 17:27:01 -08:00
main.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
poetry.lock templates: gemini-functions-agent genai package bump (#17286) 2024-02-08 19:47:58 -08:00
pyproject.toml templates: gemini-functions-agent genai package bump (#17286) 2024-02-08 19:47:58 -08:00
README.md templates: gemini-functions-agent readme update (#17288) 2024-02-09 14:10:23 -08:00

gemini-functions-agent

This template creates an agent that uses Google Gemini function calling to communicate its decisions on what actions to take.

This example creates an agent that can optionally look up information on the internet using Tavily's search engine.

See an example LangSmith trace here

Environment Setup

The following environment variables need to be set:

Set the TAVILY_API_KEY environment variable to access Tavily

Set the GOOGLE_API_KEY environment variable to access the Google Gemini APIs.

Usage

To use this package, you should first have the LangChain CLI installed:

pip install -U langchain-cli

To create a new LangChain project and install this as the only package, you can do:

langchain app new my-app --package gemini-functions-agent

If you want to add this to an existing project, you can just run:

langchain app add gemini-functions-agent

And add the following code to your server.py file:

from gemini_functions_agent import agent_executor as gemini_functions_agent_chain

add_routes(app, gemini_functions_agent_chain, path="/openai-functions-agent")

(Optional) Let's now configure LangSmith. LangSmith will help us trace, monitor and debug LangChain applications. LangSmith is currently in private beta, you can sign up here. If you don't have access, you can skip this section

export LANGCHAIN_TRACING_V2=true
export LANGCHAIN_API_KEY=<your-api-key>
export LANGCHAIN_PROJECT=<your-project>  # if not specified, defaults to "default"

If you are inside this directory, then you can spin up a LangServe instance directly by:

langchain serve

This will start the FastAPI app with a server is running locally at http://localhost:8000

We can see all templates at http://127.0.0.1:8000/docs We can access the playground at http://127.0.0.1:8000/gemini-functions-agent/playground

We can access the template from code with:

from langserve.client import RemoteRunnable

runnable = RemoteRunnable("http://localhost:8000/gemini-functions-agent")