@hupan56/wlkj 3.1.32 → 3.3.0

package/templates/qoder/scripts/domain/kg/extract/mybatis/relation_builder.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+"""Mapper ↔ Entity ↔ Table 三向关联构建器。
+
+把 mapper_parser 的结构化信息 + sql_extractor 的表名 →
+生成 entities (TABLE/COLUMN) + edges (queries/maps_to/returns)。
+"""
+from __future__ import annotations
+from . import mapper_parser as mp
+from . import sql_extractor as sq
+
+
+def build_relations(parsed: dict, repo_id: str, filepath: str):
+    """从单个 mapper XML 的解析结果构建实体 + 边。
+
+    Returns: (entities, edges)
+      entities: TABLE / COLUMN 实体
+      edges:    queries (Mapper→Table) / column_maps (COLUMN 关联)
+    """
+    entities, edges = [], []
+    ns_short = parsed.get('namespace_short', '')
+    ns_full = parsed.get('namespace', '')
+    if not ns_short:
+        return entities, edges
+
+    # Mapper 实体 id (对齐 GOAL 2 的 entity_id 规则: MAP:repo:Name)
+    mapper_id = 'MAP:%s:%s' % (repo_id, ns_short)
+
+    # ===== queries 边: Mapper → 每个被访问的 TABLE =====
+    all_tables = set()
+    for stmt in parsed.get('statements', []):
+        for tbl in stmt.get('tables', []):
+            all_tables.add(tbl)
+
+    for tbl in all_tables:
+        tbl_id = 'TBL:%s:%s' % (repo_id, tbl)
+        # TABLE 实体 (如果 GOAL 2 没建过, 这里补建)
+        entities.append({
+            'id': tbl_id, 'repo_id': repo_id, 'type': 'TABLE',
+            'canonical': tbl, 'cn': '',
+            'props': {'from': 'mapper_xml', 'file': filepath},
+        })
+        # queries 边
+        edges.append({
+            'from_id': mapper_id, 'to_id': tbl_id,
+            'edge_type': 'queries',
+            'from_repo': repo_id, 'to_repo': repo_id,
+            'confidence': 1.0, 'source': 'mapper_xml',
+            'props': {'via': 'mybatis_xml'},
+        })
+
+    # ===== COLUMN 实体 (从 resultMap 提取) =====
+    for rm_id, cols in parsed.get('result_maps', {}).items():
+        for col_map in cols:
+            col_name = col_map.get('column', '')
+            if not col_name or '$' in col_name:
+                continue
+            # COLUMN 实体 (关联到第一个被访问的表, 兜底)
+            hint_table = list(all_tables)[:1]
+            tbl_for_col = hint_table[0] if hint_table else 'unknown'
+            col_id = 'COL:%s:%s.%s' % (repo_id, tbl_for_col, col_name)
+            entities.append({
+                'id': col_id, 'repo_id': repo_id, 'type': 'COLUMN',
+                'canonical': col_name, 'cn': '',
+                'props': {'table': tbl_for_col,
+                          'property': col_map.get('property', ''),
+                          'from': 'result_map', 'file': filepath},
+            })
+
+    return entities, edges

