erlangshen 0.1.0

package/knowledge/trading_strategies.md

@@ -0,0 +1,114 @@
+# 交易策略库
+
+## 趋势跟踪策略
+
+### 均线策略
+
+#### 双均线策略
+- **参数**: MA20 + MA60
+- **信号**: 
+  - 金叉 (MA20上穿MA60) → 买入
+  - 死叉 (MA20下穿MA60) → 卖出
+- **周期**: 日线为主
+
+#### 多均线策略
+- **参数**: MA20, MA60, MA120, MA200
+- **信号**: 价格站上所有均线且均线多头排列 → 买入
+
+### MACD策略
+- **参数**: 12, 26, 9
+- **信号**:
+  - MACD线金叉Signal线 → 买入
+  - MACD线死叉Signal线 → 卖出
+  - MACD柱由负转正 → 买入
+
+### 布林带策略
+- **参数**: 20日均线, ±2标准差
+- **信号**:
+  - 价格触及下轨 → 买入
+  - 价格触及上轨 → 卖出
+  - 布林带收口 → 突破方向确认
+
+## 均值回归策略
+
+### RSI策略
+- **参数**: 14日RSI
+- **信号**:
+  - RSI<30 → 超卖, 买入信号
+  - RSI>70 → 超买, 卖出信号
+  - RSI<20 → 极度超卖
+
+### KD指标策略
+- **参数**: 9, 3, 3
+- **信号**:
+  - K值<20且金叉D值 → 买入
+  - K值>80且死叉D值 → 卖出
+
+### 布林带均值回归
+- **参数**: 20日均线, ±2标准差
+- **信号**:
+  - 价格触及下轨 → 买入
+  - 价格触及上轨 → 卖出
+  - 中轨作为止损参考
+
+## 动量策略
+
+### 动量因子
+- **定义**: 过去N日收益率
+- **信号**: 买入动量最强, 卖出动量最弱
+- **周期**: 1-12个月
+
+### 相对强弱策略
+- **计算**: 个股收益 - 市场收益
+- **信号**: 相对强弱高于阈值 → 买入
+
+## 套利策略
+
+### 跨期套利
+- **标的**: 期货不同到期合约
+- **信号**: 价差偏离历史均值 → 买入低, 卖出高
+
+### 跨市场套利
+- **标的**: 同一资产不同市场
+- **标的**: A/H股, ETF/成分股
+
+### 期现套利
+- **标的**: 期货 vs 现货指数
+- **信号**: 基差 > 持有成本 → 买入期货
+
+## 事件驱动策略
+
+### 美股财报交易
+- **方法**:
+  - 财报前: 隐含波动率高, 卖出期权
+  - 财报后: 方向确定后买入
+- **风险**: 财报意外
+
+### 宏观事件策略
+- **事件**: 央行会议, GDP发布, 非农
+- **方法**: 事件前减仓, 事件后顺势
+
+## 风控原则
+
+### 仓位管理
+| 方法 | 描述 | 适用场景 |
+|------|------|----------|
+| 固定仓位 | 每笔固定金额 | 趋势跟踪 |
+| 等权仓位 | 各标的等权重 | 多策略 |
+| 风险平价 | 按波动率倒数分配 | 组合投资 |
+| 凯利公式 | f = (bp-q)/b | 优势明显时 |
+
+### 止损方法
+| 方法 | 描述 | 参数 |
+|------|------|------|
+| 固定止损 | 价格跌破固定比例 | 5%-10% |
+| 跟踪止损 | 最高点回落固定比例 | 5%-20% |
+| 时间止损 | 持仓超过N天 | 5-20日 |
+| 波动率止损 | ATR倍数止损 | 2-3倍ATR |
+
+### 风险指标
+- **最大回撤**: 组合历史最大亏损
+- **夏普比率**: (收益-无风险)/波动率
+- **卡玛比率**: 年化收益/最大回撤
+- **胜率**: 盈利交易占比
+- **盈亏比**: 平均盈利/平均亏损

package/package.json

