blade-code 0.6.3 → 0.7.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +1 -1
- package/dist/blade.js +2 -2
- package/dist/{chunk-s53zeref.js → chunk-0afbeec0.js} +1 -1
- package/dist/chunk-2nnak657.js +6 -0
- package/dist/chunk-31chcjw6.js +2 -0
- package/dist/chunk-64jv62vs.js +1 -0
- package/dist/chunk-6f63v1mk.js +2 -0
- package/dist/{chunk-gn1acna6.js → chunk-7z6ba6t9.js} +1 -1
- package/dist/chunk-93xhspe5.js +1 -0
- package/dist/{chunk-8fq1wzv3.js → chunk-96w1szbp.js} +1 -1
- package/dist/{chunk-zfy765hj.js → chunk-c2b6ekqx.js} +1 -1
- package/dist/chunk-e21htmg4.js +2 -0
- package/dist/chunk-f76d6jxg.js +1 -0
- package/dist/chunk-fehae50v.js +1 -0
- package/dist/chunk-ft5nt5b5.js +6 -0
- package/dist/chunk-gertm8dw.js +157 -0
- package/dist/{chunk-b5nhr3na.js → chunk-gscq4fmx.js} +2 -2
- package/dist/{chunk-m7xm1ssb.js → chunk-gxnwrec6.js} +1 -1
- package/dist/{chunk-y57hcnk6.js → chunk-hc08zg54.js} +1 -1
- package/dist/chunk-hscq9623.js +1 -0
- package/dist/chunk-k4behavv.js +2 -0
- package/dist/{chunk-99gfk9b2.js → chunk-kk8akm7s.js} +1 -1
- package/dist/chunk-kpq18y6v.js +6 -0
- package/dist/{chunk-67dzm8te.js → chunk-m22zeqdy.js} +4 -4
- package/dist/{chunk-21zps3hb.js → chunk-mnw1xcwn.js} +2 -2
- package/dist/{chunk-j1ehsp0x.js → chunk-mt1c5da4.js} +2 -2
- package/dist/chunk-nsbhtvqd.js +1 -0
- package/dist/chunk-p9s685ta.js +1 -0
- package/dist/{chunk-63954vzc.js → chunk-pf33bn8s.js} +1 -1
- package/dist/chunk-phnararg.js +1 -0
- package/dist/{chunk-r9vneqaw.js → chunk-pkzy724w.js} +6 -6
- package/dist/{chunk-g54k7rea.js → chunk-q6c9qs9b.js} +2 -2
- package/dist/chunk-rzz6r1g9.js +41 -0
- package/dist/{chunk-7rjrzngk.js → chunk-sw6ngzbg.js} +1 -1
- package/dist/chunk-tn3xjczg.js +4 -0
- package/dist/chunk-v97f7je6.js +2138 -0
- package/dist/chunk-vzebrztg.js +9 -0
- package/dist/{chunk-39h8jdm3.js → chunk-wre40mf1.js} +1 -1
- package/dist/chunk-x46rv9sv.js +6 -0
- package/dist/chunk-ygta7118.js +1 -0
- package/dist/chunk-zd9stbct.js +618 -0
- package/dist/chunk-zjh9qqvq.js +1 -0
- package/dist/{chunk-9m9nqfc4.js → chunk-zmp3vdcj.js} +1 -1
- package/dist/web/assets/{FilePreview-R1mRzyRz.js → FilePreview-DLnUh-wQ.js} +1 -1
- package/dist/web/assets/{MarkdownRenderer-B46lJYAn.js → MarkdownRenderer-C5KTk9XG.js} +1 -1
- package/dist/web/assets/{McpModal-QRv5Ij61.js → McpModal-B9qJ4ewt.js} +1 -1
- package/dist/web/assets/{SettingsModal-D-mi6mUR.js → SettingsModal-DpeZ6cJb.js} +1 -1
- package/dist/web/assets/{SettingsStore-Dmo3sZ_R.js → SettingsStore-IzKtyAnM.js} +1 -1
- package/dist/web/assets/{SkillsModal-DNXjbjEs.js → SkillsModal-BDfzgJIj.js} +1 -1
- package/dist/web/assets/{TerminalPanel-DG_Q4UKr.js → TerminalPanel-B2iTF_CR.js} +1 -1
- package/dist/web/assets/{dialog-CsiChiYt.js → dialog-sI3ihAze.js} +1 -1
- package/dist/web/assets/{index-BeYCsOvo.js → index-D8mo0jzF.js} +8 -8
- package/dist/web/index.html +1 -1
- package/package.json +5 -2
- package/dist/chunk-08h3wmqh.js +0 -1
- package/dist/chunk-0b3f8r1e.js +0 -6
- package/dist/chunk-3ak4p05x.js +0 -1
- package/dist/chunk-563zye36.js +0 -1
- package/dist/chunk-5v2jzxpw.js +0 -2
- package/dist/chunk-8a2z6mgp.js +0 -6
- package/dist/chunk-8zh508xk.js +0 -2125
- package/dist/chunk-ambewjsk.js +0 -6
- package/dist/chunk-caw1bxy6.js +0 -1
- package/dist/chunk-drrgdzrf.js +0 -1
- package/dist/chunk-g7bw2nkz.js +0 -613
- package/dist/chunk-g7xh4e99.js +0 -1
- package/dist/chunk-gp3rn8f8.js +0 -1
- package/dist/chunk-kds0h72f.js +0 -6
- package/dist/chunk-kevt8v3j.js +0 -1
- package/dist/chunk-m1tnx0je.js +0 -157
- package/dist/chunk-n9mwp62v.js +0 -1
- package/dist/chunk-nrh2jage.js +0 -2
- package/dist/chunk-qjhxk0yp.js +0 -1
- package/dist/chunk-t344cq6m.js +0 -4
- package/dist/chunk-t44946na.js +0 -41
- package/dist/chunk-t7hd45rg.js +0 -8
package/dist/chunk-n9mwp62v.js
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
import{g as a,h as b,i as c,j as d,k as e}from"./chunk-zfy765hj.js";import"./chunk-s53zeref.js";export{b as setSkipUntilVersion,d as performUpgrade,c as getUpgradeCommand,e as checkVersionOnStartup,a as checkVersion};
|
package/dist/chunk-nrh2jage.js
DELETED
|
@@ -1,2 +0,0 @@
|
|
|
1
|
-
var e={name:"blade-code",version:"0.6.3",private:!1,description:"🗡️ Blade Code - 智能代码助手命令行工具",type:"module",bin:{blade:"dist/blade.js"},files:["dist","vendor/ripgrep/**","README.md"],scripts:{dev:"node scripts/run-bun.js --watch src/blade.tsx","dev:serve":"node scripts/run-bun.js --watch src/blade.tsx serve --port 4097",build:"rm -rf dist && node scripts/run-bun.js run scripts/build.ts",start:"node scripts/run-bun.js run dist/blade.js",test:"node scripts/test.js","test:all":"vitest run --config vitest.config.ts","test:unit":"node scripts/test.js unit","test:integration":"node scripts/test.js integration","test:real-api":"node scripts/test.js realApi","test:cli":"node scripts/test.js cli","test:headless-core":"node scripts/test.js headlessCore","test:e2e":"node scripts/test.js e2e","test:performance":"node scripts/test.js performance","test:snapshot":"node scripts/test.js snapshot","test:security":"node scripts/test.js security","test:coverage":"node scripts/test.js all --coverage","test:watch":"vitest --watch --config vitest.config.ts","benchmark:repo":"node scripts/run-bun.js run scripts/run-real-repo-benchmark.ts","test:update-snapshots":"node scripts/test.js snapshot --update",lint:"biome lint src tests","lint:fix":"biome lint --write src tests","type-check":"tsc --noEmit","format:check":"biome format src tests",ready:"node scripts/run-bun.js run scripts/ready.ts","detect-unused":"node scripts/run-bun.js run scripts/detect-unused.ts",release:"node scripts/release.js","release:dry":"node scripts/release.js --dry-run","release:major":"node scripts/release.js --major","release:minor":"node scripts/release.js --minor","release:patch":"node scripts/release.js --patch"},keywords:["cli","blade","ai","assistant","agent","llm"],author:"echoVic",license:"MIT",engines:{node:">=20.0.0"},optionalDependencies:{"@vscode/ripgrep":"^1.17.0","bun-pty":"^0.4.8","node-pty":"1.0.0"},dependencies:{"@agentclientprotocol/sdk":"^0.12.0","@ai-sdk/anthropic":"^3.0.7","@ai-sdk/azure":"^3.0.5","@ai-sdk/deepseek":"^2.0.4","@ai-sdk/google":"^3.0.4","@ai-sdk/openai":"^3.0.26","@ai-sdk/openai-compatible":"^2.0.4","@inkjs/ui":"^2.0.0","@modelcontextprotocol/sdk":"^1.17.4",ahooks:"^3.9.6",ai:"^6.0.39","ansi-escapes":"^7.2.0","async-mutex":"^0.5.0",axios:"^1.12.2",chalk:"^5.4.1",diff:"^8.0.2","fast-glob":"^3.3.3","fuse.js":"^7.1.0","gray-matter":"^4.0.3",hono:"^4.7.10",ink:"npm:@jrichman/ink@6.4.10","ink-big-text":"^2.0.0","ink-gradient":"^3.0.0","ink-select-input":"^6.2.0","ink-spinner":"^5.0.0","ink-text-input":"^6.0.0","js-tiktoken":"^1.0.21","lodash-es":"^4.17.21",lowlight:"^3.3.0","lru-cache":"^11.2.4",nanoid:"^5.1.6",open:"^10.1.2",openai:"^6.2.0",picomatch:"^4.0.3",react:"^19.1.1","react-dom":"^19.1.1",semver:"^7.7.3","string-width":"^8.1.0",undici:"^7.16.0","write-file-atomic":"^7.0.0",ws:"^8.18.0",yaml:"^2.8.1",yargs:"^18.0.0",zod:"^3.24.2","zod-to-json-schema":"^3.24.6",zustand:"^5.0.9"},devDependencies:{"@biomejs/biome":"^2.2.4","@types/bun":"^1.3.4","@types/json-schema":"^7.0.15","@types/lodash-es":"^4.17.12","@types/node":"^22.15.24","@types/picomatch":"^4.0.2","@types/react":"^19.1.1","@types/react-dom":"^19.1.1","@types/semver":"^7.7.1","@types/write-file-atomic":"^4.0.3","@types/ws":"^8.5.12","@types/yargs":"^17.0.33","@vitest/coverage-v8":"^3.0.0",jsdom:"^26.0.0",typescript:"^5.9.2",vitest:"^3.0.0"}};function r(){return{name:e.name,version:e.version,description:e.description}}function n(){return e.version}function o(){return e.name}function c(){return e.description}function d(){return`v${e.version}`}function a(){return`v${e.version} © 2025 Blade Code`}
|
|
2
|
-
export{r as aa,n as ba,o as ca,c as da,d as ea,a as fa};
|
package/dist/chunk-qjhxk0yp.js
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
import{$a as k,Ra as a,Sa as b,Ta as c,Ua as d,Va as e,Wa as f,Xa as g,Ya as h,Za as i,_a as j,ab as l,bb as m,cb as n,db as o,gb as p,hb as q,ib as r}from"./chunk-g7bw2nkz.js";import"./chunk-3ak4p05x.js";import"./chunk-t44946na.js";import"./chunk-t7hd45rg.js";import"./chunk-39h8jdm3.js";import"./chunk-j1ehsp0x.js";import"./chunk-s53zeref.js";export{g as saveUserMessage,i as saveToolUse,j as saveToolResult,k as saveCompaction,h as saveAssistantMessage,a as isRootSystemPrompt,l as handleTaskListUpdate,m as handleSkillActivation,n as extractModelSwitch,r as executeLoopGenerator,c as drainLoop,f as checkStopHook,d as checkOutputRecovery,e as checkIncompleteIntent,q as checkAndCompactInLoop,o as applyToolDomainEffects,p as StreamingToolExecutor,b as ConversationState};
|
package/dist/chunk-t344cq6m.js
DELETED
|
@@ -1,4 +0,0 @@
|
|
|
1
|
-
import{N as l,O as c,P as m,S as g,U as d}from"./chunk-8zh508xk.js";import{yb as u}from"./chunk-39h8jdm3.js";async function C(){if((await l().initialize(u(),[])).plugins.length>0)await c()}async function h(t){let e=t.message||t._?.[0];if(typeof e==="string"&&e.trim())return e;let n=t.stdin??process.stdin;if(!n.isTTY){let a=[];for await(let i of n)a.push(i);return Buffer.concat(a).toString("utf-8").trim()}if(t.defaultMessage!==void 0)return t.defaultMessage;throw Error("No input provided")}async function R(t){if(!g(t))return{mode:"agent",content:t};let e=await d(t,{cwd:u(),workspaceRoot:u()});if(!e.success)return{mode:"output",content:`Error: ${e.error||"未知错误"}`,exitCode:1};let n=e.data;if(n?.action==="invoke_skill"){let{skillName:a,skillArgs:i}=n;return{mode:"agent",content:i?`Please use the "${a}" skill to help me with: ${i}`:`Please use the "${a}" skill.`}}if(n?.action==="invoke_custom_command"||n?.action==="invoke_plugin_command"){let a=n.processedContent;return{mode:"agent",content:`# Custom Command: /${n.commandName}
|
|
2
|
-
|
|
3
|
-
${a}`}}return{mode:"output",content:e.content||e.message||"",exitCode:0}}function r(t){if(!(t instanceof Error))return!1;return t.message.includes("未找到会话")||t.message.includes("ENOENT")||t.message.includes("no such file")}async function s(t){try{return await m.loadSession(t,u())}catch(e){if(r(e))return null;throw e}}async function N(t){if(t.resume===!0||t.resume==="true")throw Error("--resume without a session ID is only supported in interactive UI mode");if(typeof t.resume==="string"&&t.resume.length>0)return{sessionId:t.resume,messages:await m.loadSession(t.resume)};if(t.continue){let e=await m.listSessions();if(e.length>0){let n=e[0].sessionId;return{sessionId:n,messages:await m.loadSession(n)}}}if(t.sessionId)return{sessionId:t.sessionId,messages:await s(t.sessionId)??[]};return{sessionId:`${t.fallbackSessionPrefix}-${Date.now()}`,messages:[]}}
|
|
4
|
-
export{C as p,h as q,R as r,N as s};
|
package/dist/chunk-t44946na.js
DELETED
|
@@ -1,41 +0,0 @@
|
|
|
1
|
-
import{yb as l}from"./chunk-39h8jdm3.js";import{Ib as x}from"./chunk-j1ehsp0x.js";import{Kb as N}from"./chunk-s53zeref.js";import{nanoid as j}from"nanoid";var U;((R)=>{R.PreToolUse="PreToolUse";R.PostToolUse="PostToolUse";R.PostToolUseFailure="PostToolUseFailure";R.PermissionRequest="PermissionRequest";R.UserPromptSubmit="UserPromptSubmit";R.SessionStart="SessionStart";R.SessionEnd="SessionEnd";R.Stop="Stop";R.SubagentStop="SubagentStop";R.Notification="Notification";R.Compaction="Compaction"})(U||={});var w;((W)=>{W.Approve="approve";W.Block="block";W.Async="async"})(w||={});var M;((W)=>{W.Allow="allow";W.Deny="deny";W.Ask="ask"})(M||={});var C={enabled:!1,defaultTimeout:60,timeoutBehavior:"ignore",failureBehavior:"ignore",maxConcurrentHooks:5,httpPolicy:{},PreToolUse:[],PostToolUse:[{name:"builtin:code-review-sensor",matcher:{tools:"Edit|Write"},hooks:[{type:"prompt",prompt:"审查此代码变更,重点检查:"+"1) 安全漏洞(命令注入、XSS、SQL 注入、硬编码密钥/密码)"+"2) 明显的逻辑错误(无限循环、off-by-one、空引用)"+"3) 类型安全问题。"+"如果发现严重问题,将问题描述放在 hookSpecificOutput.additionalContext 中。"+"如果没有严重问题,返回 approve。",model:void 0,timeout:15}]}],PostToolUseFailure:[],PermissionRequest:[],UserPromptSubmit:[],SessionStart:[],SessionEnd:[],Stop:[],SubagentStop:[],Notification:[],Compaction:[]};function S(Q,Y){return{...Q,...Y,PreToolUse:Y.PreToolUse??Q.PreToolUse,PostToolUse:Y.PostToolUse??Q.PostToolUse,PostToolUseFailure:Y.PostToolUseFailure??Q.PostToolUseFailure,PermissionRequest:Y.PermissionRequest??Q.PermissionRequest,UserPromptSubmit:Y.UserPromptSubmit??Q.UserPromptSubmit,SessionStart:Y.SessionStart??Q.SessionStart,SessionEnd:Y.SessionEnd??Q.SessionEnd,Stop:Y.Stop??Q.Stop,SubagentStop:Y.SubagentStop??Q.SubagentStop,Notification:Y.Notification??Q.Notification,Compaction:Y.Compaction??Q.Compaction}}function m(){let Q={};if(process.env.BLADE_DISABLE_HOOKS==="true")Q.enabled=!1;if(process.env.BLADE_HOOK_TIMEOUT){let Y=parseInt(process.env.BLADE_HOOK_TIMEOUT,10);if(!isNaN(Y)&&Y>0)Q.defaultTimeout=Y}return Q}class E{executedHooks=new Map;canExecute(Q,Y){if(!this.executedHooks.has(Q))this.executedHooks.set(Q,new Set);if(this.executedHooks.get(Q).has(Y))return console.warn(`[HookGuard] Hook ${Y} for tool ${Q} already executed, skipping`),!1;return!0}markExecuted(Q,Y){let Z=this.executedHooks.get(Q);if(Z)Z.add(Y)}cleanup(Q){this.executedHooks.delete(Q)}cleanupAll(){this.executedHooks.clear()}}import{z as $}from"zod";var L=$.object({hook_event_name:$.nativeEnum(U),hook_execution_id:$.string(),timestamp:$.string(),project_dir:$.string(),session_id:$.string(),permission_mode:$.enum(["default","autoEdit","yolo","plan"]),_metadata:$.object({blade_version:$.string(),hook_timeout_ms:$.number()}).optional()}),u=L.extend({hook_event_name:$.literal("PreToolUse"),tool_name:$.string(),tool_use_id:$.string(),tool_input:$.record($.unknown())}),i=L.extend({hook_event_name:$.literal("PostToolUse"),tool_name:$.string(),tool_use_id:$.string(),tool_input:$.record($.unknown()),tool_response:$.unknown()}),a=L.extend({hook_event_name:$.literal("Stop"),reason:$.string().optional()}),c=L.extend({hook_event_name:$.literal("PostToolUseFailure"),tool_name:$.string(),tool_use_id:$.string(),tool_input:$.record($.unknown()),error:$.string(),error_type:$.string().optional(),is_interrupt:$.boolean(),is_timeout:$.boolean()}),s=L.extend({hook_event_name:$.literal("PermissionRequest"),tool_name:$.string(),tool_use_id:$.string(),tool_input:$.record($.unknown())}),r=L.extend({hook_event_name:$.literal("UserPromptSubmit"),user_prompt:$.string(),has_images:$.boolean(),image_count:$.number()}),n=L.extend({hook_event_name:$.literal("SessionStart"),is_resume:$.boolean(),resume_session_id:$.string().optional()}),o=L.extend({hook_event_name:$.literal("SessionEnd"),reason:$.enum(["user_exit","error","max_turns","idle_timeout","ctrl_c","clear","logout","other"])}),e=L.extend({hook_event_name:$.literal("SubagentStop"),agent_type:$.string(),task_description:$.string().optional(),success:$.boolean(),result_summary:$.string().optional(),error:$.string().optional()}),t=L.extend({hook_event_name:$.literal("Notification"),notification_type:$.enum(["permission_prompt","idle_prompt","auth_success","elicitation_dialog","info","warning","error"]),title:$.string().optional(),message:$.string()}),QQ=L.extend({hook_event_name:$.literal("Compaction"),trigger:$.enum(["manual","auto"]),messages_before:$.number(),tokens_before:$.number()}),CQ=$.discriminatedUnion("hook_event_name",[u,i,a,c,s,r,n,o,e,t,QQ]),YQ=$.object({hookEventName:$.literal("PreToolUse"),permissionDecision:$.nativeEnum(M).optional(),permissionDecisionReason:$.string().optional(),updatedInput:$.record($.unknown()).optional()}),ZQ=$.object({hookEventName:$.literal("PostToolUse"),additionalContext:$.string().optional(),updatedOutput:$.unknown().optional()}),WQ=$.object({hookEventName:$.literal("Stop"),continue:$.boolean().optional(),continueReason:$.string().optional()}),XQ=$.object({hookEventName:$.literal("SubagentStop"),continue:$.boolean().optional(),continueReason:$.string().optional(),additionalContext:$.string().optional()}),$Q=$.object({hookEventName:$.literal("PermissionRequest"),permissionDecision:$.enum(["approve","deny","ask"]).optional(),permissionDecisionReason:$.string().optional()}),qQ=$.object({hookEventName:$.literal("UserPromptSubmit"),updatedPrompt:$.string().optional(),contextInjection:$.string().optional()}),JQ=$.object({hookEventName:$.literal("SessionStart"),env:$.record($.string()).optional()}),GQ=$.object({hookEventName:$.literal("Compaction"),blockCompaction:$.boolean().optional(),blockReason:$.string().optional()}),FQ=$.object({decision:$.object({behavior:$.nativeEnum(w)}).optional(),systemMessage:$.string().optional(),hookSpecificOutput:$.discriminatedUnion("hookEventName",[YQ,ZQ,WQ,XQ,$Q,qQ,JQ,GQ]).optional(),suppressOutput:$.boolean().optional()}),KQ=$.object({type:$.literal("command"),command:$.string(),timeout:$.number().positive().optional(),statusMessage:$.string().optional()}),VQ=$.object({type:$.literal("prompt"),prompt:$.string(),model:$.string().optional(),timeout:$.number().positive().optional()}),RQ=$.discriminatedUnion("type",[KQ,VQ]),I=$.union([$.string(),$.array($.string())]),_Q=$.object({tools:I.optional(),paths:I.optional(),commands:I.optional()}),D=$.object({name:$.string().optional(),matcher:_Q.optional(),hooks:$.array(RQ)}),SQ=$.object({enabled:$.boolean().optional(),defaultTimeout:$.number().positive().optional(),timeoutBehavior:$.enum(["ignore","deny","ask"]).optional(),failureBehavior:$.enum(["ignore","deny","ask"]).optional(),maxConcurrentHooks:$.number().positive().optional(),PreToolUse:$.array(D).optional(),PostToolUse:$.array(D).optional(),PostToolUseFailure:$.array(D).optional(),PermissionRequest:$.array(D).optional(),UserPromptSubmit:$.array(D).optional(),SessionStart:$.array(D).optional(),SessionEnd:$.array(D).optional(),Stop:$.array(D).optional(),SubagentStop:$.array(D).optional(),Notification:$.array(D).optional(),Compaction:$.array(D).optional()});function k(Q){let Y=FQ.safeParse(Q);if(Y.success)return{success:!0,data:Y.data};return{success:!1,error:Y.error}}class f{parse(Q,Y,Z){if(Q.timedOut){let X=Z?.timeoutBehavior||"ignore",q="Hook timeout";if(X==="deny")return{success:!1,blocking:!0,error:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else if(X==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:"Hook timeout. Continue?",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else return{success:!1,blocking:!1,warning:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}let W=this.tryParseJSON(Q.stdout);if(W){let X=k(W);if(!X.success){let J="error"in X?X.error.message:"Unknown validation error",G=Z?.failureBehavior||"ignore",F=`Invalid hook output JSON: ${J}`;if(G==="deny")return{success:!1,blocking:!0,error:F,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else if(G==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:`${F}. Continue?`,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};else return{success:!1,blocking:!1,warning:F,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}let q=X.data;if(q.decision?.behavior==="block")return{success:!1,blocking:!0,error:q.systemMessage||"Hook blocked execution",output:q,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};if(q.decision?.behavior==="async")return{success:!0,output:q,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y};return{success:!0,output:q,stdout:Q.stdout,stderr:Q.stderr,exitCode:Q.exitCode,hook:Y}}return this.parseByExitCode(Q,Y,Z)}parseByExitCode(Q,Y,Z){let W=Q.exitCode;switch(W){case 0:return{success:!0,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};case 2:return{success:!1,blocking:!0,error:Q.stderr||Q.stdout||"Hook returned exit code 2",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};case 124:{let X=Z?.timeoutBehavior||"ignore",q="Hook timeout";if(X==="deny")return{success:!1,blocking:!0,error:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else if(X==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:"Hook timeout. Continue?",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else return{success:!1,blocking:!1,warning:"Hook timeout",stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y}}default:{let X=Z?.failureBehavior||"ignore",q=Q.stderr||Q.stdout||`Hook failed with exit code ${W}`;if(X==="deny")return{success:!1,blocking:!0,error:q,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else if(X==="ask")return{success:!1,blocking:!1,needsConfirmation:!0,warning:`${q}. Continue?`,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y};else return{success:!1,blocking:!1,warning:q,stdout:Q.stdout,stderr:Q.stderr,exitCode:W,hook:Y}}}}tryParseJSON(Q){try{let Y=Q.trim();if(!Y)return null;let Z=Y.match(/^\s*(\{[\s\S]*\})\s*$/);if(!Z)return null;return JSON.parse(Z[1])}catch{return null}}}import{spawn as DQ}from"child_process";class z{content="";maxSize;constructor(Q){this.maxSize=Q}append(Q){if(this.content.length<this.maxSize){let Y=this.maxSize-this.content.length;this.content+=Q.substring(0,Y)}}getContent(){return this.content}isFull(){return this.content.length>=this.maxSize}}class H{MAX_STDOUT_SIZE=1048576;MAX_STDERR_SIZE=1048576;MAX_INPUT_SIZE=102400;async execute(Q,Y,Z,W){let X=JSON.stringify(Y);if(X.length>this.MAX_INPUT_SIZE)throw Error(`Hook input too large: ${X.length} bytes (max ${this.MAX_INPUT_SIZE})`);let q=this.createSafeEnv(Y),J=DQ(Q,[],{shell:!0,env:q,cwd:Z.projectDir,timeout:W}),G=new z(this.MAX_STDOUT_SIZE),F=new z(this.MAX_STDERR_SIZE);J.stdout.setEncoding("utf8"),J.stderr.setEncoding("utf8"),J.stdout.on("data",(K)=>{G.append(K)}),J.stderr.on("data",(K)=>{F.append(K)});try{J.stdin.write(X),J.stdin.end()}catch(K){throw J.kill("SIGTERM"),Error(`Failed to write hook input: ${K}`)}return new Promise((K,O)=>{let R=!1,_=!1,V=null,A=()=>{if(V&&Z.abortSignal)Z.abortSignal.removeEventListener("abort",V),V=null},B=setTimeout(()=>{R=!0,J.kill("SIGKILL")},W);if(J.on("close",(b)=>{if(_)return;_=!0,clearTimeout(B),A(),K({stdout:G.getContent(),stderr:F.getContent(),exitCode:R?124:b??1,timedOut:R})}),J.on("error",(b)=>{if(_)return;_=!0,clearTimeout(B),A(),O(b)}),Z.abortSignal)V=()=>{if(_)return;_=!0,clearTimeout(B),A(),J.kill("SIGTERM"),K({stdout:G.getContent(),stderr:"Hook cancelled by abort signal",exitCode:1,timedOut:!1})},Z.abortSignal.addEventListener("abort",V)})}createSafeEnv(Q){return{BLADE_PROJECT_DIR:Q.project_dir,BLADE_SESSION_ID:Q.session_id,BLADE_HOOK_EVENT:Q.hook_event_name,BLADE_TOOL_NAME:"tool_name"in Q?Q.tool_name:"",BLADE_TOOL_USE_ID:"tool_use_id"in Q?Q.tool_use_id:"",PATH:process.env.PATH||"",HOME:process.env.HOME||"",USER:process.env.USER||"",SHELL:process.env.SHELL||"/bin/sh"}}}class P extends Error{constructor(Q){super(Q);this.name="HttpHookSecurityError"}}function h(Q,Y={}){let Z;try{Z=new URL(Q)}catch{throw new P(`Invalid URL: ${Q}`)}if(Z.protocol!=="https:"&&Z.protocol!=="http:")throw new P(`Only http/https supported, got: ${Z.protocol}`);let W=Z.hostname.toLowerCase();if(LQ(W,Y.allowedHosts))return;if(Z.protocol==="http:"&&!Y.allowHttp)throw new P(`HTTP (non-TLS) blocked for ${W}; use https:// or add to allowedHosts`);if(jQ(W)&&!Y.allowLoopback)throw new P(`Loopback address blocked: ${W}; enable policy.allowLoopback or add to allowedHosts`);if(AQ(W)&&!Y.allowPrivateRanges)throw new P(`Private/link-local address blocked: ${W}; enable policy.allowPrivateRanges or add to allowedHosts`)}function LQ(Q,Y){if(!Y||Y.length===0)return!1;return Y.some((Z)=>{let W=Z.toLowerCase();if(W===Q)return!0;if(W.startsWith("*.")){let X=W.slice(1);return Q.endsWith(X)&&Q.length>X.length}return!1})}function jQ(Q){if(Q==="localhost")return!0;if(Q==="::1"||Q==="[::1]")return!0;if(/^127\.\d+\.\d+\.\d+$/.test(Q))return!0;return!1}function AQ(Q){let Y=Q.match(/^(\d+)\.(\d+)\.(\d+)\.(\d+)$/);if(!Y)return!1;let[Z,W]=Y.slice(1).map(Number);if(Z===10)return!0;if(Z===172&&W>=16&&W<=31)return!0;if(Z===192&&W===168)return!0;if(Z===169&&W===254)return!0;return!1}function p(Q){if(!Q)return{};let Y={};for(let[Z,W]of Object.entries(Q))Y[Z]=W.replace(/\$\{([A-Z_][A-Z0-9_]*)\}/gi,(X,q)=>{return process.env[q]??`\${${q}}`});return Y}var OQ=x("Execution"),BQ={PreToolUse:'{ "permissionDecision": "approve" | "deny" | "ask", "permissionDecisionReason": "...", "updatedInput": { ... } }',PostToolUse:'{ "additionalContext": "注入到工具结果的额外信息", "updatedOutput": "修改后的工具输出" }',Stop:'{ "continue": true, "continueReason": "继续执行的原因" } // continue: true 表示阻止停止',SubagentStop:'{ "continue": true, "continueReason": "...", "additionalContext": "..." }',PermissionRequest:'{ "permissionDecision": "approve" | "deny" | "ask", "permissionDecisionReason": "..." }',UserPromptSubmit:'{ "updatedPrompt": "修改后的提示词", "contextInjection": "注入的上下文" }',Compaction:'{ "blockCompaction": true, "blockReason": "阻止压缩的原因" }'};class v{processExecutor=new H;outputParser=new f;chatServiceCache=new Map;async executePreToolHooks(Q,Y,Z){if(Q.length===0)return{decision:"allow"};let W="tool_input"in Y?Y.tool_input:{},X=[];for(let q of Q)try{let J={...Y,...W&&{tool_input:W}},G=await this.executeHook(q,J,Z);if(!G.success){if(G.blocking)return{decision:"deny",reason:G.error};if(G.needsConfirmation)return{decision:"ask",reason:G.warning||G.error};if(G.warning)X.push(G.warning);continue}let F=G.output?.hookSpecificOutput;if(F&&"permissionDecision"in F){switch(F.permissionDecision){case"deny":return{decision:"deny",reason:F.permissionDecisionReason};case"ask":return{decision:"ask",reason:F.permissionDecisionReason};case"allow":break}if("updatedInput"in F&&F.updatedInput)W={...W,...F.updatedInput}}}catch(J){let G=J instanceof Error?J.message:String(J);if(X.push(`Hook failed: ${G}`),Z.config.failureBehavior==="deny")return{decision:"deny",reason:G};else if(Z.config.failureBehavior==="ask")return{decision:"ask",reason:`Hook failed: ${G}. Continue?`}}return{decision:"allow",modifiedInput:W,warning:X.length>0?X.join(`
|
|
2
|
-
`):void 0}}async executePostToolHooks(Q,Y,Z){if(Q.length===0)return{};let W=Z.config.maxConcurrentHooks||5,X=await this.executeHooksConcurrently(Q,Y,Z,W),q=[],J=void 0,G=[];for(let F of X){if(!F.success&&F.warning){G.push(F.warning);continue}let K=F.output?.hookSpecificOutput;if(K&&"additionalContext"in K){if(K.additionalContext)q.push(K.additionalContext);if("updatedOutput"in K&&K.updatedOutput!==void 0)J=K.updatedOutput}}return{additionalContext:q.length>0?q.join(`
|
|
3
|
-
|
|
4
|
-
`):void 0,modifiedOutput:J,warning:G.length>0?G.join(`
|
|
5
|
-
`):void 0}}async executeStopHooks(Q,Y,Z){if(Q.length===0)return{shouldStop:!0};let W=[];for(let X of Q)try{let q=await this.executeHook(X,Y,Z);if(!q.success){if(q.warning)W.push(q.warning);continue}let J=q.output?.hookSpecificOutput;if(J&&"continue"in J&&J.continue===!1)return{shouldStop:!1,continueReason:J.continueReason,warning:W.length>0?W.join(`
|
|
6
|
-
`):void 0}}catch(q){let J=q instanceof Error?q.message:String(q);W.push(`Hook failed: ${J}`)}return{shouldStop:!0,warning:W.length>0?W.join(`
|
|
7
|
-
`):void 0}}async executeSubagentStopHooks(Q,Y,Z){if(Q.length===0)return{shouldStop:!0};let W=[],X=[];for(let q of Q)try{let J=await this.executeHook(q,Y,Z);if(!J.success){if(J.warning)W.push(J.warning);continue}let G=J.output?.hookSpecificOutput;if(G&&"continue"in G){if(G.continue===!1)return{shouldStop:!1,continueReason:G.continueReason,warning:W.length>0?W.join(`
|
|
8
|
-
`):void 0};if("additionalContext"in G&&G.additionalContext)X.push(G.additionalContext)}}catch(J){let G=J instanceof Error?J.message:String(J);W.push(`Hook failed: ${G}`)}return{shouldStop:!0,additionalContext:X.length>0?X.join(`
|
|
9
|
-
|
|
10
|
-
`):void 0,warning:W.length>0?W.join(`
|
|
11
|
-
`):void 0}}async executePermissionRequestHooks(Q,Y,Z){if(Q.length===0)return{decision:"ask"};let W=[];for(let X of Q)try{let q=await this.executeHook(X,Y,Z);if(!q.success){if(q.warning)W.push(q.warning);continue}let J=q.output?.hookSpecificOutput;if(J&&"permissionDecision"in J){let G=J.permissionDecision;if(G==="approve"||G==="deny")return{decision:G,reason:J.permissionDecisionReason,warning:W.length>0?W.join(`
|
|
12
|
-
`):void 0}}}catch(q){let J=q instanceof Error?q.message:String(q);W.push(`Hook failed: ${J}`)}return{decision:"ask",warning:W.length>0?W.join(`
|
|
13
|
-
`):void 0}}async executeUserPromptSubmitHooks(Q,Y,Z){if(Q.length===0)return{proceed:!0};let W=[],X=[],q;for(let J of Q)try{let G=await this.executeHook(J,Y,Z);if(!G.success){if(G.blocking)return{proceed:!1,warning:G.error};if(G.warning)W.push(G.warning);continue}if(G.stdout&&G.stdout.trim())X.push(G.stdout.trim());let F=G.output?.hookSpecificOutput;if(F&&"updatedPrompt"in F){if(F.updatedPrompt)q=F.updatedPrompt;if(F.contextInjection)X.push(F.contextInjection)}}catch(G){let F=G instanceof Error?G.message:String(G);W.push(`Hook failed: ${F}`)}return{proceed:!0,updatedPrompt:q,contextInjection:X.length>0?X.join(`
|
|
14
|
-
|
|
15
|
-
`):void 0,warning:W.length>0?W.join(`
|
|
16
|
-
`):void 0}}async executeSessionStartHooks(Q,Y,Z){if(Q.length===0)return{proceed:!0};let W=[],X={};for(let q of Q)try{let J=await this.executeHook(q,Y,Z);if(!J.success){if(J.blocking)return{proceed:!1,warning:J.error};if(J.warning)W.push(J.warning);continue}let G=J.output?.hookSpecificOutput;if(G&&"env"in G&&G.env)Object.assign(X,G.env)}catch(J){let G=J instanceof Error?J.message:String(J);W.push(`Hook failed: ${G}`)}return{proceed:!0,env:Object.keys(X).length>0?X:void 0,warning:W.length>0?W.join(`
|
|
17
|
-
`):void 0}}async executeSessionEndHooks(Q,Y,Z){if(Q.length===0)return{};let W=[],X=Z.config.maxConcurrentHooks||5,q=await this.executeHooksConcurrently(Q,Y,Z,X);for(let J of q)if(!J.success&&J.warning)W.push(J.warning);return{warning:W.length>0?W.join(`
|
|
18
|
-
`):void 0}}async executePostToolUseFailureHooks(Q,Y,Z){if(Q.length===0)return{};let W=[],X=[],q=Z.config.maxConcurrentHooks||5,J=await this.executeHooksConcurrently(Q,Y,Z,q);for(let G of J){if(!G.success&&G.warning){W.push(G.warning);continue}if(G.stdout&&G.stdout.trim())X.push(G.stdout.trim())}return{additionalContext:X.length>0?X.join(`
|
|
19
|
-
|
|
20
|
-
`):void 0,warning:W.length>0?W.join(`
|
|
21
|
-
`):void 0}}async executeNotificationHooks(Q,Y,Z){let W="message"in Y?Y.message:"";if(Q.length===0)return{suppress:!1,message:W};let X=[],q=!1,J=W;for(let G of Q)try{let F=await this.executeHook(G,Y,Z);if(!F.success){if(F.warning)X.push(F.warning);continue}if(F.output?.suppressOutput){q=!0;break}if(F.stdout&&F.stdout.trim())J=F.stdout.trim()}catch(F){let K=F instanceof Error?F.message:String(F);X.push(`Hook failed: ${K}`)}return{suppress:q,message:J,warning:X.length>0?X.join(`
|
|
22
|
-
`):void 0}}async executeCompactionHooks(Q,Y,Z){if(Q.length===0)return{blockCompaction:!1};let W=[];for(let X of Q)try{let q=await this.executeHook(X,Y,Z);if(!q.success){if(q.warning)W.push(q.warning);continue}let J=q.output?.hookSpecificOutput;if(J&&"blockCompaction"in J&&J.blockCompaction)return{blockCompaction:!0,blockReason:J.blockReason,warning:W.length>0?W.join(`
|
|
23
|
-
`):void 0}}catch(q){let J=q instanceof Error?q.message:String(q);W.push(`Hook failed: ${J}`)}return{blockCompaction:!1,warning:W.length>0?W.join(`
|
|
24
|
-
`):void 0}}async executeHook(Q,Y,Z){if(Q.type==="command")return this.executeCommandHook(Q,Y,Z);if(Q.type==="prompt")return this.executePromptHook(Q,Y,Z);if(Q.type==="function")return this.executeFunctionHook(Q,Y,Z);if(Q.type==="http")return this.executeHttpHook(Q,Y,Z);throw Error(`Hook type ${Q.type} not supported`)}async executeCommandHook(Q,Y,Z){let W=(Q.timeout??Z.config.defaultTimeout??60)*1000;try{let X=await this.processExecutor.execute(Q.command,Y,Z,W);return this.outputParser.parse(X,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch(X){return{success:!1,blocking:!1,error:X instanceof Error?X.message:String(X),hook:Q}}}async executePromptHook(Q,Y,Z){let W=(Q.timeout??Z.config.defaultTimeout??60)*1000;try{let X=await this.getOrCreateChatService(Q.model),q="hook_event_name"in Y?String(Y.hook_event_name):"Unknown",J=this.buildPromptHookSystemMessage(Q,q),G=JSON.stringify(Y,null,2),F=new AbortController,K=setTimeout(()=>F.abort(),W),O;try{O=(await X.chat([{role:"system",content:J},{role:"user",content:G}],void 0,F.signal)).content}catch(_){if(F.signal.aborted||_ instanceof Error&&_.name==="AbortError")return this.outputParser.parse({stdout:"",stderr:"",exitCode:0,timedOut:!0},Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior});throw _}finally{clearTimeout(K)}let R=this.extractJsonFromLLMResponse(O);return this.outputParser.parse({stdout:R,stderr:"",exitCode:0,timedOut:!1},Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch(X){return OQ.warn(`[PromptHook] 执行失败: ${X instanceof Error?X.message:String(X)}`),{success:!1,blocking:!1,error:X instanceof Error?X.message:String(X),hook:Q}}}async executeFunctionHook(Q,Y,Z){let W=(Q.timeout??10)*1000;try{let X=await this.runFunctionWithTimeout(Q.handler,Y,Z,W);if(X===void 0||X===null)return{success:!0,hook:Q};let q={stdout:JSON.stringify(X),stderr:"",exitCode:0,timedOut:!1};return this.outputParser.parse(q,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch(X){if(X.isTimeout){let q={stdout:"",stderr:`Function hook timeout after ${W}ms`,exitCode:124,timedOut:!0};return this.outputParser.parse(q,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}return{success:!1,blocking:!1,error:X instanceof Error?X.message:String(X),hook:Q}}}runFunctionWithTimeout(Q,Y,Z,W){return new Promise((X,q)=>{let J=!1,G=setTimeout(()=>{if(J)return;J=!0;let K=Object.assign(Error(`Function hook timed out after ${W}ms`),{isTimeout:!0});q(K)},W),F=()=>{if(J)return;J=!0,clearTimeout(G),q(Error("Function hook aborted"))};Z.abortSignal?.addEventListener("abort",F,{once:!0}),Promise.resolve().then(()=>Q(Y,Z)).then((K)=>{if(J)return;J=!0,clearTimeout(G),Z.abortSignal?.removeEventListener("abort",F),X(K)},(K)=>{if(J)return;J=!0,clearTimeout(G),Z.abortSignal?.removeEventListener("abort",F),q(K)})})}async executeHttpHook(Q,Y,Z){let W=(Q.timeout??10)*1000,X=Q.maxResponseBytes??262144,q=Math.max(1,(Q.retries??0)+1);try{h(Q.url,Z.config.httpPolicy)}catch(K){return{success:!1,blocking:!0,error:K instanceof Error?K.message:String(K),hook:Q}}let J={"Content-Type":"application/json",...p(Q.headers)},G=JSON.stringify(Y),F;for(let K=0;K<q;K++){if(K>0){let V=Math.min(2**K*200,5000);await new Promise((A)=>setTimeout(A,V))}let O=new AbortController,R=setTimeout(()=>O.abort(),W),_=()=>O.abort();Z.abortSignal?.addEventListener("abort",_,{once:!0});try{let V=await fetch(Q.url,{method:"POST",headers:J,body:G,redirect:"manual",signal:O.signal});if(V.status>=300&&V.status<400){F=Error(`HTTP hook received redirect (${V.status}); redirects are disabled`);continue}if(!V.ok){if(F=Error(`HTTP hook returned ${V.status} ${V.statusText}`),V.status>=400&&V.status<500)break;continue}let B={stdout:await PQ(V,X),stderr:"",exitCode:0,timedOut:!1};return this.outputParser.parse(B,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior})}catch(V){let A=V.name==="AbortError"||V.code==="ABORT_ERR";if(A&&Z.abortSignal?.aborted){F=Error("HTTP hook aborted by caller");break}if(A){let B={stdout:"",stderr:`HTTP hook timeout after ${W}ms`,exitCode:124,timedOut:!0};if(K===q-1)return this.outputParser.parse(B,Q,{timeoutBehavior:Z.config.timeoutBehavior,failureBehavior:Z.config.failureBehavior});F=Error(`HTTP hook timeout attempt ${K+1}`);continue}F=V}finally{clearTimeout(R),Z.abortSignal?.removeEventListener("abort",_)}}return{success:!1,blocking:!1,error:F?.message??"HTTP hook failed",hook:Q}}async getOrCreateChatService(Q){let Y=Q||"__default__",Z=this.chatServiceCache.get(Y);if(Z)return Z;let{ensureStoreInitialized:W,getCurrentModel:X,getModelById:q}=await import("./chunk-kevt8v3j.js"),{createChatServiceAsync:J}=await import("./chunk-caw1bxy6.js");await W();let G=Q?q(Q)||X():X();if(!G)throw Error("PromptHook: 无法获取模型配置。请确保至少配置了一个模型。");let F=await J({provider:G.provider,apiKey:G.apiKey,baseUrl:G.baseUrl,model:G.model,temperature:G.temperature,maxContextTokens:G.maxContextTokens,maxOutputTokens:G.maxOutputTokens});return this.chatServiceCache.set(Y,F),F}buildPromptHookSystemMessage(Q,Y){let Z=BQ[Y]||"{ ... 根据事件类型返回相应字段 }";return`你是一个代码质量评估器,作为 Hook 在 ${Y} 事件中被触发。
|
|
25
|
-
|
|
26
|
-
`+`## 评估指令
|
|
27
|
-
${Q.prompt}
|
|
28
|
-
|
|
29
|
-
`+`## 输出格式
|
|
30
|
-
`+`必须返回一个 JSON 对象(不要包含任何其他文本):
|
|
31
|
-
`+`{
|
|
32
|
-
"decision": { "behavior": "approve" | "block" },
|
|
33
|
-
`+` "systemMessage": "可选的说明信息",
|
|
34
|
-
`+` "hookSpecificOutput": ${Z}
|
|
35
|
-
}
|
|
36
|
-
|
|
37
|
-
`+`重要规则:
|
|
38
|
-
`+`- 只输出 JSON,不要包含 markdown 代码块或其他文本
|
|
39
|
-
`+`- 如果没有发现问题,使用 "approve"
|
|
40
|
-
`+'- 只在发现严重问题时使用 "block"'}extractJsonFromLLMResponse(Q){let Y=Q.trim(),Z=Y.match(/^```(?:json)?\s*\n?([\s\S]*?)\n?\s*```$/);if(Z)return Z[1].trim();return Y}async executeHooksConcurrently(Q,Y,Z,W){let X=[],q=new Set;for(let J of Q){if(q.size>=W)await Promise.race(q);let G=this.executeHook(J,Y,Z).catch((K)=>({success:!1,blocking:!1,error:K instanceof Error?K.message:String(K),hook:J})),F=G.then(()=>{q.delete(F)}).catch(()=>{q.delete(F)});q.add(F),X.push(G)}return Promise.all(X)}}async function PQ(Q,Y){let Z=Q.body?.getReader();if(!Z)return"";let W=new TextDecoder,X=0,q="";while(!0){let{value:J,done:G}=await Z.read();if(G)break;if(X+=J.byteLength,X>Y){try{Z.cancel()}catch{}q+=W.decode(J.slice(0,Y-(X-J.byteLength)));break}q+=W.decode(J,{stream:!0})}return q+=W.decode(),q}import g from"picomatch";var TQ=/^([A-Za-z0-9_|]+)\((.+)\)$/;class d{matches(Q,Y){if(!Q)return!0;if(Q.tools&&Y.toolName){if(!this.matchTools(Q.tools,Y))return!1}if(Q.paths&&Y.filePath){if(!this.matchPaths(Q.paths,Y.filePath))return!1}if(Q.commands&&Y.command){if(!this.matchCommands(Q.commands,Y.command))return!1}return!0}matchTools(Q,Y){if(Array.isArray(Q))return Q.some((Z)=>this.matchToolWithParams(Z,Y));return this.matchToolWithParams(Q,Y)}matchPaths(Q,Y){if(Array.isArray(Q))return Q.some((W)=>{return g(W)(Y)});return g(Q)(Y)}matchCommands(Q,Y){if(Array.isArray(Q))return Q.some((Z)=>this.matchPattern(Y,Z));return this.matchPattern(Y,Q)}matchToolWithParams(Q,Y){let{toolName:Z,command:W,filePath:X}=Y,q=TQ.exec(Q);if(!q)return this.matchPattern(Z,Q);let[,J,G]=q;if(!this.matchPattern(Z,J))return!1;let F=this.getArgValue(Z,W,X);if(!F)return!1;return this.matchGlobOrPattern(F,G)}getArgValue(Q,Y,Z){if(Q==="Bash"||Q==="BashTool")return Y;if(["Read","Edit","Write","Glob","Grep"].includes(Q))return Z;return Y||Z}matchGlobOrPattern(Q,Y){if(/[*?[\]{}!]/.test(Y))try{return g(Y,{bash:!0,dot:!0})(Q)}catch{}if(Y.endsWith("*")){let Z=Y.slice(0,-1);return Q.startsWith(Z)}return Q===Y}matchPattern(Q,Y){if(Y==="*")return!0;if(!Y.includes("|")&&!/[.*+?^${}()|[\]\\]/.test(Y))return Q===Y;if(Y.includes("|"))return Y.split("|").map((W)=>W.trim()).includes(Q);try{return new RegExp(Y).test(Q)}catch{return Q===Y}}}class T{static instance=null;config=C;executor=new v;guard=new E;matcher=new d;sessionDisabled=!1;constructor(){}static getInstance(){if(!T.instance)T.instance=new T;return T.instance}loadConfig(Q){let Y=S(C,Q),Z=m();Y=S(Y,Z),this.config=Y}isEnabled(){if(!this.config.enabled)return!1;if(this.sessionDisabled)return!1;return!0}disable(){this.sessionDisabled=!0,console.log("[HookManager] Hooks disabled for this session")}enable(){this.sessionDisabled=!1,console.log("[HookManager] Hooks enabled for this session")}getConfig(){return this.config}registerFunction(Q,Y,Z,W){let X={type:"function",handler:Z,timeout:W?.timeout},q={name:W?.name??`inline-${Q}-${Date.now()}`,matcher:Y,hooks:[X]},J=this.config[Q]??[];return this.config[Q]=[...J,q],()=>{let G=this.config[Q];if(!G)return;this.config[Q]=G.filter((F)=>F!==q)}}async reloadConfig(){let Q=await import("node:fs/promises"),Y=await import("node:path");try{let Z=Y.join(l(),".blade","settings.local.json"),W=await Q.readFile(Z,"utf-8"),X=JSON.parse(W);if(X.hooks)this.loadConfig(X.hooks)}catch{}}async executePreToolHooks(Q,Y,Z,W){if(!this.isEnabled())return{decision:"allow"};if(W.permissionMode==="plan")return{decision:"allow"};if(!this.guard.canExecute(Y,"PreToolUse"))return{decision:"allow"};let X={hook_event_name:"PreToolUse",hook_execution_id:j(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,project_dir:W.projectDir,session_id:W.sessionId,permission_mode:W.permissionMode},q=this.getMatchingHooks("PreToolUse",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(q.length===0)return{decision:"allow"};let J={projectDir:W.projectDir,sessionId:W.sessionId,permissionMode:W.permissionMode,config:this.config,abortSignal:W.abortSignal};try{let G=await this.executor.executePreToolHooks(q,X,J);if(this.guard.markExecuted(Y,"PreToolUse"),W.permissionMode==="yolo"){if(G.decision==="deny")return G;return{decision:"allow",modifiedInput:G.modifiedInput,warning:G.warning,reason:G.reason}}return G}catch(G){return console.error("[HookManager] Error executing PreToolUse hooks:",G),{decision:"allow",warning:`Hook execution failed: ${G instanceof Error?G.message:String(G)}`}}}async executePostToolHooks(Q,Y,Z,W,X){if(!this.isEnabled())return{};if(X.permissionMode==="plan")return{};if(!this.guard.canExecute(Y,"PostToolUse"))return{};let q={hook_event_name:"PostToolUse",hook_execution_id:j(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,tool_response:W,project_dir:X.projectDir,session_id:X.sessionId,permission_mode:X.permissionMode},J=this.getMatchingHooks("PostToolUse",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(J.length===0)return{};let G={projectDir:X.projectDir,sessionId:X.sessionId,permissionMode:X.permissionMode,config:this.config,abortSignal:X.abortSignal};try{let F=await this.executor.executePostToolHooks(J,q,G);return this.guard.markExecuted(Y,"PostToolUse"),F}catch(F){return console.error("[HookManager] Error executing PostToolUse hooks:",F),{warning:`Hook execution failed: ${F instanceof Error?F.message:String(F)}`}}finally{this.guard.cleanup(Y)}}async executeStopHooks(Q){if(!this.isEnabled())return{shouldStop:!0};let Y={hook_event_name:"Stop",hook_execution_id:j(),timestamp:new Date().toISOString(),project_dir:Q.projectDir,session_id:Q.sessionId,permission_mode:Q.permissionMode,reason:Q.reason},Z=this.getMatchingHooks("Stop",{});if(Z.length===0)return{shouldStop:!0};let W={projectDir:Q.projectDir,sessionId:Q.sessionId,permissionMode:Q.permissionMode,config:this.config,abortSignal:Q.abortSignal};try{return await this.executor.executeStopHooks(Z,Y,W)}catch(X){return console.error("[HookManager] Error executing Stop hooks:",X),{shouldStop:!0,warning:`Hook execution failed: ${X instanceof Error?X.message:String(X)}`}}}async executeSubagentStopHooks(Q,Y){if(!this.isEnabled())return{shouldStop:!0};let Z={hook_event_name:"SubagentStop",hook_execution_id:j(),timestamp:new Date().toISOString(),project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode,agent_type:Q,task_description:Y.taskDescription,success:Y.success,result_summary:Y.resultSummary,error:Y.error},W=this.getMatchingHooks("SubagentStop",{});if(W.length===0)return{shouldStop:!0};let X={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeSubagentStopHooks(W,Z,X)}catch(q){return console.error("[HookManager] Error executing SubagentStop hooks:",q),{shouldStop:!0,warning:`Hook execution failed: ${q instanceof Error?q.message:String(q)}`}}}async executePermissionRequestHooks(Q,Y,Z,W){if(!this.isEnabled())return{decision:"ask"};let X={hook_event_name:"PermissionRequest",hook_execution_id:j(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,project_dir:W.projectDir,session_id:W.sessionId,permission_mode:W.permissionMode},q=this.getMatchingHooks("PermissionRequest",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(q.length===0)return{decision:"ask"};let J={projectDir:W.projectDir,sessionId:W.sessionId,permissionMode:W.permissionMode,config:this.config,abortSignal:W.abortSignal};try{return await this.executor.executePermissionRequestHooks(q,X,J)}catch(G){return console.error("[HookManager] Error executing PermissionRequest hooks:",G),{decision:"ask",warning:`Hook execution failed: ${G instanceof Error?G.message:String(G)}`}}}async executeUserPromptSubmitHooks(Q,Y){if(!this.isEnabled())return{proceed:!0};let Z={hook_event_name:"UserPromptSubmit",hook_execution_id:j(),timestamp:new Date().toISOString(),user_prompt:Q,has_images:Y.hasImages,image_count:Y.imageCount,project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode},W=this.getMatchingHooks("UserPromptSubmit",{});if(W.length===0)return{proceed:!0};let X={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeUserPromptSubmitHooks(W,Z,X)}catch(q){return console.error("[HookManager] Error executing UserPromptSubmit hooks:",q),{proceed:!0,warning:`Hook execution failed: ${q instanceof Error?q.message:String(q)}`}}}async executeSessionStartHooks(Q){if(!this.isEnabled())return{proceed:!0};let Y={hook_event_name:"SessionStart",hook_execution_id:j(),timestamp:new Date().toISOString(),project_dir:Q.projectDir,session_id:Q.sessionId,permission_mode:Q.permissionMode,is_resume:Q.isResume,resume_session_id:Q.resumeSessionId},Z=this.getMatchingHooks("SessionStart",{});if(Z.length===0)return{proceed:!0};let W={projectDir:Q.projectDir,sessionId:Q.sessionId,permissionMode:Q.permissionMode,config:this.config,abortSignal:Q.abortSignal};try{return await this.executor.executeSessionStartHooks(Z,Y,W)}catch(X){return console.error("[HookManager] Error executing SessionStart hooks:",X),{proceed:!0,warning:`Hook execution failed: ${X instanceof Error?X.message:String(X)}`}}}async executeSessionEndHooks(Q,Y){if(!this.isEnabled())return{};let Z={hook_event_name:"SessionEnd",hook_execution_id:j(),timestamp:new Date().toISOString(),project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode,reason:Q},W=this.getMatchingHooks("SessionEnd",{});if(W.length===0)return{};let X={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeSessionEndHooks(W,Z,X),{}}catch(q){return console.error("[HookManager] Error executing SessionEnd hooks:",q),{warning:`Hook execution failed: ${q instanceof Error?q.message:String(q)}`}}}async executePostToolUseFailureHooks(Q,Y,Z,W,X){if(!this.isEnabled())return{};let q={hook_event_name:"PostToolUseFailure",hook_execution_id:j(),timestamp:new Date().toISOString(),tool_name:Q,tool_use_id:Y,tool_input:Z,error:W,error_type:X.errorType,is_interrupt:X.isInterrupt,is_timeout:X.isTimeout,project_dir:X.projectDir,session_id:X.sessionId,permission_mode:X.permissionMode},J=this.getMatchingHooks("PostToolUseFailure",{toolName:Q,filePath:this.extractFilePath(Z),command:this.extractCommand(Q,Z)});if(J.length===0)return{};let G={projectDir:X.projectDir,sessionId:X.sessionId,permissionMode:X.permissionMode,config:this.config,abortSignal:X.abortSignal};try{return await this.executor.executePostToolUseFailureHooks(J,q,G)}catch(F){return console.error("[HookManager] Error executing PostToolUseFailure hooks:",F),{warning:`Hook execution failed: ${F instanceof Error?F.message:String(F)}`}}}async executeNotificationHooks(Q,Y,Z){if(!this.isEnabled())return{suppress:!1,message:Y};let W={hook_event_name:"Notification",hook_execution_id:j(),timestamp:new Date().toISOString(),project_dir:Z.projectDir,session_id:Z.sessionId,permission_mode:Z.permissionMode,notification_type:Q,title:Z.title,message:Y},X=this.getMatchingHooks("Notification",{});if(X.length===0)return{suppress:!1,message:Y};let q={projectDir:Z.projectDir,sessionId:Z.sessionId,permissionMode:Z.permissionMode,config:this.config,abortSignal:Z.abortSignal};try{return await this.executor.executeNotificationHooks(X,W,q)}catch(J){return console.error("[HookManager] Error executing Notification hooks:",J),{suppress:!1,message:Y,warning:`Hook execution failed: ${J instanceof Error?J.message:String(J)}`}}}async executeCompactionHooks(Q,Y){if(!this.isEnabled())return{blockCompaction:!1};let Z={hook_event_name:"Compaction",hook_execution_id:j(),timestamp:new Date().toISOString(),project_dir:Y.projectDir,session_id:Y.sessionId,permission_mode:Y.permissionMode,trigger:Q,messages_before:Y.messagesBefore,tokens_before:Y.tokensBefore},W=this.getMatchingHooks("Compaction",{});if(W.length===0)return{blockCompaction:!1};let X={projectDir:Y.projectDir,sessionId:Y.sessionId,permissionMode:Y.permissionMode,config:this.config,abortSignal:Y.abortSignal};try{return await this.executor.executeCompactionHooks(W,Z,X)}catch(q){return console.error("[HookManager] Error executing Compaction hooks:",q),{blockCompaction:!1,warning:`Hook execution failed: ${q instanceof Error?q.message:String(q)}`}}}getMatchingHooks(Q,Y){let Z=this.config[Q]||[],W=[];for(let X of Z)if(this.matcher.matches(X.matcher,Y))W.push(...X.hooks);return W}extractFilePath(Q){let Y=["file_path","path","filePath","source","target"];for(let Z of Y){let W=Q[Z];if(typeof W==="string")return W}return}extractCommand(Q,Y){if(Q==="Bash"||Q==="BashTool"){let Z=Y.command;if(typeof Z==="string")return Z}return}cleanup(){this.guard.cleanupAll()}}
|
|
41
|
-
export{U as nb,T as ob};
|
package/dist/chunk-t7hd45rg.js
DELETED
|
@@ -1,8 +0,0 @@
|
|
|
1
|
-
import{Ib as S}from"./chunk-j1ehsp0x.js";var f=["anthropic","openai","deepseek","google-generative-ai","groq","openrouter"],u={anthropic:"",openai:"","google-generative-ai":"",deepseek:"",groq:"",openrouter:"",azure:"",ollama:"",together:"",fireworks:"",mistral:"",cohere:"",perplexity:"",xai:"",default:""},c={anthropic:"https://api.anthropic.com",openai:"https://api.openai.com/v1",google:"https://generativelanguage.googleapis.com/v1beta",deepseek:"https://api.deepseek.com/v1",groq:"https://api.groq.com/openai/v1",openrouter:"https://openrouter.ai/api/v1",togetherai:"https://api.together.xyz/v1","fireworks-ai":"https://api.fireworks.ai/inference/v1",mistral:"https://api.mistral.ai/v1",cohere:"https://api.cohere.ai/v1",perplexity:"https://api.perplexity.ai",xai:"https://api.x.ai/v1",cerebras:"https://api.cerebras.ai/v1","novita-ai":"https://api.novita.ai/v3/openai",nvidia:"https://integrate.api.nvidia.com/v1"},T={anthropic:{"anthropic-beta":"claude-code-20250219,interleaved-thinking-2025-05-14,fine-grained-tool-streaming-2025-05-14"},openrouter:{"HTTP-Referer":"https://github.com/anthropics/blade","X-Title":"Blade"},cerebras:{"X-Cerebras-3rd-Party-Integration":"blade"},vercel:{"http-referer":"https://github.com/anthropics/blade","x-title":"blade"},zenmux:{"HTTP-Referer":"https://github.com/anthropics/blade","X-Title":"blade"}},H=(Q)=>{return T[Q]||{}};import{createAnthropic as I}from"@ai-sdk/anthropic";import{createAzure as K}from"@ai-sdk/azure";import{createDeepSeek as A}from"@ai-sdk/deepseek";import{createGoogleGenerativeAI as E}from"@ai-sdk/google";import{createOpenAI as y}from"@ai-sdk/openai";import{createOpenAICompatible as V}from"@ai-sdk/openai-compatible";import{generateText as U,jsonSchema as x,streamText as M}from"ai";var j=S("Chat");function O(Q){let q=new Set;for(let X of Q)if(X.role==="assistant"&&X.tool_calls)for(let W of X.tool_calls)q.add(W.id);return Q.filter((X)=>{if(X.role==="tool"){if(!X.tool_call_id)return!1;return q.has(X.tool_call_id)}return!0})}function L(Q){if(typeof Q==="string")return Q;return Q.filter((q)=>q.type==="text").map((q)=>q.text).join(`
|
|
2
|
-
`)}function h(Q){let q=Q;return q.text??q.textDelta??q.delta}function C(Q,q={}){try{return JSON.parse(Q)}catch{return j.warn("[VercelAIChatService] Failed to parse JSON, using fallback",{str:Q}),q}}class R{model;config;constructor(Q){this.config=Q,this.model=this.createModel(Q),j.debug("[VercelAIChatService] Initialized",{provider:Q.provider,model:Q.model})}createModel(Q){let{provider:q,apiKey:X,baseUrl:W,model:Y,customHeaders:Z,apiVersion:J}=Q;switch(q){case"openai":return y({apiKey:X,baseURL:W||void 0,headers:Z})(Y);case"anthropic":return I({apiKey:X,baseURL:W||void 0,headers:Z})(Y);case"gemini":{if(W&&!this.isGeminiOfficialUrl(W))return V({name:"gemini",apiKey:X,baseURL:W,headers:Z})(Y);return E({apiKey:X,baseURL:W||void 0})(Y)}case"google":case"google-generative-ai":case"google-vertex":{if(W&&!this.isGeminiOfficialUrl(W))return V({name:q,apiKey:X,baseURL:W,headers:Z})(Y);return E({apiKey:X,baseURL:W||void 0})(Y)}case"azure-openai":{let $=this.extractAzureResourceName(W);if($)return K({apiKey:X,resourceName:$,apiVersion:J||"2024-08-01-preview"})(Y);let _=this.buildAzureBaseUrl(W,Y);return V({name:"azure-openai",apiKey:X,baseURL:_,headers:{...Z,"api-key":X},queryParams:{"api-version":J||"2024-08-01-preview"}})(Y)}case"azure":{let $=this.extractAzureResourceName(W);if($)return K({apiKey:X,resourceName:$,apiVersion:J||"2024-08-01-preview"})(Y);let _=this.buildAzureBaseUrl(W,Y);return V({name:"azure-openai",apiKey:X,baseURL:_,headers:{...Z,"api-key":X},queryParams:{"api-version":J||"2024-08-01-preview"}})(Y)}case"deepseek":return A({apiKey:X,baseURL:W||void 0,headers:Z})(Y);default:return V({name:q,apiKey:X,baseURL:W,headers:Z})(Y)}}extractAzureResourceName(Q){if(!Q)return;let q=Q.match(/https:\/\/([^.]+)\.openai\.azure(?:\.com|\.us|\.cn|\.de)/);return q?q[1]:void 0}buildAzureBaseUrl(Q,q){if(!Q)return"";let X=Q.replace(/\/$/,"").replace(/\?.*$/,"");if(X.includes("/openai/deployments/"))return X;return`${X}/openai/deployments/${q}`}isGeminiOfficialUrl(Q){return Q.includes("generativelanguage.googleapis.com")||Q.includes("aiplatform.googleapis.com")}needsThinkingReplayMetadata(){return/deepseek/i.test([this.config.provider,this.config.model,this.config.baseUrl].join(" "))}getThinkingReplayProviderOptions(Q){let q=Q?.trim();if(!q||!this.needsThinkingReplayMetadata())return;return{openaiCompatible:{reasoning_content:q}}}shouldFlattenDeepSeekThinkingToolHistory(){return this.needsThinkingReplayMetadata()}flattenDeepSeekThinkingToolHistory(Q){let q=[],X,W=()=>{if(!X)return;q.push({role:"assistant",content:L(X.content).trim()||"I will use the requested tool and continue from its result."}),X=void 0};for(let Y of Q){if(Y.role==="system"){W(),q.push({role:"user",content:`<system>
|
|
3
|
-
${L(Y.content)}
|
|
4
|
-
</system>`});continue}if(Y.role==="assistant"&&Y.tool_calls&&Y.tool_calls.length>0&&Y.reasoningContent?.trim()){W(),X=Y;continue}if(Y.role==="tool"&&X){let Z=X.tool_calls?.find(($)=>$.id===Y.tool_call_id),J=Z?.type==="function"?Z.function:void 0;q.push({role:"user",content:`Tool result for ${Y.name||J?.name||"tool"}${J?.arguments?`(${J.arguments})`:""}:
|
|
5
|
-
${L(Y.content)}
|
|
6
|
-
|
|
7
|
-
Continue from this tool result.`});continue}W(),q.push(Y)}return W(),q}convertMessages(Q){let q=[],X=this.shouldFlattenDeepSeekThinkingToolHistory()?this.flattenDeepSeekThinkingToolHistory(Q):Q;for(let W of X)if(W.role==="system")if(Array.isArray(W.content)){let Y=W.content.find((J)=>J.type==="text"),Z={role:"system",content:L(W.content)};if(Y?.providerOptions)Z.providerOptions=Y.providerOptions;q.push(Z)}else q.push({role:"system",content:W.content});else if(W.role==="user")if(Array.isArray(W.content)){let Y=W.content.map((Z)=>{if(Z.type==="text"){let J={type:"text",text:Z.text};if(Z.providerOptions)J.providerOptions=Z.providerOptions;return J}return{type:"image",image:Z.image_url.url}});q.push({role:"user",content:Y})}else q.push({role:"user",content:W.content});else if(W.role==="assistant")if(W.tool_calls&&W.tool_calls.length>0){let Y=W.tool_calls.map((_)=>{let w=_.function;return{type:"tool-call",toolCallId:_.id,toolName:w?.name||"",input:C(w?.arguments||"{}",{})}}),Z=L(W.content),J=W.reasoningContent?.trim()?[{type:"reasoning",text:W.reasoningContent}]:[],$=this.getThinkingReplayProviderOptions(W.reasoningContent);if(Z){let _={role:"assistant",content:[...J,{type:"text",text:Z},...Y]};if($)_.providerOptions=$;q.push(_)}else if(J.length>0){let _={role:"assistant",content:[...J,...Y]};if($)_.providerOptions=$;q.push(_)}else q.push({role:"assistant",content:Y})}else if(W.reasoningContent?.trim()){let Y=L(W.content),Z=[{type:"reasoning",text:W.reasoningContent}];if(Y)Z.push({type:"text",text:Y});let J={role:"assistant",content:Z},$=this.getThinkingReplayProviderOptions(W.reasoningContent);if($)J.providerOptions=$;q.push(J)}else q.push({role:"assistant",content:L(W.content)});else if(W.role==="tool")q.push({role:"tool",content:[{type:"tool-result",toolCallId:W.tool_call_id,toolName:W.name||"unknown",output:{type:"text",value:L(W.content)}}]});if(this.config.enablePromptCaching&&this.config.provider==="anthropic"){let W=0;for(let Y of q)if(Y.role==="system"&&W<2)Y.providerOptions={anthropic:{cacheControl:{type:"ephemeral"}}},W++;for(let Y=q.length-1;Y>=0;Y--)if(q[Y].role==="user"){q[Y].providerOptions={anthropic:{cacheControl:{type:"ephemeral"}}};break}}return q}convertTools(Q){if(!Q||Q.length===0)return;let q={};for(let X of Q)q[X.name]={description:X.description,inputSchema:x(X.parameters)};return q}convertToolCalls(Q){return Q.map((q)=>({id:q.toolCallId,type:"function",function:{name:q.toolName,arguments:JSON.stringify(q.args??q.input??{})}}))}convertUsage(Q,q){if(!Q)return;let X=Q.promptTokens??Q.inputTokens??0,W=Q.completionTokens??Q.outputTokens??0,Y={promptTokens:X,completionTokens:W,totalTokens:Q.totalTokens??X+W};if(Q.reasoningTokens)Y.reasoningTokens=Q.reasoningTokens;if(q?.anthropic){if(q.anthropic.cacheCreationInputTokens!==void 0)Y.cacheCreationInputTokens=q.anthropic.cacheCreationInputTokens;if(q.anthropic.cacheReadInputTokens!==void 0)Y.cacheReadInputTokens=q.anthropic.cacheReadInputTokens}return Y}isFallbackableError(Q){if(!(Q instanceof Error))return!1;let q=Q.message.toLowerCase();if(q.includes("timeout")||q.includes("econnrefused")||q.includes("enotfound"))return!0;let X=q.match(/status[:\s]*(\d{3})/),W=X?parseInt(X[1],10):Q.status;return W!==void 0&&[429,529,503,502,500].includes(W)}async retryWithBackoff(Q,q,X=2,W=1000){let Y;for(let Z=0;Z<=X;Z++)try{return await Q()}catch(J){if(Y=J,Z>=X)break;if(!this.isFallbackableError(J))throw J;if(q?.aborted)throw J;let $=W*Math.pow(2,Z);j.debug(`[VercelAIChatService] Retry ${Z+1}/${X} after ${$}ms`),await new Promise((_)=>setTimeout(_,$))}throw Y}getFallbackModelIds(){let Q=[];if(this.config.fallbackModels&&this.config.fallbackModels.length>0)Q.push(...this.config.fallbackModels);else if(this.config.fallbackModel)Q.push(this.config.fallbackModel);return Q}getThinkingProviderOptions(){if(!this.config.supportsThinking)return;let Q=this.config.thinkingMode??"budget";if(Q==="off")return;let q=Q==="adaptive",X=this.config.thinkingBudget??1e4;if(this.config.provider==="anthropic")return{anthropic:{thinking:q?{type:"enabled",budgetTokens:0}:{type:"enabled",budgetTokens:X}}};if(this.config.provider==="deepseek"||/deepseek/i.test(this.config.model))return{deepseek:{thinking:q?{type:"enabled",budgetTokens:0}:{type:"enabled",budgetTokens:X}}};return}getStreamTextOptions(Q,q,X,W){let Y=W;if(this.config.timeout&&this.config.timeout>0){let $=AbortSignal.timeout(this.config.timeout);if(W)Y=AbortSignal.any([W,$]);else Y=$}let Z={model:Q,messages:q,tools:X,maxOutputTokens:this.config.maxOutputTokens,temperature:this.config.temperature??0,abortSignal:Y,allowSystemInMessages:!0},J=this.getThinkingProviderOptions();if(J)Z.providerOptions=J;return Z}async chat(Q,q,X){let W=Date.now();j.debug("[VercelAIChatService] Starting chat request");let Y=O(Q),Z=this.convertMessages(Y),J=this.convertTools(q),$=async(_)=>{let w=await U(this.getStreamTextOptions(_,Z,J,X)),N=w.toolCalls&&w.toolCalls.length>0?this.convertToolCalls(w.toolCalls):void 0,z=Array.isArray(w.reasoning)?w.reasoning.map((B)=>B.text).join(""):void 0;return{content:w.text,reasoningContent:z,toolCalls:N,usage:this.convertUsage(w.usage,w.providerMetadata),finishReason:w.finishReason}};try{return await this.retryWithBackoff(()=>$(this.model),X,this.config.maxRetries)}catch(_){let w=Date.now()-W;if(j.error("[VercelAIChatService] Chat failed after",w,"ms"),!this.isFallbackableError(_))throw _;let N=this.getFallbackModelIds();if(N.length===0)throw _;for(let z of N){j.warn(`[VercelAIChatService] Trying fallback model: ${z}`);let B=this.createModel({...this.config,model:z});try{return await $(B)}catch(F){if(j.warn(`[VercelAIChatService] Fallback ${z} failed: ${F instanceof Error?F.message:F}`),!this.isFallbackableError(F))throw F}}throw _}}async*streamChat(Q,q,X){let W=Date.now();j.debug("[VercelAIChatService] Starting stream request");let Y=O(Q),Z=this.convertMessages(Y),J=this.convertTools(q),$=async function*(B,F){let D=M(B.getStreamTextOptions(F,Z,J,X)),P=0;for await(let G of D.fullStream)switch(G.type){case"text-delta":yield{content:h(G)};break;case"reasoning-delta":yield{reasoningContent:h(G)};break;case"tool-call":yield{toolCalls:[{index:P++,id:G.toolCallId,type:"function",function:{name:G.toolName,arguments:JSON.stringify(G.args??G.input??{})}}]};break;case"finish":yield{finishReason:G.finishReason,usage:B.convertUsage(G.totalUsage,G.providerMetadata)};break}},_=this.config.maxRetries??2,w;for(let B=0;B<=_;B++)try{yield*$(this,this.model);let F=Date.now()-W;j.debug("[VercelAIChatService] Stream completed in",F,"ms");return}catch(F){if(w=F,!this.isFallbackableError(F))throw F;if(B<_){let D=1000*Math.pow(2,B);j.debug(`[VercelAIChatService] Stream retry ${B+1}/${_} after ${D}ms`),await new Promise((P)=>setTimeout(P,D));continue}}let N=Date.now()-W;j.error("[VercelAIChatService] Stream failed after",N,"ms");let z=this.getFallbackModelIds();if(z.length===0)throw w;for(let B of z){j.warn(`[VercelAIChatService] Stream fallback: ${B}`),yield{modelFallback:!0};let F=this.createModel({...this.config,model:B});try{yield*$(this,F);return}catch(D){if(j.warn(`[VercelAIChatService] Stream fallback ${B} failed`),!this.isFallbackableError(D))throw D}}throw w}getConfig(){return{...this.config}}updateConfig(Q){j.debug("[VercelAIChatService] Updating configuration"),this.config={...this.config,...Q},this.model=this.createModel(this.config),j.debug("[VercelAIChatService] Configuration updated")}}var v=S("Service");async function qq(Q){let q=Q;if(q.provider){let X=H(q.provider);if(Object.keys(X).length>0)q={...q,customHeaders:{...X,...q.customHeaders}},v.debug(`Injected ${q.provider} specific headers:`,Object.keys(X))}return b(q)}function b(Q){return new R(Q)}
|
|
8
|
-
export{f as pb,u as qb,c as rb,qq as sb};
|