Skip to main content
Open In Colab在 GitHub 上打开

谷歌财经

此笔记本介绍了如何使用 Google Finance Tool 从 Google Finance 页面获取信息。

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

要将该工具与 Langchain 一起使用,请安装以下软件包

%pip install --upgrade --quiet  google-search-results langchain-community

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

import os

from langchain_community.tools.google_finance import GoogleFinanceQueryRun
from langchain_community.utilities.google_finance import GoogleFinanceAPIWrapper

os.environ["SERPAPI_API_KEY"] = "[your serpapi key]"
tool = GoogleFinanceQueryRun(api_wrapper=GoogleFinanceAPIWrapper())
tool.run("Google")
'\nQuery: Google\nstock: GOOGL:NASDAQ\nprice: $161.96\npercentage: 1.68\nmovement: Up\n'

要创建使用 Google Finance 工具的代理,请安装 Langgraph

%pip install --upgrade --quiet langgraph langchain-openai

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

import os

from langchain.agents import load_tools
from langchain.chat_models import init_chat_model
from langgraph.prebuilt import create_react_agent

os.environ["OPENAI_API_KEY"] = "[your openai key]"
os.environ["SERP_API_KEY"] = "[your serpapi key]"

llm = init_chat_model("gpt-4o-mini", model_provider="openai")
tools = load_tools(["google-scholar", "google-finance"], llm=llm)
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()
================================ Human Message =================================

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


Query: Google
stock: GOOGL:NASDAQ
price: $161.96
percentage: 1.68
movement: Up

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

Google's stock (Ticker: GOOGL) is currently priced at $161.96, showing an increase of 1.68%.