提交

backend/app/screener/llm.py

@@ -23,7 +23,7 @@ SYSTEM_PROMPT = """你是 A 股选股条件解析器。把用户的自然语言
 
 【indicator 数组】技术指标条件，元素字段：
 - "indicator": 指标名，白名单：kdj_k / kdj_d / kdj_j（KDJ 的 K/D/J 值）、rsi、macd_dif / macd_dea / macd_hist（MACD 的 DIF/DEA/柱）、ma（收盘价均线）、boll_upper / boll_mid / boll_lower（布林轨道）、zhixing_dkx（知行多空线，四条收盘价均线的均值）、zhixing_trend（知行短期趋势线）、close（收盘价）、pct_chg（日涨跌幅%）
-- "params": 指标参数（可选），默认：KDJ {"n":9,"m1":3,"m2":3}；RSI {"period":14}；MACD {"fast":12,"slow":26,"signal":9}；MA {"period":20}；BOLL {"period":20,"std":2}；知行多空线 {"m1":14,"m2":28,"m3":57,"m4":114}
+- "params": 指标参数（可选），默认：KDJ {"n":9,"m1":3,"m2":3}；RSI {"period":14}；MACD {"fast":12,"slow":26,"signal":9}；MA {"period":20}；BOLL {"period":20,"std":2}；知行多空线 {"m1":14,"m2":28,"m3":57,"m4":114}；知行趋势线无参数（固定算法），不要给它填 params
 - "op": "gt" | "ge" | "lt" | "le" | "between"
 - "value": 比较数值（between 时为下界），"value2": between 上界
 - "value_indicator": 可选。指标与指标比较时填另一指标名（同白名单），如 "DIF大于DEA" -> indicator=macd_dif, op=gt, value_indicator=macd_dea, value=0；"股价在布林带下轨之下" -> indicator=close, op=lt, value_indicator=boll_lower, value=0
@@ -57,7 +57,11 @@ SYSTEM_PROMPT = """你是 A 股选股条件解析器。把用户的自然语言
 
 示例4：
 输入：近一个月股价曾经站上知行多空线的股票
-输出：{"indicator":[{"indicator":"close","op":"gt","value":0,"value_indicator":"zhixing_dkx","params":{"m1":14,"m2":28,"m3":57,"m4":114},"lookback":20,"match":"any"}],"exclude_st":true,"exclude_delisted":true,"exclude_bj":true}"""
+输出：{"indicator":[{"indicator":"close","op":"gt","value":0,"value_indicator":"zhixing_dkx","params":{"m1":14,"m2":28,"m3":57,"m4":114},"lookback":20,"match":"any"}],"exclude_st":true,"exclude_delisted":true,"exclude_bj":true}
+
+示例5：
+输入：今天股价在知行趋势线上方的股票
+输出：{"indicator":[{"indicator":"close","op":"gt","value":0,"value_indicator":"zhixing_trend","lookback":1,"match":"all"}],"exclude_st":true,"exclude_delisted":true,"exclude_bj":true}"""
 
 
 class ScreenerError(RuntimeError):
@@ -68,10 +72,14 @@ def _endpoint(base_url: str) -> str:
     """归一化 base_url -> 完整 chat/completions URL。
 
     兼容多种写法：DeepSeek/OpenAI 的 .../v1、智谱 GLM 的 .../v4、或直接给完整路径。
+    .../anthropic（DeepSeek 的 Anthropic 协议端点）剥后缀走本模块的 OpenAI 兼容协议，
+    key 两端点通用。
     """
     base = base_url.rstrip("/")
     if base.endswith("/chat/completions"):
         return base
+    if base.endswith("/anthropic"):
+        base = base[: -len("/anthropic")]
     if not re.search(r"/v\d+$", base):   # 未带版本段则补 /v1（DeepSeek/OpenAI 惯例）
         base += "/v1"
     return f"{base}/chat/completions"
@@ -164,7 +172,7 @@ EVENT_SYSTEM_PROMPT = """你是 A 股事件回测参数解析器。用户描述
 
 【entry.indicator 数组】入场信号条件（必填，至少 1 条），元素字段与白名单：
 - "indicator": kdj_k / kdj_d / kdj_j（KDJ 的 K/D/J 值）、rsi、macd_dif / macd_dea / macd_hist（MACD 的 DIF/DEA/柱）、ma（收盘价均线）、boll_upper / boll_mid / boll_lower（布林轨道）、zhixing_dkx（知行多空线）、zhixing_trend（知行短期趋势线）、close（收盘价）、pct_chg（日涨跌幅%）
-- "params": 指标参数（可选），默认：KDJ {"n":9,"m1":3,"m2":3}；RSI {"period":14}；MACD {"fast":12,"slow":26,"signal":9}；MA {"period":20}；BOLL {"period":20,"std":2}；知行多空线 {"m1":14,"m2":28,"m3":57,"m4":114}
+- "params": 指标参数（可选），默认：KDJ {"n":9,"m1":3,"m2":3}；RSI {"period":14}；MACD {"fast":12,"slow":26,"signal":9}；MA {"period":20}；BOLL {"period":20,"std":2}；知行多空线 {"m1":14,"m2":28,"m3":57,"m4":114}；知行趋势线无参数（固定算法），不要给它填 params
 - "op": "gt" | "ge" | "lt" | "le" | "between"；"value"（between 时为下界）、"value2"（上界）
 - "value_indicator": 指标与指标比较时填另一指标名（同白名单），如 "DIF 大于 DEA" -> indicator=macd_dif, op=gt, value_indicator=macd_dea, value=0
 - "value_params": 比较对象指标参数不同时指定，如 "MA5 上穿 MA20" -> indicator=ma, params={"period":5}, op=gt, value_indicator=ma, value_params={"period":20}, value=0