雅虎财经模型上下文协议 (MCP) for LLaMA 3.2 3B

此仓库包含一个模型上下文协议 (MCP) 的实现，该协议将雅虎财经 API 与 LLaMA 3.2 3B 集成。 MCP 允许模型检索实时财务数据、股票信息和市场新闻，同时保留 LLaMA 的所有常规功能。

特性

• 直接在 LLaMA 3.2 3B 中获取实时股票价格
• 检索详细的公司信息（行业、部门、市值等）
• 获取具有可自定义时间段和间隔的历史股票数据
• 获取最新的市场新闻标题
• 无缝增强 LLaMA 的功能，而不会干扰非金融查询

要求

• Python 3.8+
• Ollama - 用于在本地运行 LLaMA 3.2 3B
• 在 Ollama 中拉取的 LLaMA 3.2 3B 模型
• Python 包：
  • ollama
  • yfinance
  • requests
  • pandas

安装

1. 从 https://ollama.ai/ 安装 Ollama

2. 使用 Ollama 拉取 LLaMA 3.2 3B 模型：

   ollama pull llama3.2:3b
   

3. 克隆此仓库：

   git clone <repository-url>
   cd mcp_yahoo_finance
   

4. 安装所需的 Python 依赖项：

   pip install -r requirements.txt
   

用法

运行带有 Ollama 集成的 MCP

要通过 Ollama 将雅虎财经 MCP 与 LLaMA 3.2 3B 一起使用：

python mcp_ollama_integration.py

这将启动一个交互式会话，您可以在其中：

• 提出将通过雅虎财经数据丰富的财务问题
• 提出任何其他问题，这些问题将由 LLaMA 3.2 3B 正常处理

示例查询

财务查询（通过雅虎财经数据增强）

• "苹果股票的当前价格是多少？"
• "告诉我关于特斯拉这家公司"
• "微软的股票在过去一个月表现如何？"
• "最新的市场新闻标题是什么？"
• "亚马逊的 52 周高点是多少？"
• "英伟达在哪个行业运营？"
• "标准普尔 500 指数今年表现如何？"

非财务查询（由 LLaMA 正常处理）

• "法国的首都是什么？"
• "解释量子计算"
• "写一首关于秋天的诗"
• "什么是勾股定理？"

使用不同的 LLaMA 模型

您可以使用 --model 参数指定不同的模型：

python mcp_ollama_integration.py --model llama3.2:8b

工作原理

1. MCP 分析每个用户查询以确定它是否与财务相关
2. 对于财务查询，它：
   • 识别要调用的相关财务函数（价格、信息、历史记录、新闻）
   • 通过 MCP 调用雅虎财经 API
   • 格式化实时数据并将其作为上下文提供给 LLaMA 3.2 3B
   • LLaMA 3.2 3B 提供包含财务数据的自然响应
3. 对于非财务查询，它会将它们直接传递给 LLaMA 3.2 3B，无需修改

这种方法无缝地增强了 LLaMA 的实时财务数据功能，同时保留了其所有原始功能。

高级用法

直接 API 函数

如果您想在代码中直接使用雅虎财经 MCP 函数：

from yahoo_finance_mcp import YahooFinanceMCP

# 初始化 MCP
mcp = YahooFinanceMCP()

# 获取股票价格
price_data = mcp.execute_function("get_stock_price", {"symbol": "AAPL"})

# 获取公司信息
company_data = mcp.execute_function("get_stock_info", {"symbol": "TSLA"})

# 获取历史数据
history_data = mcp.execute_function("get_stock_history", {"symbol": "MSFT", "period": "1mo"})

# 获取市场新闻
news_data = mcp.execute_function("get_market_news", {"limit": 5})

故障排除

• "Error connecting to Ollama"（连接到 Ollama 时出错）：确保已安装并运行 Ollama
• Company not found（未找到公司）：尝试使用官方股票代码而不是公司名称
• LLaMA 3.2 3B model not found（未找到 LLaMA 3.2 3B 模型）：运行 ollama pull llama3.2:3b 下载模型

致谢

本项目使用 yfinance 库检索雅虎财经数据，并使用 Ollama 在本地运行 LLaMA 3.2 3B。