package/templates/qoder/scripts/domain/kg/extract/mybatis/sql_extractor.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+"""SQL 文本提取器: 从 SQL 提取表名 + 列引用 + 操作类型。
+
+正则匹配 MyBatis 动态 SQL 里残留的 ${ew.customSqlSegment} 等,
+但 FROM/INTO/UPDATE/JOIN 后的表名通常稳定可提。
+"""
+from __future__ import annotations
+import re
+
+# 表名提取 (FROM/JOIN/INTO/UPDATE 后跟标识符)
+_TABLE_PATTERNS = [
+    re.compile(r'\bFROM\s+`?([a-z_][a-z0-9_]*)`?', re.IGNORECASE),
+    re.compile(r'\bJOIN\s+`?([a-z_][a-z0-9_]*)`?', re.IGNORECASE),
+    re.compile(r'\bINTO\s+`?([a-z_][a-z0-9_]*)`?', re.IGNORECASE),
+    re.compile(r'\bUPDATE\s+`?([a-z_][a-z0-9_]*)`?\s+SET', re.IGNORECASE),
+]
+
+# 排除: 显然不是表名的 (变量/占位符/SQL关键字)
+_NOT_TABLES = {'dual', 'select', 'where', 'set', 'values', 'from', 'join',
+               'and', 'or', 'not', 'in', 'as', 'on', 'left', 'right', 'inner',
+               'outer', 'group', 'order', 'limit', 'distinct', 'case', 'when',
+               'then', 'else', 'end', 'null', 'true', 'false'}
+
+
+def extract_table_names(sql: str) -> list:
+    """从 SQL 文本提取所有被访问的表名。
+
+    Returns: ['test_demo', 'sys_user'] (去重保序)
+    """
+    if not sql:
+        return []
+    found = []
+    seen = set()
+    for pat in _TABLE_PATTERNS:
+        for m in pat.finditer(sql):
+            tbl = m.group(1).lower()
+            if tbl in _NOT_TABLES or tbl.startswith('$') or tbl.startswith('#'):
+                continue
+            # 排除太短或含特殊字符的 (变量插值残留)
+            if len(tbl) < 2 or '$' in tbl or '{' in tbl:
+                continue
+            if tbl not in seen:
+                seen.add(tbl)
+                found.append(tbl)
+    return found
+
+
+def extract_column_refs(sql: str) -> list:
+    """提取 SQL 里引用的列名 (粗粒度, 主要看 SELECT 字段列表)。
+
+    Returns: ['id', 'user_id', 'name'] (去重)
+    """
+    if not sql:
+        return []
+    cols = set()
+    # SELECT field1, field2 FROM ... (字段列表)
+    m = re.search(r'SELECT\s+(DISTINCT\s+)?(.*?)\s+FROM', sql, re.IGNORECASE | re.DOTALL)
+    if m:
+        field_part = m.group(2)
+        if field_part.strip() != '*':
+            for token in re.split(r'[,\s]+', field_part):
+                token = token.strip().strip('`')
+                # 别名 as xx → 取前半
+                token = token.split(' as ')[0].split(' AS ')[0].strip()
+                if re.match(r'^[a-z_][a-z0-9_]*$', token, re.I) and len(token) >= 2:
+                    cols.add(token.lower())
+    # xx.column 形式 (表别名.列)
+    for m in re.finditer(r'\b[a-z]+\s*\.\s*([a-z_][a-z0-9_]*)', sql, re.IGNORECASE):
+        cols.add(m.group(1).lower())
+    return sorted(cols)
+
+
+def classify_operation(stmt_type: str) -> str:
+    """MyBatis statement 类型 → 操作分类。"""
+    return {
+        'select': 'READ', 'insert': 'WRITE',
+        'update': 'WRITE', 'delete': 'WRITE',
+    }.get(stmt_type.lower(), 'UNKNOWN')

package/templates/qoder/scripts/domain/kg/extract/prd/__init__.py

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+"""PRD 业务知识提取 (GOAL 4)。
+
+子模块:
+  prd_llm_extract  PRD 正文 → LLM 抽功能点/字段/规则/端点/权限
+  prd_chunk_embed  PRD 分块 + embedding (LightRAG vector 路)
+  req_anchor       REQ-ID 强关联 (PRD/commit/代码注释 → anchors 表)
+"""

package/templates/qoder/scripts/domain/kg/extract/prd/prd_chunk_embed.py

