Upload app.py

app.py

@@ -54,14 +54,17 @@ def execute_tool(tool_name, **kwargs):
         return query_financial_data(kwargs.get("company_name"), kwargs.get("query_type"))
     return f"Unknown tool: {tool_name}"
 # 创建超链接
-def create_source_link(source_form, cik=None):
-    """为Source Form创建SEC EDGAR超链接"""
-    if not source_form or source_form == 'N/A' or not cik:
+def create_source_link(source_form, source_url=None):
+    """为Source Form创建超链接，使用MCP后端返回的URL"""
+    if not source_form or source_form == 'N/A':
         return source_form
     
-    # 构建SEC EDGAR链接（这是一个示例，实际链接需要更多参数）
-    edgar_url = f"https://www.sec.gov/edgar/browse/?CIK={cik}"
-    return f"[{source_form}]({edgar_url})"
+    # 如果后端提供了URL，使用后端的URL
+    if source_url and source_url != 'N/A':
+        return f"[{source_form}]({source_url})"
+    
+    # 如果没有URL，只显示文本
+    return source_form
 
 def query_financial_data(company_name, query_type):
     """查询财务数据的主函数"""
@@ -139,7 +142,10 @@ def query_financial_data(company_name, query_type):
             result += f"- **Earnings Per Share**: {format_value(eps, False)}\n"
             result += f"- **Operating Expenses**: {format_value(opex)}\n"
             result += f"- **Operating Cash Flow**: {format_value(ocf)}\n"
-            result += f"- **Source Form**: {create_source_link(data.get('source_form', 'N/A'), cik)}\n"
+            # 使用后端返回的 source_url
+            source_form = data.get('source_form', 'N/A')
+            source_url = data.get('source_url', None)  # 从后端获取URL
+            result += f"- **Source Form**: {create_source_link(source_form, source_url)}\n"
         
         elif internal_query_type == "3年趋势":
             metrics_resp = requests.post(
@@ -165,6 +171,9 @@ def query_financial_data(company_name, query_type):
             # 显示所有数据（包括年度和季度）
             all_data = metrics.get('metrics', [])
             
+            # 按期间降序排序，确保显示最近的3年数据
+            all_data = sorted(all_data, key=lambda x: x.get('period', '0000'), reverse=True)
+            
             result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) | Source Form |\n"
             result += "|--------|-------------|----------------|-----|------------------------|-------------------------|-------------|\n"
             
@@ -176,11 +185,11 @@ def query_financial_data(company_name, query_type):
                 opex = (m.get('operating_expenses') or 0) / 1e9
                 ocf = (m.get('operating_cash_flow') or 0) / 1e9
                 source_form = m.get('source_form', 'N/A')
-                cik_val = m.get('cik') or cik
+                source_url = m.get('source_url', None)  # 从后端获取URL
                 
                 # 区分年度和季度
                 period_prefix = "FY" if 'Q' not in period else ""
-                source_link = create_source_link(source_form, cik_val)
+                source_link = create_source_link(source_form, source_url)
                 
                 result += f"| {period_prefix}{period} | {format_value(rev)} | {format_value(inc)} | {format_value(eps_val, False)} | {format_value(opex)} | {format_value(ocf)} | {source_link} |\n"
         
@@ -206,6 +215,9 @@ def query_financial_data(company_name, query_type):
             # 显示所有数据（包括年度和季度）
             all_data = metrics.get('metrics', [])
             
+            # 按期间降序排序，确保显示最近的5年数据
+            all_data = sorted(all_data, key=lambda x: x.get('period', '0000'), reverse=True)
+            
             result += f"## 5-Year Financial Trends ({metrics.get('count', 0)} periods)\n\n"
             result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) | Source Form |\n"
             result += "|--------|-------------|----------------|-----|------------------------|-------------------------|-------------|\n"
@@ -218,11 +230,11 @@ def query_financial_data(company_name, query_type):
                 opex = (m.get('operating_expenses') or 0) / 1e9
                 ocf = (m.get('operating_cash_flow') or 0) / 1e9
                 source_form = m.get('source_form', 'N/A')
-                cik_val = m.get('cik') or cik
+                source_url = m.get('source_url', None)  # 从后端获取URL
                 
                 # 区分年度和季度
                 period_prefix = "FY" if 'Q' not in period else ""
-                source_link = create_source_link(source_form, cik_val)
+                source_link = create_source_link(source_form, source_url)
                 
                 result += f"| {period_prefix}{period} | {format_value(rev)} | {format_value(inc)} | {format_value(eps_val, False)} | {format_value(opex)} | {format_value(ocf)} | {source_link} |\n"
         
@@ -257,14 +269,12 @@ def query_financial_data(company_name, query_type):
                 filing_date = filing.get('filing_date', 'N/A')
                 accession_num = filing.get('accession_number', 'N/A')
                 primary_doc = filing.get('primary_document', 'N/A')
+                filing_url = filing.get('filing_url', None)  # 从后端获取URL
                 
-                # 创建 SEC EDGAR 原始文档链接
-                if accession_num != 'N/A' and primary_doc != 'N/A':
-                    # 移除 accession_number 中的连字符以匹配 SEC URL 格式
-                    acc_no_clean = accession_num.replace('-', '')
-                    doc_url = f"https://www.sec.gov/Archives/edgar/data/{cik}/{acc_no_clean}/{primary_doc}"
-                    form_link = f"[{form_type}]({doc_url})"
-                    primary_doc_link = f"[{primary_doc}]({doc_url})"
+                # 使用后端返回的URL创建链接
+                if filing_url and filing_url != 'N/A':
+                    form_link = f"[{form_type}]({filing_url})"
+                    primary_doc_link = f"[{primary_doc}]({filing_url})"
                 else:
                     form_link = form_type
                     primary_doc_link = primary_doc