Upload app.py

app.py

@@ -28,6 +28,19 @@ def format_value(value, value_type="money"):
     else:  # number
         return f"{value:.2f}"
 
+def normalize_cik(cik):
+    """
+    格式化 CIK 为标准的 10 位格式
+    """
+    if not cik:
+        return None
+    # 转换为字符串并移除非数字字符
+    cik_str = str(cik).replace('-', '').replace(' ', '')
+    # 仅保留数字
+    cik_str = ''.join(c for c in cik_str if c.isdigit())
+    # 填充前导 0 至 10 位
+    return cik_str.zfill(10) if cik_str else None
+
 # MCP 工具定义
 def create_mcp_tools():
     """创建 MCP 工具列表"""
@@ -128,7 +141,10 @@ def query_financial_data(company_name, query_type):
         result += f"**Stock Symbol**: {company.get('tickers', ['N/A'])[0] if company.get('tickers') else 'N/A'}\n"
         result += f"**Industry**: {company.get('sic_description', 'N/A')}\n\n---\n\n"
         
-        cik = company.get('cik')
+        # 获取并格式化 CIK 为 10 位标准格式
+        cik = normalize_cik(company.get('cik'))
+        if not cik:
+            return result + "❌ Error: Invalid CIK from company search"
         
         # 根据查询类型获取数据
         if internal_query_type == "最新财务数据":