@pippit-dev/cli 1.0.24 → 1.0.25

package/skills/xyq-nest-skill/scripts/xyq_common.py

@@ -1,162 +0,0 @@
-"""小云雀 agent-im OpenAPI 公共模块：查询会话（鉴权为 Authorization: Bearer <access_key>）"""
-
-import json
-import os
-import sys
-import urllib.request
-import urllib.error
-import urllib.parse
-
-# Credentials may only be sent to the fixed production HTTPS origin.
-XYQ_BASE = "https://xyq.jianying.com"
-ACCESS_KEY = os.environ.get("XYQ_ACCESS_KEY", "")
-
-# API 路径常量
-GET_THREAD_PATH = "/api/biz/v1/skill/get_thread"
-HTTP_TIMEOUT_SECONDS = 30 * 60
-
-
-class _NoRedirect(urllib.request.HTTPRedirectHandler):
-    def redirect_request(self, req, fp, code, msg, headers, newurl):
-        # Never forward credentials or request bodies through a redirect.
-        raise urllib.error.HTTPError(req.full_url, code, "API 重定向已拒绝", headers, fp)
-
-
-def authenticated_open(req):
-    target = urllib.parse.urlsplit(req.full_url)
-    if (target.scheme != "https" or target.hostname != "xyq.jianying.com"
-            or target.port not in (None, 443) or target.username is not None
-            or target.password is not None):
-        raise urllib.error.URLError("仅允许小云雀生产 HTTPS 地址")
-    return urllib.request.build_opener(_NoRedirect()).open(req, timeout=HTTP_TIMEOUT_SECONDS)
-
-
-def redact_error(value):
-    text = str(value)
-    return text.replace(ACCESS_KEY, "[REDACTED]") if ACCESS_KEY else text
-
-
-def _headers():
-    if not ACCESS_KEY:
-        print("错误：请设置 XYQ_ACCESS_KEY 环境变量", file=sys.stderr)
-        sys.exit(1)
-    return {
-        "Authorization": f"Bearer {ACCESS_KEY}",
-        "Content-Type": "application/json",
-    }
-
-
-def api_post(path: str, body: dict) -> dict:
-    """POST 请求 agent-im OpenAPI"""
-    url = f"{XYQ_BASE.rstrip('/')}{path}"
-    data = json.dumps(body).encode("utf-8")
-    req = urllib.request.Request(
-        url,
-        data=data,
-        method="POST",
-        headers=_headers(),
-    )
-    try:
-        with authenticated_open(req) as resp:
-            return json.loads(resp.read().decode("utf-8"))
-    except urllib.error.HTTPError as e:
-        err_body = e.read().decode("utf-8") if e.fp else ""
-        print(f"API 错误 {e.code}: {redact_error(err_body)}", file=sys.stderr)
-        sys.exit(1)
-    except urllib.error.URLError as e:
-        print(f"网络错误: {redact_error(e.reason)}", file=sys.stderr)
-        sys.exit(1)
-
-
-def api_get(path: str) -> dict:
-    """GET 请求 agent-im OpenAPI"""
-    url = f"{XYQ_BASE.rstrip('/')}{path}"
-    req = urllib.request.Request(url, method="GET", headers=_headers())
-    try:
-        with authenticated_open(req) as resp:
-            return json.loads(resp.read().decode("utf-8"))
-    except urllib.error.HTTPError as e:
-        err_body = e.read().decode("utf-8") if e.fp else ""
-        print(f"API 错误 {e.code}: {redact_error(err_body)}", file=sys.stderr)
-        sys.exit(1)
-    except urllib.error.URLError as e:
-        print(f"网络错误: {redact_error(e.reason)}", file=sys.stderr)
-        sys.exit(1)
-
-
-def parse_response(resp: dict) -> dict:
-    """
-    分析 API 响应。
-    响应结构：{"ret":"0","errmsg":"","data":{}}
-    如果 ret 不是 "0"，打印错误信息并退出；否则返回 data。
-    """
-    ret = resp.get("ret", "")
-    if ret != "0":
-        errmsg = resp.get("errmsg", "未知错误")
-        print(f"错误码: {redact_error(ret)}, 错误信息: {redact_error(errmsg)}", file=sys.stderr)
-        sys.exit(1)
-    return resp.get("data", {})
-
-
-def get_thread(thread_id: str, run_id: str = "", after_seq: int = 0) -> dict:
-    """
-    查询会话消息列表。
-    返回 data: { messages: [...] }。
-    """
-    body = {}
-    if thread_id:
-        body["thread_id"] = thread_id
-    if run_id:
-        body["run_id"] = run_id
-    body["after_seq"] = after_seq
-    resp = api_post(GET_THREAD_PATH, body)
-    resp = parse_response(resp)
-    thread = resp.get("thread", {})
-    run_list = thread.get("run_list", [])
-    if len(run_list) == 0:
-        print("错误：未返回 run_list", file=sys.stderr)
-        sys.exit(1)
-    run = run_list[0]
-    run_state = run.get("state", "")
-
-    # 判断 run_state
-    if run_state == 3:
-        # 成功
-        print("成功：本次创作已完成", file=sys.stderr)
-        return run
-    elif run_state == 4:
-        # 失败
-        fail_reason = run.get("fail_reason", "未知失败原因")
-        print(f"错误：{redact_error(fail_reason)}", file=sys.stderr)
-        sys.exit(1)
-    elif run_state == 5:
-        # 取消
-        print("错误：本次创作已被终止", file=sys.stderr)
-        sys.exit(1)
-    else:
-        print("本次创作进行中", file=sys.stdout)
-        return run
-
-
-def extract_entries_from_run(run: dict) -> list:
-    """
-    从 Run 的 EntryList 中提取符合条件的 entry。
-    """
-    matched = []
-    for entry in run.get("entry_list") or []:
-        e = {}
-        message = entry.get("message")
-        artifact = entry.get("artifact")
-        if message:
-            e["id"] = message.get("message_id", "")
-            e["role"] = message.get("role", "")
-            e["content"] = message.get("content", [])
-            client_tool_calls = message.get("client_tool_calls", [])
-            if len(client_tool_calls) > 0:
-                e["content"].extend(client_tool_calls)  
-        if artifact:
-            e["id"] = artifact.get("artifact_id", "")
-            e["role"] = artifact.get("role", "")
-            e["content"] = artifact.get("content", [])
-        matched.append(e)
-    return matched