@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+"""PRD 分块 + embedding → LightRAG vector 路。
+
+让 rag_search 能搜到 PRD 正文 (语义召回), 不再只搜标题。
+复用工作台 rag.py 的 embed_texts (DashScope text-embedding-v3, 1024维)。
+"""
+from __future__ import annotations
+import os
+import sys
+
+_THIS = os.path.dirname(os.path.abspath(__file__))
+for _i in range(8):
+    _p = os.path.dirname(_THIS)
+    if os.path.isfile(os.path.join(_p, 'foundation', 'bootstrap.py')):
+        sys.path.insert(0, _p); break
+    _THIS = _p
+
+
+def chunk_text(text: str, size: int = 500, overlap: int = 50) -> list:
+    """滑窗分块。Returns: [chunk1, chunk2, ...]"""
+    if not text:
+        return []
+    chunks = []
+    i = 0
+    while i < len(text):
+        chunk = text[i:i + size]
+        chunks.append(chunk)
+        i += size - overlap
+    return chunks
+
+
+def embed_prd(prd_id: str, content_md: str, repo_id: str = ''):
+    """PRD 分块 + embedding → 写 PG embeddings 表。
+
+    Returns: {chunks, embedded} 失败返回 {'chunks':0,'embedded':0}
+    """
+    chunks = chunk_text(content_md)
+    if not chunks:
+        return {'chunks': 0, 'embedded': 0}
+
+    # 复用工作台 rag.py 的 embed_texts
+    try:
+        ws_backend = os.path.join(os.getcwd(), 'wlinkj-workspace', 'backend')
+        sys.path.insert(0, ws_backend)
+        from app.rag import embed_texts
+    except Exception:
+        # 兜底: 直接调 DashScope embedding API
+        return _embed_fallback(prd_id, chunks, repo_id)
+
+    vecs = embed_texts(chunks)
+    if not vecs:
+        return {'chunks': len(chunks), 'embedded': 0}
+
+    # 写 PG embeddings 表
+    _upsert_embeddings(prd_id, chunks, vecs, repo_id)
+    return {'chunks': len(chunks), 'embedded': len(vecs)}
+
+
+def _embed_fallback(prd_id, chunks, repo_id):
+    """直接调 DashScope text-embedding-v3 (rag.py 不可用时)。"""
+    import json, urllib.request
+    key = os.environ.get('DASHSCOPE_API_KEY', '')
+    if not key:
+        return {'chunks': len(chunks), 'embedded': 0}
+    # DashScope embedding API
+    url = 'https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings'
+    try:
+        req = urllib.request.Request(
+            url,
+            data=json.dumps({
+                'model': 'text-embedding-v3',
+                'input': chunks[:10],  # 单次上限10条
+                'dimensions': 1024,
+            }).encode('utf-8'),
+            headers={'Authorization': 'Bearer ' + key, 'Content-Type': 'application/json'},
+        )
+        with urllib.request.urlopen(req, timeout=30) as resp:
+            r = json.loads(resp.read())
+        vecs = [d['embedding'] for d in r['data']]
+        _upsert_embeddings(prd_id, chunks[:len(vecs)], vecs, repo_id)
+        return {'chunks': len(chunks), 'embedded': len(vecs)}
+    except Exception:
+        return {'chunks': len(chunks), 'embedded': 0}
+
+
+def _upsert_embeddings(prd_id, chunks, vecs, repo_id):
+    """写 PG embeddings 表 (pgvector)。"""
+    try:
+        from domain.kg.extract.java import pg_upsert
+        engine, _ = pg_upsert._get_pg_engine()
+        if engine is None:
+            return
+        from sqlalchemy import text
+        import json
+        with engine.begin() as conn:
+            for chunk, vec in zip(chunks, vecs):
+                vec_str = '[' + ','.join(str(v) for v in vec) + ']'
+                eid = 'PRDEMB:%s:%d' % (prd_id[:12], hash(chunk) & 0xffff)
+                conn.execute(text("""
+                    INSERT INTO embeddings (entity_id, repo_id, kind, content, vec, embedded_at)
+                    VALUES (:id, :repo, 'prd', :content, CAST(:vec AS VECTOR), now())
+                    ON CONFLICT (entity_id) DO UPDATE SET content=EXCLUDED.content, vec=EXCLUDED.vec, embedded_at=now()
+                """), {'id': eid, 'repo': repo_id, 'content': chunk[:500], 'vec': vec_str})
+    except Exception as e:
+        print('  [prd-embed] 写入跳过:', str(e)[:60])

package/templates/qoder/scripts/domain/kg/extract/prd/prd_llm_extract.py