@@ -0,0 +1,42 @@
+{
+  "name": "erlangshen",
+  "version": "0.1.0",
+  "description": "二郎神 - 全知全能的AI投资智能体",
+  "keywords": [
+    "investment",
+    "ai-agent",
+    "finance",
+    "trading",
+    "macro",
+    "portfolio"
+  ],
+  "author": "xiaoerdata",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/xiaoerdata/xiaoer-erlangshen.git"
+  },
+  "homepage": "https://github.com/xiaoerdata/xiaoer-erlangshen",
+  "engines": {
+    "node": ">=16.0.0",
+    "python": ">=3.9"
+  },
+  "bin": {
+    "erlangshen": "./bin/cli.js"
+  },
+  "scripts": {
+    "postinstall": "node scripts/postinstall.js",
+    "preuninstall": "node scripts/preuninstall.js"
+  },
+  "os": [
+    "darwin",
+    "linux",
+    "win32"
+  ],
+  "dependencies": {
+    "chalk": "^4.1.2",
+    "ora": "^5.4.1",
+    "prompts": "^2.4.2",
+    "boxen": "^5.1.2"
+  }
+}

package/requirements.txt

@@ -0,0 +1,14 @@
+openai>=1.0.0
+pydantic>=2.0.0
+psycopg2-binary>=2.9.0
+lark-oapi>=1.0.0
+rich>=13.0.0
+aiohttp>=3.9.0
+asyncio
+beautifulsoup4>=4.12.0
+lxml>=4.9.0
+loguru>=0.7.0
+python-dateutil>=2.8.0
+httpx>=0.25.0
+fastapi>=0.104.0
+uvicorn>=0.24.0

package/scripts/postinstall.js

@@ -0,0 +1,188 @@
+#!/usr/bin/env node
+
+/**
+ * postinstall.js - npm 安装后脚本
+ * 自动安装 Python 依赖
+ */
+
+const { spawn, execSync } = require('child_process');
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+const installDir = path.dirname(require.main.filename);
+const projectRoot = installDir;
+
+console.log('📦 二郎神安装中...');
+console.log(`   安装目录: ${installDir}`);
+
+// 检查 Python 版本
+function checkPython() {
+  console.log('\n🔍 检查 Python 环境...');
+  
+  try {
+    const python = os.platform() === 'win32' ? 'python' : 'python3';
+    const version = execSync(`${python} --version`, { encoding: 'utf-8' }).trim();
+    console.log(`   ✓ ${version}`);
+    
+    // 检查版本号
+    const match = version.match(/Python (\d+)\.(\d+)/);
+    if (match) {
+      const major = parseInt(match[1]);
+      const minor = parseInt(match[2]);
+      
+      if (major < 3 || (major === 3 && minor < 9)) {
+        console.error('   ✗ Python 3.9+  required');
+        process.exit(1);
+      }
+    }
+    
+    return python;
+  } catch (e) {
+    console.error('   ✗ Python 未安装');
+    console.error('\n请先安装 Python 3.9+:');
+    console.error('   macOS: brew install python3');
+    console.error('   Linux: sudo apt install python3 python3-pip');
+    console.error('   Windows: https://www.python.org/downloads/');
+    process.exit(1);
+  }
+}
+
+// 安装 Python 依赖
+function installDependencies(python) {
+  console.log('\n📚 安装 Python 依赖...');
+  
+  const requirementsPath = path.join(projectRoot, 'requirements.txt');
+  
+  if (!fs.existsSync(requirementsPath)) {
+    console.log('   ⚠ requirements.txt not found, skipping');
+    return;
+  }
+  
+  return new Promise((resolve, reject) => {
+    const pip = os.platform() === 'win32' ? 'pip' : 'pip3';
+    
+    // 使用 pip install -r requirements.txt
+    const proc = spawn(python, ['-m', pip, 'install', '-r', requirementsPath, '--quiet'], {
+      stdio: 'inherit',
+      cwd: projectRoot
+    });
+    
+    proc.on('close', (code) => {
+      if (code === 0) {
+        console.log('   ✓ 依赖安装完成');
+        resolve();
+      } else {
+        console.error('   ✗ 依赖安装失败');
+        reject(new Error(`pip exited with code ${code}`));
+      }
+    });
+    
+    proc.on('error', reject);
+  });
+}
+
+// 创建配置目录
+function createConfigDir() {
+  console.log('\n⚙️  初始化配置目录...');
+  
+  const configDir = os.homedir();
+  const erlangshenDir = path.join(configDir, '.erlangshen');
+  
+  if (!fs.existsSync(erlangshenDir)) {
+    fs.mkdirSync(erlangshenDir, { recursive: true });
+    console.log(`   ✓ 创建配置目录: ${erlangshenDir}`);
+  } else {
+    console.log(`   ✓ 配置目录已存在: ${erlangshenDir}`);
+  }
+  
+  // 创建示例配置
+  const configPath = path.join(erlangshenDir, 'config.toml');
+  if (!fs.existsSync(configPath)) {
+    const configTemplate = `# 二郎神配置文件
+# https://github.com/xiaoerdata/xiaoer-erlangshen
+
+# DeepSeek API Key (必需)
+deepseek_api_key = ""
+
+# 数据库配置 (可选，用于连接远程行情数据库)
+[database]
+host = ""
+port = 3306
+user = ""
+password = ""
+
+# 飞书配置 (可选，用于消息推送)
+[feishu]
+app_id = ""
+app_secret = ""
+
+# 代理配置 (可选)
+[proxy]
+enabled = false
+http = ""
+https = ""
+`;
+    
+    fs.writeFileSync(configPath, configTemplate);
+    console.log(`   ✓ 创建配置文件: ${configPath}`);
+  }
+}
+
+// 验证安装
+function verifyInstall(python) {
+  console.log('\n✅ 验证安装...');
+  
+  try {
+    // 测试导入二郎神
+    const testCode = `
+import sys
+sys.path.insert(0, '${projectRoot.replace(/\\/g, '\\\\')}')
+from src.core.brain import Brain
+from src.core.investment_universe import get_universe
+print('ok')
+`;
+    
+    const result = execSync(python, ['-c', testCode], { encoding: 'utf-8' }).trim();
+    
+    if (result === 'ok') {
+      console.log('   ✓ 二郎神验证通过');
+      return true;
+    }
+  } catch (e) {
+    console.log('   ⚠ 验证时出现警告 (可能缺少 API Key)');
+  }
+  
+  return false;
+}
+
+// 主函数
+async function main() {
+  console.log('\n========================================');
+  console.log('   二郎神 v0.1.0 - AI投资智能体');
+  console.log('   https://github.com/xiaoerdata/xiaoer-erlangshen');
+  console.log('========================================\n');
+  
+  try {
+    const python = checkPython();
+    await installDependencies(python);
+    createConfigDir();
+    verifyInstall(python);
+    
+    console.log('\n========================================');
+    console.log('   ✓ 安装完成!');
+    console.log('========================================');
+    console.log('\n📖 使用方法:');
+    console.log('   erlangshen --cli          # 启动CLI');
+    console.log('   erlangshen --api           # 启动API服务');
+    console.log('\n⚠️  首次使用请配置 API Key:');
+    console.log('   vim ~/.erlangshen/config.toml');
+    console.log('');
+    
+  } catch (e) {
+    console.error('\n❌ 安装失败:', e.message);
+    process.exit(1);
+  }
+}
+
+main();

