A_Share_investment_Agent: 一款利用AI来辅助投资决策的系统

这是一个基于人工智能的投资系统概念验证项目。项目目标是探索如何使用 AI 来辅助投资决策。本项目仅用于教育目的，不适用于实际交易或投资。

系统由以下几个协同工作的 agent 组成：

  1. 1. Market Data Analyst - 负责收集和预处理市场数据

  2. 2. Valuation Agent - 计算股票内在价值并生成交易信号

  3. 3. Sentiment Agent - 分析市场情绪并生成交易信号

  4. 4. Fundamentals Agent - 分析基本面数据并生成交易信号

  5. 5. Technical Analyst - 分析技术指标并生成交易信号

  6. 6. Risk Manager - 计算风险指标并设置仓位限制

  7. 7. Portfolio Manager - 制定最终交易决策并生成订单

效果展示

基本运行

正在获取 xxx 的财务指标数据...
获取实时行情...
成功获取实时行情数据

获取利润表数据...
成功获取利润表数据
.........
Final Result:
{
  "action": "hold",
  "quantity": 0,
  "confidence": 0.4,
  "agent_signals": [
    {
      "agent": "Technical Analysis",
      "signal": "neutral",
      "confidence": 0.19
    },
 .......
  ],
  "reasoning": "Risk Management........"
}

回测功能

正在获取 301157 的历史行情数据...
开始日期：2024-12-08
结束日期：2025-01-07
警告：获取到的数据量（21条）不足以计算所有技术指标（需要至少120条）
尝试获取更长时间范围的数据...
成功获取历史行情数据，共 446 条记录

总收益率: -5.25%
INFO:backtest:
==================================================
INFO:backtest:回测结果汇总
INFO:backtest:==================================================
INFO:backtest:初始资金: 100,000.00
INFO:backtest:最终总值: 94,750.33
INFO:backtest:总收益率: -5.25%
INFO:backtest:夏普比率: -0.92
INFO:backtest:最大回撤: -19.14%

快速安装

克隆仓库

git clone https://github.com/quanzhankf/A_Share_investment_Agent.git
cd A_Share_investment_Agent

安装依赖

pip install -r requirements.txt

配置Gemini

# Create .env file for your API keys
cp .env.example .env

# Get your Gemini API key from https://aistudio.google.com/

运行项目

python src/main.py --ticker 301155

总结

项目其实很简单，但思路很好。 把股票交易数据和新闻信息发送给AI, 让AI来判断投资决策。

里面默认使用了Gemini，有兴趣的小伙伴可以改成DeepSeek啥的。 不会部署的朋友， 评论区留言代码，我把Final Result发您~

推荐阅读