解锁金融数据的力量：使用FinancialDatasets Toolkit进行高级市场分析

本文链接：https://blog.csdn.net/afTFODguAKBF/article/details/143061151

解锁金融数据的力量：使用FinancialDatasets Toolkit进行高级市场分析

引言

在金融分析领域，获取准确和丰富的数据是成功的关键。FinancialDatasets Toolkit提供了一个强大的接口，可以访问超过16,000家公司的30多年金融数据。这篇文章将引导你如何设置和使用这个工具包，并为你提供实用的代码示例和最佳实践。

主要内容

1. 环境设置

要使用FinancialDatasets Toolkit，你需要两个API密钥：

FINANCIAL_DATASETS_API_KEY: 从financialdatasets.ai获取。
OPENAI_API_KEY: 从OpenAI获取。

import getpass
import os

# 设置API密钥
os.environ["FINANCIAL_DATASETS_API_KEY"] = getpass.getpass("Enter your financial datasets API key: ")
os.environ["OPENAI_API_KEY"] = getpass.getpass("Enter your OpenAI API key: ")

2. 安装工具包

工具包在langchain-community包中。

%pip install -qU langchain-community

3. 实例化工具包

创建FinancialDatasetsToolkit实例。

from langchain_community.agent_toolkits.financial_datasets.toolkit import FinancialDatasetsToolkit
from langchain_community.utilities.financial_datasets import FinancialDatasetsAPIWrapper

api_wrapper = FinancialDatasetsAPIWrapper(
    financial_datasets_api_key=os.environ["FINANCIAL_DATASETS_API_KEY"]
)
toolkit = FinancialDatasetsToolkit(api_wrapper=api_wrapper)

4. 使用工具进行财务分析

工具包提供了多个工具，例如财务报表分析。

# 查看可用工具
tools = toolkit.get_tools()

代码示例

以下示例展示如何用FinancialDatasetsToolkit分析公司的收入和债务数据。

from langchain_core.tools import tool
from langchain_openai import ChatOpenAI
from langchain.agents import AgentExecutor, create_tool_calling_agent
from langchain_core.prompts import ChatPromptTemplate

# 定义系统提示
system_prompt = """
你是一个高级财务分析AI助手，具备访问和分析金融数据的专业工具。
可以帮助用户进行财务分析，并检索公开交易公司的财务报表。
"""

# 实例化LLM
model = ChatOpenAI(model="gpt-4o")
prompt = ChatPromptTemplate.from_messages(
    [
        ("system", system_prompt),
        ("human", "{input}"),
        ("placeholder", "{agent_scratchpad}"),
    ]
)

# 创建代理
agent = create_tool_calling_agent(model, tools, prompt)
agent_executor = AgentExecutor(agent=agent, tools=tools)

# 查询代理
query = "What was AAPL's revenue in 2023? What about its total debt in Q1 2024?"
agent_executor.invoke({"input": query})