@crewx/sdk 0.8.9-rc.12 → 0.8.9-rc.13
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/esm/index.js +124 -46
- package/dist/esm/plugins/index.js +98 -20
- package/dist/esm/repository/index.js +96 -18
- package/dist/index.js +124 -46
- package/dist/plugins/index.js +97 -19
- package/dist/provider/acp/meta.d.ts +3 -0
- package/dist/repository/index.js +96 -18
- package/dist/repository/task.repository.d.ts +2 -0
- package/package.json +1 -1
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import*as
|
|
1
|
+
import*as D from'path';import D__default,{join,dirname}from'path';import {fileURLToPath}from'url';import {existsSync,mkdirSync,writeFileSync,appendFileSync,readFileSync}from'fs';import Bt,{homedir}from'os';import {createHash}from'crypto';import {sql,eq,desc,and,or,isNull,like,asc,inArray,gte,lt}from'drizzle-orm';import {sqliteTable,text,integer,real,index,unique}from'drizzle-orm/sqlite-core';var G=(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 Nt=()=>fileURLToPath(import.meta.url),vt=()=>D__default.dirname(Nt()),g=vt();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 rt=class extends v{name="file-logger";unsubs=[];logFiles=new Map;logsDir;version;constructor(t){super(),this.logsDir=join(t?.workspaceRoot??process.cwd(),".crewx","logs"),this.version=t?.version??"unknown";}attach(t){this.unsubs.push(t.on("task:start",e=>{try{existsSync(this.logsDir)||mkdirSync(this.logsDir,{recursive:!0});let s=_t(e.timestamp),n=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}
|
|
@@ -8,11 +8,11 @@ Message: ${e.message}
|
|
|
8
8
|
`;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();appendFileSync(s,`[${n}] STDOUT: ${e.output}
|
|
9
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",o=`[${n}] INFO: Task ${r} in ${e.durationMs}ms
|
|
10
10
|
[${n}] INFO: Process closed with exit code: ${e.error?1:0}
|
|
11
|
-
`;appendFileSync(s,o,"utf8"),this.logFiles.delete(e.traceId);}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[],this.logFiles.clear();}};function
|
|
11
|
+
`;appendFileSync(s,o,"utf8"),this.logFiles.delete(e.traceId);}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[],this.logFiles.clear();}};function Mt(d){let t=D.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 gt(d){let t=Mt(d);return createHash("sha256").update(t).digest("hex")}var U=class{resolveDbPath(){return process.env.CREWX_DB?process.env.CREWX_DB:process.env.CREWX_TRACES_DB?process.env.CREWX_TRACES_DB:join(Bt.homedir(),".crewx","crewx.db")}resolveDbPaths(){return [this.resolveDbPath()]}isMissingTableError(t){return t instanceof Error&&/no such table:/i.test(t.message)}dbExists(t){return existsSync(t??this.resolveDbPath())}};function S(d){let t=G("better-sqlite3"),{drizzle:e}=G("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 kt=new Set,zt={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 ot(d,t){return (d.get(`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='${t}'`)?.cnt??0)>0}function Ht(d,t){if(t>0||!ot(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(zt))s.has(n)||d.run(`ALTER TABLE tasks ADD COLUMN ${n} ${r}`);}var qt={"0002_normalize_task_names":{workspace_name:"TEXT",project_name:"TEXT"}};function jt(d,t,e){if(!ot(d,"__drizzle_migrations")||!ot(d,"tasks"))return;let s=d.all(e`SELECT hash FROM __drizzle_migrations`),n=new Set(s.map(o=>o.hash)),r=JSON.parse(readFileSync(D__default.join(t,"meta/_journal.json"),"utf-8"));for(let o of r.entries){let i=qt[o.tag];if(!i)continue;let l=D__default.join(t,`${o.tag}.sql`);if(!existsSync(l))continue;let p=readFileSync(l,"utf-8"),_=createHash("sha256").update(p).digest("hex");if(n.has(_))continue;let k=d.all("PRAGMA table_info(tasks)"),w=new Set(k.map(y=>y.name));for(let[y,K]of Object.entries(i))w.has(y)||(d.run(`ALTER TABLE tasks ADD COLUMN ${y} ${K}`),w.add(y));}}function Wt(d,t,e){let s=d.all(e`SELECT hash FROM __drizzle_migrations`),n=new Set(s.map(o=>o.hash)),r=JSON.parse(readFileSync(D__default.join(t,"meta/_journal.json"),"utf-8"));for(let o of r.entries){let i=D__default.join(t,`${o.tag}.sql`);if(!existsSync(i))continue;let l=readFileSync(i,"utf-8"),p=createHash("sha256").update(l).digest("hex");if(n.has(p))continue;let _=/ALTER\s+TABLE\s+[`"]?(\w+)[`"]?\s+ADD\s+[`"]?(\w+)[`"]?/gi,k=[],w;for(;(w=_.exec(l))!==null;)k.push({table:w[1],column:w[2]});if(k.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;k.every(({table:W,column:At})=>d.all(`PRAGMA table_info("${W}")`).some(Ct=>Ct.name===At))&&d.run(e`INSERT INTO __drizzle_migrations (hash, created_at) VALUES (${p}, ${o.when})`);}}function Kt(d){let{migrate:t}=G("drizzle-orm/better-sqlite3/migrator"),{sql:e}=G("drizzle-orm"),s=[D__default.join(g,"../migrations"),D__default.join(g,"migrations"),D__default.join(g,"../../../../drizzle/migrations"),D__default.join(process.cwd(),"drizzle/migrations")],n=s.find(p=>existsSync(D__default.join(p,"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'`),o=0;r?.cnt&&(o=d.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),
|
|
13
|
+
`)}`);let r=d.get(e`SELECT count(*) as cnt FROM sqlite_master WHERE type='table' AND name='__drizzle_migrations'`),o=0;r?.cnt&&(o=d.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),Ht(d,o),r?.cnt&&(Wt(d,n,e),jt(d,n,e)),t(d,{migrationsFolder:n});let l=(d.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0)-o;if(l>0){let p=r?.cnt?"Database migrated":"Database initialized";console.log(`[crewx] ${p} (${l} migration${l>1?"s":""} applied).`);}}function J(d,t){kt.has(t)||(Kt(d),kt.add(t));}var u=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 F=sqliteTable("workspaces",{id:text("id").primaryKey(),slug:text("slug").notNull().unique(),name:text("name").notNull(),workspace_path:text("workspace_path"),description:text("description"),is_active:integer("is_active").notNull().default(1),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull()});var a=sqliteTable("tasks",{id:text("id").primaryKey(),agent_id:text("agent_id").notNull(),user_id:text("user_id"),prompt:text("prompt").notNull(),mode:text("mode").notNull().default("execute"),status:text("status").notNull().default("running"),result:text("result"),error:text("error"),started_at:text("started_at").notNull(),completed_at:text("completed_at"),duration_ms:integer("duration_ms"),metadata:text("metadata"),workspace_id:text("workspace_id"),trace_id:text("trace_id"),parent_task_id:text("parent_task_id"),caller_agent_id:text("caller_agent_id"),model:text("model"),platform:text("platform").default("cli"),crewx_version:text("crewx_version"),input_tokens:integer("input_tokens").default(0),output_tokens:integer("output_tokens").default(0),cost_usd:real("cost_usd").default(0),pid:integer("pid"),rendered_prompt:text("rendered_prompt"),command:text("command"),coding_agent_command:text("coding_agent_command"),exit_code:integer("exit_code"),logs:text("logs"),thread_id:text("thread_id"),workspace_ref:text("workspace_ref"),project_id:text("project_id"),project_ref:text("project_ref"),cached_input_tokens:integer("cached_input_tokens").default(0)},d=>({idx_tasks_agent_id:index("idx_tasks_agent_id").on(d.agent_id),idx_tasks_status:index("idx_tasks_status").on(d.status),idx_tasks_started_at:index("idx_tasks_started_at").on(d.started_at),idx_tasks_trace_id:index("idx_tasks_trace_id").on(d.trace_id),idx_tasks_parent_task_id:index("idx_tasks_parent_task_id").on(d.parent_task_id),idx_tasks_crewx_version:index("idx_tasks_crewx_version").on(d.crewx_version),idx_tasks_pid:index("idx_tasks_pid").on(d.pid),idx_tasks_thread_id:index("idx_tasks_thread_id").on(d.thread_id),idx_tasks_workspace_id:index("idx_tasks_workspace_id").on(d.workspace_id),idx_tasks_workspace_ref:index("idx_tasks_workspace_ref").on(d.workspace_ref),idx_tasks_project_id:index("idx_tasks_project_id").on(d.project_id),idx_tasks_ws_started:index("idx_tasks_ws_started").on(d.workspace_id,d.started_at)}));var c=sqliteTable("threads",{id:text("id").primaryKey(),workspace_id:text("workspace_id").references(()=>F.id,{onDelete:"set null"}),platform:text("platform").notNull().default("cli"),title:text("title"),first_message:text("first_message"),last_message:text("last_message"),message_count:integer("message_count").notNull().default(0),created_at:text("created_at").notNull(),updated_at:text("updated_at").notNull(),metadata:text("metadata"),title_locked:integer("title_locked").notNull().default(0),pinned:integer("pinned").notNull().default(0),starred:integer("starred").notNull().default(0)},d=>({idx_threads_updated_at:index("idx_threads_updated_at").on(d.updated_at),idx_threads_workspace_id:index("idx_threads_workspace_id").on(d.workspace_id)}));var te=sqliteTable("spans",{id:text("id").primaryKey(),task_id:text("task_id").references(()=>a.id,{onDelete:"set null"}),parent_span_id:text("parent_span_id").references(()=>te.id,{onDelete:"set null"}),name:text("name").notNull(),kind:text("kind").notNull().default("internal"),status:text("status").notNull().default("ok"),started_at:text("started_at").notNull(),completed_at:text("completed_at"),duration_ms:integer("duration_ms"),input:text("input"),output:text("output"),error:text("error"),attributes:text("attributes")},d=>({idx_spans_task_id:index("idx_spans_task_id").on(d.task_id),idx_spans_parent_span_id:index("idx_spans_parent_span_id").on(d.parent_span_id)}));sqliteTable("tool_calls",{id:text("id").primaryKey(),task_id:text("task_id").references(()=>a.id,{onDelete:"cascade"}),session_id:text("session_id"),tool_name:text("tool_name").notNull(),files:text("files"),input:text("input"),output:text("output"),duration_ms:integer("duration_ms"),timestamp:text("timestamp").notNull()},d=>({idx_tool_calls_task_id:index("idx_tool_calls_task_id").on(d.task_id),idx_tool_calls_tool_name:index("idx_tool_calls_tool_name").on(d.tool_name),idx_tool_calls_timestamp:index("idx_tool_calls_timestamp").on(d.timestamp)}));sqliteTable("thread_boxes",{id:text("id").primaryKey(),thread_id:text("thread_id").notNull().references(()=>c.id,{onDelete:"cascade"}),seq:integer("seq").notNull(),first_task_id:text("first_task_id").notNull(),mid_task_id:text("mid_task_id").notNull(),last_task_id:text("last_task_id").notNull(),task_count:integer("task_count").notNull(),summary:text("summary"),source_tokens:integer("source_tokens").notNull(),summary_tokens:integer("summary_tokens"),created_at:text("created_at").notNull()},d=>({idx_thread_boxes_thread_id:index("idx_thread_boxes_thread_id").on(d.thread_id),idx_thread_boxes_seq:index("idx_thread_boxes_seq").on(d.thread_id,d.seq),uniq_thread_boxes_thread_seq:unique().on(d.thread_id,d.seq)}));sqliteTable("request_logs",{id:text("id").primaryKey(),path:text("path").notNull(),method:text("method").notNull(),status_code:integer("status_code").notNull(),duration_ms:integer("duration_ms").notNull(),ip:text("ip"),request_headers:text("request_headers"),response_headers:text("response_headers"),request_body:text("request_body"),response_body:text("response_body"),query:text("query"),user_id:text("user_id"),project_id:text("project_id"),partition_key:text("partition_key").notNull(),timestamp:text("timestamp").notNull().default(sql`(datetime('now'))`),metadata:text("metadata")},d=>({idx_request_logs_timestamp:index("idx_request_logs_timestamp").on(d.timestamp),idx_request_logs_path:index("idx_request_logs_path").on(d.path),idx_request_logs_status_code:index("idx_request_logs_status_code").on(d.status_code),idx_request_logs_partition_key:index("idx_request_logs_partition_key").on(d.partition_key)}));var H="2026-05-09",pe="0.8.9-rc.13",q=10,j=parseInt(pe.split("rc.")[1]),et=class extends U{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let n=dirname(e);existsSync(n)||mkdirSync(n,{recursive:true});}else if(!existsSync(e))throw new u("NOT_FOUND","Database not found");let s=S(e);if(t)try{J(s.db,e);}catch(n){throw s.close(),n}return s}startTask(t){let e=this.openHandle(true);try{e.db.insert(a).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 u?s:new u("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 u?s:new u("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:a.logs}).from(a).where(eq(a.id,t)).limit(1).get(),o=r?.logs?JSON.parse(r.logs):[];o.push(e),n.update(a).set({logs:JSON.stringify(o)}).where(eq(a.id,t)).run();},{behavior:"immediate"});}catch(n){throw n instanceof u?n:new u("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(a).where(eq(a.status,"running")).orderBy(desc(a.started_at)).all()}catch(e){throw new u("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(a).orderBy(desc(a.started_at)).limit(100).all()}catch(e){throw new u("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(a).where(eq(a.id,t)).limit(1).get()??void 0}catch(s){throw new u("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:a.id,status:a.status,pid:a.pid}).from(a).where(eq(a.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(a).set({status:"failed",error:"Killed by user",completed_at:new Date().toISOString()}).where(and(eq(a.id,t),eq(a.status,"running"))).run(),{killed:!0,pid:s.pid??void 0}}catch(s){throw s instanceof u?s:new u("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:a.id,pid:a.pid}).from(a).where(eq(a.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(a).set({status:"failed",error:"Reaped: process not found (orphaned task)",completed_at:new Date().toISOString()}).where(and(eq(a.id,n.id),eq(a.status,"running"))).run(),s++);}return s}finally{t.close();}}findTaskStatus(t,e){let s=this.resolveDbPaths();for(let n of s){if(!existsSync(n))continue;let r=
|
|
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 u?s:new u("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:a.logs}).from(a).where(eq(a.id,t)).limit(1).get(),o=r?.logs?JSON.parse(r.logs):[];o.push(e),n.update(a).set({logs:JSON.stringify(o)}).where(eq(a.id,t)).run();},{behavior:"immediate"});}catch(n){throw n instanceof u?n:new u("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(a).where(eq(a.status,"running")).orderBy(desc(a.started_at)).all()}catch(e){throw new u("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(a).orderBy(desc(a.started_at)).limit(100).all()}catch(e){throw new u("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(a).where(eq(a.id,t)).limit(1).get()??void 0}catch(s){throw new u("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:a.id,status:a.status,pid:a.pid}).from(a).where(eq(a.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(a).set({status:"failed",error:"Killed by user",completed_at:new Date().toISOString()}).where(and(eq(a.id,t),eq(a.status,"running"))).run(),{killed:!0,pid:s.pid??void 0}}catch(s){throw s instanceof u?s:new u("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:a.id,pid:a.pid}).from(a).where(eq(a.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(a).set({status:"failed",error:"Reaped: process not found (orphaned task)",completed_at:new Date().toISOString()}).where(and(eq(a.id,n.id),eq(a.status,"running"))).run(),s++);}return s}finally{t.close();}}findTaskStatus(t,e){let s=this.resolveDbPaths();for(let n of s){if(!existsSync(n))continue;let r=S(n);try{let o=e?eq(a.workspace_id,e):void 0,i=o?and(eq(a.id,t),o):eq(a.id,t),l=r.db.select().from(a).where(i).limit(1).get()??void 0;if(!l){let p=or(eq(a.thread_id,t),and(isNull(a.thread_id),like(a.command,`%--thread=${t}%`))),_=o?and(p,o):p;l=r.db.select().from(a).where(_).orderBy(desc(a.started_at)).limit(1).get()??void 0;}if(l)return l}catch(o){throw new u("DB_ERROR","Failed to find task status",o)}finally{r.close();}}}findChildTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=S(o);try{let l=e?and(eq(a.parent_task_id,t),eq(a.workspace_id,e)):eq(a.parent_task_id,t),p=i.db.select().from(a).where(l).orderBy(asc(a.started_at)).all();for(let _ of p)n.has(_.id)||(n.add(_.id),r.push(_));}catch(l){throw new u("DB_ERROR","Failed to find child tasks",l)}finally{i.close();}}return r}getWorkspaceUsageSummary(t){if(!this.dbExists())return [];let e=this.openHandle(false);try{return e.db.all(t?sql`
|
|
16
16
|
SELECT
|
|
17
17
|
COALESCE(workspace_id, 'unknown') AS workspace_id,
|
|
18
18
|
COALESCE(SUM(input_tokens), 0) AS input_tokens,
|
|
@@ -33,12 +33,24 @@ ${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 u("DB_ERROR","Failed to get workspace usage summary",s)}finally{e.close();}}getThreadTokenUsage(t,e){let s=this.resolveDbPaths(),n=new Set,r=0,o=0,i=0;for(let l of s){if(!existsSync(l))continue;let p=
|
|
36
|
+
`)}catch(s){throw new u("DB_ERROR","Failed to get workspace usage summary",s)}finally{e.close();}}getThreadTokenUsage(t,e){let s=this.resolveDbPaths(),n=new Set,r=0,o=0,i=0;for(let l of s){if(!existsSync(l))continue;let p=S(l);try{let _=or(eq(a.thread_id,t),and(isNull(a.thread_id),like(a.command,`%--thread=${t}%`))),k=e?and(_,eq(a.workspace_id,e)):_,w=p.db.select({id:a.id,input_tokens:a.input_tokens,output_tokens:a.output_tokens,cost_usd:a.cost_usd}).from(a).where(k).all();for(let y of w)n.has(y.id)||(n.add(y.id),r+=y.input_tokens??0,o+=y.output_tokens??0,i+=y.cost_usd??0);}catch(_){throw new u("DB_ERROR","Failed to get thread token usage",_)}finally{p.close();}}return {inputTokens:r,outputTokens:o,costUsd:i}}findTasksByThread(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=S(o);try{let l=or(eq(a.thread_id,t),and(isNull(a.thread_id),like(a.command,`%--thread=${t}%`))),p=e?and(l,eq(a.workspace_id,e)):l,_=i.db.select().from(a).where(p).orderBy(asc(a.started_at)).all();for(let k of _)n.has(k.id)||(n.add(k.id),r.push(k));}catch(l){throw new u("DB_ERROR","Failed to find tasks by thread",l)}finally{i.close();}}return r}findAllTasks(t){if(!this.dbExists())return {rows:[],total:0};let e=this.openHandle(false);try{let s=[];t.workspaceId&&s.push(eq(a.workspace_id,t.workspaceId));let n=t.agents&&t.agents.length>0?t.agents:t.agentId?[t.agentId]:null;n&&s.push(inArray(a.agent_id,n));let r=t.statuses&&t.statuses.length>0?t.statuses:t.status?[t.status]:null;r&&s.push(inArray(a.status,r));let o=t.q??t.search;o&&s.push(like(a.prompt,`%${o}%`)),t.from&&s.push(gte(a.started_at,t.from)),t.to&&s.push(lt(a.started_at,t.to));let i=s.length>0?and(...s):void 0,l=e.db.select({count:sql`count(*)`}).from(a).where(i).get(),p=(t.sortDir??"DESC")==="ASC"?asc(a.started_at):desc(a.started_at);return {rows:e.db.select().from(a).where(i).orderBy(p).limit(t.limit).offset(t.offset).all(),total:l?.count??0}}catch(s){throw new u("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?sql`
|
|
37
37
|
SELECT
|
|
38
38
|
t.agent_id,
|
|
39
39
|
t.workspace_id,
|
|
40
40
|
COUNT(*) AS total_tasks,
|
|
41
|
-
COALESCE(SUM(
|
|
41
|
+
COALESCE(SUM(
|
|
42
|
+
COALESCE(t.input_tokens, 0)
|
|
43
|
+
+ CASE
|
|
44
|
+
WHEN t.started_at >= ${H}
|
|
45
|
+
AND (
|
|
46
|
+
t.crewx_version IS NULL
|
|
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, ${q}) AS INTEGER) < ${j})
|
|
49
|
+
)
|
|
50
|
+
THEN COALESCE(t.cached_input_tokens, 0)
|
|
51
|
+
ELSE 0
|
|
52
|
+
END
|
|
53
|
+
), 0) AS input_tokens,
|
|
42
54
|
COALESCE(SUM(t.output_tokens), 0) AS output_tokens,
|
|
43
55
|
COALESCE(SUM(t.cached_input_tokens), 0) AS cached_input_tokens,
|
|
44
56
|
COALESCE(SUM(t.cost_usd), 0) AS cost_usd
|
|
@@ -48,13 +60,40 @@ ${s.join(`
|
|
|
48
60
|
AND t.started_at < ${e}
|
|
49
61
|
AND t.workspace_id = ${s}
|
|
50
62
|
GROUP BY t.agent_id, t.workspace_id
|
|
51
|
-
ORDER BY (
|
|
63
|
+
ORDER BY (
|
|
64
|
+
COALESCE(SUM(
|
|
65
|
+
COALESCE(t.input_tokens, 0)
|
|
66
|
+
+ CASE
|
|
67
|
+
WHEN t.started_at >= ${H}
|
|
68
|
+
AND (
|
|
69
|
+
t.crewx_version IS NULL
|
|
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, ${q}) AS INTEGER) < ${j})
|
|
72
|
+
)
|
|
73
|
+
THEN COALESCE(t.cached_input_tokens, 0)
|
|
74
|
+
ELSE 0
|
|
75
|
+
END
|
|
76
|
+
), 0)
|
|
77
|
+
+ COALESCE(SUM(t.output_tokens), 0)
|
|
78
|
+
) DESC
|
|
52
79
|
`:sql`
|
|
53
80
|
SELECT
|
|
54
81
|
t.agent_id,
|
|
55
82
|
t.workspace_id,
|
|
56
83
|
COUNT(*) AS total_tasks,
|
|
57
|
-
COALESCE(SUM(
|
|
84
|
+
COALESCE(SUM(
|
|
85
|
+
COALESCE(t.input_tokens, 0)
|
|
86
|
+
+ CASE
|
|
87
|
+
WHEN t.started_at >= ${H}
|
|
88
|
+
AND (
|
|
89
|
+
t.crewx_version IS NULL
|
|
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, ${q}) AS INTEGER) < ${j})
|
|
92
|
+
)
|
|
93
|
+
THEN COALESCE(t.cached_input_tokens, 0)
|
|
94
|
+
ELSE 0
|
|
95
|
+
END
|
|
96
|
+
), 0) AS input_tokens,
|
|
58
97
|
COALESCE(SUM(t.output_tokens), 0) AS output_tokens,
|
|
59
98
|
COALESCE(SUM(t.cached_input_tokens), 0) AS cached_input_tokens,
|
|
60
99
|
COALESCE(SUM(t.cost_usd), 0) AS cost_usd
|
|
@@ -63,12 +102,39 @@ ${s.join(`
|
|
|
63
102
|
AND t.started_at >= ${t}
|
|
64
103
|
AND t.started_at < ${e}
|
|
65
104
|
GROUP BY t.agent_id, t.workspace_id
|
|
66
|
-
ORDER BY (
|
|
67
|
-
|
|
105
|
+
ORDER BY (
|
|
106
|
+
COALESCE(SUM(
|
|
107
|
+
COALESCE(t.input_tokens, 0)
|
|
108
|
+
+ CASE
|
|
109
|
+
WHEN t.started_at >= ${H}
|
|
110
|
+
AND (
|
|
111
|
+
t.crewx_version IS NULL
|
|
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, ${q}) AS INTEGER) < ${j})
|
|
114
|
+
)
|
|
115
|
+
THEN COALESCE(t.cached_input_tokens, 0)
|
|
116
|
+
ELSE 0
|
|
117
|
+
END
|
|
118
|
+
), 0)
|
|
119
|
+
+ COALESCE(SUM(t.output_tokens), 0)
|
|
120
|
+
) DESC
|
|
121
|
+
`).map(o=>({agentId:o.agent_id,workspaceId:o.workspace_id??null,totalTasks:o.total_tasks,inputTokens:o.input_tokens,outputTokens:o.output_tokens,cachedInputTokens:o.cached_input_tokens,costUsd:o.cost_usd,totalTokens:o.input_tokens+o.output_tokens}))}catch(r){throw new u("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?sql`
|
|
68
122
|
SELECT
|
|
69
123
|
date(t.started_at) AS date,
|
|
70
124
|
t.agent_id,
|
|
71
|
-
COALESCE(SUM(
|
|
125
|
+
COALESCE(SUM(
|
|
126
|
+
COALESCE(t.input_tokens, 0)
|
|
127
|
+
+ CASE
|
|
128
|
+
WHEN t.started_at >= ${H}
|
|
129
|
+
AND (
|
|
130
|
+
t.crewx_version IS NULL
|
|
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, ${q}) AS INTEGER) < ${j})
|
|
133
|
+
)
|
|
134
|
+
THEN COALESCE(t.cached_input_tokens, 0)
|
|
135
|
+
ELSE 0
|
|
136
|
+
END
|
|
137
|
+
), 0) AS input_tokens,
|
|
72
138
|
COALESCE(SUM(t.output_tokens), 0) AS output_tokens,
|
|
73
139
|
COALESCE(SUM(t.cached_input_tokens), 0) AS cached_input_tokens,
|
|
74
140
|
COALESCE(SUM(t.cost_usd), 0) AS cost_usd
|
|
@@ -83,7 +149,19 @@ ${s.join(`
|
|
|
83
149
|
SELECT
|
|
84
150
|
date(t.started_at) AS date,
|
|
85
151
|
t.agent_id,
|
|
86
|
-
COALESCE(SUM(
|
|
152
|
+
COALESCE(SUM(
|
|
153
|
+
COALESCE(t.input_tokens, 0)
|
|
154
|
+
+ CASE
|
|
155
|
+
WHEN t.started_at >= ${H}
|
|
156
|
+
AND (
|
|
157
|
+
t.crewx_version IS NULL
|
|
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, ${q}) AS INTEGER) < ${j})
|
|
160
|
+
)
|
|
161
|
+
THEN COALESCE(t.cached_input_tokens, 0)
|
|
162
|
+
ELSE 0
|
|
163
|
+
END
|
|
164
|
+
), 0) AS input_tokens,
|
|
87
165
|
COALESCE(SUM(t.output_tokens), 0) AS output_tokens,
|
|
88
166
|
COALESCE(SUM(t.cached_input_tokens), 0) AS cached_input_tokens,
|
|
89
167
|
COALESCE(SUM(t.cost_usd), 0) AS cost_usd
|
|
@@ -93,7 +171,7 @@ ${s.join(`
|
|
|
93
171
|
AND t.started_at < ${e}
|
|
94
172
|
GROUP BY date(t.started_at), t.agent_id
|
|
95
173
|
ORDER BY date(t.started_at) ASC
|
|
96
|
-
`).map(o=>({date:o.date,agentId:o.agent_id,inputTokens:o.input_tokens,outputTokens:o.output_tokens,cachedInputTokens:o.cached_input_tokens,costUsd:o.cost_usd}))}catch(r){throw new u("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(a).where(and(eq(a.id,t),eq(a.workspace_id,e))).limit(1).get()??void 0}catch(n){throw new u("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(),o=s?and(eq(a.id,t),eq(a.status,"running"),eq(a.workspace_id,s)):and(eq(a.id,t),eq(a.status,"running"));n.db.update(a).set({status:"failed",error:e,completed_at:r}).where(o).run();}catch(r){throw r instanceof u?r:new u("DB_ERROR","Failed to mark task failed",r)}finally{n.close();}}findTasksByPromptHint(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=
|
|
174
|
+
`).map(o=>({date:o.date,agentId:o.agent_id,inputTokens:o.input_tokens,outputTokens:o.output_tokens,cachedInputTokens:o.cached_input_tokens,costUsd:o.cost_usd,totalTokens:o.input_tokens+o.output_tokens}))}catch(r){throw new u("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(a).where(and(eq(a.id,t),eq(a.workspace_id,e))).limit(1).get()??void 0}catch(n){throw new u("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(),o=s?and(eq(a.id,t),eq(a.status,"running"),eq(a.workspace_id,s)):and(eq(a.id,t),eq(a.status,"running"));n.db.update(a).set({status:"failed",error:e,completed_at:r}).where(o).run();}catch(r){throw r instanceof u?r:new u("DB_ERROR","Failed to mark task failed",r)}finally{n.close();}}findTasksByPromptHint(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=S(o);try{let l=e?and(like(a.prompt,`%${t}%`),eq(a.workspace_id,e)):like(a.prompt,`%${t}%`),p=i.db.select().from(a).where(l).orderBy(asc(a.started_at)).all();for(let _ of p)n.has(_.id)||(n.add(_.id),r.push(_));}catch(l){throw new u("DB_ERROR","Failed to find tasks by prompt hint",l)}finally{i.close();}}return r}};var ut=class extends v{name="sqlite-tracing";unsubs=[];dbPath;version;constructor(t){super(),this.dbPath=join(t?.dbRoot??homedir(),".crewx","crewx.db"),this.version=t?.version??"unknown";}attach(t){let e=new et({dbPath:this.dbPath}),s=process.cwd(),r=existsSync(join(s,"crewx.yaml"))||existsSync(join(s,"crewx.yml"))?gt(s):null,o=process.argv.join(" ");this.unsubs.push(t.on("task:start",i=>{try{let l=i.callerAgentId??null,p=i.parentTaskId??null,_=i.rootTraceId??i.traceId,k=i.metadata?JSON.stringify(i.metadata):JSON.stringify({provider:i.provider??"cli/claude"});e.startTask({id:i.traceId,agentId:i.agentRef.replace(/^@/,""),prompt:i.message,mode:i.mode,status:"running",pid:i.pid??null,startedAt:i.timestamp.toISOString(),crewxVersion:this.version,platform:i.platform??"cli",model:i.model??null,renderedPrompt:i.renderedPrompt??null,command:o,codingAgentCommand:i.codingAgentCommand??null,workspaceId:i.workspaceId??r,callerAgentId:l,parentTaskId:p,traceId:_,metadata:k,threadId:i.threadId??null});}catch{}}),t.on("task:output",i=>{try{e.appendLog(i.traceId,{timestamp:i.timestamp.toISOString(),level:i.level??"stdout",message:i.output});}catch{}}),t.on("task:end",i=>{try{e.finishTask({id:i.traceId,status:i.error?"failed":"success",result:i.result??null,error:i.error?JSON.stringify(i.error):null,completedAt:i.timestamp.toISOString(),durationMs:i.durationMs,exitCode:i.exitCode??null,inputTokens:i.inputTokens??0,outputTokens:i.outputTokens??0,cachedInputTokens:i.cachedInputTokens??0,costUsd:i.costUsd??0,model:i.model??null});}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[];}};var st=class extends U{dbPath;constructor(t={}){super(),t.dbPath?this.dbPath=t.dbPath:t.dbRoot&&(this.dbPath=join(t.dbRoot,".crewx","crewx.db"));}resolveDbPath(){return this.dbPath?this.dbPath:super.resolveDbPath()}openHandle(t){let e=this.resolveDbPath();if(t){let n=dirname(e);existsSync(n)||mkdirSync(n,{recursive:true});}else if(!existsSync(e))throw new u("NOT_FOUND","Database not found");let s=S(e);if(t)try{J(s.db,e);}catch(n){throw s.close(),n}return s}validateWorkspaceId(t,e){return t.db.select({id:F.id}).from(F).where(eq(F.id,e)).limit(1).get()?e:null}topLevelTaskPredicateSql(t="child"){return sql.raw(`(
|
|
97
175
|
${t}.parent_task_id IS NULL
|
|
98
176
|
OR ${t}.parent_task_id = ''
|
|
99
177
|
OR NOT EXISTS (
|
|
@@ -101,7 +179,7 @@ ${s.join(`
|
|
|
101
179
|
WHERE parent.id = ${t}.parent_task_id
|
|
102
180
|
AND parent.thread_id = ${t}.thread_id
|
|
103
181
|
)
|
|
104
|
-
)`)}findAllThreads(t){let e=this.resolveDbPaths(),s=new Set,n=[];for(let r of e){if(!existsSync(r))continue;let o=
|
|
182
|
+
)`)}findAllThreads(t){let e=this.resolveDbPaths(),s=new Set,n=[];for(let r of e){if(!existsSync(r))continue;let o=S(r);try{let i=t?eq(c.workspace_id,t):void 0,l=o.db.select().from(c).where(i).orderBy(desc(c.updated_at)).all();for(let p of l)s.has(p.id)||(s.add(p.id),n.push(p));}catch(i){throw new u("DB_ERROR","Failed to find all threads",i)}finally{o.close();}}return n}findThreadById(t,e){let s=this.resolveDbPaths();for(let n of s){if(!existsSync(n))continue;let r=S(n);try{let o=eq(c.id,t),i=e?and(o,eq(c.workspace_id,e)):o,l=r.db.select().from(c).where(i).limit(1).get()??void 0;if(l)return l}catch(o){throw new u("DB_ERROR","Failed to find thread by id",o)}finally{r.close();}}}threadExists(t,e){let s=this.resolveDbPaths();for(let n of s){if(!existsSync(n))continue;let r=S(n);try{let o=eq(c.id,t),i=e?and(o,eq(c.workspace_id,e)):o;if(r.db.select({id:c.id}).from(c).where(i).limit(1).get())return !0}catch(o){throw new u("DB_ERROR","Failed to check thread existence",o)}finally{r.close();}}return false}aggregateTaskStats(t,e){let s=this.resolveDbPaths(),n=0,r=0,o=0,i=0,l=0,p=new Set;for(let _ of s){if(!existsSync(_))continue;let k=S(_);try{let w=k.db.get(sql`
|
|
105
183
|
SELECT
|
|
106
184
|
count(*) AS cnt,
|
|
107
185
|
COALESCE(SUM(child.input_tokens), 0) AS total_input,
|
|
@@ -112,19 +190,19 @@ ${s.join(`
|
|
|
112
190
|
WHERE child.thread_id = ${t}
|
|
113
191
|
AND ${this.topLevelTaskPredicateSql()}
|
|
114
192
|
${e?sql`AND child.workspace_id = ${e}`:sql``}
|
|
115
|
-
`);w&&(n+=w.cnt,r+=w.total_input,o+=w.total_output,i+=w.total_cached,l+=w.total_cost);let
|
|
193
|
+
`);w&&(n+=w.cnt,r+=w.total_input,o+=w.total_output,i+=w.total_cached,l+=w.total_cost);let y=k.db.all(sql`
|
|
116
194
|
SELECT DISTINCT child.agent_id FROM tasks child
|
|
117
195
|
WHERE child.thread_id = ${t}
|
|
118
196
|
AND child.agent_id IS NOT NULL AND child.agent_id != ''
|
|
119
197
|
AND ${this.topLevelTaskPredicateSql()}
|
|
120
198
|
${e?sql`AND child.workspace_id = ${e}`:sql``}
|
|
121
|
-
`);for(let
|
|
199
|
+
`);for(let K of y)p.add(K.agent_id);}catch(w){throw new u("DB_ERROR","Failed to aggregate task stats",w)}finally{k.close();}}return {taskCount:n,inputTokens:r,outputTokens:o,cachedInputTokens:i,costUsd:l,agentIds:Array.from(p)}}findTopLevelTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=S(o);try{let l=i.db.all(sql`
|
|
122
200
|
SELECT child.* FROM tasks child
|
|
123
201
|
WHERE child.thread_id = ${t}
|
|
124
202
|
AND ${this.topLevelTaskPredicateSql()}
|
|
125
203
|
${e?sql`AND child.workspace_id = ${e}`:sql``}
|
|
126
204
|
ORDER BY child.started_at ASC
|
|
127
|
-
`);for(let p of l)n.has(p.id)||(n.add(p.id),r.push(p));}catch(l){throw new u("DB_ERROR","Failed to find top-level tasks",l)}finally{i.close();}}return r}findAllTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=
|
|
205
|
+
`);for(let p of l)n.has(p.id)||(n.add(p.id),r.push(p));}catch(l){throw new u("DB_ERROR","Failed to find top-level tasks",l)}finally{i.close();}}return r}findAllTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let o of s){if(!existsSync(o))continue;let i=S(o);try{let l=eq(a.thread_id,t),p=e?and(l,eq(a.workspace_id,e)):l,_=i.db.select().from(a).where(p).orderBy(asc(a.started_at)).all();for(let k of _)n.has(k.id)||(n.add(k.id),r.push(k));}catch(l){throw new u("DB_ERROR","Failed to find all tasks for thread",l)}finally{i.close();}}return r}findTaskById(t,e,s){let n=this.resolveDbPaths();for(let r of n){if(!existsSync(r))continue;let o=S(r);try{let i=and(eq(a.id,e),eq(a.thread_id,t)),l=s?and(i,eq(a.workspace_id,s)):i,p=o.db.select().from(a).where(l).limit(1).get();if(!p)continue;let _=o.db.select().from(a).where(eq(a.parent_task_id,p.id)).orderBy(asc(a.started_at)).all();return {task:p,children:_}}catch(i){throw new u("DB_ERROR","Failed to find task by id",i)}finally{o.close();}}}batchFetchTasks(t,e){let s=new Map;if(t.length===0)return s;let n=this.resolveDbPaths();for(let r of n){if(!existsSync(r))continue;let o=S(r);try{let i=o.db.all(sql`
|
|
128
206
|
SELECT child.* FROM tasks child
|
|
129
207
|
WHERE child.thread_id IN (${sql.join(t.map(l=>sql`${l}`),sql`, `)})
|
|
130
208
|
AND ${this.topLevelTaskPredicateSql()}
|
|
@@ -138,7 +216,7 @@ ${s.join(`
|
|
|
138
216
|
message_count = message_count + 1,
|
|
139
217
|
updated_at = ${r}
|
|
140
218
|
WHERE id = ${t}
|
|
141
|
-
`),p},{behavior:"immediate"})}}catch(r){throw r instanceof u?r:new u("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(eq(c.id,t)).run();}catch(r){throw r instanceof u?r:new u("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(eq(c.id,t)).run();}catch(n){throw n instanceof u?n:new u("DB_ERROR","Failed to update thread",n)}finally{s.close();}}togglePin(t,e){let s=this.openHandle(true);try{let n=e?and(eq(c.id,t),eq(c.workspace_id,e)):eq(c.id,t),r=s.db.select({pinned:c.pinned,metadata:c.metadata}).from(c).where(n).get();if(!r)return null;let o=r.pinned?0:1,i=r.metadata?JSON.parse(r.metadata):{};if(o){let l=e?and(eq(c.pinned,1),eq(c.workspace_id,e)):eq(c.pinned,1),p=s.db.select({metadata:c.metadata}).from(c).where(l).all(),_=0;for(let k of p){let w=k.metadata?JSON.parse(k.metadata):{};typeof w.pinOrder=="number"&&w.pinOrder>_&&(_=w.pinOrder);}i.pinOrder=_+1;}else delete i.pinOrder;return s.db.update(c).set({pinned:o,metadata:Object.keys(i).length>0?JSON.stringify(i):null}).where(n).run(),{pinned:!!o}}catch(n){throw n instanceof u?n:new u("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?and(eq(c.id,t[n]),eq(c.workspace_id,e)):eq(c.id,t[n]),o=s.db.select({metadata:c.metadata}).from(c).where(r).get();if(!o)continue;let i=o.metadata?JSON.parse(o.metadata):{};i.pinOrder=n+1,s.db.update(c).set({metadata:JSON.stringify(i)}).where(r).run();}}catch(n){throw n instanceof u?n:new u("DB_ERROR","Failed to reorder pins",n)}finally{s.close();}}toggleStar(t,e){let s=this.openHandle(true);try{let n=e?and(eq(c.id,t),eq(c.workspace_id,e)):eq(c.id,t),r=s.db.select({starred:c.starred}).from(c).where(n).get();if(!r)return null;let o=r.starred?0:1;return s.db.update(c).set({starred:o}).where(n).run(),{starred:!!o}}catch(n){throw n instanceof u?n:new u("DB_ERROR","Failed to toggle star",n)}finally{s.close();}}};function
|
|
219
|
+
`),p},{behavior:"immediate"})}}catch(r){throw r instanceof u?r:new u("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(eq(c.id,t)).run();}catch(r){throw r instanceof u?r:new u("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(eq(c.id,t)).run();}catch(n){throw n instanceof u?n:new u("DB_ERROR","Failed to update thread",n)}finally{s.close();}}togglePin(t,e){let s=this.openHandle(true);try{let n=e?and(eq(c.id,t),eq(c.workspace_id,e)):eq(c.id,t),r=s.db.select({pinned:c.pinned,metadata:c.metadata}).from(c).where(n).get();if(!r)return null;let o=r.pinned?0:1,i=r.metadata?JSON.parse(r.metadata):{};if(o){let l=e?and(eq(c.pinned,1),eq(c.workspace_id,e)):eq(c.pinned,1),p=s.db.select({metadata:c.metadata}).from(c).where(l).all(),_=0;for(let k of p){let w=k.metadata?JSON.parse(k.metadata):{};typeof w.pinOrder=="number"&&w.pinOrder>_&&(_=w.pinOrder);}i.pinOrder=_+1;}else delete i.pinOrder;return s.db.update(c).set({pinned:o,metadata:Object.keys(i).length>0?JSON.stringify(i):null}).where(n).run(),{pinned:!!o}}catch(n){throw n instanceof u?n:new u("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?and(eq(c.id,t[n]),eq(c.workspace_id,e)):eq(c.id,t[n]),o=s.db.select({metadata:c.metadata}).from(c).where(r).get();if(!o)continue;let i=o.metadata?JSON.parse(o.metadata):{};i.pinOrder=n+1,s.db.update(c).set({metadata:JSON.stringify(i)}).where(r).run();}}catch(n){throw n instanceof u?n:new u("DB_ERROR","Failed to reorder pins",n)}finally{s.close();}}toggleStar(t,e){let s=this.openHandle(true);try{let n=e?and(eq(c.id,t),eq(c.workspace_id,e)):eq(c.id,t),r=s.db.select({starred:c.starred}).from(c).where(n).get();if(!r)return null;let o=r.starred?0:1;return s.db.update(c).set({starred:o}).where(n).run(),{starred:!!o}}catch(n){throw n instanceof u?n:new u("DB_ERROR","Failed to toggle star",n)}finally{s.close();}}};function Ee(d){return d.replace(/<conversation_history[^>]*>[\s\S]*?<\/conversation_history>/g,"").split(`
|
|
142
220
|
`).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(`
|
|
143
|
-
`).trim()}function
|
|
144
|
-
`):e&&typeof e=="object"&&e.result!==void 0&&(t=e.result||"");}catch{t=
|
|
221
|
+
`).trim()}function Se(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=Ee(t);}return t}var nt=class{dbPath;constructor(t){this.dbPath=t??join(homedir(),".crewx","crewx.db");}getThreadRepo(){return new st({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),o=n.findTopLevelTasks(t),i=new Set(["done","completed","success"]);o=o.filter(_=>(!_.status||i.has(_.status))&&(!e?.currentTraceId||_.trace_id!==e.currentTraceId)),o=o.slice(0,s);let l=r?.platform??"cli",p=this.rowsToMessages(o);return {threadId:t,platform:l,messages:p,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=Se(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 pt=class extends v{name="conversation";_provider;unsubStart=null;unsubEnd=null;constructor(t){super(),this._provider=new nt(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?.();}};export{pt as ConversationPlugin,rt as FileLoggerPlugin,ut as SqliteTracingPlugin};
|