@@ -0,0 +1,153 @@
+# -*- coding: utf-8 -*-
+"""PRD 正文 → LLM 抽取功能点/字段/规则/端点/权限。
+
+输入: PRD markdown 正文
+输出: 结构化知识 (FEATURE/CONSTRAINT/PERMISSION 实体)
+用 DashScope qwen-plus (已接入) 做结构化抽取。
+
+被 asset/prd_importer.py 调用 (灌资产中心 prds 表时, 每个PRD正文过一遍LLM)。
+"""
+from __future__ import annotations
+import os
+import sys
+import json
+import re
+import urllib.request
+
+_THIS = os.path.dirname(os.path.abspath(__file__))
+for _i in range(8):
+    _p = os.path.dirname(_THIS)
+    if os.path.isfile(os.path.join(_p, 'foundation', 'bootstrap.py')):
+        sys.path.insert(0, _p); break
+    _THIS = _p
+
+PRD_EXTRACT_PROMPT = """从以下PRD正文提取结构化知识, 只返回严格JSON(不要markdown代码块, 不要解释), 格式:
+{"features":[{"name":"资产列表分页","desc":"支持按页查看"}],"fields":[{"name":"owner_id","label":"资产归属人","rule":"必填"}],"rules":[{"rule":"金额>1万需二次审批","type":"business"}],"endpoints":[{"path":"/asset/list","method":"GET","desc":"资产列表"}],"permissions":[{"code":"asset:list","desc":"资产查看权限"}]}
+若某类提取不到就返回空数组。PRD正文:
+---
+%s
+---"""
+
+
+def _call_qwen(prompt: str, timeout: int = 60) -> str:
+    """调 DashScope qwen-plus, 返回文本。失败返回 ''。"""
+    key = os.environ.get('DASHSCOPE_API_KEY', '')
+    base = os.environ.get('ALI_CHAT_BASE', '').rstrip('/')
+    model = os.environ.get('ALI_CHAT_MODEL', 'qwen-plus')
+    if not key or not base:
+        return ''
+    req = urllib.request.Request(
+        base + '/chat/completions',
+        data=json.dumps({
+            'model': model,
+            'messages': [{'role': 'user', 'content': prompt}],
+            'max_tokens': 2000, 'temperature': 0.1,
+        }).encode('utf-8'),
+        headers={'Authorization': 'Bearer ' + key, 'Content-Type': 'application/json'},
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=timeout) as resp:
+            r = json.loads(resp.read())
+            return r['choices'][0]['message']['content']
+    except Exception:
+        return ''
+
+
+def _parse_json_lenient(text: str) -> dict:
+    """宽松解析 LLM 返回的 JSON (可能有 markdown 包裹或前后缀)。"""
+    if not text:
+        return {}
+    # 剥 markdown 代码块
+    text = re.sub(r'^```(?:json)?\s*', '', text.strip())
+    text = re.sub(r'\s*```$', '', text.strip())
+    # 找第一个 { 到最后一个 }
+    s = text.find('{')
+    e = text.rfind('}')
+    if s >= 0 and e > s:
+        text = text[s:e + 1]
+    try:
+        return json.loads(text)
+    except Exception:
+        return {}
+
+
+def extract_prd_knowledge(content_md: str) -> dict:
+    """PRD 正文 → LLM 抽取 → 结构化 dict。
+
+    Returns: {features, fields, rules, endpoints, permissions}
+    """
+    if not content_md or len(content_md) < 50:
+        return {'features': [], 'fields': [], 'rules': [], 'endpoints': [], 'permissions': []}
+    # 截断过长正文 (避免超 token)
+    content = content_md[:6000]
+    raw = _call_qwen(PRD_EXTRACT_PROMPT % content)
+    parsed = _parse_json_lenient(raw)
+    # 兜底确保字段齐全
+    return {
+        'features': parsed.get('features', []),
+        'fields': parsed.get('fields', []),
+        'rules': parsed.get('rules', []),
+        'endpoints': parsed.get('endpoints', []),
+        'permissions': parsed.get('permissions', []),
+    }
+
+
+def to_entities_edges(extracted: dict, prd_id: str, project_id: str, repo_id: str = ''):
+    """把 LLM 抽取结果转成 entities + edges。
+
+    entities: FEATURE / CONSTRAINT / PERMISSION
+    edges: 全部锚到 prd_id (强关联, confidence=1.0)
+    """
+    entities, edges = [], []
+    base = 'FEAT:%s' % prd_id[:12] if prd_id else 'FEAT:unknown'
+
+    # 功能点 → FEATURE
+    for f in extracted.get('features', []):
+        fname = f.get('name', '').strip()
+        if not fname:
+            continue
+        fid = '%s:%s' % (base, fname[:20])
+        entities.append({
+            'id': fid, 'repo_id': repo_id, 'type': 'FEATURE',
+            'canonical': fname, 'cn': f.get('desc', ''),
+            'props': {'source': 'prd_llm', 'prd_id': prd_id},
+        })
+        edges.append(_anchor_edge(fid, prd_id, repo_id))
+
+    # 业务规则 → CONSTRAINT
+    for r in extracted.get('rules', []):
+        rule = r.get('rule', '').strip()
+        if not rule:
+            continue
+        cid = 'CONST:%s:%s' % (prd_id[:8], rule[:20])
+        entities.append({
+            'id': cid, 'repo_id': repo_id, 'type': 'CONSTRAINT',
+            'canonical': rule[:50], 'cn': r.get('type', ''),
+            'props': {'source': 'prd_llm', 'prd_id': prd_id, 'rule': rule},
+        })
+        edges.append(_anchor_edge(cid, prd_id, repo_id))
+
+    # 权限 → PERMISSION
+    for p in extracted.get('permissions', []):
+        code = p.get('code', '').strip()
+        if not code:
+            continue
+        pid = 'PERM:PRD:%s:%s' % (prd_id[:8], code[:20])
+        entities.append({
+            'id': pid, 'repo_id': repo_id, 'type': 'PERMISSION',
+            'canonical': code, 'cn': p.get('desc', ''),
+            'props': {'source': 'prd_llm', 'prd_id': prd_id},
+        })
+        edges.append(_anchor_edge(pid, prd_id, repo_id))
+
+    return entities, edges
+
+
+def _anchor_edge(from_id: str, prd_id: str, repo_id: str) -> dict:
+    """实体 → PRD 的强关联边。"""
+    return {
+        'from_id': from_id, 'to_id': 'prd:%s' % prd_id,
+        'edge_type': 'anchors', 'from_repo': repo_id, 'to_repo': repo_id,
+        'confidence': 1.0, 'source': 'explicit',
+        'props': {'via': 'prd_llm'},
+    }

