Skip to main content
Open In ColabOpen on GitHub

GraphQL

GraphQL 是一种用于API的查询语言，也是一个用于针对您的数据执行这些查询的运行时。GraphQL 提供了对API中数据的完整且易于理解的描述，赋予客户端精确请求所需内容的能力，使API随着时间的推移更容易演进，并支持强大的开发者工具。

通过在提供给代理的工具列表中加入BaseGraphQLTool，您可以授予您的代理从GraphQL API查询数据的能力，以满足您的任何需求。

这个 Jupyter Notebook 演示了如何将 GraphQLAPIWrapper 组件与 Agent 一起使用。

在这个例子中，我们将使用公开的Star Wars GraphQL API，该API可在以下端点获取：https://swapi-graphql.netlify.app/.netlify/functions/index。

首先，你需要安装 httpx 和 gql Python 包。

pip install httpx gql > /dev/null
%pip install --upgrade --quiet  langchain-community

现在，让我们使用指定的星球大战API端点创建一个BaseGraphQLTool实例，并使用该工具初始化一个代理。

from langchain.agents import AgentType, initialize_agent, load_tools
from langchain_openai import OpenAI

llm = OpenAI(temperature=0)

tools = load_tools(
    ["graphql"],
    graphql_endpoint="https://swapi-graphql.netlify.app/.netlify/functions/index",
)

agent = initialize_agent(
    tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True
)

现在，我们可以使用Agent来对Star Wars GraphQL API运行查询。让我们要求Agent列出所有星球大战电影及其上映日期。

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 "


agent.run(suffix + graphql_fields)


> Entering new AgentExecutor chain...
 I need to query the graphql database to get the titles of all the star wars films
Action: query_graphql
Action Input: query { allFilms { films { title } } }
Observation: "{\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}"
Thought: I now know the titles of all the star wars films
Final Answer: The titles of all the star wars films are: A New Hope, The Empire Strikes Back, Return of the Jedi, The Phantom Menace, Attack of the Clones, and Revenge of the Sith.

> Finished chain.
'The titles of all the star wars films are: A New Hope, The Empire Strikes Back, Return of the Jedi, The Phantom Menace, Attack of the Clones, and Revenge of the Sith.'

这个页面有帮助吗？