package/scripts/preuninstall.js

@@ -0,0 +1,22 @@
+#!/usr/bin/env node
+
+/**
+ * preuninstall.js - npm 卸载前脚本
+ * 清理配置（可选）
+ */
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+const configDir = path.join(os.homedir(), '.erlangshen');
+
+console.log('\n🧹 二郎神卸载中...');
+
+// 检查是否要保留配置
+if (process.env.ERLANG_SHEN_KEEP_CONFIG !== 'false') {
+  console.log('\n⚠️  配置目录保留在:', configDir);
+  console.log('   如需完全删除，请运行: rm -rf ~/.erlangshen');
+}
+
+console.log('\n👋 感谢使用二郎神，期待再见!\n');

package/src/__init__.py

@@ -0,0 +1,4 @@
+"""
+二郎神 - 投资分析智能体
+"""
+__version__ = "0.1.0"

package/src/agents/__init__.py

@@ -0,0 +1,3 @@
+"""
+Agents 模块
+"""

package/src/agents/base.py

@@ -0,0 +1,103 @@
+"""
+基础 Agent 类
+"""
+
+from abc import ABC, abstractmethod
+from typing import Any, Dict, Optional
+from src.brain import Brain
+from src.mcp.registry import MCPRegistry
+
+
+class BaseAgent(ABC):
+    """
+    基础 Agent 抽象类
+    
+    所有专业 Agent 的基类
+    """
+    
+    def __init__(self, brain: Brain, mcp: MCPRegistry):
+        self.brain = brain
+        self.mcp = mcp
+        self.name = self.__class__.__name__
+        self.role = "分析师"
+        self.system_prompt = self._build_system_prompt()
+    
+    @abstractmethod
+    def _build_system_prompt(self) -> str:
+        """构建系统提示"""
+        pass
+    
+    @abstractmethod
+    async def process(self, query: str, **kwargs) -> str:
+        """
+        处理查询
+        
+        Args:
+            query: 用户查询
+            **kwargs: 其他参数
+        
+        Returns:
+            处理结果
+        """
+        pass
+    
+    async def think(
+        self,
+        prompt: str,
+        temperature: Optional[float] = None
+    ) -> str:
+        """
+        统一思考接口
+        
+        Args:
+            prompt: 思考提示
+            temperature: 温度参数
+        
+        Returns:
+            思考结果
+        """
+        return await self.brain.think(
+            prompt=prompt,
+            system=self.system_prompt,
+            temperature=temperature
+        )
+    
+    async def analyze(
+        self,
+        query: str,
+        framework: Optional[str] = None,
+        data: Optional[Dict[str, Any]] = None
+    ) -> str:
+        """
+        统一分析接口
+        
+        Args:
+            query: 分析查询
+            framework: 分析框架
+            data: 相关数据
+        
+        Returns:
+            分析结果
+        """
+        return await self.brain.analyze(
+            query=query,
+            framework=framework,
+            data=data
+        )
+    
+    async def call_mcp_tool(self, tool_name: str, **kwargs) -> Any:
+        """
+        调用 MCP 工具
+        
+        Args:
+            tool_name: 工具名称
+            **kwargs: 工具参数
+        
+        Returns:
+            工具执行结果
+        """
+        return await self.mcp.call_tool(tool_name, **kwargs)
+    
+    def get_available_tools(self) -> list:
+        """获取可用工具列表"""
+        return self.mcp.list_tools()