package/templates/qoder/scripts/domain/kg/extract/prd/req_anchor.py

@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+"""REQ-ID 强关联: PRD/commit/代码注释 三种锚定来源。
+
+复用 archive_prd.py 的正则 (REQ-2026-005 / REQ-012 两种格式)。
+锚点写 anchors 表, 关联写 edges (confidence=1.0, source=explicit)。
+"""
+from __future__ import annotations
+import os
+import sys
+import re
+
+_THIS = os.path.dirname(os.path.abspath(__file__))
+for _i in range(8):
+    _p = os.path.dirname(_THIS)
+    if os.path.isfile(os.path.join(_p, 'foundation', 'bootstrap.py')):
+        sys.path.insert(0, _p); break
+    _THIS = _p
+
+# 复用 archive_prd.py 的正则 (兼容 REQ-2026-005 标准格式 + REQ-012 老格式)
+_REQ_RE = re.compile(r'REQ-(\d{2,4})(?:-(\d{2,4}))?', re.IGNORECASE)
+
+
+def format_req_id(match) -> str:
+    """格式化 REQ-ID (标准化)。"""
+    g1, g2 = match.group(1), match.group(2)
+    if g2:
+        return 'REQ-%s-%s' % (g1, g2)
+    return 'REQ-%s' % g1
+
+
+def find_req_ids(text: str) -> list:
+    """从文本提取所有 REQ-ID (去重保序)。"""
+    seen = set()
+    out = []
+    for m in _REQ_RE.finditer(text or ''):
+        rid = format_req_id(m)
+        if rid not in seen:
+            seen.add(rid)
+            out.append(rid)
+    return out
+
+
+# ============ 三种锚定 ============
+
+def anchor_prd(prd_id: str, title: str, project_id: str, repo_id: str = ''):
+    """时机1: PRD 创建时锚定。
+
+    PRD 标题/内容含 REQ-ID → 锚到 anchors 表。
+    没有 REQ-ID → 用 PRD id 生成锚点 (REQ-PRD-xxx)。
+    Returns: (anchor_entities, anchor_edges)
+    """
+    entities, edges = [], []
+    req_ids = find_req_ids(title)
+    if not req_ids:
+        # 无 REQ-ID, 用 PRD 生成一个锚点
+        req_ids = ['REQ-PRD-%s' % prd_id[:8]]
+
+    for rid in req_ids:
+        # anchors 表实体 (type=ANCHOR)
+        entities.append({
+            'id': rid, 'repo_id': repo_id, 'type': 'ANCHOR',
+            'canonical': rid, 'cn': title[:60],
+            'props': {'kind': 'feature', 'prd_id': prd_id, 'project_id': project_id},
+        })
+        # PRD → 锚点 强关联边
+        edges.append({
+            'from_id': 'prd:%s' % prd_id, 'to_id': rid,
+            'edge_type': 'anchors', 'from_repo': repo_id, 'to_repo': repo_id,
+            'confidence': 1.0, 'source': 'explicit',
+            'props': {'via': 'prd_create'},
+        })
+    return entities, edges
+
+
+def anchor_commit(commit_msg: str, changed_files: list, repo_id: str = ''):
+    """时机2: commit message 带 REQ-ID → 代码文件锚定。
+
+    feat(REQ-2026-005): 资产列表分页
+    → changed_files 里的代码 → 锚到 REQ-2026-005
+    Returns: edges (不产生新实体, 只关联已有代码实体到锚点)
+    """
+    edges = []
+    req_ids = find_req_ids(commit_msg)
+    if not req_ids:
+        return edges
+
+    for rid in req_ids:
+        for filepath in changed_files:
+            # 文件级锚定 (文件可能对应多个实体, 这里先建文件→锚点边)
+            file_entity_id = 'FILE:%s:%s' % (repo_id, filepath.replace('\\', '/'))
+            edges.append({
+                'from_id': file_entity_id, 'to_id': rid,
+                'edge_type': 'implements_req', 'from_repo': repo_id, 'to_repo': repo_id,
+                'confidence': 1.0, 'source': 'commit',
+                'props': {'commit_msg': commit_msg[:100]},
+            })
+    return edges
+
+
+def anchor_inline_code(content: str, filepath: str, repo_id: str = ''):
+    """时机3: 代码注释里的 REQ-ID 锚定。
+
+    // REQ-2026-005 资产列表分页
+    /* REQ-047 老逻辑 */
+    Returns: edges
+    """
+    edges = []
+    req_ids = find_req_ids(content)
+    if not req_ids:
+        return edges
+
+    file_entity_id = 'FILE:%s:%s' % (repo_id, filepath.replace('\\', '/'))
+    for rid in req_ids:
+        edges.append({
+            'from_id': file_entity_id, 'to_id': rid,
+            'edge_type': 'implements_req', 'from_repo': repo_id, 'to_repo': repo_id,
+            'confidence': 1.0, 'source': 'code_comment',
+            'props': {'via': 'inline_comment'},
+        })
+    return edges

