GraphQL
GraphQL is a query language for APIs and a runtime for executing those queries against your data.
GraphQLprovides a complete and understandable description of the data in your API, gives clients the power to ask for exactly what they need and nothing more, makes it easier to evolve APIs over time, and enables powerful developer tools.
By including aBaseGraphQLTool in the list of tools provided to an Agent, you can grant your Agent the ability to query data from GraphQL APIs for any purposes you need.
This Jupyter Notebook demonstrates how to use theGraphQLAPIWrapper component with an Agent.
In this example, we'll be using the publicStar Wars GraphQL API available at the following endpoint:https://swapi-graphql.netlify.app/graphql .
First, you need to installhttpx andgql Python packages.
pip install httpx gql>/dev/null
Note: you may need to restart the kernel to use updated packages.
%pip install--upgrade--quiet langchain-community
Note: you may need to restart the kernel to use updated packages.
Now, let's create a BaseGraphQLTool instance with the specified Star Wars API endpoint and initialize an Agent with the tool.
import os
os.environ["OPENAI_API_KEY"]=""
from langchain_community.agent_toolkits.load_toolsimport load_tools
tools= load_tools(
["graphql"],
graphql_endpoint="https://swapi-graphql.netlify.app/graphql",
)
from langgraph.prebuiltimport create_react_agent
agent= create_react_agent("openai:gpt-4.1-mini", tools)
Now, we can use the Agent to run queries against the Star Wars GraphQL API. Let's ask the Agent to list all the Star Wars films and their release dates.
graphql_fields="""allFilms {
films {
title
director
releaseDate
speciesConnection {
species {
name
classification
homeworld {
name
}
}
}
}
}
"""
suffix="Search for the titles of all the stawars films stored in the graphql database that has this schema "
input_message={
"role":"user",
"content": suffix+ graphql_fields,
}
for stepin agent.stream(
{"messages":[input_message]},
stream_mode="values",
):
step["messages"][-1].pretty_print()
================================[1m Human Message [0m=================================
Search for the titles of all the stawars films stored in the graphql database that has this schema allFilms {
films {
title
director
releaseDate
speciesConnection {
species {
name
classification
homeworld {
name
}
}
}
}
}
==================================[1m Ai Message [0m==================================
Tool Calls:
query_graphql (call_tN5A0dBbfOMewuw8Yy13bYpW)
Call ID: call_tN5A0dBbfOMewuw8Yy13bYpW
Args:
tool_input: query { allFilms { films { title } } }
=================================[1m Tool Message [0m=================================
Name: query_graphql
"{\n \"allFilms\": {\n \"films\": [\n {\n \"title\": \"A New Hope\"\n },\n {\n \"title\": \"The Empire Strikes Back\"\n },\n {\n \"title\": \"Return of the Jedi\"\n },\n {\n \"title\": \"The Phantom Menace\"\n },\n {\n \"title\": \"Attack of the Clones\"\n },\n {\n \"title\": \"Revenge of the Sith\"\n }\n ]\n }\n}"
==================================[1m Ai Message [0m==================================
The titles of all the Star Wars films stored in the database are:
1. A New Hope
2. The Empire Strikes Back
3. Return of the Jedi
4. The Phantom Menace
5. Attack of the Clones
6. Revenge of the Sith
If you would like more information about any of these films, please let me know!
Related
- Toolconceptual guide
- Toolhow-to guides