Repository navigation
Expand file tree
/
Copy pathrouter.py
More file actions
1440 lines (1239 loc) · 56 KB
/
Copy pathrouter.py
File metadata and controls
1440 lines (1239 loc) · 56 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
#!/usr/bin/env python3
"""Org Discussion Router — LLM 意图分类 + JOIN / QA / OTHER 分派处理。
由 ``org-router.yml`` workflow 在收到 ``org_msg_router`` repository_dispatch
或手动 workflow_dispatch 时调用。
路由流程
────────
1. 解析 CLI 参数（discussion_number, comment_body, comment_author, category_name）
2. LLM 三分类：JOIN / QA / OTHER（失败时降级正则）
3. JOIN → 条件检测（关注组织 + Star 仓库）→ 通过则邀请
4. QA → Q&A 分类下用 CSM_QA 回答，否则引导用户去 Q&A 区
5. OTHER → 友好引导回复
"""
from __future__ import annotations
import argparse
import json
import logging
import os
import re
import sys
import time
import urllib.error
import urllib.request
from typing import Any, Optional
# ── 确保包根目录在 sys.path ─────────────────────────────────────────────────
_REPO_ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
if _REPO_ROOT not in sys.path:
sys.path.insert(0, _REPO_ROOT)
from scripts._utils import configure_logging, SKIP_AUTHORS # noqa: E402
from scripts._github import GitHubGraphQL # noqa: E402
logger = logging.getLogger("org_router")
# ── 常量 ────────────────────────────────────────────────────────────────────
GITHUB_API_URL = "https://api.github.com"
BOT_MARKER = "<!-- csm-qa-bot -->"
BOT_FOOTER = (
"\n\n---\n"
"> 🤖 此回复由 [CSM-QA-Robot](https://github.com/NEVSTOP-LAB/CSM-QA-Robot) 自动处理。"
)
# JOIN 条件：从 GitHub Actions vars 注入，逗号分隔
JOIN_FOLLOW_ORG = os.getenv("JOIN_FOLLOW_ORG", "NEVSTOP-LAB")
JOIN_STAR_REPOS = [
r.strip() for r in
os.getenv("JOIN_STAR_REPOS", "Communicable-State-Machine,CSM-API-String-Arguments-Support,CSM-MassData-Parameter-Support,CSM-INI-Static-Variable-Support").split(",")
if r.strip()
]
JOIN_STAR_OWNER = JOIN_FOLLOW_ORG # Star 仓库所属组织与关注组织一致
JOIN_DEFAULT_TEAM = os.getenv("JOIN_DEFAULT_TEAM", "csm-community")
# LLM 分类提示词
INTENT_CLASSIFY_PROMPT = """你是一个 GitHub 讨论区路由助手。请判断以下评论属于哪一类意图，只回复一个标签：
标签说明：
- JOIN：用户**本人**表达了加入组织的意愿（例如"我想加入"、"申请加入"、"希望能成为成员"、"想参与贡献"等，无需特定命令）。**注意**：建议/邀请/欢迎**他人**加入（如"你可以加入"、"欢迎加入我们"、"推荐你申请"）不属于 JOIN，应归类为 OTHER；用户以组织成员身份回复他人时也不属于 JOIN。
- QA：用户提出技术问题或框架使用问题（可能包含 join/加入 等技术术语）
- OTHER：其他与上述无关的评论
用户评论：
'''
{comment_body}
'''
只回复标签名（JOIN / QA / OTHER），不要任何解释。"""
# 带 thread 上下文的分类提示词：将整个讨论历史传给 LLM，使其能理解追问的语境
INTENT_CLASSIFY_PROMPT_WITH_CONTEXT = """你是一个 GitHub 讨论区路由助手。以下是该 Discussion 的完整对话历史（按时间顺序，user=用户, assistant=Bot），请根据上下文判断最后一条用户评论的意图，只回复一个标签：
标签说明：
- JOIN：用户**本人**表达了加入组织的意愿（例如"我想加入"、"申请加入"、"希望能成为成员"、"想参与贡献"等，无需特定命令）。**注意**：建议/邀请/欢迎**他人**加入（如"你可以加入"、"欢迎加入我们"、"推荐你申请"）不属于 JOIN，应归类为 OTHER；用户以组织成员身份回复他人时也不属于 JOIN。
- QA：用户提出技术问题或框架使用问题（可能包含 join/加入 等技术术语）
- OTHER：其他与上述无关的评论
对话历史：
---
{history_text}
---
最后一条用户评论（需要你判断意图）：
'''
{comment_body}
'''
请仅根据最后这条评论在整个对话中的上下文来判断其意图。
只回复标签名（JOIN / QA / OTHER），不要任何解释。"""
def _build_history_text(history: list[dict[str, str]]) -> str:
"""将 history 列表格式化为可读的对话文本。
限制策略：最多保留最近 20 条消息，且总字符数不超过 8000，
防止超长线程撑爆 LLM 上下文窗口。"""
# 最多保留最近 N 条消息（原帖首条始终保留）
MAX_ENTRIES = 20
if len(history) > MAX_ENTRIES:
# 保留首条（原帖）+ 最近 MAX_ENTRIES-1 条
history = history[:1] + history[-(MAX_ENTRIES - 1):]
MAX_CHARS = 8000
lines: list[str] = []
for entry in history:
role = entry.get("role", "")
if role == "user":
role_label = "用户"
elif role == "assistant":
role_label = "Bot"
else:
role_label = "用户" # 未知角色按用户处理
content = entry.get("content", "").strip()
if not content:
continue
# 截断每条消息
lines.append(f"[{role_label}]: {content[:500]}")
# 总长度限制
text = "\n\n".join(lines)
if len(text) > MAX_CHARS:
text = text[-MAX_CHARS:]
# 寻找第一个换行符后的位置作为自然截断点
first_nl = text.find("\n")
if first_nl >= 0 and first_nl < len(text) - 1:
text = text[first_nl + 1:]
text = "…[上文已省略]…\n\n" + text
return text
# DeepSeek API（兼容 OpenAI 格式）
LLM_API_BASE = os.getenv("LLM_API_BASE", "https://api.deepseek.com/v1")
LLM_API_KEY = os.getenv("LLM_API_KEY", "")
LLM_MODEL = os.getenv("LLM_MODEL", "deepseek-chat")
# QA 分类名（大小写精确匹配）
QA_CATEGORY_NAME = "Q&A"
# 降级正则：JOIN 匹配在 _fallback_classify 内按强弱分层
_RE_QA = re.compile(r"问|？|\?|怎么|如何|是什么|报错|bug|error|请教|求助", re.IGNORECASE)
def _repo_link(repo: str) -> str:
"""把仓库名转为 Markdown 链接。"""
return f"[{repo}](https://github.com/{JOIN_STAR_OWNER}/{repo})"
# App 安装信息（用于发送组织邀请，需要 org Members 写权限）
_GITHUB_APP_ID = os.getenv("GITHUB_APP_ID", "")
_GITHUB_APP_PRIVATE_KEY = os.getenv("GITHUB_APP_PRIVATE_KEY", "")
_GITHUB_APP_INSTALL_ID = os.getenv("GITHUB_APP_INSTALL_ID", "")
# ── REST helper ──────────────────────────────────────────────────────────────
# ── App Installation Token ──────────────────────────────────────────────────
def _get_app_installation_token(owner: str, repo: str) -> Optional[str]:
"""获取 GitHub App installation token（含 org Members 权限）。
邀请 API ``POST /orgs/{org}/invitations`` 需要 org:write 权限，
``CSM_QA_GH_TOKEN``（Fine-grained PAT）只有 discussions 权限，
因此需用 App 私钥签发 JWT 换取 installation token。
"""
app_id = os.getenv("GITHUB_APP_ID", "")
key_pem = os.getenv("GITHUB_APP_PRIVATE_KEY", "")
if not app_id or not key_pem:
return None
try:
import jwt
now = int(time.time())
jwt_token = jwt.encode(
{"iat": now - 60, "exp": now + 10 * 60, "iss": str(app_id)},
key_pem,
algorithm="RS256",
)
except Exception as exc:
logger.warning("App JWT 签发失败: %s", exc)
return None
try:
resp = _rest_req(jwt_token, "GET", f"/repos/{owner}/{repo}/installation")
install = json.loads(resp.read())
install_id = install.get("id")
if not install_id:
raise RuntimeError("未找到 installation id")
except Exception as exc:
logger.warning("获取 installation id 失败: %s", exc)
return None
try:
data = json.dumps({}).encode()
req = urllib.request.Request(
f"{GITHUB_API_URL}/app/installations/{install_id}/access_tokens",
data=data,
headers={
"Authorization": f"Bearer {jwt_token}",
"Accept": "application/vnd.github.v3+json",
"Content-Type": "application/json",
"User-Agent": "org-router/1.0",
},
method="POST",
)
with urllib.request.urlopen(req, timeout=15) as resp:
result = json.loads(resp.read())
token = result.get("token")
if token:
perms = result.get("permissions", {})
logger.info(
"App installation token 获取成功 (id=%s, permissions=%s)",
install_id, json.dumps(perms),
)
else:
logger.warning("installation token 响应为空: %s", json.dumps(result)[:300])
return token
except Exception as exc:
logger.warning("换取 installation token 失败: %s", exc)
return None
# ── REST helper ──────────────────────────────────────────────────────────────
def _rest_req(token: str, method: str, path: str) -> Any:
"""发送 GitHub REST API 请求，返回 HTTP response 对象。
Raises:
RuntimeError: 网络错误或超时。
urllib.error.HTTPError: HTTP 4xx/5xx。
"""
quoted = urllib.parse.quote(path, safe="/?:&=#")
url = f"{GITHUB_API_URL}{quoted}"
req = urllib.request.Request(
url,
method=method,
headers={
"Authorization": f"Bearer {token}",
"Accept": "application/vnd.github.v3+json",
"User-Agent": "org-router/1.0",
},
)
try:
return urllib.request.urlopen(req, timeout=15)
except urllib.error.HTTPError:
raise # 4xx/5xx 原样抛出，由 _check_* 解析 404 语义
except urllib.error.URLError as exc:
raise RuntimeError(f"REST 请求失败 {method} {path}: {exc}") from exc
# ── LLM 意图分类 ────────────────────────────────────────────────────────────
def classify_intent(comment_body: str, history: Optional[list[dict[str, str]]] = None) -> str:
"""对评论正文做 LLM 三分类，返回 ``"JOIN"`` / ``"QA"`` / ``"OTHER"``。
当提供 ``history``（整个 discussion 的对话历史）时，使用带上下文的 prompt，
帮助 LLM 更准确地理解追问类消息的意图。
失败时降级为正则匹配（强弱分层）：
- 强关键词（加入/申请加入/成为成员/想加入/参与贡献）→ JOIN
- 弱关键词 join + QA 模式共存（如 "怎么 join 数组"）→ QA
- 仅弱关键词 join → JOIN
- QA 关键词 → QA
- 其余 → OTHER
"""
text = comment_body.strip()
if not text:
return "OTHER"
if not LLM_API_KEY:
logger.warning("LLM_API_KEY 未配置，使用正则降级分类")
return _fallback_classify(text)
# 构建 LLM 消息：有历史上下文时使用带上下文的 prompt
# 先转义用户内容中的花括号，避免 str.format() KeyError
safe_text = text[:800].replace("{", "{{").replace("}", "}}")
try:
if history and len(history) > 0:
history_text = _build_history_text(history).replace("{", "{{").replace("}", "}}")
prompt = INTENT_CLASSIFY_PROMPT_WITH_CONTEXT.format(
history_text=history_text,
comment_body=safe_text,
)
else:
prompt = INTENT_CLASSIFY_PROMPT.format(comment_body=safe_text)
payload = json.dumps({
"model": LLM_MODEL,
"messages": [
{"role": "user", "content": prompt}
],
"temperature": 0,
"max_tokens": 10,
}).encode()
req = urllib.request.Request(
f"{LLM_API_BASE}/chat/completions",
data=payload,
headers={
"Authorization": f"Bearer {LLM_API_KEY}",
"Content-Type": "application/json",
},
method="POST",
)
with urllib.request.urlopen(req, timeout=15) as resp:
result = json.loads(resp.read())
content = result.get("choices", [{}])[0].get("message", {}).get("content", "")
label = content.strip().upper()
for tag in ("JOIN", "QA", "OTHER"):
if tag in label:
logger.info("LLM 分类结果: %s", tag)
return tag
logger.warning("LLM 返回无法解析: %r，降级正则分类", label)
except Exception as exc:
logger.warning("LLM 调用失败: %s，降级正则分类", exc)
return _fallback_classify(text)
def _fallback_classify(text: str) -> str:
# 第二人称建议加入模式（建议/邀请他人加入，不是评论者自己要加入）
# re.DOTALL：用户可能用换行分隔"你可以"和"加入"，跨行匹配避免漏判
_RE_SUGGEST_JOIN = re.compile(
r"(你可以|建议你|推荐你|欢迎你|欢迎|邀请你|你试试|你去|你能够)"
r".{0,15}?"
r"(加入|申请|参与|成为成员)",
re.IGNORECASE | re.DOTALL,
)
if _RE_SUGGEST_JOIN.search(text):
logger.info("正则降级分类: OTHER（第二人称建议加入模式）")
return "OTHER"
# 强 JOIN 模式（中文明确表达加入意图）
_RE_JOIN_STRONG = re.compile(r"加入|申请加入|成为成员|想加入|申请成为|参与贡献", re.IGNORECASE)
has_strong_join = _RE_JOIN_STRONG.search(text)
has_qa = _RE_QA.search(text)
has_weak_join = re.search(r"\bjoin\b", text, re.IGNORECASE) if not has_strong_join else None
if has_strong_join:
logger.info("正则降级分类: JOIN（强关键词）")
return "JOIN"
if has_qa and has_weak_join:
# "LabVIEW 怎么 join 数组" → QA 优先于弱 join
logger.info("正则降级分类: QA（弱 join + QA 模式共存）")
return "QA"
if has_weak_join:
logger.info("正则降级分类: JOIN（弱 join）")
return "JOIN"
if has_qa:
logger.info("正则降级分类: QA")
return "QA"
logger.info("正则降级分类: OTHER")
return "OTHER"
# ── JOIN 逻辑 ────────────────────────────────────────────────────────────────
def _check_following(token: str, username: str, org: str) -> tuple[bool, str]:
"""检查用户是否已关注组织。返回 ``(passed, detail)``。"""
try:
resp = _rest_req(token, "GET", f"/users/{username}/following/{org}")
# 请求成功（未抛 HTTPError）→ 已关注
return True, f"已关注 @{org}"
except urllib.error.HTTPError as exc:
if exc.code == 404:
return False, f"未关注 @{org}"
raise RuntimeError(f"检查关注失败: HTTP {exc.code}") from exc
def _is_org_member(token: str, org: str, username: str) -> bool:
"""通过 REST API 检查用户是否已在组织内。
App installation token 需要 org Members:Read 权限。
204 = 是成员，404 = 非成员，其他状态码 = 查询失败返回 False。
"""
try:
_rest_req(token, "GET", f"/orgs/{org}/members/{username}")
logger.info("_is_org_member: %s 已在 %s 组织内 (204)", username, org)
return True
except urllib.error.HTTPError as exc:
if exc.code == 404:
logger.info("_is_org_member: %s 不在 %s 组织内 (404)", username, org)
return False
logger.warning(
"_is_org_member 查询失败: %s org=%s HTTP %d body=%s",
username, org, exc.code,
exc.read().decode("utf-8", errors="replace")[:200],
)
return False
except Exception as exc:
logger.warning("_is_org_member 网络异常: %s %s %s", username, org, exc)
return False
def _get_starred_repos(token: str, username: str) -> set[str]:
"""获取用户所有 Star 仓库的 full_name 集合（仅第一页 100 条）。
``GET /users/{username}/starred/{owner}/{repo}`` 端点在 PAT 认证下
不可靠（始终返回 404），改用列表+过滤方式。
"""
starred: set[str] = set()
page = 1
while page <= 5: # 最多 500 条，够用
try:
resp = _rest_req(
token, "GET",
f"/users/{username}/starred?per_page=100&page={page}",
)
data = json.loads(resp.read())
if not data:
break
for item in data:
starred.add(item.get("full_name", ""))
page += 1
except urllib.error.HTTPError as exc:
raise RuntimeError(
f"获取 Star 列表失败 {username}: HTTP {exc.code}"
) from exc
return starred
def check_all_conditions(token: str, username: str) -> tuple[bool, list[dict[str, Any]]]:
"""汇总所有加入条件检测结果。
Returns:
``(all_met, results)`` 其中 ``results`` 为条件列表，每项含
``name`` / ``icon`` / ``passed`` / ``detail``。
"""
results: list[dict[str, Any]] = []
# 条件 1：关注组织
ok_follow, detail_follow = _check_following(token, username, JOIN_FOLLOW_ORG)
results.append({
"name": f"关注 @{JOIN_FOLLOW_ORG}",
"icon": "👀",
"passed": ok_follow,
"detail": detail_follow,
})
# 条件 2：Star 全部指定仓库 — 先拉取全部 Star 列表，再逐个比对
try:
starred = _get_starred_repos(token, username)
except RuntimeError as exc:
logger.error("获取 Star 列表失败: %s", exc)
starred = set()
star_details: list[str] = []
for repo in JOIN_STAR_REPOS:
full = f"{JOIN_STAR_OWNER}/{repo}"
if full not in starred:
star_details.append(repo)
star_passed = len(star_details) == 0
if star_passed:
star_detail = "已 Star 全部"
else:
star_detail = "缺少：" + ", ".join(_repo_link(r) for r in star_details)
results.append({
"name": "Star 指定仓库",
"icon": "⭐",
"passed": star_passed,
"detail": star_detail,
})
all_met = all(r["passed"] for r in results)
return all_met, results
def build_condition_report(
username: str,
all_met: bool,
results: list[dict[str, Any]],
) -> str:
"""生成条件检测 Markdown 报告。"""
passed_count = sum(1 for r in results if r["passed"])
total = len(results)
lines = [
f"## 📋 @{username} 的加入申请审核结果",
"",
"| 条件 | 状态 | 详情 |",
"|------|:----:|------|",
]
for r in results:
icon = r["icon"]
emoji = "✅" if r["passed"] else "❌"
lines.append(f"| {icon} {r['name']} | {emoji} | {r['detail']} |")
lines.append("")
if all_met:
lines.append(
f"🎉 全部通过 ({passed_count}/{total})！正在发送邀请…"
)
else:
lines.append(
f"🔴 需要全部满足，当前 {passed_count}/{total} 项通过。"
f"满足后请再次发送申请，无需特殊格式，说明来意即可。"
)
lines.append("")
lines.append(
f"> ⭐ 需 Star 的仓库：{', '.join(_repo_link(r) for r in JOIN_STAR_REPOS)}"
)
if all_met:
lines.append("")
lines.append(
"\n---\n"
"> ### 🏷️ 团队分组说明\n"
"> 加入组织后默认邀请至 **CSM-Community**（CSM 社区爱好者）团队。\n"
"> \n"
"> | 团队 | 说明 |\n"
"> |------|------|\n"
"> | **CSM-Community** | CSM 社区爱好者（默认加入） |\n"
"> | ├─ **CSM-Module-Author** | CSM 模块的贡献者 |\n"
"> | └─ **CSM-Developer** | CSM 开发人员 |\n"
"> \n"
"> 团队分组权限不会主动提升，@NEVSTOP-LAB/csm-committee 会根据贡献"
"提高对应的项目权限。\n"
"> \n"
"> ---\n"
"> ### 💡 温馨提示\n"
"> \n"
"> - 💬 Q&A 中提问 CSM 相关问题，机器人会自动回复；你也可以创建仓库上传"
"你的 CSM 模块，在 Issue 中 @nevstop 或 @NEVSTOP-LAB/csm-committee，"
"我们会帮你 Review 并提出改进建议\n"
"> - 📋 [项目任务看板](https://github.com/orgs/NEVSTOP-LAB/projects/18) "
"中是主要需要完成的任务列表，欢迎认领并完成对应的 Issue\n"
"> - ⚠️ 成员需每月有公开贡献（commit / Issue / PR），"
"长期无贡献将被自动移出组织。请保持活跃，为社区做出贡献！"
)
return "\n".join(lines)
def _resolve_user_id(token: str, username: str) -> int:
"""通过 REST API 获取用户数字 ID（邀请 API 需要）。"""
try:
resp = _rest_req(token, "GET", f"/users/{username}")
data = json.loads(resp.read())
user_id = data.get("id")
if not user_id:
raise RuntimeError(f"GET /users/{username} 未返回 id 字段")
return int(user_id)
except urllib.error.HTTPError as exc:
raise RuntimeError(
f"获取用户 {username} 信息失败: HTTP {exc.code}"
) from exc
def send_invitation(token: str, org: str, user_id: int) -> bool:
"""发送组织邀请。返回 True 表示成功。"""
try:
# 标记 token 类型方便日志排查
token_type = "App" if token != os.environ.get("CSM_QA_GH_TOKEN", "") else "PAT"
data = json.dumps({"invitee_id": user_id}).encode()
req = urllib.request.Request(
f"{GITHUB_API_URL}/orgs/{org}/invitations",
data=data,
headers={
"Authorization": f"Bearer {token}",
"Accept": "application/vnd.github.v3+json",
"Content-Type": "application/json",
"User-Agent": "org-router/1.0",
},
method="POST",
)
with urllib.request.urlopen(req, timeout=15) as resp:
resp.read()
logger.info(
"%s token 邀请成功: org=%s user_id=%d status=%d",
token_type, org, user_id, resp.status,
)
return resp.status in (201, 200)
except urllib.error.HTTPError as exc:
body = exc.read().decode("utf-8", errors="replace")
logger.error(
"%s token 邀请失败: org=%s user_id=%d HTTP %d body=%s",
token_type, org, user_id, exc.code, body[:400],
)
return False
def _add_team_membership(token: str, org: str, team_slug: str, username: str) -> bool:
"""将用户添加到指定团队。需要 org Members:Write 权限。
``PUT /orgs/{org}/teams/{team_slug}/memberships/{username}``
成功返回 True，失败返回 False。
"""
try:
team_url = f"/orgs/{org}/teams/{team_slug}/memberships/{username}"
_rest_req(token, "PUT", team_url)
logger.info(
"用户已添加到团队: %s → %s/%s",
username, org, team_slug,
)
return True
except urllib.error.HTTPError as exc:
if exc.code == 409:
# 用户已在团队中 → 视为成功
logger.info("用户已在团队中: %s/%s", username, team_slug)
return True
body = exc.read().decode("utf-8", errors="replace")
logger.error(
"添加团队失败: %s → %s/%s HTTP %d body=%s",
username, org, team_slug, exc.code, body[:300],
)
return False
except Exception as exc:
logger.error("添加团队网络异常: %s/%s %s", username, team_slug, exc)
return False
def post_reply(token: str, discussion_id: str, body: str) -> str:
"""向 Discussion 发布评论，返回新评论的 node ID。"""
gql_client = GitHubGraphQL(token, user_agent="org-router/1.0")
full_body = f"{body}{BOT_FOOTER}\n{BOT_MARKER}"
gql = """
mutation($discussionId: ID!, $body: String!) {
addDiscussionComment(input: {discussionId: $discussionId, body: $body}) {
comment {
id
url
}
}
}
"""
data = gql_client.query(gql, {"discussionId": discussion_id, "body": full_body})
comment = data.get("addDiscussionComment", {}).get("comment", {})
comment_url = comment.get("url", "")
logger.info("评论已发布: %s", comment_url)
return comment.get("id", "")
# ── QA 逻辑 ──────────────────────────────────────────────────────────────────
def _handle_qa(
token: str,
discussion_number: int,
category_name: str,
dry_run: bool,
comment_body: str = "",
comment_author: str = "",
) -> None:
"""处理 QA 意图：Q&A 分类下调用 CSM_QA 回答，否则引导。
当 CSM_QA 初始化失败或生成回答出错时，发布错误提示回复而非崩溃，
确保 workflow 不会因临时故障（网络、模型下载等）整体失败。
模拟模式（discussion_number=0）时跳过 Discussion API 调用：
直接使用 --comment-body 作为问题，调用 CSM_QA 生成回答并打印到 stdout。
Args:
comment_author: 评论作者（当前仅透传，SKIP_AUTHORS 检查已在 main() 完成）。
"""
source_owner, source_repo = _get_source_repo_parts()
# ── 模拟模式（discussion_number=0）───────────────────────────────────
if discussion_number == 0:
simulate_question = comment_body.strip()
if not simulate_question:
logger.error("[SIMULATE] 模拟模式需要提供 --comment-body（提问正文）")
return
logger.info("[SIMULATE] QA 模拟模式：question=%s", simulate_question[:100])
# 延迟导入 CSM_QA
try:
from csm_llm_qa import CSM_QA # noqa: F811
except Exception:
logger.exception("[SIMULATE] 导入 CSM_QA 失败")
return
try:
qa_engine = CSM_QA.from_env(temperature=0, max_tokens=2048)
except Exception:
logger.exception("[SIMULATE] CSM_QA.from_env() 初始化失败")
return
try:
answer = qa_engine.ask(simulate_question)
# 模拟模式下将回答打印到 stdout，供 workflow 日志查看
print("=" * 60)
print(" SIMULATE QA Answer")
print("=" * 60)
print(f"Question: {simulate_question}")
print("-" * 40)
print(answer)
print("=" * 60)
logger.info("[SIMULATE] QA 回答已生成（%d chars）", len(answer))
except Exception:
logger.exception("[SIMULATE] 生成 QA 回答失败")
return
if category_name != QA_CATEGORY_NAME:
# 检查 thread 中是否已有 Skip 用户介入
if _has_human_intervention(
token, discussion_number, source_owner, source_repo,
):
logger.info(
"Discussion #%d 已有 SKIP_AUTHORS 用户介入，跳过 Q&A 引导回复",
discussion_number,
)
return
# 非 Q&A 分类 → 引导到 Q&A 区
try:
gql_client = GitHubGraphQL(token, user_agent="org-router/1.0")
discussion = fetch_discussion(gql_client, source_owner, source_repo, discussion_number)
disc_id = discussion.get("id", "")
qa_url = f"https://github.com/orgs/{source_owner}/discussions/categories/q-a"
guide_body = (
f"💡 技术问题请在 [Q&A 分类]({qa_url}) 下提出，"
f"那里的 Bot 会自动为你解答。感谢理解！"
)
if not dry_run:
post_reply(token, disc_id, guide_body)
else:
logger.info("[DRY-RUN] 将引导至 Q&A 分类: discussion_id=%s", disc_id)
except Exception:
logger.exception("非 Q&A 引导回复失败")
return
# ── Q&A 分类 → 延迟导入并初始化 CSM_QA ──────────────────────────────
logger.info("Q&A 分类下的 QA 请求，初始化 CSM_QA…")
# 先尝试获取 discussion 元信息（失败时可发布错误回复）
try:
gql_client = GitHubGraphQL(token, user_agent="org-router/1.0")
discussion = fetch_discussion(gql_client, source_owner, source_repo, discussion_number)
disc_id = discussion.get("id", "")
except Exception:
logger.exception("获取 discussion #%d 失败", discussion_number)
return
# 延迟导入 CSM_QA 和 discussion_bot 函数（JOIN/OTHER 路径不触发）
try:
from scripts.discussion_bot import ( # type: ignore[import-not-found]
compute_reply_plan,
build_reply,
post_comment,
fetch_discussion as fetch_disc,
resolve_skip_logins,
)
from csm_llm_qa import CSM_QA
except Exception:
logger.exception("导入 CSM_QA / discussion_bot 模块失败")
if not dry_run and disc_id:
try:
post_reply(
token, disc_id,
"⚠️ QA Bot 初始化失败（依赖模块导入错误），请稍后重试。"
"若问题持续，请联系管理员。",
)
except Exception:
logger.exception("发布错误提示失败")
return
client = GitHubGraphQL(token)
# 初始化 RAG 问答引擎（可能下载模型/构建向量库，耗时较长）
try:
qa_engine = CSM_QA.from_env(temperature=0, max_tokens=2048)
except Exception:
logger.exception("CSM_QA.from_env() 初始化失败")
if not dry_run and disc_id:
try:
post_reply(
token, disc_id,
"⚠️ QA Bot 初始化失败（模型加载或向量库构建出错），请稍后重试。"
"若问题持续，请联系管理员。",
)
except Exception:
logger.exception("发布错误提示失败")
return
# 获取 Bot 自身的登录名（用于 compute_reply_plan 作者校验）
try:
viewer_data = client.query("query { viewer { login } }")
bot_login = viewer_data.get("viewer", {}).get("login")
except Exception:
bot_login = None
# 解析完整跳过名单（SKIP_AUTHORS + csm-committee 团队成员）
effective_skip = resolve_skip_logins(client, source_owner)
# 检查 discussion 作者是否在跳过列表中（GitHub login 大小写不敏感）
author_login = (discussion.get("author") or {}).get("login", "").casefold()
if author_login in effective_skip:
logger.info(
"Discussion #%d 作者 %r 在跳过列表中，跳过", discussion_number, author_login
)
return
plan = compute_reply_plan(discussion, bot_login, skip_logins=effective_skip)
if plan is None:
logger.info("无需回复（已回复且无追问）")
return
question, history = plan
logger.info("生成回答中 (question=%s chars, history=%d turns)", len(question), len(history))
if not dry_run:
try:
answer = qa_engine.ask(question, history=history)
reply_body = build_reply(answer) # build_reply 已含 footer + marker
post_comment(client, disc_id, reply_body)
except Exception:
logger.exception("生成 QA 回答失败")
try:
post_reply(
token, disc_id,
"⚠️ 生成回答时出错，请稍后重试。若问题持续，请联系管理员。",
)
except Exception:
logger.exception("发布错误提示失败")
else:
logger.info("[DRY-RUN] 将生成 QA 回答: question=%.100s", question)
# ── 辅助函数 ─────────────────────────────────────────────────────────────────
def _get_repo_parts() -> tuple[str, str]:
"""从 GITHUB_REPOSITORY 解析 owner/repo。"""
repo_env = os.environ.get("GITHUB_REPOSITORY", "")
if "/" not in repo_env:
raise ValueError(f"GITHUB_REPOSITORY 格式不正确: {repo_env!r}，期望 'owner/repo'")
owner, repo = repo_env.split("/", 1)
return owner, repo
def _get_source_repo_parts() -> tuple[str, str]:
"""获取组织 Discussion 实际归属的源仓库 owner/repo。
默认 ``<org>/.github``，其中 ``<org>`` 取自 ``GITHUB_REPOSITORY``。
可通过环境变量 ``DISCUSSION_SOURCE_REPO`` 覆盖（格式 ``owner/repo``）。
"""
source_env = (os.environ.get("DISCUSSION_SOURCE_REPO") or "").strip()
if source_env:
if "/" not in source_env:
logger.warning(
"DISCUSSION_SOURCE_REPO=%r 格式不正确（期望 owner/repo），回退默认",
source_env,
)
else:
parts = source_env.split("/", 1)
if parts[0] and parts[1]:
return parts[0], parts[1]
logger.warning(
"DISCUSSION_SOURCE_REPO=%r 含空 owner 或 repo，回退默认",
source_env,
)
src_owner, _ = _get_repo_parts()
src_repo = ".github"
logger.info("使用默认源仓库: %s/%s", src_owner, src_repo)
return src_owner, src_repo
def _build_classify_history(
token: str, discussion_number: int, classify_input: str
) -> Optional[list[dict[str, str]]]:
"""为分类构建 discussion thread 历史上下文。
拉取 discussion 数据后，将标题+正文作为首条 user 消息，后续评论按
user/assistant 角色组装为对话历史。Bot 评论通过 BOT_MARKER 识别。
返回的 history 不包含当前要分类的评论本身（即 classify_input），
格式与 compute_reply_plan 一致：``[{"role": "user"|"assistant", "content": str}]``。
"""
source_owner, source_repo = _get_source_repo_parts()
gql_client = GitHubGraphQL(token, user_agent="org-router/1.0")
discussion = fetch_discussion(gql_client, source_owner, source_repo, discussion_number)
title = (discussion.get("title") or "").strip()
body = (discussion.get("body") or "").strip()
original_question = f"{title}\n\n{body}".strip() if body else title
comments = discussion.get("comments", {}).get("nodes", []) or []
# 构建对话历史
history: list[dict[str, str]] = []
if original_question:
history.append({"role": "user", "content": original_question})
# 当前要分类的消息文本（strip 后用于比较）
classify_stripped = classify_input.strip()
for c in comments:
c_body = (c.get("body") or "").strip()
if not c_body:
continue
# 跳过当前要分类的评论本身（避免重复喂给 LLM）
# 对于 discussion_comment 事件，新评论总是最后一条；同时用
# strip 后的文本比较，容忍空白差异
if c_body == classify_stripped:
continue
is_bot = BOT_MARKER in c_body
role = "assistant" if is_bot else "user"
history.append({"role": role, "content": c_body})
return history if history else None
def fetch_discussion(
client: Any, owner: str, repo: str, number: int
) -> dict[str, Any]:
"""拉取指定 discussion 的详情（含所有评论，自动分页）。"""
gql = """
query($owner: String!, $repo: String!, $number: Int!) {
repository(owner: $owner, name: $repo) {
discussion(number: $number) {
id
number
title
body
url
closed
author { login }
category {
id
name
}
comments(first: 100) {
nodes {
id
body
createdAt
author { login }
}
pageInfo {
hasNextPage
endCursor
}
}
}
}
}
"""
data = client.query(gql, {"owner": owner, "repo": repo, "number": number})
repository = data.get("repository") or {}
disc = repository.get("discussion")
if not disc:
raise RuntimeError(f"Discussion #{number} 不存在或无权限访问")
# 分页拉取剩余评论
while disc["comments"]["pageInfo"]["hasNextPage"]:
cursor = disc["comments"]["pageInfo"]["endCursor"]
more_gql = """
query($discussionId: ID!, $cursor: String!) {
node(id: $discussionId) {
... on Discussion {
comments(first: 100, after: $cursor) {
nodes {
id
body
createdAt
author { login }
}
pageInfo {
hasNextPage
endCursor
}
}
}
}
}
"""
more_data = client.query(more_gql, {"discussionId": disc["id"], "cursor": cursor})
node = more_data.get("node") or {}
more = node.get("comments", {})
disc["comments"]["nodes"].extend(more.get("nodes", []))
disc["comments"]["pageInfo"] = more.get(
"pageInfo", {"hasNextPage": False, "endCursor": None}
)
return disc
def _handle_join(
token: str,
discussion_number: int,
comment_author: str,
dry_run: bool,
) -> None:
"""处理 JOIN 意图：条件检测 → 报告 → 通过则邀请。
模拟模式（discussion_number=0）时跳过所有 API 调用，仅打印检测报告。