>j*,UdZddlmZddlZddlZddlZddlZddlZddl Zddl m Z m Z m Z mZddlmZmZmZmZmZmZejeZdddd d d d d d d dddd d dZded<dZddddZdZdZdZd(dZ d)d Z!d*d#Z"Gd$d%eZ#d+d'Z$dS),u 七牛云 Kling 图像生成后端. 使用七牛云 AI 大模型推理 API 的 Kling 系列模型进行文生图。 接口采用异步模式: 提交任务 → 轮询状态 → 下载图片。 支持的模型: - kling-v1: 文生图 V1.0 - kling-v1-5: 文生图 V1.5 (支持角色/人脸参考) - kling-v2: 文生图 V2.0 (推荐) 图片以 URL 形式返回 (有效期 7 天), 本插件自动下载到本地缓存。 ) annotationsN)AnyDictListOptional)DEFAULT_ASPECT_RATIOImageGenProvidererror_responseresolve_aspect_ratiosave_url_imagesuccess_responsezKling v2z~60su#最新文生图, 画面质量最高u 按量计费)displayspeed strengthspricez Kling v1.5z~45su%支持角色特征参考/人脸参考zKling v1z~30su基础文生图, 速度快)kling-v2z kling-v1-5zkling-v1zDict[str, Dict[str, Any]]_MODELSr16:9z9:16z1:1) landscapeportraitsquarei,zhttps://api.qnaigc.com/v1return Optional[str]cdD]:}tj|d}|r|cS; ddlm}|}|di}|di}t |tr,|dd}|r|Sn#t$rYnwxYwdS) u获取七牛云 API Key.) QINIU_API_KEYOPENROUTER_API_KEYr load_config providersqiniuapi_keyN) osenvirongetstriphermes_cli.configr isinstancedict Exception)env_varkeyr cfgr! qiniu_cfgs 8/home/ubuntu/.hermes/plugins/image_gen/qiniu/__init__.py _get_api_keyr1Ls;jnnWb))//11  JJJ   111111kmmGGK,, MM'2.. i & & -- 2..4466C       4sA>"5r::@@BBL3 //W\222 A111111kmm*4S$*?*?G#''+&&&R,6w,E,EMGKK(((2 i & & 3 }}Wb117799H 3H//!2228B7D8Q8QYGKK,,22444WY  1g--gi00 0 AAA :C@@@@@@@@A '-0 00s B>E' AE'' F1FF hermes_aspectstrc8t|dS)u4将 Hermes 抽象比例映射为 Kling 原生比例.r) _ASPECT_MAPr&)r@s r0_kling_aspect_ratiorDzs ??=& 1 11cneZdZdZeddZeddZddZdd Zdd Z dd Z e fddZ dS)QiniuImageGenProvideru 七牛云 Kling 文生图后端.rrAcdS)Nr"selfs r0namezQiniuImageGenProvider.nameswrEcdS)N七牛云 KlingrIrJs r0 display_namez"QiniuImageGenProvider.display_names  rEboolc8ttSN)rPr1rJs r0 is_availablez"QiniuImageGenProvider.is_availablesLNN###rEList[Dict[str, Any]]cHdtDS)NcTg|]%\}}||d|d|d|dd&S)rrrr)idrrrrrI).0r<metas r0 z5QiniuImageGenProvider.list_models..sU    $  ?g!+.g       rE)ritemsrJs r0 list_modelsz!QiniuImageGenProvider.list_modelss-   #*--//    rErctSrR)r9rJs r0 default_modelz#QiniuImageGenProvider.default_modelsrEDict[str, Any]cdddddddgdS) NrNpaidu@七牛云 Kling 系列文生图 (kling-v1/v1-5/v2), 异步接口ru七牛云 AI API Keyzhttps://console.qnaigc.com/)r-prompturl)rLbadgetagenv_varsrIrJs r0get_setup_schemaz&QiniuImageGenProvider.get_setup_schemas4%U+48   rErb aspect_ratiokwargsrc  |pd}t|}|stddd|St}|stddd|St \}}t |}t j||dd|d d d } tj td | d |dd} tj | d} t j| } | d} n#tjj$rs}|dd}t(d|td|jd|dd|||cYd}~Sd}~wt.$r"}td|dd|||cYd}~Sd}~wwxYw| stddd|||St(d| |t3jt4z}d}t3j|krt3jt8tj td| dd |i } tj |d!}t j|}n3#t.$r&}t(d"| |Yd}~d}~wwxYw|d#d}|d$kr`|d%g}|rGt=|t>r2|d&}t=|t@r|d'}nJ|d(kr,td)|d*d+d,d|||St3j|k| td-t4d.| d/d|||S tC|d0|d12}nJ#t.$r=}t(d3|tE||||d| d4d56cYd}~Sd}~wwxYwtEtG||||dd| i6S)7NrzPrompt is requiredinvalid_argumentr")error error_typeproviderrhuh七牛云 API Key 未设置. 请设置环境变量 QINIU_API_KEY 或通过 hermes auth add qiniu 添加. auth_requiredi )r6rbrhnzutf-8z/images/generationszBearer zapplication/json) Authorizationz Content-Type)dataheaders)timeouttask_idizKling submit failed: %suKling 提交失败 (HTTP z): api_error)rlrmrnr6rbrhuKling 提交异常: uKling 返回了空的 task_idempty_responsez#Kling task submitted: %s (model=%s)z/images/tasks/rr)rt zKling poll error for %s: %sstatussucceedrsrrcfaileduKling 生成失败: status_messageu 未知错误generation_faileduKling 任务超时 (>zs): rvkling_<)prefixrvzFailed to cache Kling image: %su%图片缓存失败，URL有效期7天)rwnote)imager6rbrhrnextra)$r'r r r1r?rDjsondumpsencodeurllibrequestRequest _API_BASEurlopenloadsreadr&rl HTTPErrordecoder7r8coder+infotime _POLL_TIMEOUTsleep_POLL_INTERVALwarningr)listr*r r rA)rKrbrhriaspectr#r<rY kling_ratiopayloadreqrespresultrwebodyr>deadline image_urlreq2resp2r{srsfirst saved_paths r0generatezQiniuImageGenProvider.generates ,B%%''%l33 !*- #  .. !>+ # ())$)&11 *UdUm'     6'?? n$$ - - -!47!4!4 2%   >))#r)::DZ ,,Fjj++GG|%   6688??$$TcT*D LL2D 9 9 9!C!&CCTCC& #          !2S22& #         !5+ #    97HMMM9;;.#' ikkH$$ J~ & & &>))55G55(*=G*=*=>*D  ..tR.@@EJJLL11   rs   #"""""  ,,,,,,,,,,,,  8 $ $:  < 1 &&*   ' ,111102222N N N N N ,N N N l======rE