@crewx/sdk 0.8.9-rc.15 → 0.8.9-rc.17
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/dist/activity-log/builder.d.ts +0 -4
- package/dist/config/pricing.d.ts +9 -0
- package/dist/esm/index.js +88 -52
- package/dist/esm/plugins/index.js +63 -27
- package/dist/esm/repository/index.js +61 -25
- package/dist/index.js +88 -52
- package/dist/plugins/index.js +63 -27
- package/dist/repository/index.d.ts +1 -1
- package/dist/repository/index.js +61 -25
- package/dist/repository/task.repository.d.ts +12 -0
- package/package.json +1 -1
package/dist/plugins/index.js
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
'use strict';var fs=require('fs'),
|
|
1
|
+
'use strict';var fs=require('fs'),D=require('path'),Pt=require('os'),crypto=require('crypto'),drizzleOrm=require('drizzle-orm'),sqliteCore=require('drizzle-orm/sqlite-core');function _interopDefault(e){return e&&e.__esModule?e:{default:e}}function _interopNamespace(e){if(e&&e.__esModule)return e;var n=Object.create(null);if(e){Object.keys(e).forEach(function(k){if(k!=='default'){var d=Object.getOwnPropertyDescriptor(e,k);Object.defineProperty(n,k,d.get?d:{enumerable:true,get:function(){return e[k]}});}})}n.default=e;return Object.freeze(n)}var D__namespace=/*#__PURE__*/_interopNamespace(D);var Pt__default=/*#__PURE__*/_interopDefault(Pt);var K=(d=>typeof require<"u"?require:typeof Proxy<"u"?new Proxy(d,{get:(t,e)=>(typeof require<"u"?require:t)[e]}):d)(function(d){if(typeof require<"u")return require.apply(this,arguments);throw Error('Dynamic require of "'+d+'" is not supported')});var v=class{detach(t){}};function _t(d){let t=e=>String(e).padStart(2,"0");return `${d.getFullYear()}${t(d.getMonth()+1)}${t(d.getDate())}T${t(d.getHours())}${t(d.getMinutes())}${t(d.getSeconds())}`}var nt=class extends v{name="file-logger";unsubs=[];logFiles=new Map;logsDir;version;constructor(t){super(),this.logsDir=D.join(t?.workspaceRoot??process.cwd(),".crewx","logs"),this.version=t?.version??"unknown";}attach(t){this.unsubs.push(t.on("task:start",e=>{try{fs.existsSync(this.logsDir)||fs.mkdirSync(this.logsDir,{recursive:!0});let s=_t(e.timestamp),n=D.join(this.logsDir,`${s}_${e.traceId}.log`);this.logFiles.set(e.traceId,n);let r=`=== TASK LOG: ${e.traceId} ===
|
|
2
2
|
CrewX Version: ${this.version}
|
|
3
3
|
Mode: ${e.mode}
|
|
4
4
|
Agent: ${e.agentRef}
|
|
@@ -6,13 +6,13 @@ Started: ${e.timestamp.toLocaleString()}
|
|
|
6
6
|
Message: ${e.message}
|
|
7
7
|
|
|
8
8
|
`;fs.writeFileSync(n,r,{encoding:"utf8",mode:384});}catch{}}),t.on("task:output",e=>{try{let s=this.logFiles.get(e.traceId);if(!s)return;let n=new Date().toISOString();fs.appendFileSync(s,`[${n}] STDOUT: ${e.output}
|
|
9
|
-
`,"utf8");}catch{}}),t.on("task:end",e=>{try{let s=this.logFiles.get(e.traceId);if(!s)return;let n=new Date().toLocaleString(),r=e.error?`failed: ${e.error.message}`:"completed successfully",
|
|
9
|
+
`,"utf8");}catch{}}),t.on("task:end",e=>{try{let s=this.logFiles.get(e.traceId);if(!s)return;let n=new Date().toLocaleString(),r=e.error?`failed: ${e.error.message}`:"completed successfully",i=`[${n}] INFO: Task ${r} in ${e.durationMs}ms
|
|
10
10
|
[${n}] INFO: Process closed with exit code: ${e.error?1:0}
|
|
11
|
-
`;fs.appendFileSync(s,
|
|
11
|
+
`;fs.appendFileSync(s,i,"utf8"),this.logFiles.delete(e.traceId);}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[],this.logFiles.clear();}};function vt(d){let t=D__namespace.resolve(d);return process.platform==="win32"&&(t=t.replace(/\\/g,"/"),t=t.replace(/^([A-Z]):/,(e,s)=>`${s.toLowerCase()}:`)),t.length>1&&!/^[a-zA-Z]:\/$/.test(t)&&(t=t.replace(/\/+$/,"")),t}function ft(d){let t=vt(d);return crypto.createHash("sha256").update(t).digest("hex")}var F=class{resolveDbPath(){return process.env.CREWX_DB?process.env.CREWX_DB:process.env.CREWX_TRACES_DB?process.env.CREWX_TRACES_DB:D.join(Pt__default.default.homedir(),".crewx","crewx.db")}resolveDbPaths(){return [this.resolveDbPath()]}isMissingTableError(t){return t instanceof Error&&/no such table:/i.test(t.message)}dbExists(t){return fs.existsSync(t??this.resolveDbPath())}};function E(d){let t=K("better-sqlite3"),{drizzle:e}=K("drizzle-orm/better-sqlite3"),s=new t(d);return s.exec("PRAGMA journal_mode = WAL"),s.exec("PRAGMA busy_timeout = 5000"),s.exec("PRAGMA foreign_keys = ON"),{db:e(s),runRaw:(n,r=[])=>s.prepare(n).run(...r),close:()=>s.close()}}var gt=new Set,It={agent_id:"TEXT",status:"TEXT DEFAULT 'running'",started_at:"TEXT",trace_id:"TEXT",parent_task_id:"TEXT",crewx_version:"TEXT",pid:"INTEGER",thread_id:"TEXT",workspace_id:"TEXT",workspace_ref:"TEXT",workspace_name:"TEXT",project_id:"TEXT",project_name:"TEXT"};function rt(d,t){return (d.get(`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='${t}'`)?.cnt??0)>0}function Mt(d,t){if(t>0||!rt(d,"tasks"))return;let e=d.all("PRAGMA table_info(tasks)"),s=new Set(e.map(n=>n.name));for(let[n,r]of Object.entries(It))s.has(n)||d.run(`ALTER TABLE tasks ADD COLUMN ${n} ${r}`);}var Ut={"0002_normalize_task_names":{workspace_name:"TEXT",project_name:"TEXT"}};function Bt(d,t,e){if(!rt(d,"__drizzle_migrations")||!rt(d,"tasks"))return;let s=d.all(e`SELECT hash FROM __drizzle_migrations`),n=new Set(s.map(i=>i.hash)),r=JSON.parse(fs.readFileSync(D__namespace.default.join(t,"meta/_journal.json"),"utf-8"));for(let i of r.entries){let a=Ut[i.tag];if(!a)continue;let l=D__namespace.default.join(t,`${i.tag}.sql`);if(!fs.existsSync(l))continue;let u=fs.readFileSync(l,"utf-8"),p=crypto.createHash("sha256").update(u).digest("hex");if(n.has(p))continue;let g=d.all("PRAGMA table_info(tasks)"),k=new Set(g.map(b=>b.name));for(let[b,j]of Object.entries(a))k.has(b)||(d.run(`ALTER TABLE tasks ADD COLUMN ${b} ${j}`),k.add(b));}}function Ft(d,t,e){let s=d.all(e`SELECT hash FROM __drizzle_migrations`),n=new Set(s.map(i=>i.hash)),r=JSON.parse(fs.readFileSync(D__namespace.default.join(t,"meta/_journal.json"),"utf-8"));for(let i of r.entries){let a=D__namespace.default.join(t,`${i.tag}.sql`);if(!fs.existsSync(a))continue;let l=fs.readFileSync(a,"utf-8"),u=crypto.createHash("sha256").update(l).digest("hex");if(n.has(u))continue;let p=/ALTER\s+TABLE\s+[`"]?(\w+)[`"]?\s+ADD\s+[`"]?(\w+)[`"]?/gi,g=[],k;for(;(k=p.exec(l))!==null;)g.push({table:k[1],column:k[2]});if(g.length===0||!l.split(/-->\s*statement-breakpoint/).map(W=>W.trim()).filter(Boolean).every(W=>/^ALTER\s+TABLE\s+.+\s+ADD\s+/i.test(W)))continue;g.every(({table:W,column:xt})=>d.all(`PRAGMA table_info("${W}")`).some(At=>At.name===xt))&&d.run(e`INSERT INTO __drizzle_migrations (hash, created_at) VALUES (${u}, ${i.when})`);}}function Ht(d){let{migrate:t}=K("drizzle-orm/better-sqlite3/migrator"),{sql:e}=K("drizzle-orm"),s=[D__namespace.default.join(__dirname,"../migrations"),D__namespace.default.join(__dirname,"migrations"),D__namespace.default.join(__dirname,"../../../../drizzle/migrations"),D__namespace.default.join(process.cwd(),"drizzle/migrations")],n=s.find(u=>fs.existsSync(D__namespace.default.join(u,"meta/_journal.json")));if(!n)throw new Error(`migrations folder not found. Searched:
|
|
12
12
|
${s.join(`
|
|
13
|
-
`)}`);let r=d.get(e`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),
|
|
13
|
+
`)}`);let r=d.get(e`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),i=0;r?.cnt&&(i=d.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),Mt(d,i),r?.cnt&&(Ft(d,n,e),Bt(d,n,e)),t(d,{migrationsFolder:n});let l=(d.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-i;if(l>0){let u=r?.cnt?"Database migrated":"Database initialized";console.log(`[crewx] ${u} (${l} migration${l>1?"s":""} applied).`);}}function X(d,t){gt.has(t)||(Ht(d),gt.add(t));}var _=class extends Error{code;cause;constructor(t,e,s){super(e),this.name="RepositoryError",this.code=t,this.cause=s,Object.setPrototypeOf(this,new.target.prototype);}};var H=sqliteCore.sqliteTable("workspaces",{id:sqliteCore.text("id").primaryKey(),slug:sqliteCore.text("slug").notNull().unique(),name:sqliteCore.text("name").notNull(),workspace_path:sqliteCore.text("workspace_path"),description:sqliteCore.text("description"),is_active:sqliteCore.integer("is_active").notNull().default(1),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull()});var o=sqliteCore.sqliteTable("tasks",{id:sqliteCore.text("id").primaryKey(),agent_id:sqliteCore.text("agent_id").notNull(),user_id:sqliteCore.text("user_id"),prompt:sqliteCore.text("prompt").notNull(),mode:sqliteCore.text("mode").notNull().default("execute"),status:sqliteCore.text("status").notNull().default("running"),result:sqliteCore.text("result"),error:sqliteCore.text("error"),started_at:sqliteCore.text("started_at").notNull(),completed_at:sqliteCore.text("completed_at"),duration_ms:sqliteCore.integer("duration_ms"),metadata:sqliteCore.text("metadata"),workspace_id:sqliteCore.text("workspace_id"),trace_id:sqliteCore.text("trace_id"),parent_task_id:sqliteCore.text("parent_task_id"),caller_agent_id:sqliteCore.text("caller_agent_id"),model:sqliteCore.text("model"),platform:sqliteCore.text("platform").default("cli"),crewx_version:sqliteCore.text("crewx_version"),input_tokens:sqliteCore.integer("input_tokens").default(0),output_tokens:sqliteCore.integer("output_tokens").default(0),cost_usd:sqliteCore.real("cost_usd").default(0),pid:sqliteCore.integer("pid"),rendered_prompt:sqliteCore.text("rendered_prompt"),command:sqliteCore.text("command"),coding_agent_command:sqliteCore.text("coding_agent_command"),exit_code:sqliteCore.integer("exit_code"),logs:sqliteCore.text("logs"),thread_id:sqliteCore.text("thread_id"),workspace_ref:sqliteCore.text("workspace_ref"),project_id:sqliteCore.text("project_id"),project_ref:sqliteCore.text("project_ref"),cached_input_tokens:sqliteCore.integer("cached_input_tokens").default(0)},d=>({idx_tasks_agent_id:sqliteCore.index("idx_tasks_agent_id").on(d.agent_id),idx_tasks_status:sqliteCore.index("idx_tasks_status").on(d.status),idx_tasks_started_at:sqliteCore.index("idx_tasks_started_at").on(d.started_at),idx_tasks_trace_id:sqliteCore.index("idx_tasks_trace_id").on(d.trace_id),idx_tasks_parent_task_id:sqliteCore.index("idx_tasks_parent_task_id").on(d.parent_task_id),idx_tasks_crewx_version:sqliteCore.index("idx_tasks_crewx_version").on(d.crewx_version),idx_tasks_pid:sqliteCore.index("idx_tasks_pid").on(d.pid),idx_tasks_thread_id:sqliteCore.index("idx_tasks_thread_id").on(d.thread_id),idx_tasks_workspace_id:sqliteCore.index("idx_tasks_workspace_id").on(d.workspace_id),idx_tasks_workspace_ref:sqliteCore.index("idx_tasks_workspace_ref").on(d.workspace_ref),idx_tasks_project_id:sqliteCore.index("idx_tasks_project_id").on(d.project_id),idx_tasks_ws_started:sqliteCore.index("idx_tasks_ws_started").on(d.workspace_id,d.started_at)}));var c=sqliteCore.sqliteTable("threads",{id:sqliteCore.text("id").primaryKey(),workspace_id:sqliteCore.text("workspace_id").references(()=>H.id,{onDelete:"set null"}),platform:sqliteCore.text("platform").notNull().default("cli"),title:sqliteCore.text("title"),first_message:sqliteCore.text("first_message"),last_message:sqliteCore.text("last_message"),message_count:sqliteCore.integer("message_count").notNull().default(0),created_at:sqliteCore.text("created_at").notNull(),updated_at:sqliteCore.text("updated_at").notNull(),metadata:sqliteCore.text("metadata"),title_locked:sqliteCore.integer("title_locked").notNull().default(0),pinned:sqliteCore.integer("pinned").notNull().default(0),starred:sqliteCore.integer("starred").notNull().default(0)},d=>({idx_threads_updated_at:sqliteCore.index("idx_threads_updated_at").on(d.updated_at),idx_threads_workspace_id:sqliteCore.index("idx_threads_workspace_id").on(d.workspace_id)}));var Yt=sqliteCore.sqliteTable("spans",{id:sqliteCore.text("id").primaryKey(),task_id:sqliteCore.text("task_id").references(()=>o.id,{onDelete:"set null"}),parent_span_id:sqliteCore.text("parent_span_id").references(()=>Yt.id,{onDelete:"set null"}),name:sqliteCore.text("name").notNull(),kind:sqliteCore.text("kind").notNull().default("internal"),status:sqliteCore.text("status").notNull().default("ok"),started_at:sqliteCore.text("started_at").notNull(),completed_at:sqliteCore.text("completed_at"),duration_ms:sqliteCore.integer("duration_ms"),input:sqliteCore.text("input"),output:sqliteCore.text("output"),error:sqliteCore.text("error"),attributes:sqliteCore.text("attributes")},d=>({idx_spans_task_id:sqliteCore.index("idx_spans_task_id").on(d.task_id),idx_spans_parent_span_id:sqliteCore.index("idx_spans_parent_span_id").on(d.parent_span_id)}));sqliteCore.sqliteTable("tool_calls",{id:sqliteCore.text("id").primaryKey(),task_id:sqliteCore.text("task_id").references(()=>o.id,{onDelete:"cascade"}),session_id:sqliteCore.text("session_id"),tool_name:sqliteCore.text("tool_name").notNull(),files:sqliteCore.text("files"),input:sqliteCore.text("input"),output:sqliteCore.text("output"),duration_ms:sqliteCore.integer("duration_ms"),timestamp:sqliteCore.text("timestamp").notNull()},d=>({idx_tool_calls_task_id:sqliteCore.index("idx_tool_calls_task_id").on(d.task_id),idx_tool_calls_tool_name:sqliteCore.index("idx_tool_calls_tool_name").on(d.tool_name),idx_tool_calls_timestamp:sqliteCore.index("idx_tool_calls_timestamp").on(d.timestamp)}));sqliteCore.sqliteTable("thread_boxes",{id:sqliteCore.text("id").primaryKey(),thread_id:sqliteCore.text("thread_id").notNull().references(()=>c.id,{onDelete:"cascade"}),seq:sqliteCore.integer("seq").notNull(),first_task_id:sqliteCore.text("first_task_id").notNull(),mid_task_id:sqliteCore.text("mid_task_id").notNull(),last_task_id:sqliteCore.text("last_task_id").notNull(),task_count:sqliteCore.integer("task_count").notNull(),summary:sqliteCore.text("summary"),source_tokens:sqliteCore.integer("source_tokens").notNull(),summary_tokens:sqliteCore.integer("summary_tokens"),created_at:sqliteCore.text("created_at").notNull()},d=>({idx_thread_boxes_thread_id:sqliteCore.index("idx_thread_boxes_thread_id").on(d.thread_id),idx_thread_boxes_seq:sqliteCore.index("idx_thread_boxes_seq").on(d.thread_id,d.seq),uniq_thread_boxes_thread_seq:sqliteCore.unique().on(d.thread_id,d.seq)}));sqliteCore.sqliteTable("request_logs",{id:sqliteCore.text("id").primaryKey(),path:sqliteCore.text("path").notNull(),method:sqliteCore.text("method").notNull(),status_code:sqliteCore.integer("status_code").notNull(),duration_ms:sqliteCore.integer("duration_ms").notNull(),ip:sqliteCore.text("ip"),request_headers:sqliteCore.text("request_headers"),response_headers:sqliteCore.text("response_headers"),request_body:sqliteCore.text("request_body"),response_body:sqliteCore.text("response_body"),query:sqliteCore.text("query"),user_id:sqliteCore.text("user_id"),project_id:sqliteCore.text("project_id"),partition_key:sqliteCore.text("partition_key").notNull(),timestamp:sqliteCore.text("timestamp").notNull().default(drizzleOrm.sql`(datetime('now'))`),metadata:sqliteCore.text("metadata")},d=>({idx_request_logs_timestamp:sqliteCore.index("idx_request_logs_timestamp").on(d.timestamp),idx_request_logs_path:sqliteCore.index("idx_request_logs_path").on(d.path),idx_request_logs_status_code:sqliteCore.index("idx_request_logs_status_code").on(d.status_code),idx_request_logs_partition_key:sqliteCore.index("idx_request_logs_partition_key").on(d.partition_key)}));var M="2026-05-09",ae="0.8.9-rc.13",U=10,B=parseInt(ae.split("rc.")[1]),tt=class extends F{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=D.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let n=D.dirname(e);fs.existsSync(n)||fs.mkdirSync(n,{recursive:true});}else if(!fs.existsSync(e))throw new _("NOT_FOUND","Database not found");let s=E(e);if(t)try{X(s.db,e);}catch(n){throw s.close(),n}return s}startTask(t){let e=this.openHandle(true);try{e.db.insert(o).values({id:t.id,agent_id:t.agentId,prompt:t.prompt,mode:t.mode,status:t.status,started_at:t.startedAt,pid:t.pid??null,parent_task_id:t.parentTaskId??null,caller_agent_id:t.callerAgentId??null,trace_id:t.traceId??null,command:t.command??null,metadata:t.metadata??null,workspace_id:t.workspaceId??null,platform:t.platform??"cli",crewx_version:t.crewxVersion??null,thread_id:t.threadId??null,model:t.model??null,rendered_prompt:t.renderedPrompt??null,coding_agent_command:t.codingAgentCommand??null}).onConflictDoNothing().run();}catch(s){throw s instanceof _?s:new _("DB_ERROR","Failed to start task",s)}finally{e.close();}}finishTask(t){let e=this.openHandle(true);try{e.runRaw(`UPDATE tasks SET status=?, result=?, error=?, completed_at=?, duration_ms=?,
|
|
14
14
|
exit_code=?, input_tokens=?, output_tokens=?, cached_input_tokens=?, cost_usd=?,
|
|
15
|
-
model=COALESCE(?, model) WHERE id=?`,[t.status,t.result??null,t.error??null,t.completedAt,t.durationMs??null,t.exitCode??null,t.inputTokens??0,t.outputTokens??0,t.cachedInputTokens??0,t.costUsd??0,t.model??null,t.id]);}catch(s){throw s instanceof
|
|
15
|
+
model=COALESCE(?, model) WHERE id=?`,[t.status,t.result??null,t.error??null,t.completedAt,t.durationMs??null,t.exitCode??null,t.inputTokens??0,t.outputTokens??0,t.cachedInputTokens??0,t.costUsd??0,t.model??null,t.id]);}catch(s){throw s instanceof _?s:new _("DB_ERROR","Failed to finish task",s)}finally{e.close();}}appendLog(t,e){let s=this.openHandle(true);try{s.db.transaction(n=>{let r=n.select({logs:o.logs}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get(),i=r?.logs?JSON.parse(r.logs):[];i.push(e),n.update(o).set({logs:JSON.stringify(i)}).where(drizzleOrm.eq(o.id,t)).run();},{behavior:"immediate"});}catch(n){throw n instanceof _?n:new _("DB_ERROR","Failed to append log",n)}finally{s.close();}}getRunningTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).where(drizzleOrm.eq(o.status,"running")).orderBy(drizzleOrm.desc(o.started_at)).all()}catch(e){throw new _("DB_ERROR","Failed to get running tasks",e)}finally{t.close();}}getAllTasks(){if(!this.dbExists())return [];let t=this.openHandle(false);try{return t.db.select().from(o).orderBy(drizzleOrm.desc(o.started_at)).limit(100).all()}catch(e){throw new _("DB_ERROR","Failed to get all tasks",e)}finally{t.close();}}getTask(t){if(!this.dbExists())return;let e=this.openHandle(false);try{return e.db.select().from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get()??void 0}catch(s){throw new _("DB_ERROR","Failed to get task",s)}finally{e.close();}}killTask(t){if(!this.dbExists())return {killed:false};let e=this.openHandle(true);try{let s=e.db.select({id:o.id,status:o.status,pid:o.pid}).from(o).where(drizzleOrm.eq(o.id,t)).limit(1).get();if(!s||s.status!=="running")return {killed:!1};if(s.pid)try{process.kill(s.pid,"SIGTERM");}catch{}return e.db.update(o).set({status:"failed",error:"Killed by user",completed_at:new Date().toISOString()}).where(drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"))).run(),{killed:!0,pid:s.pid??void 0}}catch(s){throw s instanceof _?s:new _("DB_ERROR","Failed to kill task",s)}finally{e.close();}}reapOrphanedTasks(){if(!this.dbExists())return 0;let t=this.openHandle(true);try{let e=t.db.select({id:o.id,pid:o.pid}).from(o).where(drizzleOrm.eq(o.status,"running")).all(),s=0;for(let n of e){if(!n.pid)continue;let r=!1;try{process.kill(n.pid,0),r=!0;}catch{}r||(t.db.update(o).set({status:"failed",error:"Reaped: process not found (orphaned task)",completed_at:new Date().toISOString()}).where(drizzleOrm.and(drizzleOrm.eq(o.id,n.id),drizzleOrm.eq(o.status,"running"))).run(),s++);}return s}finally{t.close();}}findTaskStatus(t,e){let s=this.resolveDbPaths();for(let n of s){if(!fs.existsSync(n))continue;let r=E(n);try{let i=e?drizzleOrm.eq(o.workspace_id,e):void 0,a=i?drizzleOrm.and(drizzleOrm.eq(o.id,t),i):drizzleOrm.eq(o.id,t),l=r.db.select().from(o).where(a).limit(1).get()??void 0;if(!l){let u=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),p=i?drizzleOrm.and(u,i):u;l=r.db.select().from(o).where(p).orderBy(drizzleOrm.desc(o.started_at)).limit(1).get()??void 0;}if(l)return l}catch(i){throw new _("DB_ERROR","Failed to find task status",i)}finally{r.close();}}}findChildTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let i of s){if(!fs.existsSync(i))continue;let a=E(i);try{let l=e?drizzleOrm.and(drizzleOrm.eq(o.parent_task_id,t),drizzleOrm.eq(o.workspace_id,e)):drizzleOrm.eq(o.parent_task_id,t),u=a.db.select().from(o).where(l).orderBy(drizzleOrm.asc(o.started_at)).all();for(let p of u)n.has(p.id)||(n.add(p.id),r.push(p));}catch(l){throw new _("DB_ERROR","Failed to find child tasks",l)}finally{a.close();}}return r}getWorkspaceUsageSummary(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.all(t?drizzleOrm.sql`
|
|
16
16
|
SELECT
|
|
17
17
|
COALESCE(workspace_id, 'unknown') AS workspace_id,
|
|
18
18
|
COALESCE(SUM(input_tokens), 0) AS input_tokens,
|
|
@@ -33,7 +33,7 @@ ${s.join(`
|
|
|
33
33
|
FROM tasks
|
|
34
34
|
GROUP BY workspace_id
|
|
35
35
|
ORDER BY (COALESCE(SUM(input_tokens), 0) + COALESCE(SUM(output_tokens), 0)) DESC
|
|
36
|
-
`)}catch(s){throw new
|
|
36
|
+
`)}catch(s){throw new _("DB_ERROR","Failed to get workspace usage summary",s)}finally{e.close();}}getThreadTokenUsage(t,e){let s=this.resolveDbPaths(),n=new Set,r=0,i=0,a=0;for(let l of s){if(!fs.existsSync(l))continue;let u=E(l);try{let p=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),g=e?drizzleOrm.and(p,drizzleOrm.eq(o.workspace_id,e)):p,k=u.db.select({id:o.id,input_tokens:o.input_tokens,output_tokens:o.output_tokens,cost_usd:o.cost_usd}).from(o).where(g).all();for(let b of k)n.has(b.id)||(n.add(b.id),r+=b.input_tokens??0,i+=b.output_tokens??0,a+=b.cost_usd??0);}catch(p){throw new _("DB_ERROR","Failed to get thread token usage",p)}finally{u.close();}}return {inputTokens:r,outputTokens:i,costUsd:a}}findTasksByThread(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let i of s){if(!fs.existsSync(i))continue;let a=E(i);try{let l=drizzleOrm.or(drizzleOrm.eq(o.thread_id,t),drizzleOrm.and(drizzleOrm.isNull(o.thread_id),drizzleOrm.like(o.command,`%--thread=${t}%`))),u=e?drizzleOrm.and(l,drizzleOrm.eq(o.workspace_id,e)):l,p=a.db.select().from(o).where(u).orderBy(drizzleOrm.asc(o.started_at)).all();for(let g of p)n.has(g.id)||(n.add(g.id),r.push(g));}catch(l){throw new _("DB_ERROR","Failed to find tasks by thread",l)}finally{a.close();}}return r}findAllTasks(t){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let s=[];t.workspaceId&&s.push(drizzleOrm.eq(o.workspace_id,t.workspaceId));let n=t.agents&&t.agents.length>0?t.agents:t.agentId?[t.agentId]:null;n&&s.push(drizzleOrm.inArray(o.agent_id,n));let r=t.statuses&&t.statuses.length>0?t.statuses:t.status?[t.status]:null;r&&s.push(drizzleOrm.inArray(o.status,r));let i=t.q??t.search;i&&s.push(drizzleOrm.like(o.prompt,`%${i}%`)),t.from&&s.push(drizzleOrm.gte(o.started_at,t.from)),t.to&&s.push(drizzleOrm.lt(o.started_at,t.to));let a=s.length>0?drizzleOrm.and(...s):void 0,l=e.db.select({count:drizzleOrm.sql`count(*)`}).from(o).where(a).get(),u=(t.sortDir??"DESC")==="ASC"?drizzleOrm.asc(o.started_at):drizzleOrm.desc(o.started_at);return {rows:e.db.select().from(o).where(a).orderBy(u).limit(t.limit).offset(t.offset).all(),total:l?.count??0}}catch(s){throw new _("DB_ERROR","Failed to find all tasks",s)}finally{e.close();}}getAgentUsage(t,e,s){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.all(s?drizzleOrm.sql`
|
|
37
37
|
SELECT
|
|
38
38
|
t.agent_id,
|
|
39
39
|
t.workspace_id,
|
|
@@ -41,11 +41,11 @@ ${s.join(`
|
|
|
41
41
|
COALESCE(SUM(
|
|
42
42
|
COALESCE(t.input_tokens, 0)
|
|
43
43
|
+ CASE
|
|
44
|
-
WHEN t.started_at >= ${
|
|
44
|
+
WHEN t.started_at >= ${M}
|
|
45
45
|
AND (
|
|
46
46
|
t.crewx_version IS NULL
|
|
47
47
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
48
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
48
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${U}) AS INTEGER) < ${B})
|
|
49
49
|
)
|
|
50
50
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
51
51
|
ELSE 0
|
|
@@ -64,11 +64,11 @@ ${s.join(`
|
|
|
64
64
|
COALESCE(SUM(
|
|
65
65
|
COALESCE(t.input_tokens, 0)
|
|
66
66
|
+ CASE
|
|
67
|
-
WHEN t.started_at >= ${
|
|
67
|
+
WHEN t.started_at >= ${M}
|
|
68
68
|
AND (
|
|
69
69
|
t.crewx_version IS NULL
|
|
70
70
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
71
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
71
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${U}) AS INTEGER) < ${B})
|
|
72
72
|
)
|
|
73
73
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
74
74
|
ELSE 0
|
|
@@ -84,11 +84,11 @@ ${s.join(`
|
|
|
84
84
|
COALESCE(SUM(
|
|
85
85
|
COALESCE(t.input_tokens, 0)
|
|
86
86
|
+ CASE
|
|
87
|
-
WHEN t.started_at >= ${
|
|
87
|
+
WHEN t.started_at >= ${M}
|
|
88
88
|
AND (
|
|
89
89
|
t.crewx_version IS NULL
|
|
90
90
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
91
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
91
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${U}) AS INTEGER) < ${B})
|
|
92
92
|
)
|
|
93
93
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
94
94
|
ELSE 0
|
|
@@ -106,11 +106,11 @@ ${s.join(`
|
|
|
106
106
|
COALESCE(SUM(
|
|
107
107
|
COALESCE(t.input_tokens, 0)
|
|
108
108
|
+ CASE
|
|
109
|
-
WHEN t.started_at >= ${
|
|
109
|
+
WHEN t.started_at >= ${M}
|
|
110
110
|
AND (
|
|
111
111
|
t.crewx_version IS NULL
|
|
112
112
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
113
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
113
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${U}) AS INTEGER) < ${B})
|
|
114
114
|
)
|
|
115
115
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
116
116
|
ELSE 0
|
|
@@ -118,18 +118,18 @@ ${s.join(`
|
|
|
118
118
|
), 0)
|
|
119
119
|
+ COALESCE(SUM(t.output_tokens), 0)
|
|
120
120
|
) DESC
|
|
121
|
-
`).map(
|
|
121
|
+
`).map(i=>({agentId:i.agent_id,workspaceId:i.workspace_id??null,totalTasks:i.total_tasks,inputTokens:i.input_tokens,outputTokens:i.output_tokens,cachedInputTokens:i.cached_input_tokens,costUsd:i.cost_usd,totalTokens:i.input_tokens+i.output_tokens}))}catch(r){throw new _("DB_ERROR","Failed to get agent usage",r)}finally{n.close();}}getAgentUsageTrendRaw(t,e,s){if(!this.dbExists())return [];let n=this.openHandle(false);try{return n.db.all(s?drizzleOrm.sql`
|
|
122
122
|
SELECT
|
|
123
123
|
date(t.started_at) AS date,
|
|
124
124
|
t.agent_id,
|
|
125
125
|
COALESCE(SUM(
|
|
126
126
|
COALESCE(t.input_tokens, 0)
|
|
127
127
|
+ CASE
|
|
128
|
-
WHEN t.started_at >= ${
|
|
128
|
+
WHEN t.started_at >= ${M}
|
|
129
129
|
AND (
|
|
130
130
|
t.crewx_version IS NULL
|
|
131
131
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
132
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
132
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${U}) AS INTEGER) < ${B})
|
|
133
133
|
)
|
|
134
134
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
135
135
|
ELSE 0
|
|
@@ -152,11 +152,11 @@ ${s.join(`
|
|
|
152
152
|
COALESCE(SUM(
|
|
153
153
|
COALESCE(t.input_tokens, 0)
|
|
154
154
|
+ CASE
|
|
155
|
-
WHEN t.started_at >= ${
|
|
155
|
+
WHEN t.started_at >= ${M}
|
|
156
156
|
AND (
|
|
157
157
|
t.crewx_version IS NULL
|
|
158
158
|
OR (t.crewx_version LIKE '0.8.%' AND t.crewx_version NOT LIKE '0.8.9%')
|
|
159
|
-
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${
|
|
159
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${U}) AS INTEGER) < ${B})
|
|
160
160
|
)
|
|
161
161
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
162
162
|
ELSE 0
|
|
@@ -171,7 +171,43 @@ ${s.join(`
|
|
|
171
171
|
AND t.started_at < ${e}
|
|
172
172
|
GROUP BY date(t.started_at), t.agent_id
|
|
173
173
|
ORDER BY date(t.started_at) ASC
|
|
174
|
-
`).map(
|
|
174
|
+
`).map(i=>({date:i.date,agentId:i.agent_id,inputTokens:i.input_tokens,outputTokens:i.output_tokens,cachedInputTokens:i.cached_input_tokens,costUsd:i.cost_usd,totalTokens:i.input_tokens+i.output_tokens}))}catch(r){throw new _("DB_ERROR","Failed to get agent usage trend",r)}finally{n.close();}}findTaskForStop(t,e){if(!this.dbExists())return;let s=this.openHandle(false);try{return s.db.select().from(o).where(drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.workspace_id,e))).limit(1).get()??void 0}catch(n){throw new _("DB_ERROR","Failed to find task for stop",n)}finally{s.close();}}markTaskFailed(t,e,s){if(!this.dbExists())return;let n=this.openHandle(true);try{let r=new Date().toISOString(),i=s?drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"),drizzleOrm.eq(o.workspace_id,s)):drizzleOrm.and(drizzleOrm.eq(o.id,t),drizzleOrm.eq(o.status,"running"));n.db.update(o).set({status:"failed",error:e,completed_at:r}).where(i).run();}catch(r){throw r instanceof _?r:new _("DB_ERROR","Failed to mark task failed",r)}finally{n.close();}}findTasksByPromptHint(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let i of s){if(!fs.existsSync(i))continue;let a=E(i);try{let l=e?drizzleOrm.and(drizzleOrm.like(o.prompt,`%${t}%`),drizzleOrm.eq(o.workspace_id,e)):drizzleOrm.like(o.prompt,`%${t}%`),u=a.db.select().from(o).where(l).orderBy(drizzleOrm.asc(o.started_at)).all();for(let p of u)n.has(p.id)||(n.add(p.id),r.push(p));}catch(l){throw new _("DB_ERROR","Failed to find tasks by prompt hint",l)}finally{a.close();}}return r}getProviderUsage(t,e,s){if(!this.dbExists())return [];let n=this.openHandle(false);try{let r=drizzleOrm.sql`
|
|
175
|
+
CASE
|
|
176
|
+
WHEN ${o.model} LIKE 'claude-%' OR ${o.model} IN ('opus', 'sonnet', 'haiku', 'opus[1m]', 'sonnet[1m]') THEN 'claude'
|
|
177
|
+
WHEN ${o.model} LIKE 'gpt-%' OR ${o.model} LIKE 'codex-%' THEN 'codex'
|
|
178
|
+
WHEN ${o.model} LIKE 'gemini-%' THEN 'gemini'
|
|
179
|
+
WHEN ${o.model} LIKE 'zai-%' OR ${o.model} LIKE 'openrouter/z-ai/%' THEN 'opencode'
|
|
180
|
+
WHEN ${o.model} LIKE 'minimax/%' THEN 'minimax'
|
|
181
|
+
WHEN ${o.model} LIKE 'qwen%' THEN 'qwen'
|
|
182
|
+
ELSE 'unknown'
|
|
183
|
+
END
|
|
184
|
+
`,i=drizzleOrm.sql`
|
|
185
|
+
COALESCE(${o.input_tokens}, 0)
|
|
186
|
+
+ CASE
|
|
187
|
+
WHEN ${o.started_at} >= ${M}
|
|
188
|
+
AND (
|
|
189
|
+
${o.crewx_version} IS NULL
|
|
190
|
+
OR (${o.crewx_version} LIKE '0.8.%' AND ${o.crewx_version} NOT LIKE '0.8.9%')
|
|
191
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${U}) AS INTEGER) < ${B})
|
|
192
|
+
)
|
|
193
|
+
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
194
|
+
ELSE 0
|
|
195
|
+
END
|
|
196
|
+
`,a=s?drizzleOrm.sql`WHERE ${o.status} IN ('completed', 'success') AND ${o.started_at} >= ${t} AND ${o.started_at} < ${e} AND ${o.workspace_id} = ${s}`:drizzleOrm.sql`WHERE ${o.status} IN ('completed', 'success') AND ${o.started_at} >= ${t} AND ${o.started_at} < ${e}`;return n.db.all(drizzleOrm.sql`
|
|
197
|
+
SELECT
|
|
198
|
+
${r} AS provider,
|
|
199
|
+
COUNT(*) AS total_tasks,
|
|
200
|
+
COALESCE(SUM(${i}), 0) AS input_tokens,
|
|
201
|
+
COALESCE(SUM(${o.output_tokens}), 0) AS output_tokens,
|
|
202
|
+
COALESCE(SUM(${o.cached_input_tokens}), 0) AS cached_input_tokens,
|
|
203
|
+
COALESCE(SUM(${o.cost_usd}), 0) AS cost_usd,
|
|
204
|
+
COALESCE(SUM(${o.duration_ms}), 0) AS active_duration_ms,
|
|
205
|
+
MAX(${o.completed_at}) AS last_active_at
|
|
206
|
+
FROM ${o}
|
|
207
|
+
${a}
|
|
208
|
+
GROUP BY provider
|
|
209
|
+
ORDER BY (COALESCE(SUM(${i}), 0) + COALESCE(SUM(${o.output_tokens}), 0)) DESC
|
|
210
|
+
`).map(u=>({provider:u.provider,totalTasks:u.total_tasks,inputTokens:u.input_tokens,outputTokens:u.output_tokens,cachedInputTokens:u.cached_input_tokens,costUsd:u.cost_usd,totalTokens:u.input_tokens+u.output_tokens,activeDurationMs:u.active_duration_ms??0,lastActiveAt:u.last_active_at??null}))}catch(r){throw new _("DB_ERROR","Failed to get provider usage",r)}finally{n.close();}}};var ct=class extends v{name="sqlite-tracing";unsubs=[];dbPath;version;constructor(t){super(),this.dbPath=D.join(t?.dbRoot??Pt.homedir(),".crewx","crewx.db"),this.version=t?.version??"unknown";}attach(t){let e=new tt({dbPath:this.dbPath}),s=process.cwd(),r=fs.existsSync(D.join(s,"crewx.yaml"))||fs.existsSync(D.join(s,"crewx.yml"))?ft(s):null,i=process.argv.join(" ");this.unsubs.push(t.on("task:start",a=>{try{let l=a.callerAgentId??null,u=a.parentTaskId??null,p=a.rootTraceId??a.traceId,g=a.metadata?JSON.stringify(a.metadata):JSON.stringify({provider:a.provider??"cli/claude"});e.startTask({id:a.traceId,agentId:a.agentRef.replace(/^@/,""),prompt:a.message,mode:a.mode,status:"running",pid:a.pid??null,startedAt:a.timestamp.toISOString(),crewxVersion:this.version,platform:a.platform??"cli",model:a.model??null,renderedPrompt:a.renderedPrompt??null,command:i,codingAgentCommand:a.codingAgentCommand??null,workspaceId:a.workspaceId??r,callerAgentId:l,parentTaskId:u,traceId:p,metadata:g,threadId:a.threadId??null});}catch{}}),t.on("task:output",a=>{try{e.appendLog(a.traceId,{timestamp:a.timestamp.toISOString(),level:a.level??"stdout",message:a.output});}catch{}}),t.on("task:end",a=>{try{e.finishTask({id:a.traceId,status:a.error?"failed":"success",result:a.result??null,error:a.error?JSON.stringify(a.error):null,completedAt:a.timestamp.toISOString(),durationMs:a.durationMs,exitCode:a.exitCode??null,inputTokens:a.inputTokens??0,outputTokens:a.outputTokens??0,cachedInputTokens:a.cachedInputTokens??0,costUsd:a.costUsd??0,model:a.model??null});}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[];}};var et=class extends F{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=D.join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let n=D.dirname(e);fs.existsSync(n)||fs.mkdirSync(n,{recursive:true});}else if(!fs.existsSync(e))throw new _("NOT_FOUND","Database not found");let s=E(e);if(t)try{X(s.db,e);}catch(n){throw s.close(),n}return s}validateWorkspaceId(t,e){return t.db.select({id:H.id}).from(H).where(drizzleOrm.eq(H.id,e)).limit(1).get()?e:null}topLevelTaskPredicateSql(t="child"){return drizzleOrm.sql.raw(`(
|
|
175
211
|
${t}.parent_task_id IS NULL
|
|
176
212
|
OR ${t}.parent_task_id = ''
|
|
177
213
|
OR NOT EXISTS (
|
|
@@ -179,7 +215,7 @@ ${s.join(`
|
|
|
179
215
|
WHERE parent.id = ${t}.parent_task_id
|
|
180
216
|
AND parent.thread_id = ${t}.thread_id
|
|
181
217
|
)
|
|
182
|
-
)`)}findAllThreads(t){let e=this.resolveDbPaths(),s=new Set,n=[];for(let r of e){if(!fs.existsSync(r))continue;let
|
|
218
|
+
)`)}findAllThreads(t){let e=this.resolveDbPaths(),s=new Set,n=[];for(let r of e){if(!fs.existsSync(r))continue;let i=E(r);try{let a=t?drizzleOrm.eq(c.workspace_id,t):void 0,l=i.db.select().from(c).where(a).orderBy(drizzleOrm.desc(c.updated_at)).all();for(let u of l)s.has(u.id)||(s.add(u.id),n.push(u));}catch(a){throw new _("DB_ERROR","Failed to find all threads",a)}finally{i.close();}}return n}findThreadById(t,e){let s=this.resolveDbPaths();for(let n of s){if(!fs.existsSync(n))continue;let r=E(n);try{let i=drizzleOrm.eq(c.id,t),a=e?drizzleOrm.and(i,drizzleOrm.eq(c.workspace_id,e)):i,l=r.db.select().from(c).where(a).limit(1).get()??void 0;if(l)return l}catch(i){throw new _("DB_ERROR","Failed to find thread by id",i)}finally{r.close();}}}threadExists(t,e){let s=this.resolveDbPaths();for(let n of s){if(!fs.existsSync(n))continue;let r=E(n);try{let i=drizzleOrm.eq(c.id,t),a=e?drizzleOrm.and(i,drizzleOrm.eq(c.workspace_id,e)):i;if(r.db.select({id:c.id}).from(c).where(a).limit(1).get())return !0}catch(i){throw new _("DB_ERROR","Failed to check thread existence",i)}finally{r.close();}}return false}aggregateTaskStats(t,e){let s=this.resolveDbPaths(),n=0,r=0,i=0,a=0,l=0,u=new Set;for(let p of s){if(!fs.existsSync(p))continue;let g=E(p);try{let k=g.db.get(drizzleOrm.sql`
|
|
183
219
|
SELECT
|
|
184
220
|
count(*) AS cnt,
|
|
185
221
|
COALESCE(SUM(child.input_tokens), 0) AS total_input,
|
|
@@ -190,25 +226,25 @@ ${s.join(`
|
|
|
190
226
|
WHERE child.thread_id = ${t}
|
|
191
227
|
AND ${this.topLevelTaskPredicateSql()}
|
|
192
228
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
193
|
-
`);k&&(n+=k.cnt,r+=k.total_input,
|
|
229
|
+
`);k&&(n+=k.cnt,r+=k.total_input,i+=k.total_output,a+=k.total_cached,l+=k.total_cost);let b=g.db.all(drizzleOrm.sql`
|
|
194
230
|
SELECT DISTINCT child.agent_id FROM tasks child
|
|
195
231
|
WHERE child.thread_id = ${t}
|
|
196
232
|
AND child.agent_id IS NOT NULL AND child.agent_id != ''
|
|
197
233
|
AND ${this.topLevelTaskPredicateSql()}
|
|
198
234
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
199
|
-
`);for(let
|
|
235
|
+
`);for(let j of b)u.add(j.agent_id);}catch(k){throw new _("DB_ERROR","Failed to aggregate task stats",k)}finally{g.close();}}return {taskCount:n,inputTokens:r,outputTokens:i,cachedInputTokens:a,costUsd:l,agentIds:Array.from(u)}}findTopLevelTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let i of s){if(!fs.existsSync(i))continue;let a=E(i);try{let l=a.db.all(drizzleOrm.sql`
|
|
200
236
|
SELECT child.* FROM tasks child
|
|
201
237
|
WHERE child.thread_id = ${t}
|
|
202
238
|
AND ${this.topLevelTaskPredicateSql()}
|
|
203
239
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
204
240
|
ORDER BY child.started_at ASC
|
|
205
|
-
`);for(let
|
|
241
|
+
`);for(let u of l)n.has(u.id)||(n.add(u.id),r.push(u));}catch(l){throw new _("DB_ERROR","Failed to find top-level tasks",l)}finally{a.close();}}return r}findAllTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let i of s){if(!fs.existsSync(i))continue;let a=E(i);try{let l=drizzleOrm.eq(o.thread_id,t),u=e?drizzleOrm.and(l,drizzleOrm.eq(o.workspace_id,e)):l,p=a.db.select().from(o).where(u).orderBy(drizzleOrm.asc(o.started_at)).all();for(let g of p)n.has(g.id)||(n.add(g.id),r.push(g));}catch(l){throw new _("DB_ERROR","Failed to find all tasks for thread",l)}finally{a.close();}}return r}findTaskById(t,e,s){let n=this.resolveDbPaths();for(let r of n){if(!fs.existsSync(r))continue;let i=E(r);try{let a=drizzleOrm.and(drizzleOrm.eq(o.id,e),drizzleOrm.eq(o.thread_id,t)),l=s?drizzleOrm.and(a,drizzleOrm.eq(o.workspace_id,s)):a,u=i.db.select().from(o).where(l).limit(1).get();if(!u)continue;let p=i.db.select().from(o).where(drizzleOrm.eq(o.parent_task_id,u.id)).orderBy(drizzleOrm.asc(o.started_at)).all();return {task:u,children:p}}catch(a){throw new _("DB_ERROR","Failed to find task by id",a)}finally{i.close();}}}batchFetchTasks(t,e){let s=new Map;if(t.length===0)return s;let n=this.resolveDbPaths();for(let r of n){if(!fs.existsSync(r))continue;let i=E(r);try{let a=i.db.all(drizzleOrm.sql`
|
|
206
242
|
SELECT child.* FROM tasks child
|
|
207
243
|
WHERE child.thread_id IN (${drizzleOrm.sql.join(t.map(l=>drizzleOrm.sql`${l}`),drizzleOrm.sql`, `)})
|
|
208
244
|
AND ${this.topLevelTaskPredicateSql()}
|
|
209
245
|
${e?drizzleOrm.sql`AND child.workspace_id = ${e}`:drizzleOrm.sql``}
|
|
210
246
|
ORDER BY child.started_at ASC
|
|
211
|
-
`);for(let l of
|
|
247
|
+
`);for(let l of a){let u=l.thread_id;s.has(u)||s.set(u,[]),s.get(u).push(l);}}catch(a){throw new _("DB_ERROR","Failed to batch fetch tasks",a)}finally{i.close();}}return s}updateThreadTitle(t,e,s){if(!this.dbExists())return;let n=this.openHandle(true);try{let r=drizzleOrm.eq(c.id,t),i=s?drizzleOrm.and(r,drizzleOrm.eq(c.workspace_id,s)):r;if(!n.db.select({id:c.id}).from(c).where(i).limit(1).get())return;n.db.update(c).set({title:e,title_locked:1,updated_at:new Date().toISOString()}).where(drizzleOrm.eq(c.id,t)).run();}catch(r){throw r instanceof _?r:new _("DB_ERROR","Failed to update thread title",r)}finally{n.close();}}upsertThread(t,e){let s=this.openHandle(true);try{let n=e.workspaceId?this.validateWorkspaceId(s,e.workspaceId):null,r=new Date().toISOString();if(s.db.select({id:c.id,message_count:c.message_count}).from(c).where(drizzleOrm.eq(c.id,t)).limit(1).get()){let a={updated_at:r};e.title!==void 0&&(a.title=e.title),e.titleLocked!==void 0&&(a.title_locked=e.titleLocked?1:0),s.db.update(c).set(a).where(drizzleOrm.eq(c.id,t)).run();}else s.db.insert(c).values({id:t,platform:e.platform,workspace_id:n,title:e.title??null,title_locked:e.titleLocked?1:0,message_count:0,created_at:r,updated_at:r}).run();}catch(n){throw n instanceof _?n:new _("DB_ERROR","Failed to upsert thread",n)}finally{s.close();}}ensureThread(t,e,s){let n=this.openHandle(true);try{let r=s?this.validateWorkspaceId(n,s):null,i=n.db.select({id:c.id,platform:c.platform,workspace_id:c.workspace_id}).from(c).where(drizzleOrm.eq(c.id,t)).limit(1).get();if(i){r&&!i.workspace_id&&n.db.update(c).set({workspace_id:r}).where(drizzleOrm.eq(c.id,t)).run();return}let a=new Date().toISOString();n.db.insert(c).values({id:t,platform:e,workspace_id:r,message_count:0,created_at:a,updated_at:a}).run();}catch(r){throw r instanceof _?r:new _("DB_ERROR","Failed to ensure thread",r)}finally{n.close();}}saveUserMessage(t,e,s){if(!this.dbExists())return {firstMessage:false};let n=this.openHandle(true);try{let r=new Date().toISOString();return {firstMessage:n.db.transaction(a=>{let u=a.select({message_count:c.message_count}).from(c).where(drizzleOrm.eq(c.id,t)).limit(1).get()?.message_count===0;return a.run(drizzleOrm.sql`
|
|
212
248
|
UPDATE threads
|
|
213
249
|
SET first_message = COALESCE(first_message, ${e}),
|
|
214
250
|
title = CASE WHEN title_locked = 0 AND title IS NULL THEN substr(${e}, 1, 60) ELSE title END,
|
|
@@ -216,7 +252,7 @@ ${s.join(`
|
|
|
216
252
|
message_count = message_count + 1,
|
|
217
253
|
updated_at = ${r}
|
|
218
254
|
WHERE id = ${t}
|
|
219
|
-
`),
|
|
255
|
+
`),u},{behavior:"immediate"})}}catch(r){throw r instanceof _?r:new _("DB_ERROR","Failed to save user message",r)}finally{n.close();}}saveAssistantMessage(t,e,s){if(!this.dbExists())return;let n=this.openHandle(true);try{let r=new Date().toISOString();n.db.update(c).set({last_message:e,updated_at:r}).where(drizzleOrm.eq(c.id,t)).run();}catch(r){throw r instanceof _?r:new _("DB_ERROR","Failed to save assistant message",r)}finally{n.close();}}updateThread(t,e){if(!this.dbExists())return;let s=this.openHandle(true);try{let n={updated_at:new Date().toISOString()};e.title!==void 0&&(n.title=e.title,n.title_locked=1),e.titleLocked!==void 0&&(n.title_locked=e.titleLocked?1:0),s.db.update(c).set(n).where(drizzleOrm.eq(c.id,t)).run();}catch(n){throw n instanceof _?n:new _("DB_ERROR","Failed to update thread",n)}finally{s.close();}}togglePin(t,e){let s=this.openHandle(true);try{let n=e?drizzleOrm.and(drizzleOrm.eq(c.id,t),drizzleOrm.eq(c.workspace_id,e)):drizzleOrm.eq(c.id,t),r=s.db.select({pinned:c.pinned,metadata:c.metadata}).from(c).where(n).get();if(!r)return null;let i=r.pinned?0:1,a=r.metadata?JSON.parse(r.metadata):{};if(i){let l=e?drizzleOrm.and(drizzleOrm.eq(c.pinned,1),drizzleOrm.eq(c.workspace_id,e)):drizzleOrm.eq(c.pinned,1),u=s.db.select({metadata:c.metadata}).from(c).where(l).all(),p=0;for(let g of u){let k=g.metadata?JSON.parse(g.metadata):{};typeof k.pinOrder=="number"&&k.pinOrder>p&&(p=k.pinOrder);}a.pinOrder=p+1;}else delete a.pinOrder;return s.db.update(c).set({pinned:i,metadata:Object.keys(a).length>0?JSON.stringify(a):null}).where(n).run(),{pinned:!!i}}catch(n){throw n instanceof _?n:new _("DB_ERROR","Failed to toggle pin",n)}finally{s.close();}}reorderPins(t,e){let s=this.openHandle(true);try{for(let n=0;n<t.length;n++){let r=e?drizzleOrm.and(drizzleOrm.eq(c.id,t[n]),drizzleOrm.eq(c.workspace_id,e)):drizzleOrm.eq(c.id,t[n]),i=s.db.select({metadata:c.metadata}).from(c).where(r).get();if(!i)continue;let a=i.metadata?JSON.parse(i.metadata):{};a.pinOrder=n+1,s.db.update(c).set({metadata:JSON.stringify(a)}).where(r).run();}}catch(n){throw n instanceof _?n:new _("DB_ERROR","Failed to reorder pins",n)}finally{s.close();}}toggleStar(t,e){let s=this.openHandle(true);try{let n=e?drizzleOrm.and(drizzleOrm.eq(c.id,t),drizzleOrm.eq(c.workspace_id,e)):drizzleOrm.eq(c.id,t),r=s.db.select({starred:c.starred}).from(c).where(n).get();if(!r)return null;let i=r.starred?0:1;return s.db.update(c).set({starred:i}).where(n).run(),{starred:!!i}}catch(n){throw n instanceof _?n:new _("DB_ERROR","Failed to toggle star",n)}finally{s.close();}}};function me(d){return d.replace(/<conversation_history[^>]*>[\s\S]*?<\/conversation_history>/g,"").split(`
|
|
220
256
|
`).filter(n=>!(n.startsWith("Loaded ")&&n.includes("layouts from")||n.includes("[dotenv@")||n.includes("[Nest]")&&n.includes("DEBUG")||n.startsWith("Registered custom layout:")||n.startsWith("Updated custom layout:"))).join(`
|
|
221
257
|
`).trim()}function fe(d){if(!d)return "";let t=d;try{let e=JSON.parse(t);Array.isArray(e)?t=e.filter(s=>s?.type==="text"&&s?.text).map(s=>s.text).join(`
|
|
222
|
-
`):e&&typeof e=="object"&&e.result!==void 0&&(t=e.result||"");}catch{t=me(t);}return t}var st=class{dbPath;constructor(t){this.dbPath=t??
|
|
258
|
+
`):e&&typeof e=="object"&&e.result!==void 0&&(t=e.result||"");}catch{t=me(t);}return t}var st=class{dbPath;constructor(t){this.dbPath=t??D.join(Pt.homedir(),".crewx","crewx.db");}getThreadRepo(){return new et({dbPath:this.dbPath})}updateThread(t,e){this.getThreadRepo().updateThread(t,{title:e.title});}async ensureThread(t,e,s){let n=this.getThreadRepo(),r=n.findThreadById(t);if(r){if(r.platform!==e)throw new Error(`Thread '${t}' already exists with platform '${r.platform}' \u2014 cannot change to '${e}' (platform is immutable)`);return {created:false}}return n.ensureThread(t,e,s),{created:true}}async fetchHistory(t,e){let s=e?.limit??100,n=this.getThreadRepo(),r=n.findThreadById(t),i=n.findTopLevelTasks(t),a=new Set(["done","completed","success"]);i=i.filter(p=>(!p.status||a.has(p.status))&&(!e?.currentTraceId||p.trace_id!==e.currentTraceId)),i=i.slice(0,s);let l=r?.platform??"cli",u=this.rowsToMessages(i);return {threadId:t,platform:l,messages:u,metadata:{title:r?.title??void 0,firstMessage:r?.first_message??void 0,lastMessage:r?.last_message??void 0,messageCount:r?.message_count??0,updatedAt:r?.updated_at?new Date(r.updated_at).getTime():void 0}}}async saveUserMessage(t,e,s,n){let{firstMessage:r}=this.getThreadRepo().saveUserMessage(t,e);return {id:t,firstMessage:r}}async saveAssistantMessage(t,e,s,n){return this.getThreadRepo().saveAssistantMessage(t,e),{id:t}}close(){}rowsToMessages(t){let e=[];for(let s of t){s.prompt&&e.push({id:`${s.id}-user`,text:s.prompt,isAssistant:false,timestamp:new Date(s.started_at).getTime(),metadata:{caller_agent_id:s.caller_agent_id}});let n=fe(s.result);n&&e.push({id:`${s.id}-assistant`,text:n,isAssistant:true,timestamp:new Date(s.started_at).getTime(),metadata:{agent_id:s.agent_id,task_id:s.id}});}return e}};var ut=class extends v{name="conversation";_provider;unsubStart=null;unsubEnd=null;constructor(t){super(),this._provider=new st(t?.dbPath);}get conversationProvider(){return this._provider}async afterUserMessage(t,e,s,n){}async afterAssistantMessage(t,e,s){}attach(t){this.unsubStart=t.on("task:start",async e=>{if(!e.threadId)return;let s=e.platform??"cli";try{let n=await this._provider.ensureThread(e.threadId,s,e.workspaceId),r=await this._provider.saveUserMessage(e.threadId,e.message??"");await this.afterUserMessage(e.threadId,r.id,n.created||r.firstMessage,e);}catch{}}),this.unsubEnd=t.on("task:end",async e=>{if(!e.result)return;let s=e.metadata?.threadId;if(!s)return;let n=e.agentRef?.replace(/^@/,"")??"";try{let{id:r}=await this._provider.saveAssistantMessage(s,e.result,n);await this.afterAssistantMessage(s,r,e);}catch{}});}detach(t){this.unsubStart?.(),this.unsubStart=null,this.unsubEnd?.(),this.unsubEnd=null,this._provider.close?.();}};exports.ConversationPlugin=ut;exports.FileLoggerPlugin=nt;exports.SqliteTracingPlugin=ct;
|
|
@@ -10,7 +10,7 @@ export { pushSchema } from './push.js';
|
|
|
10
10
|
export type { PushResult } from './push.js';
|
|
11
11
|
export type { Workspace, NewWorkspace } from '../schema/index.js';
|
|
12
12
|
export { TaskRepository } from './task.repository.js';
|
|
13
|
-
export type { TaskRow, NewTask, AgentUsageRow, TrendRow } from './task.repository.js';
|
|
13
|
+
export type { TaskRow, NewTask, AgentUsageRow, TrendRow, ProviderUsageRow } from './task.repository.js';
|
|
14
14
|
export { ThreadRepository } from './thread.repository.js';
|
|
15
15
|
export type { ThreadRow, NewThread } from './thread.repository.js';
|
|
16
16
|
export { SpanRepository } from './span.repository.js';
|