package/templates/qoder/scripts/domain/kg/extract/ts_extract.py

@@ -65,6 +65,117 @@ def _find_calls_in_function_ast(func_node, source_bytes):
     return calls
 
 
+# ── P1: calls / imports 边抽取 ────────────────────────────────────────────
+# 把 tree-sitter 已能识别的 call_expression 和 import 语句导出成图边,
+# 让知识图谱从"按钮→接口→控制器"单条业务链升级为带函数级调用的代码图。
+# 返回结构供 kg_build / kg_incremental 写成 entities(FUNCTION) + edges(calls/imports)。
+
+# 函数定义节点类型 (TS/JS): 命名函数 / const fn = () => {} / 类方法
+_FN_NODE_TYPES = (
+    'function_declaration',      # function foo() {}
+    'method_definition',         # class { foo() {} }
+    # lexical_declaration 内的 arrow_function 由下面单独处理 (export const fn = () => {})
+)
+
+
+def _fn_name_from_node(node, source_bytes):
+    """Extract a function's name from a definition node, or None."""
+    if node.type == 'function_declaration':
+        name_node = node.child_by_field_name('name')
+        return _node_text(name_node, source_bytes) if name_node else None
+    if node.type == 'method_definition':
+        name_node = node.child_by_field_name('name')
+        return _node_text(name_node, source_bytes) if name_node else None
+    if node.type == 'lexical_declaration':
+        # export const fnName = (...) => { ... }  /  const fnName = function() {}
+        for child in node.children:
+            if child.type == 'variable_declarator':
+                name_node = child.child_by_field_name('name')
+                value = child.child_by_field_name('value')
+                if name_node and value and value.type in ('arrow_function', 'function_expression'):
+                    return _node_text(name_node, source_bytes)
+    return None
+
+
+def ts_extract_calls(content):
+    """Extract function-level call edges from TS/JS/Vue-script source.
+
+    Walks every function definition in the AST, then reuses
+    _find_calls_in_function_ast to collect each function's callees.
+
+    Returns:
+        [(from_fn, callee), ...] — caller function name and callee text.
+        Callee text may be qualified (e.g. 'obj.method', 'useStore()');
+        callers choose how to resolve. Tree-sitter unavailable → [] .
+    """
+    if not _TS_AVAILABLE or not content:
+        return []
+    try:
+        source = content.encode('utf-8')
+        tree = _PARSER.parse(source)
+    except Exception:
+        return []
+
+    edges = []
+    # Collect every function definition (named + arrow + method).
+    fns = []
+
+    def collect_fns(node):
+        if node.type in _FN_NODE_TYPES:
+            fns.append(node)
+        # lexical_declaration may hold `const fn = () => {}` — handled in _fn_name_from_node
+        if node.type == 'lexical_declaration':
+            fns.append(node)
+        for child in node.children:
+            collect_fns(child)
+
+    collect_fns(tree.root_node)
+
+    for fn_node in fns:
+        caller = _fn_name_from_node(fn_node, source)
+        if not caller:
+            continue
+        for callee, _full in _find_calls_in_function_ast(fn_node, source):
+            # Skip trivial self-references and empty callees.
+            callee = callee.strip()
+            if callee and callee != caller:
+                edges.append((caller, callee))
+    return edges
+
+
+def ts_extract_imports(content):
+    """Extract import edges from TS/JS source (local module imports only).
+
+    Returns:
+        [(imported_name, source_path), ...] for local imports (relative paths
+        or @/ #/ alias). Third-party bare specifiers (e.g. 'vue', 'axios') are
+        filtered out — they aren't part of the project graph.
+    """
+    if not content:
+        return []
+    out = []
+    # Generic local import: import { a, b } from './x' | import x from '@/y' | import * as z from '../z'
+    for m in re.finditer(
+        r"import\s+(?:\{([^}]*)\}|\*\s+as\s+(\w+)|(\w+))\s*(?:,\s*\{([^}]*)\})?\s+from\s+['\"]([^'\"]+)['\"]",
+        content,
+    ):
+        names_group = m.group(1) or m.group(4)  # named imports {a, b}
+        default_name = m.group(2) or m.group(3)  # * as z  or  default x
+        src = (m.group(5) or '').strip()
+        # Only local imports (relative or alias); skip bare packages like 'vue'.
+        if not src or not (src.startswith('.') or src.startswith('@/') or src.startswith('#/')):
+            continue
+        names = []
+        if names_group:
+            names.extend(n.strip().split(' as ')[-1].strip() for n in names_group.split(',') if n.strip())
+        if default_name:
+            names.append(default_name.strip())
+        for name in names:
+            if name:
+                out.append((name, src))
+    return out
+
+
 def ts_build_api_fn_table(api_dir):
     """Tree-sitter version: build {fnName: {url, verb, file}} from api/*.ts.
 

package/templates/qoder/scripts/domain/kg/graph/kg_semantic.py

@@ -179,9 +179,11 @@ def _load_model():
     # (本进程早先走过 bootstrap), sys.modules 里缓存的是错的, 要一并清掉。
     saved_paths = [p for p in sys.path]
     scripts_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    # Note: sys.path may contain pathlib.Path objects (not just str) in some
+    # environments — coerce to str before .lower() to avoid AttributeError.
     sys.path[:] = [p for p in sys.path
-                   if not (p and p.lower().startswith(scripts_root.lower()))
-                   and os.path.basename(p) not in ('foundation', 'kg', 'mcp', 'report', 'test', 'setup')]
+                   if not (p and str(p).lower().startswith(scripts_root.lower()))
+                   and os.path.basename(str(p)) not in ('foundation', 'kg', 'mcp', 'report', 'test', 'setup')]
     # 清掉被本地模块污染的 sys.modules 缓存 (filelock 等), 让第三方重新 import 干净版
     _purged = {}
     for mod_name in list(sys.modules):