package/src/agents/base_agent.py

@@ -0,0 +1,86 @@
+"""
+Base Agent - 基础智能体
+所有智能体的基类
+"""
+from abc import ABC, abstractmethod
+from typing import Optional, Any
+from loguru import logger
+
+
+class BaseAgent(ABC):
+    """
+    基础智能体抽象类
+
+    提供：
+    - 名称和描述
+    - 工具绑定
+    - 消息处理
+    - 日志记录
+    """
+
+    def __init__(
+        self,
+        name: str,
+        description: str = "",
+        tools: Optional[dict[str, Any]] = None,
+    ):
+        self.name = name
+        self.description = description
+        self.tools = tools or {}
+        logger.info(f"Agent '{name}' initialized")
+
+    def bind_tool(self, name: str, tool: Any) -> None:
+        """绑定工具"""
+        self.tools[name] = tool
+
+    async def call_tool(self, name: str, **kwargs) -> Any:
+        """调用工具 - 支持 dotted 格式如 'market_tools.get_stock_price'"""
+        # 支持 dotted 格式：tool_instance.method_name
+        if "." in name:
+            parts = name.split(".", 1)
+            tool_key = parts[0]
+            method_name = parts[1]
+            
+            if tool_key not in self.tools:
+                return {"error": f"Tool '{tool_key}' not found"}
+            
+            tool = self.tools[tool_key]
+            method = getattr(tool, method_name, None)
+            if method is None:
+                return {"error": f"Method '{method_name}' not found on tool '{tool_key}'"}
+            if callable(method):
+                return await method(**kwargs)
+            return {"error": f"Method '{method_name}' on '{tool_key}' is not callable"}
+        
+        # 直接格式
+        if name not in self.tools:
+            return {"error": f"Tool '{name}' not found"}
+
+        tool = self.tools[name]
+        if hasattr(tool, "execute"):
+            return await tool.execute(**kwargs)
+        elif callable(tool):
+            return await tool(**kwargs)
+        return {"error": f"Tool '{name}' is not callable"}
+
+    @abstractmethod
+    async def process(self, query: str, context: Optional[dict] = None) -> dict:
+        """
+        处理查询 - 子类必须实现
+
+        Args:
+            query: 输入查询
+            context: 上下文
+
+        Returns:
+            dict 处理结果
+        """
+        pass
+
+    def get_info(self) -> dict:
+        """获取智能体信息"""
+        return {
+            "name": self.name,
+            "description": self.description,
+            "tools": list(self.tools.keys()),
+        }