解锁 LangGraph和OpenAI，打造金融分析智能体

利用 LangChain、LangGraph 和 Yahoo Finance 打造股票绩效分析智能体。

微信搜索关注《AI科技论谈》

AI 赋能股票分析

在股票交易中，投资者要依靠各种工具和方法来做出明智决策。其中一种方法是基本面分析，它通过评估公司财务与股票表现提供操作建议。随着AI和机器学习的发展，股票分析如今在很大程度上可以实现自动化。

本文介绍如何用 LangChain、LangGraph 和 Yahoo Finance 构建股票绩效分析智能体，利用实时数据和关键指标分析。这个教程能帮你创建自己的智能体，下面就开始吧！

金融分析智能体的功能

• 使用 Yahoo Finance 获取股票价格数据。

• 对相对强弱指数（RSI）、指数平滑异同移动平均线（MACD）、成交量加权平均价格（VWAP）等技术指标进行运算。

• 评估财务指标，如市盈率、资产负债率和利润率。

• 运用 OpenAI 强大的语言模型给出结构化的人工智能分析结果。

所需工具

1. LangGraph：用于编排工具和构建对话智能体的库。

2. OpenAI GPT-4：用于生成智能且结构化的金融见解。

3. yfinance：用于获取股票价格和财务比率。

4. ta（技术分析库）：用于计算关键技术指标。

5. Python 库（pandas、dotenv、datetime）：用于数据处理和环境设置。

步骤 1：环境设置

首先安装所需的库：

pip install -U langgraph langchain langchain_openai pandas ta python-dotenv yfinance

接着，设置一个.env 文件，以便安全存储 OpenAI API 密钥：

OPENAI_API_KEY=你的 OpenAI API 密钥

步骤 2：分析工具

获取股票价格

此工具的作用是获取股票的历史数据并计算多个技术指标。

from typing import Union, Dict, Set, List, TypedDict, Annotated
import pandas as pd
from langchain_core.tools import tool
import yfinance as yf
from ta.momentum import RSIIndicator, StochasticOscillator
from ta.trend import SMAIndicator, EMAIndicator, MACD
from ta.volume import volume_weighted_average_price

@tool
def get_stock_prices(ticker: str) -> Union[Dict, str]:
    """获取指定股票代码的历史股票价格数据和技术指标"""
    try:
        data = yf.download(
            ticker,
            start=dt.datetime.now() - dt.timedelta(weeks=24*3),
            end=dt.datetime.now(),
            interval='1wk'
        )
        df= data.copy()
        data.reset_index(i