@crewx/sdk 0.8.9 → 0.9.0-rc.1
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 +51 -47
- package/dist/esm/plugins/index.js +30 -29
- package/dist/esm/repository/index.js +20 -19
- package/dist/index.browser.js +1 -1
- package/dist/index.js +51 -47
- package/dist/migrations/0006_add_task_run_epoch.sql +1 -0
- package/dist/migrations/meta/_journal.json +7 -0
- package/dist/plugins/index.js +19 -18
- package/dist/provider/bridge.d.ts +1 -0
- package/dist/provider/cli/adapter.types.d.ts +8 -0
- package/dist/provider/cli/meta.d.ts +2 -0
- package/dist/repository/index.d.ts +1 -0
- package/dist/repository/index.js +20 -19
- package/dist/repository/task.repository.d.ts +1 -0
- package/dist/schema/tasks.d.ts +17 -0
- package/dist/types/index.d.ts +2 -0
- package/dist/types/task-log.types.d.ts +1 -1
- package/package.json +4 -4
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import*as N from'path';import N__default,{join,dirname}from'path';import {fileURLToPath}from'url';import {existsSync,mkdirSync,writeFileSync,appendFileSync,readFileSync}from'fs';import Ft,{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=()=>N__default.dirname(Nt()),g=vt();var L=class{detach(t){}};function
|
|
1
|
+
import*as N from'path';import N__default,{join,dirname}from'path';import {fileURLToPath}from'url';import {existsSync,mkdirSync,writeFileSync,appendFileSync,readFileSync}from'fs';import Ft,{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=()=>N__default.dirname(Nt()),g=vt();var L=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 L{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,12 @@ 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",i=`[${n}] INFO: Task ${r} in ${e.durationMs}ms
|
|
10
10
|
[${n}] INFO: Process closed with exit code: ${e.error?1:0}
|
|
11
|
-
`;appendFileSync(s,i,"utf8"),this.logFiles.delete(e.traceId);}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[],this.logFiles.clear();}};function Mt(d){let t=N.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
|
|
11
|
+
`;appendFileSync(s,i,"utf8"),this.logFiles.delete(e.traceId);}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[],this.logFiles.clear();}};function Mt(d){let t=N.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 z=class{resolveDbPath(){return process.env.CREWX_DB?process.env.CREWX_DB:process.env.CREWX_TRACES_DB?process.env.CREWX_TRACES_DB:join(Ft.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 T(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,Ht={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 zt(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(Ht))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 Wt(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(i=>i.hash)),r=JSON.parse(readFileSync(N__default.join(t,"meta/_journal.json"),"utf-8"));for(let i of r.entries){let a=qt[i.tag];if(!a)continue;let l=N__default.join(t,`${i.tag}.sql`);if(!existsSync(l))continue;let c=readFileSync(l,"utf-8"),_=createHash("sha256").update(c).digest("hex");if(n.has(_))continue;let k=d.all("PRAGMA table_info(tasks)"),w=new Set(k.map(R=>R.name));for(let[R,K]of Object.entries(a))w.has(R)||(d.run(`ALTER TABLE tasks ADD COLUMN ${R} ${K}`),w.add(R));}}function jt(d,t,e){let s=d.all(e`SELECT hash FROM __drizzle_migrations`),n=new Set(s.map(i=>i.hash)),r=JSON.parse(readFileSync(N__default.join(t,"meta/_journal.json"),"utf-8"));for(let i of r.entries){let a=N__default.join(t,`${i.tag}.sql`);if(!existsSync(a))continue;let l=readFileSync(a,"utf-8"),c=createHash("sha256").update(l).digest("hex");if(n.has(c))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(j=>j.trim()).filter(Boolean).every(j=>/^ALTER\s+TABLE\s+.+\s+ADD\s+/i.test(j)))continue;k.every(({table:j,column:At})=>d.all(`PRAGMA table_info("${j}")`).some(Ct=>Ct.name===At))&&d.run(e`INSERT INTO __drizzle_migrations (hash, created_at) VALUES (${c}, ${i.when})`);}}function Kt(d){let{migrate:t}=G("drizzle-orm/better-sqlite3/migrator"),{sql:e}=G("drizzle-orm"),s=[N__default.join(g,"../migrations"),N__default.join(g,"migrations"),N__default.join(g,"../../../../drizzle/migrations"),N__default.join(process.cwd(),"drizzle/migrations")],n=s.find(c=>existsSync(N__default.join(c,"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'`),i=0;r?.cnt&&(i=d.get(e`SELECT count(*) as cnt FROM __drizzle_migrations`)?.cnt??0),zt(d,i),r?.cnt&&(jt(d,n,e),Wt(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 c=r?.cnt?"Database migrated":"Database initialized";console.log(`[crewx] ${c} (${l} migration${l>1?"s":""} applied).`);}}function Y(d,t){kt.has(t)||(Kt(d),kt.add(t));}var
|
|
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),zt(d,i),r?.cnt&&(jt(d,n,e),Wt(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 c=r?.cnt?"Database migrated":"Database initialized";console.log(`[crewx] ${c} (${l} migration${l>1?"s":""} applied).`);}}function Y(d,t){kt.has(t)||(Kt(d),kt.add(t));}var p=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 q=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 o=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),run_epoch:integer("run_epoch").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 u=sqliteTable("threads",{id:text("id").primaryKey(),workspace_id:text("workspace_id").references(()=>q.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(()=>o.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(()=>o.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(()=>u.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 B="2026-05-09",pe="0.8.9-rc.13",F=10,H=parseInt(pe.split("rc.")[1]),et=class extends z{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 p("NOT_FOUND","Database not found");let s=T(e);if(t)try{Y(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 p?s:new p("DB_ERROR","Failed to start task",s)}finally{e.close();}}finishTask(t){let e=this.openHandle(true);try{let s=t.runEpoch??null;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)
|
|
15
|
+
model=COALESCE(?, model)
|
|
16
|
+
WHERE id=? AND status='running' AND COALESCE(run_epoch, 0) = COALESCE(?, 0)`,[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,s]);}catch(s){throw s instanceof p?s:new p("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(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(eq(o.id,t)).run();},{behavior:"immediate"});}catch(n){throw n instanceof p?n:new p("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(eq(o.status,"running")).orderBy(desc(o.started_at)).all()}catch(e){throw new p("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(desc(o.started_at)).limit(100).all()}catch(e){throw new p("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(eq(o.id,t)).limit(1).get()??void 0}catch(s){throw new p("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(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(and(eq(o.id,t),eq(o.status,"running"))).run(),{killed:!0,pid:s.pid??void 0}}catch(s){throw s instanceof p?s:new p("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(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(and(eq(o.id,n.id),eq(o.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=T(n);try{let i=e?eq(o.workspace_id,e):void 0,a=i?and(eq(o.id,t),i):eq(o.id,t),l=r.db.select().from(o).where(a).limit(1).get()??void 0;if(!l){let c=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),_=i?and(c,i):c;l=r.db.select().from(o).where(_).orderBy(desc(o.started_at)).limit(1).get()??void 0;}if(l)return l}catch(i){throw new p("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(!existsSync(i))continue;let a=T(i);try{let l=e?and(eq(o.parent_task_id,t),eq(o.workspace_id,e)):eq(o.parent_task_id,t),c=a.db.select().from(o).where(l).orderBy(asc(o.started_at)).all();for(let _ of c)n.has(_.id)||(n.add(_.id),r.push(_));}catch(l){throw new p("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?sql`
|
|
16
17
|
SELECT
|
|
17
18
|
COALESCE(workspace_id, 'unknown') AS workspace_id,
|
|
18
19
|
COALESCE(SUM(input_tokens), 0) AS input_tokens,
|
|
@@ -33,7 +34,7 @@ ${s.join(`
|
|
|
33
34
|
FROM tasks
|
|
34
35
|
GROUP BY workspace_id
|
|
35
36
|
ORDER BY (COALESCE(SUM(input_tokens), 0) + COALESCE(SUM(output_tokens), 0)) DESC
|
|
36
|
-
`)}catch(s){throw new
|
|
37
|
+
`)}catch(s){throw new p("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(!existsSync(l))continue;let c=T(l);try{let _=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),k=e?and(_,eq(o.workspace_id,e)):_,w=c.db.select({id:o.id,input_tokens:o.input_tokens,output_tokens:o.output_tokens,cost_usd:o.cost_usd}).from(o).where(k).all();for(let R of w)n.has(R.id)||(n.add(R.id),r+=R.input_tokens??0,i+=R.output_tokens??0,a+=R.cost_usd??0);}catch(_){throw new p("DB_ERROR","Failed to get thread token usage",_)}finally{c.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(!existsSync(i))continue;let a=T(i);try{let l=or(eq(o.thread_id,t),and(isNull(o.thread_id),like(o.command,`%--thread=${t}%`))),c=e?and(l,eq(o.workspace_id,e)):l,_=a.db.select().from(o).where(c).orderBy(asc(o.started_at)).all();for(let k of _)n.has(k.id)||(n.add(k.id),r.push(k));}catch(l){throw new p("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(eq(o.workspace_id,t.workspaceId));let n=t.agents&&t.agents.length>0?t.agents:t.agentId?[t.agentId]:null;n&&s.push(inArray(o.agent_id,n));let r=t.statuses&&t.statuses.length>0?t.statuses:t.status?[t.status]:null;r&&s.push(inArray(o.status,r));let i=t.q??t.search;i&&s.push(like(o.prompt,`%${i}%`)),t.from&&s.push(gte(o.started_at,t.from)),t.to&&s.push(lt(o.started_at,t.to));let a=s.length>0?and(...s):void 0,l=e.db.select({count:sql`count(*)`}).from(o).where(a).get(),c=(t.sortDir??"DESC")==="ASC"?asc(o.started_at):desc(o.started_at);return {rows:e.db.select().from(o).where(a).orderBy(c).limit(t.limit).offset(t.offset).all(),total:l?.count??0}}catch(s){throw new p("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
38
|
SELECT
|
|
38
39
|
t.agent_id,
|
|
39
40
|
t.workspace_id,
|
|
@@ -41,11 +42,11 @@ ${s.join(`
|
|
|
41
42
|
COALESCE(SUM(
|
|
42
43
|
COALESCE(t.input_tokens, 0)
|
|
43
44
|
+ CASE
|
|
44
|
-
WHEN t.started_at >= ${
|
|
45
|
+
WHEN t.started_at >= ${B}
|
|
45
46
|
AND (
|
|
46
47
|
t.crewx_version IS NULL
|
|
47
48
|
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, ${
|
|
49
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${F}) AS INTEGER) < ${H})
|
|
49
50
|
)
|
|
50
51
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
51
52
|
ELSE 0
|
|
@@ -64,11 +65,11 @@ ${s.join(`
|
|
|
64
65
|
COALESCE(SUM(
|
|
65
66
|
COALESCE(t.input_tokens, 0)
|
|
66
67
|
+ CASE
|
|
67
|
-
WHEN t.started_at >= ${
|
|
68
|
+
WHEN t.started_at >= ${B}
|
|
68
69
|
AND (
|
|
69
70
|
t.crewx_version IS NULL
|
|
70
71
|
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, ${
|
|
72
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${F}) AS INTEGER) < ${H})
|
|
72
73
|
)
|
|
73
74
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
74
75
|
ELSE 0
|
|
@@ -84,11 +85,11 @@ ${s.join(`
|
|
|
84
85
|
COALESCE(SUM(
|
|
85
86
|
COALESCE(t.input_tokens, 0)
|
|
86
87
|
+ CASE
|
|
87
|
-
WHEN t.started_at >= ${
|
|
88
|
+
WHEN t.started_at >= ${B}
|
|
88
89
|
AND (
|
|
89
90
|
t.crewx_version IS NULL
|
|
90
91
|
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, ${
|
|
92
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${F}) AS INTEGER) < ${H})
|
|
92
93
|
)
|
|
93
94
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
94
95
|
ELSE 0
|
|
@@ -106,11 +107,11 @@ ${s.join(`
|
|
|
106
107
|
COALESCE(SUM(
|
|
107
108
|
COALESCE(t.input_tokens, 0)
|
|
108
109
|
+ CASE
|
|
109
|
-
WHEN t.started_at >= ${
|
|
110
|
+
WHEN t.started_at >= ${B}
|
|
110
111
|
AND (
|
|
111
112
|
t.crewx_version IS NULL
|
|
112
113
|
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, ${
|
|
114
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${F}) AS INTEGER) < ${H})
|
|
114
115
|
)
|
|
115
116
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
116
117
|
ELSE 0
|
|
@@ -118,18 +119,18 @@ ${s.join(`
|
|
|
118
119
|
), 0)
|
|
119
120
|
+ COALESCE(SUM(t.output_tokens), 0)
|
|
120
121
|
) DESC
|
|
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
|
|
122
|
+
`).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 p("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`
|
|
122
123
|
SELECT
|
|
123
124
|
date(t.started_at) AS date,
|
|
124
125
|
t.agent_id,
|
|
125
126
|
COALESCE(SUM(
|
|
126
127
|
COALESCE(t.input_tokens, 0)
|
|
127
128
|
+ CASE
|
|
128
|
-
WHEN t.started_at >= ${
|
|
129
|
+
WHEN t.started_at >= ${B}
|
|
129
130
|
AND (
|
|
130
131
|
t.crewx_version IS NULL
|
|
131
132
|
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, ${
|
|
133
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${F}) AS INTEGER) < ${H})
|
|
133
134
|
)
|
|
134
135
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
135
136
|
ELSE 0
|
|
@@ -152,11 +153,11 @@ ${s.join(`
|
|
|
152
153
|
COALESCE(SUM(
|
|
153
154
|
COALESCE(t.input_tokens, 0)
|
|
154
155
|
+ CASE
|
|
155
|
-
WHEN t.started_at >= ${
|
|
156
|
+
WHEN t.started_at >= ${B}
|
|
156
157
|
AND (
|
|
157
158
|
t.crewx_version IS NULL
|
|
158
159
|
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, ${
|
|
160
|
+
OR (t.crewx_version LIKE '0.8.9-rc.%' AND CAST(SUBSTR(t.crewx_version, ${F}) AS INTEGER) < ${H})
|
|
160
161
|
)
|
|
161
162
|
THEN COALESCE(t.cached_input_tokens, 0)
|
|
162
163
|
ELSE 0
|
|
@@ -171,7 +172,7 @@ ${s.join(`
|
|
|
171
172
|
AND t.started_at < ${e}
|
|
172
173
|
GROUP BY date(t.started_at), t.agent_id
|
|
173
174
|
ORDER BY date(t.started_at) ASC
|
|
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
|
|
175
|
+
`).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 p("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(and(eq(o.id,t),eq(o.workspace_id,e))).limit(1).get()??void 0}catch(n){throw new p("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?and(eq(o.id,t),eq(o.status,"running"),eq(o.workspace_id,s)):and(eq(o.id,t),eq(o.status,"running"));n.db.update(o).set({status:"failed",error:e,completed_at:r}).where(i).run();}catch(r){throw r instanceof p?r:new p("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(!existsSync(i))continue;let a=T(i);try{let l=e?and(like(o.prompt,`%${t}%`),eq(o.workspace_id,e)):like(o.prompt,`%${t}%`),c=a.db.select().from(o).where(l).orderBy(asc(o.started_at)).all();for(let _ of c)n.has(_.id)||(n.add(_.id),r.push(_));}catch(l){throw new p("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=sql`
|
|
175
176
|
CASE
|
|
176
177
|
WHEN ${o.model} LIKE 'claude-%' OR ${o.model} IN ('opus', 'sonnet', 'haiku', 'opus[1m]', 'sonnet[1m]') THEN 'claude'
|
|
177
178
|
WHEN ${o.model} LIKE 'gpt-%' OR ${o.model} LIKE 'codex-%' THEN 'codex'
|
|
@@ -184,11 +185,11 @@ ${s.join(`
|
|
|
184
185
|
`,i=sql`
|
|
185
186
|
COALESCE(${o.input_tokens}, 0)
|
|
186
187
|
+ CASE
|
|
187
|
-
WHEN ${o.started_at} >= ${
|
|
188
|
+
WHEN ${o.started_at} >= ${B}
|
|
188
189
|
AND (
|
|
189
190
|
${o.crewx_version} IS NULL
|
|
190
191
|
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}, ${
|
|
192
|
+
OR (${o.crewx_version} LIKE '0.8.9-rc.%' AND CAST(SUBSTR(${o.crewx_version}, ${F}) AS INTEGER) < ${H})
|
|
192
193
|
)
|
|
193
194
|
THEN COALESCE(${o.cached_input_tokens}, 0)
|
|
194
195
|
ELSE 0
|
|
@@ -207,7 +208,7 @@ ${s.join(`
|
|
|
207
208
|
${a}
|
|
208
209
|
GROUP BY provider
|
|
209
210
|
ORDER BY (COALESCE(SUM(${i}), 0) + COALESCE(SUM(${o.output_tokens}), 0)) DESC
|
|
210
|
-
`).map(c=>({provider:c.provider,totalTasks:c.total_tasks,inputTokens:c.input_tokens,outputTokens:c.output_tokens,cachedInputTokens:c.cached_input_tokens,costUsd:c.cost_usd,totalTokens:c.input_tokens+c.output_tokens,activeDurationMs:c.active_duration_ms??0,lastActiveAt:c.last_active_at??null}))}catch(r){throw new
|
|
211
|
+
`).map(c=>({provider:c.provider,totalTasks:c.total_tasks,inputTokens:c.input_tokens,outputTokens:c.output_tokens,cachedInputTokens:c.cached_input_tokens,costUsd:c.cost_usd,totalTokens:c.input_tokens+c.output_tokens,activeDurationMs:c.active_duration_ms??0,lastActiveAt:c.last_active_at??null}))}catch(r){throw new p("DB_ERROR","Failed to get provider usage",r)}finally{n.close();}}};var ut=class extends L{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,i=process.argv.join(" ");this.unsubs.push(t.on("task:start",a=>{try{let l=a.callerAgentId??null,c=a.parentTaskId??null,_=a.rootTraceId??a.traceId,k=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:c,traceId:_,metadata:k,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{let l=typeof a.metadata?.runEpoch=="number"?a.metadata.runEpoch:null;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,runEpoch:l});}catch{}}));}detach(t){this.unsubs.forEach(e=>e()),this.unsubs=[];}};var st=class extends z{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 p("NOT_FOUND","Database not found");let s=T(e);if(t)try{Y(s.db,e);}catch(n){throw s.close(),n}return s}validateWorkspaceId(t,e){return t.db.select({id:q.id}).from(q).where(eq(q.id,e)).limit(1).get()?e:null}topLevelTaskPredicateSql(t="child"){return sql.raw(`(
|
|
211
212
|
${t}.parent_task_id IS NULL
|
|
212
213
|
OR ${t}.parent_task_id = ''
|
|
213
214
|
OR NOT EXISTS (
|
|
@@ -215,7 +216,7 @@ ${s.join(`
|
|
|
215
216
|
WHERE parent.id = ${t}.parent_task_id
|
|
216
217
|
AND parent.thread_id = ${t}.thread_id
|
|
217
218
|
)
|
|
218
|
-
)`)}findAllThreads(t){let e=this.resolveDbPaths(),s=new Set,n=[];for(let r of e){if(!existsSync(r))continue;let i=T(r);try{let a=t?eq(u.workspace_id,t):void 0,l=i.db.select().from(u).where(a).orderBy(desc(u.updated_at)).all();for(let c of l)s.has(c.id)||(s.add(c.id),n.push(c));}catch(a){throw new
|
|
219
|
+
)`)}findAllThreads(t){let e=this.resolveDbPaths(),s=new Set,n=[];for(let r of e){if(!existsSync(r))continue;let i=T(r);try{let a=t?eq(u.workspace_id,t):void 0,l=i.db.select().from(u).where(a).orderBy(desc(u.updated_at)).all();for(let c of l)s.has(c.id)||(s.add(c.id),n.push(c));}catch(a){throw new p("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(!existsSync(n))continue;let r=T(n);try{let i=eq(u.id,t),a=e?and(i,eq(u.workspace_id,e)):i,l=r.db.select().from(u).where(a).limit(1).get()??void 0;if(l)return l}catch(i){throw new p("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(!existsSync(n))continue;let r=T(n);try{let i=eq(u.id,t),a=e?and(i,eq(u.workspace_id,e)):i;if(r.db.select({id:u.id}).from(u).where(a).limit(1).get())return !0}catch(i){throw new p("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,c=new Set;for(let _ of s){if(!existsSync(_))continue;let k=T(_);try{let w=k.db.get(sql`
|
|
219
220
|
SELECT
|
|
220
221
|
count(*) AS cnt,
|
|
221
222
|
COALESCE(SUM(child.input_tokens), 0) AS total_input,
|
|
@@ -232,7 +233,7 @@ ${s.join(`
|
|
|
232
233
|
AND child.agent_id IS NOT NULL AND child.agent_id != ''
|
|
233
234
|
AND ${this.topLevelTaskPredicateSql()}
|
|
234
235
|
${e?sql`AND child.workspace_id = ${e}`:sql``}
|
|
235
|
-
`);for(let K of R)c.add(K.agent_id);}catch(w){throw new
|
|
236
|
+
`);for(let K of R)c.add(K.agent_id);}catch(w){throw new p("DB_ERROR","Failed to aggregate task stats",w)}finally{k.close();}}return {taskCount:n,inputTokens:r,outputTokens:i,cachedInputTokens:a,costUsd:l,agentIds:Array.from(c)}}findTopLevelTasks(t,e,s){let n=this.resolveDbPaths(),r=new Set,i=[];for(let a of n){if(!existsSync(a))continue;let l=T(a);try{let c;s!==void 0?c=l.db.all(sql`
|
|
236
237
|
SELECT * FROM (
|
|
237
238
|
SELECT child.*,
|
|
238
239
|
ROW_NUMBER() OVER (PARTITION BY child.agent_id ORDER BY child.started_at DESC) AS rn
|
|
@@ -249,13 +250,13 @@ ${s.join(`
|
|
|
249
250
|
AND ${this.topLevelTaskPredicateSql()}
|
|
250
251
|
${e?sql`AND child.workspace_id = ${e}`:sql``}
|
|
251
252
|
ORDER BY child.started_at ASC
|
|
252
|
-
`);for(let
|
|
253
|
+
`);for(let _ of c)r.has(_.id)||(r.add(_.id),i.push(_));}catch(c){throw new p("DB_ERROR","Failed to find top-level tasks",c)}finally{l.close();}}if(s!==void 0&&n.length>1){let a=new Map;for(let l of i){let c=l.agent_id??"";a.has(c)||a.set(c,[]),a.get(c).push(l);}i=[];for(let l of a.values())l.sort((c,_)=>(_.started_at??"").localeCompare(c.started_at??"")),i.push(...l.slice(0,s));i.sort((l,c)=>(l.started_at??"").localeCompare(c.started_at??""));}return i}findAllTasks(t,e){let s=this.resolveDbPaths(),n=new Set,r=[];for(let i of s){if(!existsSync(i))continue;let a=T(i);try{let l=eq(o.thread_id,t),c=e?and(l,eq(o.workspace_id,e)):l,_=a.db.select().from(o).where(c).orderBy(asc(o.started_at)).all();for(let k of _)n.has(k.id)||(n.add(k.id),r.push(k));}catch(l){throw new p("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(!existsSync(r))continue;let i=T(r);try{let a=and(eq(o.id,e),eq(o.thread_id,t)),l=s?and(a,eq(o.workspace_id,s)):a,c=i.db.select().from(o).where(l).limit(1).get();if(!c)continue;let _=i.db.select().from(o).where(eq(o.parent_task_id,c.id)).orderBy(asc(o.started_at)).all();return {task:c,children:_}}catch(a){throw new p("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(!existsSync(r))continue;let i=T(r);try{let a=i.db.all(sql`
|
|
253
254
|
SELECT child.* FROM tasks child
|
|
254
255
|
WHERE child.thread_id IN (${sql.join(t.map(l=>sql`${l}`),sql`, `)})
|
|
255
256
|
AND ${this.topLevelTaskPredicateSql()}
|
|
256
257
|
${e?sql`AND child.workspace_id = ${e}`:sql``}
|
|
257
258
|
ORDER BY child.started_at ASC
|
|
258
|
-
`);for(let l of a){let c=l.thread_id;s.has(c)||s.set(c,[]),s.get(c).push(l);}}catch(a){throw new
|
|
259
|
+
`);for(let l of a){let c=l.thread_id;s.has(c)||s.set(c,[]),s.get(c).push(l);}}catch(a){throw new p("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=eq(u.id,t),i=s?and(r,eq(u.workspace_id,s)):r;if(!n.db.select({id:u.id}).from(u).where(i).limit(1).get())return;n.db.update(u).set({title:e,title_locked:1,updated_at:new Date().toISOString()}).where(eq(u.id,t)).run();}catch(r){throw r instanceof p?r:new p("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:u.id,message_count:u.message_count}).from(u).where(eq(u.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(u).set(a).where(eq(u.id,t)).run();}else s.db.insert(u).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 p?n:new p("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:u.id,platform:u.platform,workspace_id:u.workspace_id}).from(u).where(eq(u.id,t)).limit(1).get();if(i){r&&!i.workspace_id&&n.db.update(u).set({workspace_id:r}).where(eq(u.id,t)).run();return}let a=new Date().toISOString();n.db.insert(u).values({id:t,platform:e,workspace_id:r,message_count:0,created_at:a,updated_at:a}).run();}catch(r){throw r instanceof p?r:new p("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 c=a.select({message_count:u.message_count}).from(u).where(eq(u.id,t)).limit(1).get()?.message_count===0;return a.run(sql`
|
|
259
260
|
UPDATE threads
|
|
260
261
|
SET first_message = COALESCE(first_message, ${e}),
|
|
261
262
|
title = CASE WHEN title_locked = 0 AND title IS NULL THEN substr(${e}, 1, 60) ELSE title END,
|
|
@@ -263,8 +264,8 @@ ${s.join(`
|
|
|
263
264
|
message_count = message_count + 1,
|
|
264
265
|
updated_at = ${r}
|
|
265
266
|
WHERE id = ${t}
|
|
266
|
-
`),c},{behavior:"immediate"})}}catch(r){throw r instanceof
|
|
267
|
+
`),c},{behavior:"immediate"})}}catch(r){throw r instanceof p?r:new p("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(u).set({last_message:e,updated_at:r}).where(eq(u.id,t)).run();}catch(r){throw r instanceof p?r:new p("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(u).set(n).where(eq(u.id,t)).run();}catch(n){throw n instanceof p?n:new p("DB_ERROR","Failed to update thread",n)}finally{s.close();}}togglePin(t,e){let s=this.openHandle(true);try{let n=e?and(eq(u.id,t),eq(u.workspace_id,e)):eq(u.id,t),r=s.db.select({pinned:u.pinned,metadata:u.metadata}).from(u).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?and(eq(u.pinned,1),eq(u.workspace_id,e)):eq(u.pinned,1),c=s.db.select({metadata:u.metadata}).from(u).where(l).all(),_=0;for(let k of c){let w=k.metadata?JSON.parse(k.metadata):{};typeof w.pinOrder=="number"&&w.pinOrder>_&&(_=w.pinOrder);}a.pinOrder=_+1;}else delete a.pinOrder;return s.db.update(u).set({pinned:i,metadata:Object.keys(a).length>0?JSON.stringify(a):null}).where(n).run(),{pinned:!!i}}catch(n){throw n instanceof p?n:new p("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(u.id,t[n]),eq(u.workspace_id,e)):eq(u.id,t[n]),i=s.db.select({metadata:u.metadata}).from(u).where(r).get();if(!i)continue;let a=i.metadata?JSON.parse(i.metadata):{};a.pinOrder=n+1,s.db.update(u).set({metadata:JSON.stringify(a)}).where(r).run();}}catch(n){throw n instanceof p?n:new p("DB_ERROR","Failed to reorder pins",n)}finally{s.close();}}toggleStar(t,e){let s=this.openHandle(true);try{let n=e?and(eq(u.id,t),eq(u.workspace_id,e)):eq(u.id,t),r=s.db.select({starred:u.starred}).from(u).where(n).get();if(!r)return null;let i=r.starred?0:1;return s.db.update(u).set({starred:i}).where(n).run(),{starred:!!i}}catch(n){throw n instanceof p?n:new p("DB_ERROR","Failed to toggle star",n)}finally{s.close();}}};function Ee(d){return d.replace(/<conversation_history[^>]*>[\s\S]*?<\/conversation_history>/g,"").split(`
|
|
267
268
|
`).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(`
|
|
268
269
|
`).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(`
|
|
269
|
-
`):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),i=n.findTopLevelTasks(t,void 0,s),a=new Set(["queued","cancelled"]);i=i.filter(
|
|
270
|
-
`)[0].slice(0,200)),e.push({id:`${s.id}-assistant`,text:n,isAssistant:true,timestamp:new Date(s.started_at).getTime(),metadata:i});}}return e}};var
|
|
270
|
+
`):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),i=n.findTopLevelTasks(t,void 0,s),a=new Set(["queued","cancelled"]);i=i.filter(_=>(!_.status||!a.has(_.status))&&(!e?.currentTraceId||_.trace_id!==e.currentTraceId));let l=r?.platform??"cli",c=this.rowsToMessages(i);return {threadId:t,platform:l,messages:c,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(){}normalizeStatus(t){if(t&&!["success","completed","done"].includes(t)){if(["failed","error"].includes(t))return "failed";if(t==="running")return "running"}}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),r=this.normalizeStatus(s.status);if(n||r==="running"||r==="failed"){let i={agent_id:s.agent_id,task_id:s.id};r&&(i.status=r),r==="failed"&&s.error&&(i.reason=s.error.split(`
|
|
271
|
+
`)[0].slice(0,200)),e.push({id:`${s.id}-assistant`,text:n,isAssistant:true,timestamp:new Date(s.started_at).getTime(),metadata:i});}}return e}};var pt=class extends L{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};
|