跳到主要内容
Open In ColabOpen on GitHub

谷歌财经

本笔记介绍了如何使用 Google 财经工具从 Google 财经页面获取信息。

要获取 SerpApi 密钥,请在以下网址注册: https://serpapi.com/users/sign_up

要在 Langchain 中使用该工具,请安装以下软件包

%pip install --upgrade --quiet google-search-results langchain-community
Note: you may need to restart the kernel to use updated packages.

然后将环境变量 SERPAPI_API_KEY 设置为您的 SerpApi 密钥,或将其作为参数传递给包装器 serp_api_key="your secret key"。

import os

os.environ["SERPAPI_API_KEY"] = ""
from langchain_community.tools.google_finance import GoogleFinanceQueryRun
from langchain_community.utilities.google_finance import GoogleFinanceAPIWrapper

tool = GoogleFinanceQueryRun(api_wrapper=GoogleFinanceAPIWrapper())
tool.run("Google")
'\nQuery: Google\nstock: GOOGL:NASDAQ\nprice: $159.96\npercentage: 0.94\nmovement: Up\nus: price = 42210.57, movement = Down\neurope: price = 23638.56, movement = Up\nasia: price = 38183.26, movement = Up\n'

为了创建使用 Google 财经工具的 Agent,请安装 Langgraph

%pip install --upgrade --quiet langgraph langchain-openai
Note: you may need to restart the kernel to use updated packages.

并使用 create_react_agent 功能来初始化一个 ReAct Agent。您还需要设置您的 OPEN_API_KEY(访问 https://platform.openai.com),以便访问 OpenAI 的聊天模型。

import os

os.environ["OPENAI_API_KEY"] = ""
os.environ["SERP_API_KEY"] = ""
from langchain.chat_models import init_chat_model

llm = init_chat_model("gpt-4o-mini", model_provider="openai")
API 参考:init_chat_model
from langchain_community.agent_toolkits.load_tools import load_tools

tools = load_tools(["google-scholar", "google-finance"], llm=llm)
API 参考:load_tools
from langgraph.prebuilt import create_react_agent

agent = create_react_agent(llm, tools)

events = agent.stream(
{"messages": [("user", "What is Google's stock?")]},
stream_mode="values",
)
for event in events:
event["messages"][-1].pretty_print()
API 参考:create_react_agent
================================ Human Message =================================

What is Google's stock?
================================== Ai Message ==================================
Tool Calls:
google_finance (call_8m0txCtxNuQaAv9UlomPhSA1)
Call ID: call_8m0txCtxNuQaAv9UlomPhSA1
Args:
query: Google
================================= Tool Message =================================
Name: google_finance


Query: Google
stock: GOOGL:NASDAQ
price: $159.96
percentage: 0.94
movement: Up
us: price = 42210.57, movement = Down
europe: price = 23638.56, movement = Up
asia: price = 38183.26, movement = Up

================================== Ai Message ==================================

Google's stock, listed as GOOGL on NASDAQ, is currently priced at $159.96, with a movement up by 0.94%.