@honor-claw/yoyo 1.2.0 → 1.2.1-beta.0

package/skills/yoyo-control/scripts/time_infer.py

@@ -0,0 +1,99 @@
+import time
+import hmac
+import hashlib
+import base64
+import json
+import requests
+from datetime import datetime
+import argparse
+
+
+def gen_headers():
+    # ak / sk 建议线上不要写死在代码里，放到环境变量或配置文件中
+    ak = "DB1178355DDE4CD2"
+    sk = "4764F7B34264E8F4343AC7363AA6D4"
+
+    # 对应：var ts = Math.floor(Date.now()).toString();
+    # Date.now() 是毫秒，这里同样用毫秒时间戳
+    ts = str(int(time.time() * 1000))
+
+    # 对应：CryptoJS.HmacSHA256(ts + ak, sk).toString().toUpperCase()
+    msg = (ts + ak).encode("utf-8")
+    key = sk.encode("utf-8")
+    digest = hmac.new(key, msg, hashlib.sha256).hexdigest().upper()
+
+    # 对应：new Buffer(...).toString('base64');
+    # 即对十六进制字符串做 base64 编码
+    sign = base64.b64encode(digest.encode("utf-8")).decode("utf-8")
+
+    headers = {
+        "accessKey": ak,
+        "ts": ts,
+        "sign": sign,
+        "Content-Type": "application/json",
+    }
+    return headers
+
+
+def get_current_local_time_str():
+    """
+    生成形如：2025-12-23 12:48:00.000 的时间字符串
+    使用当前系统时间
+    """
+    now = datetime.now()
+    # %f 是微秒，取前 3 位当毫秒
+    return now.strftime("%Y-%m-%d %H:%M:%S.%f")[:-3]
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(
+        description="调用 time-agent 接口的命令行工具"
+    )
+    parser.add_argument(
+        "--query",
+        # required=True,
+        default="大小周下周六休息7点",
+        help="语义查询内容（必填）",
+    )
+    parser.add_argument(
+        "--type",
+        dest="req_type",
+        default="time",
+        help="payload.data.type，默认值为 'time'",
+    )
+    parser.add_argument(
+        "--timeZone",
+        dest="time_zone",
+        default="+0800",
+        help="payload.data.timeZone，默认值为 '+0800'",
+    )
+    return parser.parse_args()
+
+
+def main():
+    url = "https://ai-model-access-drcn.rnd.honor.com/time-agent/v1/models/model/predict"
+    args = parse_args()
+
+    payload = {
+        "data": {
+            "type": args.req_type,
+            "timeZone": args.time_zone,
+            "query": args.query,
+            "localTime": get_current_local_time_str()
+        }
+    }
+
+    headers = gen_headers()
+
+    resp = requests.post(url, headers=headers, data=json.dumps(payload))
+    print("Status code:", resp.status_code)
+    try:
+        datas = resp.json()
+        res = json.dumps(datas.get("data", {}), separators=(",", ":"), ensure_ascii=False)
+        print({"data_time": res})
+    except Exception:
+        print("Response text:", resp.text)
+
+
+if __name__ == "__main__":
+    main()