用FinancialDatasets Toolkit掌握金融数据分析

398 阅读2分钟

用FinancialDatasets Toolkit掌握金融数据分析

引言

在金融分析领域,获取准确的市场数据是至关重要的。FinancialDatasets Toolkit提供了一个强大的API,可以访问超过16,000种股票的30多年数据。本文将介绍如何利用这个工具包进行金融数据分析。

主要内容

安装和设置

要使用FinancialDatasets Toolkit,你需要两个API密钥:

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

设置环境变量以存储这些密钥:

import getpass
import os

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: ")

安装所需的Python包:

%pip install -qU langchain-community

初始化工具包

初始化FinancialDatasets Toolkit:

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)

使用工具

查看可用的工具:

tools = toolkit.get_tools()

在代理中使用

构建能够回答金融问题的代理:

from langchain_core.tools import tool
from langchain_openai import ChatOpenAI

model = ChatOpenAI(model="gpt-4o")  # 使用API代理服务提高访问稳定性

system_prompt = """
... # 此处省略长文本,可参考上文示例
"""

query = "What was AAPL's revenue in 2023? What about its total debt in Q1 2024?"

from langchain.agents import AgentExecutor, create_tool_calling_agent
from langchain_core.prompts import ChatPromptTemplate

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)

# 发起查询
agent_executor.invoke({"input": query})

常见问题和解决方案

网络限制

由于某些地区对API访问有限制,开发者可以考虑使用API代理服务来提高访问的稳定性和速度。

数据准确性

确保在使用API时,数据的时间戳和财务年度匹配,以获得准确的分析结果。

总结和进一步学习资源

本文介绍了如何使用FinancialDatasets Toolkit进行金融数据分析。若要进一步深入了解,可以参考以下资源:

参考资料

如果这篇文章对你有帮助,欢迎点赞并关注我的博客。您的支持是我持续创作的动力!

---END---