@iamharshil/cortex 5.0.0-beta.1 → 5.0.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/bin/tui.js +19 -0
- package/dist/bin/index.js +37 -726
- package/dist/index.js +25 -25
- package/dist/tui/index.js +431 -0
- package/package.json +10 -3
package/dist/bin/index.js
CHANGED
|
@@ -1,139 +1,20 @@
|
|
|
1
1
|
#!/usr/bin/env node
|
|
2
2
|
#!/usr/bin/env node
|
|
3
3
|
|
|
4
|
-
"use strict";
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
8
|
-
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
9
|
-
var __getProtoOf = Object.getPrototypeOf;
|
|
10
|
-
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
11
|
-
var __esm = (fn, res) => function __init() {
|
|
12
|
-
return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
|
|
13
|
-
};
|
|
14
|
-
var __copyProps = (to, from, except, desc) => {
|
|
15
|
-
if (from && typeof from === "object" || typeof from === "function") {
|
|
16
|
-
for (let key of __getOwnPropNames(from))
|
|
17
|
-
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
18
|
-
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
19
|
-
}
|
|
20
|
-
return to;
|
|
21
|
-
};
|
|
22
|
-
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
|
|
23
|
-
// If the importer is in node compatibility mode or this is not an ESM
|
|
24
|
-
// file that has been converted to a CommonJS file using a Babel-
|
|
25
|
-
// compatible transform (i.e. "__esModule" has not been set), then set
|
|
26
|
-
// "default" to the CommonJS "module.exports" for node compatibility.
|
|
27
|
-
isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
|
|
28
|
-
mod
|
|
29
|
-
));
|
|
4
|
+
"use strict";var G=Object.create;var E=Object.defineProperty;var L=Object.getOwnPropertyDescriptor;var B=Object.getOwnPropertyNames;var H=Object.getPrototypeOf,K=Object.prototype.hasOwnProperty;var X=(e,t)=>()=>(e&&(t=e(e=0)),t);var Y=(e,t,s,i)=>{if(t&&typeof t=="object"||typeof t=="function")for(let o of B(t))!K.call(e,o)&&o!==s&&E(e,o,{get:()=>t[o],enumerable:!(i=L(t,o))||i.enumerable});return e};var x=(e,t,s)=>(s=e!=null?G(H(e)):{},Y(t||!e||!e.__esModule?E(s,"default",{value:e,enumerable:!0}):s,e));var he={};function k(e){switch(e.type){case"ollama":return new V(e);case"lmstudio":return new Q(e);case"openrouter":return new Z(e);default:throw new Error(`Unknown provider type: ${e.type}`)}}async function ie(e,t,s){let i=`call_${Date.now()}_${Math.random().toString(36).substr(2,9)}`;try{switch(e){case"Read":{let o=t.file_path,n=t.offset||1,r=t.limit||1e3,l=(await(0,y.readFile)(o,"utf-8")).split(`
|
|
5
|
+
`),d=l.slice(n-1,n-1+r);return{tool_call_id:i,output:d.join(`
|
|
6
|
+
`)+(l.length>n-1+r?`
|
|
30
7
|
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
switch (i.type) {
|
|
35
|
-
case "ollama":
|
|
36
|
-
return new S(i);
|
|
37
|
-
case "lmstudio":
|
|
38
|
-
return new k(i);
|
|
39
|
-
case "openrouter":
|
|
40
|
-
return new $(i);
|
|
41
|
-
default:
|
|
42
|
-
throw new Error(`Unknown provider type: ${i.type}`);
|
|
43
|
-
}
|
|
44
|
-
}
|
|
45
|
-
async function I(i, e, t) {
|
|
46
|
-
let o = `call_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
|
|
47
|
-
try {
|
|
48
|
-
switch (i) {
|
|
49
|
-
case "Read": {
|
|
50
|
-
let s = e.file_path, r = e.offset || 1, n = e.limit || 1e3, c = (await (0, import_promises4.readFile)(s, "utf-8")).split(`
|
|
51
|
-
`), l = c.slice(r - 1, r - 1 + n);
|
|
52
|
-
return { tool_call_id: o, output: l.join(`
|
|
53
|
-
`) + (c.length > r - 1 + n ? `
|
|
8
|
+
... ${l.length-(n-1+r)} more lines`:"")}}case"Write":{let o=t.file_path,n=t.content,r=(0,M.dirname)(o);return await re(r,{recursive:!0}),await(0,y.writeFile)(o,n,"utf-8"),{tool_call_id:i,output:`File written: ${o}`}}case"Edit":{let o=t.file_path,n=t.old_string,r=t.new_string,l=await(0,y.readFile)(o,"utf-8");if(!l.includes(n))return{tool_call_id:i,output:"Error: Could not find the specified text to replace. The old_string must match exactly.",is_error:!0};let d=l.replace(n,r);return await(0,y.writeFile)(o,d,"utf-8"),{tool_call_id:i,output:`File edited: ${o}`}}case"Bash":{let o=t.command,n=t.description||"",{stdout:r,stderr:l,exitCode:d}=await(0,q.execa)(o,{shell:!0,cwd:s,timeout:12e4}),c=l?`${r}
|
|
9
|
+
${l}`:r;if(d!==0&&!r)return{tool_call_id:i,output:`Command failed with exit code ${d}: ${o}
|
|
10
|
+
${l}`,is_error:!0};let w=c.length>15e3?c.slice(0,15e3)+`
|
|
54
11
|
|
|
55
|
-
... ${c.length
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
case "Edit": {
|
|
62
|
-
let s = e.file_path, r = e.old_string, n = e.new_string, a = await (0, import_promises4.readFile)(s, "utf-8");
|
|
63
|
-
if (!a.includes(r)) return { tool_call_id: o, output: "Error: Could not find the specified text to replace. The old_string must match exactly.", is_error: true };
|
|
64
|
-
let c = a.replace(r, n);
|
|
65
|
-
return await (0, import_promises4.writeFile)(s, c, "utf-8"), { tool_call_id: o, output: `File edited: ${s}` };
|
|
66
|
-
}
|
|
67
|
-
case "Bash": {
|
|
68
|
-
let s = e.command, r = e.description || "", { stdout: n, stderr: a, exitCode: c } = await (0, import_execa.execa)(s, { shell: true, cwd: t, timeout: 12e4 }), l = a ? `${n}
|
|
69
|
-
${a}` : n;
|
|
70
|
-
if (c !== 0 && !n) return { tool_call_id: o, output: `Command failed with exit code ${c}: ${s}
|
|
71
|
-
${a}`, is_error: true };
|
|
72
|
-
let u = l.length > 15e3 ? l.slice(0, 15e3) + `
|
|
73
|
-
|
|
74
|
-
... output truncated (${l.length} chars)` : l;
|
|
75
|
-
return { tool_call_id: o, output: u || `Command executed successfully (exit code: ${c})` };
|
|
76
|
-
}
|
|
77
|
-
case "Glob": {
|
|
78
|
-
let s = e.pattern, r = e.path || ".", n = (0, import_path4.resolve)(t, r), a = await (0, import_fast_glob.default)(s, { cwd: n, absolute: true, onlyFiles: true });
|
|
79
|
-
return { tool_call_id: o, output: a.length > 0 ? a.join(`
|
|
80
|
-
`) : "No files found" };
|
|
81
|
-
}
|
|
82
|
-
case "Grep": {
|
|
83
|
-
let s = e.pattern, r = e.path || ".", n = e.include, a = (0, import_path4.resolve)(t, r), c = await (0, import_fast_glob.default)(n || "**/*", { cwd: a, onlyFiles: true }), l = [];
|
|
84
|
-
for (let w of c.slice(0, 100)) try {
|
|
85
|
-
let _ = (await (0, import_promises4.readFile)(w, "utf-8")).split(`
|
|
86
|
-
`);
|
|
87
|
-
for (let f = 0; f < _.length; f++) new RegExp(s, "i").test(_[f]) && l.push(`${w}:${f + 1}: ${_[f]}`);
|
|
88
|
-
} catch {
|
|
89
|
-
}
|
|
90
|
-
let u = l.length > 100 ? l.slice(0, 100).concat([`... ${l.length - 100} more matches`]) : l;
|
|
91
|
-
return { tool_call_id: o, output: u.length > 0 ? u.join(`
|
|
92
|
-
`) : "No matches found" };
|
|
93
|
-
}
|
|
94
|
-
case "TodoWrite": {
|
|
95
|
-
let s = e.todos;
|
|
96
|
-
return { tool_call_id: o, output: `Todo list updated with ${s.length} items:
|
|
97
|
-
${s.map((r, n) => `${n + 1}. [${r.status}] ${r.content}`).join(`
|
|
98
|
-
`)}` };
|
|
99
|
-
}
|
|
100
|
-
default:
|
|
101
|
-
return { tool_call_id: o, output: `Unknown tool: ${i}`, is_error: true };
|
|
102
|
-
}
|
|
103
|
-
} catch (s) {
|
|
104
|
-
return { tool_call_id: o, output: `Error executing ${i}: ${s instanceof Error ? s.message : String(s)}`, is_error: true };
|
|
105
|
-
}
|
|
106
|
-
}
|
|
107
|
-
async function V(i, e) {
|
|
108
|
-
let { mkdirSync: t } = await import("fs");
|
|
109
|
-
t(i, e);
|
|
110
|
-
}
|
|
111
|
-
async function E() {
|
|
112
|
-
let i = [(0, import_path5.join)(process.cwd(), ".cortex", "config.json"), (0, import_path5.join)((0, import_os4.homedir)(), ".cortex", "config.json")];
|
|
113
|
-
for (let e of i) try {
|
|
114
|
-
let t = await (0, import_promises5.readFile)(e, "utf-8");
|
|
115
|
-
return JSON.parse(t);
|
|
116
|
-
} catch {
|
|
117
|
-
}
|
|
118
|
-
return {};
|
|
119
|
-
}
|
|
120
|
-
async function re() {
|
|
121
|
-
console.log(import_chalk.default.blue("Initializing Cortex project..."));
|
|
122
|
-
let i = (0, import_path5.join)(process.cwd(), ".cortex"), e = (0, import_path5.join)(i, "config.json"), t = (0, import_path5.join)(i, "mcp.json"), o = (0, import_path5.join)(process.cwd(), "CORTEX.md");
|
|
123
|
-
try {
|
|
124
|
-
await (0, import_promises5.mkdir)(i, { recursive: true });
|
|
125
|
-
} catch {
|
|
126
|
-
}
|
|
127
|
-
try {
|
|
128
|
-
await (0, import_promises5.access)(e), console.log(import_chalk.default.yellow("Cortex already initialized in this project"));
|
|
129
|
-
return;
|
|
130
|
-
} catch {
|
|
131
|
-
}
|
|
132
|
-
await (0, import_promises5.writeFile)(e, JSON.stringify({ provider: "ollama", model: "llama3.2" }, null, 2)), console.log(import_chalk.default.green("\u2705 Created .cortex/config.json"));
|
|
133
|
-
try {
|
|
134
|
-
await (0, import_promises5.access)(o);
|
|
135
|
-
} catch {
|
|
136
|
-
await (0, import_promises5.writeFile)(o, `# Cortex Project Memory
|
|
12
|
+
... output truncated (${c.length} chars)`:c;return{tool_call_id:i,output:w||`Command executed successfully (exit code: ${d})`}}case"Glob":{let o=t.pattern,n=t.path||".",r=(0,M.resolve)(s,n),l=await(0,S.default)(o,{cwd:r,absolute:!0,onlyFiles:!0});return{tool_call_id:i,output:l.length>0?l.join(`
|
|
13
|
+
`):"No files found"}}case"Grep":{let o=t.pattern,n=t.path||".",r=t.include,l=(0,M.resolve)(s,n),d=await(0,S.default)(r||"**/*",{cwd:l,onlyFiles:!0}),c=[];for(let P of d.slice(0,100))try{let b=(await(0,y.readFile)(P,"utf-8")).split(`
|
|
14
|
+
`);for(let v=0;v<b.length;v++)new RegExp(o,"i").test(b[v])&&c.push(`${P}:${v+1}: ${b[v]}`)}catch{}let w=c.length>100?c.slice(0,100).concat([`... ${c.length-100} more matches`]):c;return{tool_call_id:i,output:w.length>0?w.join(`
|
|
15
|
+
`):"No matches found"}}case"TodoWrite":{let o=t.todos;return{tool_call_id:i,output:`Todo list updated with ${o.length} items:
|
|
16
|
+
${o.map((n,r)=>`${r+1}. [${n.status}] ${n.content}`).join(`
|
|
17
|
+
`)}`}}default:return{tool_call_id:i,output:`Unknown tool: ${e}`,is_error:!0}}}catch(o){return{tool_call_id:i,output:`Error executing ${e}: ${o instanceof Error?o.message:String(o)}`,is_error:!0}}}async function re(e,t){let{mkdirSync:s}=await import("fs");s(e,t)}async function $(){let e=[(0,g.join)(process.cwd(),".cortex","config.json"),(0,g.join)((0,J.homedir)(),".cortex","config.json")];for(let t of e)try{let s=await(0,u.readFile)(t,"utf-8");return JSON.parse(s)}catch{}return{}}async function le(){console.log(a.default.blue("Initializing Cortex project..."));let e=(0,g.join)(process.cwd(),".cortex"),t=(0,g.join)(e,"config.json"),s=(0,g.join)(e,"mcp.json"),i=(0,g.join)(process.cwd(),"CORTEX.md");try{await(0,u.mkdir)(e,{recursive:!0})}catch{}try{await(0,u.access)(t),console.log(a.default.yellow("Cortex already initialized in this project"));return}catch{}await(0,u.writeFile)(t,JSON.stringify({provider:"ollama",model:"llama3.2"},null,2)),console.log(a.default.green("\u2705 Created .cortex/config.json"));try{await(0,u.access)(i)}catch{await(0,u.writeFile)(i,`# Cortex Project Memory
|
|
137
18
|
|
|
138
19
|
## Project Overview
|
|
139
20
|
<!-- Fill in your project description -->
|
|
@@ -146,223 +27,13 @@ async function re() {
|
|
|
146
27
|
|
|
147
28
|
## Architecture Notes
|
|
148
29
|
<!-- Document important architectural decisions -->
|
|
149
|
-
`),
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
\
|
|
153
|
-
}
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
try {
|
|
157
|
-
await c.initialize(), await c.run(i);
|
|
158
|
-
let l = c.getContextInfo();
|
|
159
|
-
console.log(import_chalk.default.dim(`
|
|
160
|
-
\u{1F4CA} Session: ${l.messages} messages, ${l.tokens} tokens`));
|
|
161
|
-
} catch (l) {
|
|
162
|
-
console.error(import_chalk.default.red(`
|
|
163
|
-
\u274C Error: ${l instanceof Error ? l.message : String(l)}`)), process.exit(1);
|
|
164
|
-
} finally {
|
|
165
|
-
await c.cleanup();
|
|
166
|
-
}
|
|
167
|
-
}
|
|
168
|
-
async function ae() {
|
|
169
|
-
console.log(import_chalk.default.blue(`\u{1F50D} Checking provider status...
|
|
170
|
-
`));
|
|
171
|
-
let i = await E(), e = i.provider || "ollama";
|
|
172
|
-
try {
|
|
173
|
-
let t = y({ type: e, baseUrl: i.url });
|
|
174
|
-
console.log(import_chalk.default.green(`\u2705 Provider: ${t.name}`));
|
|
175
|
-
let o = await t.listModels();
|
|
176
|
-
if (o.length > 0) {
|
|
177
|
-
console.log(import_chalk.default.cyan(`
|
|
178
|
-
\u{1F4E6} Available models:`));
|
|
179
|
-
for (let s of o.slice(0, 10)) console.log(` - ${s}`);
|
|
180
|
-
o.length > 10 && console.log(import_chalk.default.dim(` ... and ${o.length - 10} more`));
|
|
181
|
-
} else console.log(import_chalk.default.yellow("\u26A0\uFE0F No models found. Make sure your provider is running."));
|
|
182
|
-
} catch (t) {
|
|
183
|
-
console.error(import_chalk.default.red(`\u274C Provider error: ${t instanceof Error ? t.message : String(t)}`));
|
|
184
|
-
}
|
|
185
|
-
}
|
|
186
|
-
async function ce(i) {
|
|
187
|
-
console.log(import_chalk.default.blue(`\u2699\uFE0F Setting up Cortex...
|
|
188
|
-
`));
|
|
189
|
-
let e = (0, import_path5.join)(process.cwd(), ".cortex", "config.json"), t = {};
|
|
190
|
-
try {
|
|
191
|
-
let s = await (0, import_promises5.readFile)(e, "utf-8");
|
|
192
|
-
t = JSON.parse(s);
|
|
193
|
-
} catch {
|
|
194
|
-
}
|
|
195
|
-
let o = { provider: i.provider || t.provider || "ollama", model: i.model || t.model || "llama3.2", url: t.url || "http://localhost:11434" };
|
|
196
|
-
await (0, import_promises5.writeFile)(e, JSON.stringify(o, null, 2)), console.log(import_chalk.default.green("\u2705 Configuration saved!")), console.log(import_chalk.default.dim(` Provider: ${o.provider}`)), console.log(import_chalk.default.dim(` Model: ${o.model}`));
|
|
197
|
-
}
|
|
198
|
-
async function le(i) {
|
|
199
|
-
let e = await E(), t = i.provider || e.provider || "ollama";
|
|
200
|
-
try {
|
|
201
|
-
let s = await y({ type: t, baseUrl: e.url }).listModels();
|
|
202
|
-
if (s.length === 0) {
|
|
203
|
-
console.log(import_chalk.default.yellow("No models available"));
|
|
204
|
-
return;
|
|
205
|
-
}
|
|
206
|
-
console.log(import_chalk.default.cyan(`Available models on ${t}:`));
|
|
207
|
-
for (let r of s) console.log(` ${r}`);
|
|
208
|
-
} catch (o) {
|
|
209
|
-
console.error(import_chalk.default.red(`Error: ${o instanceof Error ? o.message : String(o)}`));
|
|
210
|
-
}
|
|
211
|
-
}
|
|
212
|
-
function v(i = 0) {
|
|
213
|
-
console.log(), console.log(import_chalk.default.dim(i === 0 ? "\u{1F44B} Goodbye!" : "\u274C Cancelled.")), process.exit(i);
|
|
214
|
-
}
|
|
215
|
-
var import_commander, import_chalk, import_chalk2, import_promises, import_path, import_os, import_path2, import_os2, import_promises2, import_promises3, import_path3, import_os3, import_execa, import_promises4, import_path4, import_fast_glob, import_execa2, import_events, import_promises5, import_path5, import_os4, S, k, $, G, x, W, C, U, P, T, p, me;
|
|
216
|
-
var init_dist = __esm({
|
|
217
|
-
"dist/index.js"() {
|
|
218
|
-
"use strict";
|
|
219
|
-
import_commander = require("commander");
|
|
220
|
-
import_chalk = __toESM(require("chalk"), 1);
|
|
221
|
-
import_chalk2 = __toESM(require("chalk"), 1);
|
|
222
|
-
import_promises = require("fs/promises");
|
|
223
|
-
import_path = require("path");
|
|
224
|
-
import_os = require("os");
|
|
225
|
-
import_path2 = require("path");
|
|
226
|
-
import_os2 = require("os");
|
|
227
|
-
import_promises2 = require("fs/promises");
|
|
228
|
-
import_promises3 = require("fs/promises");
|
|
229
|
-
import_path3 = require("path");
|
|
230
|
-
import_os3 = require("os");
|
|
231
|
-
import_execa = require("execa");
|
|
232
|
-
import_promises4 = require("fs/promises");
|
|
233
|
-
import_path4 = require("path");
|
|
234
|
-
import_fast_glob = __toESM(require("fast-glob"), 1);
|
|
235
|
-
import_execa2 = require("execa");
|
|
236
|
-
import_events = require("events");
|
|
237
|
-
import_promises5 = require("fs/promises");
|
|
238
|
-
import_path5 = require("path");
|
|
239
|
-
import_os4 = require("os");
|
|
240
|
-
S = class {
|
|
241
|
-
name = "ollama";
|
|
242
|
-
baseUrl;
|
|
243
|
-
defaultModel;
|
|
244
|
-
timeout;
|
|
245
|
-
constructor(e) {
|
|
246
|
-
this.baseUrl = e.baseUrl || "http://localhost:11434", this.defaultModel = e.defaultModel || "llama3.2", this.timeout = e.timeout || 12e4;
|
|
247
|
-
}
|
|
248
|
-
async chat(e, t, o) {
|
|
249
|
-
let r = { model: o || this.defaultModel, messages: this.formatMessages(e), stream: false };
|
|
250
|
-
t && t.length > 0 && (r.tools = t.map((n) => ({ type: "function", function: { name: n.name, description: n.description, parameters: n.input_schema } })));
|
|
251
|
-
try {
|
|
252
|
-
let n = await fetch(`${this.baseUrl}/api/chat`, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify(r), signal: AbortSignal.timeout(this.timeout) });
|
|
253
|
-
if (!n.ok) {
|
|
254
|
-
let c = await n.text();
|
|
255
|
-
throw new Error(`Ollama API error: ${n.status} - ${c}`);
|
|
256
|
-
}
|
|
257
|
-
let a = await n.json();
|
|
258
|
-
return { content: a.message?.content || "", tool_calls: a.message?.tool_calls?.map((c) => ({ id: c.id, type: "function", function: { name: c.function.name, arguments: c.function.arguments } })), stop_reason: a.done ? "stop" : void 0 };
|
|
259
|
-
} catch (n) {
|
|
260
|
-
throw n instanceof Error && n.name === "TimeoutError" ? new Error(`Ollama request timed out after ${this.timeout}ms`) : n;
|
|
261
|
-
}
|
|
262
|
-
}
|
|
263
|
-
async listModels() {
|
|
264
|
-
try {
|
|
265
|
-
let e = await fetch(`${this.baseUrl}/api/tags`);
|
|
266
|
-
return e.ok ? (await e.json()).models?.map((o) => o.name) || [] : [];
|
|
267
|
-
} catch {
|
|
268
|
-
return [];
|
|
269
|
-
}
|
|
270
|
-
}
|
|
271
|
-
getTokenizer() {
|
|
272
|
-
return (e) => Math.ceil(e.length / 4);
|
|
273
|
-
}
|
|
274
|
-
formatMessages(e) {
|
|
275
|
-
return e.map((t) => ({ role: t.role === "tool" ? "user" : t.role, content: t.content }));
|
|
276
|
-
}
|
|
277
|
-
};
|
|
278
|
-
k = class {
|
|
279
|
-
name = "lmstudio";
|
|
280
|
-
baseUrl;
|
|
281
|
-
defaultModel;
|
|
282
|
-
timeout;
|
|
283
|
-
constructor(e) {
|
|
284
|
-
this.baseUrl = e.baseUrl || "http://localhost:1234/v1", this.defaultModel = e.defaultModel || "llama-3.1-8b", this.timeout = e.timeout || 12e4;
|
|
285
|
-
}
|
|
286
|
-
async chat(e, t, o) {
|
|
287
|
-
let r = { model: o || this.defaultModel, messages: this.formatMessages(e), stream: false };
|
|
288
|
-
t && t.length > 0 && (r.tools = t.map((n) => ({ type: "function", function: { name: n.name, description: n.description, parameters: n.input_schema } })));
|
|
289
|
-
try {
|
|
290
|
-
let n = await fetch(`${this.baseUrl}/chat/completions`, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify(r), signal: AbortSignal.timeout(this.timeout) });
|
|
291
|
-
if (!n.ok) {
|
|
292
|
-
let l = await n.text();
|
|
293
|
-
throw new Error(`LM Studio API error: ${n.status} - ${l}`);
|
|
294
|
-
}
|
|
295
|
-
let c = (await n.json()).choices?.[0];
|
|
296
|
-
return { content: c?.message?.content || "", tool_calls: c?.message?.tool_calls?.map((l) => ({ id: l.id, type: "function", function: { name: l.function.name, arguments: l.function.arguments } })) };
|
|
297
|
-
} catch (n) {
|
|
298
|
-
throw n instanceof Error && n.name === "TimeoutError" ? new Error(`LM Studio request timed out after ${this.timeout}ms`) : n;
|
|
299
|
-
}
|
|
300
|
-
}
|
|
301
|
-
async listModels() {
|
|
302
|
-
try {
|
|
303
|
-
let e = await fetch(`${this.baseUrl.replace("/v1", "")}/api/v0/models`);
|
|
304
|
-
return e.ok ? (await e.json()).data?.map((o) => o.id) || [] : [];
|
|
305
|
-
} catch {
|
|
306
|
-
return [];
|
|
307
|
-
}
|
|
308
|
-
}
|
|
309
|
-
getTokenizer() {
|
|
310
|
-
return (e) => Math.ceil(e.length / 4);
|
|
311
|
-
}
|
|
312
|
-
formatMessages(e) {
|
|
313
|
-
return e.map((t) => ({ role: t.role === "tool" ? "user" : t.role, content: t.content }));
|
|
314
|
-
}
|
|
315
|
-
};
|
|
316
|
-
$ = class {
|
|
317
|
-
name = "openrouter";
|
|
318
|
-
apiKey;
|
|
319
|
-
baseUrl;
|
|
320
|
-
defaultModel;
|
|
321
|
-
timeout;
|
|
322
|
-
constructor(e) {
|
|
323
|
-
this.apiKey = e.apiKey, this.baseUrl = e.baseUrl || "https://openrouter.ai/api/v1", this.defaultModel = e.defaultModel || "anthropic/claude-3.5-sonnet", this.timeout = e.timeout || 12e4;
|
|
324
|
-
}
|
|
325
|
-
async chat(e, t, o) {
|
|
326
|
-
let r = { model: o || this.defaultModel, messages: this.formatMessages(e) };
|
|
327
|
-
t && t.length > 0 && (r.tools = t.map((n) => ({ type: "function", function: { name: n.name, description: n.description, parameters: n.input_schema } })));
|
|
328
|
-
try {
|
|
329
|
-
let n = await fetch(`${this.baseUrl}/chat/completions`, { method: "POST", headers: { "Content-Type": "application/json", Authorization: `Bearer ${this.apiKey}`, "HTTP-Referer": "https://cortex.dev", "X-Title": "Cortex" }, body: JSON.stringify(r), signal: AbortSignal.timeout(this.timeout) });
|
|
330
|
-
if (!n.ok) {
|
|
331
|
-
let l = await n.text();
|
|
332
|
-
throw new Error(`OpenRouter API error: ${n.status} - ${l}`);
|
|
333
|
-
}
|
|
334
|
-
let c = (await n.json()).choices?.[0];
|
|
335
|
-
return { content: c?.message?.content || "", tool_calls: c?.message?.tool_calls?.map((l) => ({ id: l.id, type: "function", function: { name: l.function.name, arguments: l.function.arguments } })), stop_reason: c?.finish_reason };
|
|
336
|
-
} catch (n) {
|
|
337
|
-
throw n instanceof Error && n.name === "TimeoutError" ? new Error(`OpenRouter request timed out after ${this.timeout}ms`) : n;
|
|
338
|
-
}
|
|
339
|
-
}
|
|
340
|
-
async listModels() {
|
|
341
|
-
try {
|
|
342
|
-
let e = await fetch(`${this.baseUrl}/models`, { headers: { Authorization: `Bearer ${this.apiKey}` } });
|
|
343
|
-
return e.ok ? (await e.json()).data?.map((o) => o.id) || [] : [];
|
|
344
|
-
} catch {
|
|
345
|
-
return [];
|
|
346
|
-
}
|
|
347
|
-
}
|
|
348
|
-
getTokenizer() {
|
|
349
|
-
return (e) => Math.ceil(e.length / 4);
|
|
350
|
-
}
|
|
351
|
-
formatMessages(e) {
|
|
352
|
-
return e.map((t) => t.tool_calls ? { role: "assistant", content: t.content || "", tool_calls: t.tool_calls.map((o) => ({ id: o.id, type: o.type, function: { name: o.function.name, arguments: o.function.arguments } })) } : t.tool_call_id ? { role: "tool", content: t.content, tool_call_id: t.tool_call_id } : { role: t.role === "tool" ? "user" : t.role, content: t.content });
|
|
353
|
-
}
|
|
354
|
-
};
|
|
355
|
-
G = { maxTokens: 18e4, systemPromptTokens: 3e3, toolDefTokens: 5e3, warningThreshold: 0.7, autoCompactThreshold: 0.95 };
|
|
356
|
-
x = class {
|
|
357
|
-
messages = [];
|
|
358
|
-
config;
|
|
359
|
-
tokenCount = 0;
|
|
360
|
-
memoryFiles = [];
|
|
361
|
-
constructor(e = {}) {
|
|
362
|
-
this.config = { ...G, ...e };
|
|
363
|
-
}
|
|
364
|
-
getSystemPrompt() {
|
|
365
|
-
return `You are Cortex, an autonomous AI coding agent. You have access to tools to read, write, and edit files, execute shell commands, and search through the codebase.
|
|
30
|
+
`),console.log(a.default.green("\u2705 Created CORTEX.md"))}await(0,u.writeFile)(s,JSON.stringify({servers:{}},null,2)),console.log(a.default.green("\u2705 Created .cortex/mcp.json")),console.log(a.default.green(`
|
|
31
|
+
\u{1F389} Cortex initialized! Run "cortex" to start coding.`))}async function ce(e,t){let s=await $(),i=t.provider||s.provider||"ollama",o=t.model||s.model,n=t.url||s.url,r=t.planMode||s.planMode,l={provider:i,model:o,providerUrl:n,planMode:r},d=new ae(l,process.cwd());try{await d.initialize(),await d.run(e);let c=d.getContextInfo();console.log(a.default.dim(`
|
|
32
|
+
\u{1F4CA} Session: ${c.messages} messages, ${c.tokens} tokens`))}catch(c){console.error(a.default.red(`
|
|
33
|
+
\u274C Error: ${c instanceof Error?c.message:String(c)}`)),process.exit(1)}finally{await d.cleanup()}}async function de(){console.log(a.default.blue(`\u{1F50D} Checking provider status...
|
|
34
|
+
`));let e=await $(),t=e.provider||"ollama";try{let s=k({type:t,baseUrl:e.url});console.log(a.default.green(`\u2705 Provider: ${s.name}`));let i=await s.listModels();if(i.length>0){console.log(a.default.cyan(`
|
|
35
|
+
\u{1F4E6} Available models:`));for(let o of i.slice(0,10))console.log(` - ${o}`);i.length>10&&console.log(a.default.dim(` ... and ${i.length-10} more`))}else console.log(a.default.yellow("\u26A0\uFE0F No models found. Make sure your provider is running."))}catch(s){console.error(a.default.red(`\u274C Provider error: ${s instanceof Error?s.message:String(s)}`))}}async function me(e){console.log(a.default.blue(`\u2699\uFE0F Setting up Cortex...
|
|
36
|
+
`));let t=(0,g.join)(process.cwd(),".cortex","config.json"),s={};try{let o=await(0,u.readFile)(t,"utf-8");s=JSON.parse(o)}catch{}let i={provider:e.provider||s.provider||"ollama",model:e.model||s.model||"llama3.2",url:s.url||"http://localhost:11434"};await(0,u.writeFile)(t,JSON.stringify(i,null,2)),console.log(a.default.green("\u2705 Configuration saved!")),console.log(a.default.dim(` Provider: ${i.provider}`)),console.log(a.default.dim(` Model: ${i.model}`))}async function pe(e){let t=await $(),s=e.provider||t.provider||"ollama";try{let i=await k({type:s,baseUrl:t.url}).listModels();if(i.length===0){console.log(a.default.yellow("No models available"));return}console.log(a.default.cyan(`Available models on ${s}:`));for(let o of i)console.log(` ${o}`)}catch(i){console.error(a.default.red(`Error: ${i instanceof Error?i.message:String(i)}`))}}function _(e=0){console.log(),console.log(a.default.dim(e===0?"\u{1F44B} Goodbye!":"\u274C Cancelled.")),process.exit(e)}async function ue(){if(F.length===0||F.includes("--tui")){let e=await $(),t=e.provider||"ollama",s=e.model,i=(0,g.join)(__dirname,"..","bin","tui.js"),o=["--provider",t];s&&o.push("--model",s);let{spawn:n}=await import("child_process");n("node",[i,...o],{cwd:process.cwd(),stdio:"inherit",shell:!1}).on("close",r=>{_(r||0)})}else f.parse()}var D,a,m,O,T,R,j,N,C,p,h,A,q,y,M,S,I,z,u,g,J,V,Q,Z,ee,te,oe,se,U,ne,ae,f,F,W=X(()=>{"use strict";D=require("commander"),a=x(require("chalk"),1),m=x(require("chalk"),1),O=require("fs/promises"),T=require("path"),R=require("os"),j=require("path"),N=require("os"),C=require("fs/promises"),p=require("fs/promises"),h=require("path"),A=require("os"),q=require("execa"),y=require("fs/promises"),M=require("path"),S=x(require("fast-glob"),1),I=require("execa"),z=require("events"),u=require("fs/promises"),g=require("path"),J=require("os"),V=class{name="ollama";baseUrl;defaultModel;timeout;constructor(e){this.baseUrl=e.baseUrl||"http://localhost:11434",this.defaultModel=e.defaultModel||"llama3.2",this.timeout=e.timeout||12e4}async chat(e,t,s){let i={model:s||this.defaultModel,messages:this.formatMessages(e),stream:!1};t&&t.length>0&&(i.tools=t.map(o=>({type:"function",function:{name:o.name,description:o.description,parameters:o.input_schema}})));try{let o=await fetch(`${this.baseUrl}/api/chat`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(i),signal:AbortSignal.timeout(this.timeout)});if(!o.ok){let r=await o.text();throw new Error(`Ollama API error: ${o.status} - ${r}`)}let n=await o.json();return{content:n.message?.content||"",tool_calls:n.message?.tool_calls?.map(r=>({id:r.id,type:"function",function:{name:r.function.name,arguments:r.function.arguments}})),stop_reason:n.done?"stop":void 0}}catch(o){throw o instanceof Error&&o.name==="TimeoutError"?new Error(`Ollama request timed out after ${this.timeout}ms`):o}}async listModels(){try{let e=await fetch(`${this.baseUrl}/api/tags`);return e.ok?(await e.json()).models?.map(t=>t.name)||[]:[]}catch{return[]}}getTokenizer(){return e=>Math.ceil(e.length/4)}formatMessages(e){return e.map(t=>({role:t.role==="tool"?"user":t.role,content:t.content}))}},Q=class{name="lmstudio";baseUrl;defaultModel;timeout;constructor(e){this.baseUrl=e.baseUrl||"http://localhost:1234/v1",this.defaultModel=e.defaultModel||"llama-3.1-8b",this.timeout=e.timeout||12e4}async chat(e,t,s){let i={model:s||this.defaultModel,messages:this.formatMessages(e),stream:!1};t&&t.length>0&&(i.tools=t.map(o=>({type:"function",function:{name:o.name,description:o.description,parameters:o.input_schema}})));try{let o=await fetch(`${this.baseUrl}/chat/completions`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(i),signal:AbortSignal.timeout(this.timeout)});if(!o.ok){let r=await o.text();throw new Error(`LM Studio API error: ${o.status} - ${r}`)}let n=(await o.json()).choices?.[0];return{content:n?.message?.content||"",tool_calls:n?.message?.tool_calls?.map(r=>({id:r.id,type:"function",function:{name:r.function.name,arguments:r.function.arguments}}))}}catch(o){throw o instanceof Error&&o.name==="TimeoutError"?new Error(`LM Studio request timed out after ${this.timeout}ms`):o}}async listModels(){try{let e=await fetch(`${this.baseUrl.replace("/v1","")}/api/v0/models`);return e.ok?(await e.json()).data?.map(t=>t.id)||[]:[]}catch{return[]}}getTokenizer(){return e=>Math.ceil(e.length/4)}formatMessages(e){return e.map(t=>({role:t.role==="tool"?"user":t.role,content:t.content}))}},Z=class{name="openrouter";apiKey;baseUrl;defaultModel;timeout;constructor(e){this.apiKey=e.apiKey,this.baseUrl=e.baseUrl||"https://openrouter.ai/api/v1",this.defaultModel=e.defaultModel||"anthropic/claude-3.5-sonnet",this.timeout=e.timeout||12e4}async chat(e,t,s){let i={model:s||this.defaultModel,messages:this.formatMessages(e)};t&&t.length>0&&(i.tools=t.map(o=>({type:"function",function:{name:o.name,description:o.description,parameters:o.input_schema}})));try{let o=await fetch(`${this.baseUrl}/chat/completions`,{method:"POST",headers:{"Content-Type":"application/json",Authorization:`Bearer ${this.apiKey}`,"HTTP-Referer":"https://cortex.dev","X-Title":"Cortex"},body:JSON.stringify(i),signal:AbortSignal.timeout(this.timeout)});if(!o.ok){let r=await o.text();throw new Error(`OpenRouter API error: ${o.status} - ${r}`)}let n=(await o.json()).choices?.[0];return{content:n?.message?.content||"",tool_calls:n?.message?.tool_calls?.map(r=>({id:r.id,type:"function",function:{name:r.function.name,arguments:r.function.arguments}})),stop_reason:n?.finish_reason}}catch(o){throw o instanceof Error&&o.name==="TimeoutError"?new Error(`OpenRouter request timed out after ${this.timeout}ms`):o}}async listModels(){try{let e=await fetch(`${this.baseUrl}/models`,{headers:{Authorization:`Bearer ${this.apiKey}`}});return e.ok?(await e.json()).data?.map(t=>t.id)||[]:[]}catch{return[]}}getTokenizer(){return e=>Math.ceil(e.length/4)}formatMessages(e){return e.map(t=>t.tool_calls?{role:"assistant",content:t.content||"",tool_calls:t.tool_calls.map(s=>({id:s.id,type:s.type,function:{name:s.function.name,arguments:s.function.arguments}}))}:t.tool_call_id?{role:"tool",content:t.content,tool_call_id:t.tool_call_id}:{role:t.role==="tool"?"user":t.role,content:t.content})}};ee={maxTokens:18e4,systemPromptTokens:3e3,toolDefTokens:5e3,warningThreshold:.7,autoCompactThreshold:.95},te=class{messages=[];config;tokenCount=0;memoryFiles=[];constructor(e={}){this.config={...ee,...e}}getSystemPrompt(){return`You are Cortex, an autonomous AI coding agent. You have access to tools to read, write, and edit files, execute shell commands, and search through the codebase.
|
|
366
37
|
|
|
367
38
|
## Your Capabilities
|
|
368
39
|
- Read, write, and edit files in the codebase
|
|
@@ -383,164 +54,21 @@ You are working in the current directory. Use absolute paths for file operations
|
|
|
383
54
|
|
|
384
55
|
## Tools
|
|
385
56
|
You have access to the following tools:
|
|
386
|
-
${
|
|
57
|
+
${oe.map(e=>`- ${e.name}: ${e.description}`).join(`
|
|
387
58
|
`)}
|
|
388
|
-
|
|
389
|
-
}
|
|
390
|
-
addMessage(e) {
|
|
391
|
-
this.messages.push(e), this.updateTokenCount();
|
|
392
|
-
}
|
|
393
|
-
getMessages() {
|
|
394
|
-
return [...this.messages];
|
|
395
|
-
}
|
|
396
|
-
getMessagesForAPI() {
|
|
397
|
-
return [{ role: "system", content: this.getSystemPrompt() }, ...this.messages];
|
|
398
|
-
}
|
|
399
|
-
async compact(e) {
|
|
400
|
-
let t = this.summarizeConversation(e), o = this.messages.filter((r) => r.role === "user").pop(), s = this.messages.filter((r) => r.role === "assistant").pop();
|
|
401
|
-
this.messages = [], o && this.messages.push({ role: "user", content: `[Previous conversation summary]: ${t}
|
|
59
|
+
`}addMessage(e){this.messages.push(e),this.updateTokenCount()}getMessages(){return[...this.messages]}getMessagesForAPI(){return[{role:"system",content:this.getSystemPrompt()},...this.messages]}async compact(e){let t=this.summarizeConversation(e),s=this.messages.filter(o=>o.role==="user").pop(),i=this.messages.filter(o=>o.role==="assistant").pop();this.messages=[],s&&this.messages.push({role:"user",content:`[Previous conversation summary]: ${t}
|
|
402
60
|
|
|
403
61
|
[Continuing from previous session]
|
|
404
62
|
|
|
405
|
-
${
|
|
406
|
-
|
|
407
|
-
|
|
408
|
-
|
|
409
|
-
|
|
410
|
-
|
|
411
|
-
|
|
412
|
-
let e = this.config.systemPromptTokens, t = this.config.toolDefTokens, o = this.estimateTokens(this.messagesToString()), s = this.memoryFiles.length * 500, r = e + t + o + s, n = this.config.maxTokens - r;
|
|
413
|
-
return { system: e, tools: t, messages: o, memory: s, available: n, total: this.config.maxTokens };
|
|
414
|
-
}
|
|
415
|
-
shouldCompact() {
|
|
416
|
-
let e = this.getTokenUsage();
|
|
417
|
-
return (this.config.maxTokens - e.available) / this.config.maxTokens > this.config.autoCompactThreshold;
|
|
418
|
-
}
|
|
419
|
-
shouldWarn() {
|
|
420
|
-
let e = this.getTokenUsage();
|
|
421
|
-
return (this.config.maxTokens - e.available) / this.config.maxTokens > this.config.warningThreshold;
|
|
422
|
-
}
|
|
423
|
-
async loadProjectMemory(e) {
|
|
424
|
-
let t = (0, import_path2.join)(e, "CORTEX.md");
|
|
425
|
-
try {
|
|
426
|
-
await (0, import_promises2.access)(t), this.memoryFiles.push(t);
|
|
427
|
-
} catch {
|
|
428
|
-
}
|
|
429
|
-
}
|
|
430
|
-
async loadUserMemory() {
|
|
431
|
-
let e = (0, import_os2.homedir)(), t = (0, import_path2.join)(e, ".cortex", "memory.md");
|
|
432
|
-
try {
|
|
433
|
-
await (0, import_promises2.access)(t), this.memoryFiles.push(t);
|
|
434
|
-
} catch {
|
|
435
|
-
}
|
|
436
|
-
}
|
|
437
|
-
clear() {
|
|
438
|
-
this.messages = [], this.tokenCount = 0;
|
|
439
|
-
}
|
|
440
|
-
updateTokenCount() {
|
|
441
|
-
let e = this.messagesToString();
|
|
442
|
-
this.tokenCount = this.estimateTokens(e);
|
|
443
|
-
}
|
|
444
|
-
messagesToString() {
|
|
445
|
-
return this.messages.map((e) => {
|
|
446
|
-
let t = `${e.role}: ${e.content}`;
|
|
447
|
-
return e.tool_calls && (t += `
|
|
448
|
-
` + JSON.stringify(e.tool_calls)), t;
|
|
449
|
-
}).join(`
|
|
450
|
-
`);
|
|
451
|
-
}
|
|
452
|
-
estimateTokens(e) {
|
|
453
|
-
return Math.ceil(e.length / 4);
|
|
454
|
-
}
|
|
455
|
-
};
|
|
456
|
-
W = [{ name: "Read", description: "Read the contents of a file from the filesystem." }, { name: "Write", description: "Create a new file or overwrite an existing file." }, { name: "Edit", description: "Make a precise edit to an existing file." }, { name: "Bash", description: "Execute a shell command in the terminal." }, { name: "Glob", description: "Find files matching a pattern." }, { name: "Grep", description: "Search for text within files." }, { name: "TodoWrite", description: "Create and manage a todo list." }];
|
|
457
|
-
C = class {
|
|
458
|
-
projectMemory = "";
|
|
459
|
-
userMemory = "";
|
|
460
|
-
sessions = [];
|
|
461
|
-
configDir;
|
|
462
|
-
projectDir = "";
|
|
463
|
-
constructor() {
|
|
464
|
-
this.configDir = (0, import_path3.join)((0, import_os3.homedir)(), ".cortex");
|
|
465
|
-
}
|
|
466
|
-
async initialize(e) {
|
|
467
|
-
this.projectDir = e, await this.ensureConfigDir(), await this.loadProjectMemory(e), await this.loadUserMemory(), await this.loadSessionHistory();
|
|
468
|
-
}
|
|
469
|
-
async ensureConfigDir() {
|
|
470
|
-
try {
|
|
471
|
-
await (0, import_promises3.mkdir)(this.configDir, { recursive: true });
|
|
472
|
-
} catch {
|
|
473
|
-
}
|
|
474
|
-
}
|
|
475
|
-
async loadProjectMemory(e) {
|
|
476
|
-
let t = (0, import_path3.join)(e, "CORTEX.md");
|
|
477
|
-
try {
|
|
478
|
-
this.projectMemory = await (0, import_promises3.readFile)(t, "utf-8");
|
|
479
|
-
} catch {
|
|
480
|
-
this.projectMemory = "";
|
|
481
|
-
}
|
|
482
|
-
}
|
|
483
|
-
async loadUserMemory() {
|
|
484
|
-
let e = (0, import_path3.join)(this.configDir, "memory.md");
|
|
485
|
-
try {
|
|
486
|
-
this.userMemory = await (0, import_promises3.readFile)(e, "utf-8");
|
|
487
|
-
} catch {
|
|
488
|
-
this.userMemory = "";
|
|
489
|
-
}
|
|
490
|
-
}
|
|
491
|
-
async loadSessionHistory() {
|
|
492
|
-
let e = (0, import_path3.join)(this.configDir, "sessions.json");
|
|
493
|
-
try {
|
|
494
|
-
let t = await (0, import_promises3.readFile)(e, "utf-8");
|
|
495
|
-
this.sessions = JSON.parse(t);
|
|
496
|
-
} catch {
|
|
497
|
-
this.sessions = [];
|
|
498
|
-
}
|
|
499
|
-
}
|
|
500
|
-
async saveSession(e) {
|
|
501
|
-
this.sessions.push(e), this.sessions.length > 50 && (this.sessions = this.sessions.slice(-50));
|
|
502
|
-
let t = (0, import_path3.join)(this.configDir, "sessions.json");
|
|
503
|
-
await (0, import_promises3.writeFile)(t, JSON.stringify(this.sessions, null, 2));
|
|
504
|
-
}
|
|
505
|
-
async updateProjectMemory(e) {
|
|
506
|
-
if (!this.projectDir) return;
|
|
507
|
-
let t = (0, import_path3.join)(this.projectDir, "CORTEX.md");
|
|
508
|
-
await (0, import_promises3.writeFile)(t, e, "utf-8"), this.projectMemory = e;
|
|
509
|
-
}
|
|
510
|
-
async updateUserMemory(e) {
|
|
511
|
-
let t = (0, import_path3.join)(this.configDir, "memory.md");
|
|
512
|
-
await (0, import_promises3.writeFile)(t, e, "utf-8"), this.userMemory = e;
|
|
513
|
-
}
|
|
514
|
-
getProjectMemory() {
|
|
515
|
-
return this.projectMemory;
|
|
516
|
-
}
|
|
517
|
-
getUserMemory() {
|
|
518
|
-
return this.userMemory;
|
|
519
|
-
}
|
|
520
|
-
getMemoryForContext() {
|
|
521
|
-
let e = [];
|
|
522
|
-
if (this.projectMemory && e.push(`## Project Memory (${this.projectDir})
|
|
523
|
-
${this.projectMemory}`), this.userMemory && e.push(`## User Memory
|
|
524
|
-
${this.userMemory}`), this.sessions.length > 0) {
|
|
525
|
-
let t = this.sessions.slice(-5);
|
|
526
|
-
e.push(`## Recent Sessions
|
|
527
|
-
${t.map((o) => `- ${new Date(o.endTime).toLocaleDateString()}: ${o.summary}`).join(`
|
|
528
|
-
`)}`);
|
|
529
|
-
}
|
|
530
|
-
return e.join(`
|
|
63
|
+
${s.content}`}),i&&this.messages.push({role:"assistant",content:i.content}),this.updateTokenCount()}summarizeConversation(e){let t=e?500:300,s=this.messages.filter(r=>r.tool_calls).map(r=>r.tool_calls?.map(l=>l.function.name).join(", ")).join("; "),i=this.messages.filter(r=>r.role==="tool"&&r.content?.includes("File edited")).length,o=this.messages.filter(r=>r.role==="tool"&&r.content?.includes("File written")).length,n=`Session summary: ${i} files edited, ${o} files created`;return s&&(n+=`. Tools used: ${s}`),n.slice(0,t)}getTokenUsage(){let e=this.config.systemPromptTokens,t=this.config.toolDefTokens,s=this.estimateTokens(this.messagesToString()),i=this.memoryFiles.length*500,o=e+t+s+i,n=this.config.maxTokens-o;return{system:e,tools:t,messages:s,memory:i,available:n,total:this.config.maxTokens}}shouldCompact(){let e=this.getTokenUsage();return(this.config.maxTokens-e.available)/this.config.maxTokens>this.config.autoCompactThreshold}shouldWarn(){let e=this.getTokenUsage();return(this.config.maxTokens-e.available)/this.config.maxTokens>this.config.warningThreshold}async loadProjectMemory(e){let t=(0,j.join)(e,"CORTEX.md");try{await(0,C.access)(t),this.memoryFiles.push(t)}catch{}}async loadUserMemory(){let e=(0,N.homedir)(),t=(0,j.join)(e,".cortex","memory.md");try{await(0,C.access)(t),this.memoryFiles.push(t)}catch{}}clear(){this.messages=[],this.tokenCount=0}updateTokenCount(){let e=this.messagesToString();this.tokenCount=this.estimateTokens(e)}messagesToString(){return this.messages.map(e=>{let t=`${e.role}: ${e.content}`;return e.tool_calls&&(t+=`
|
|
64
|
+
`+JSON.stringify(e.tool_calls)),t}).join(`
|
|
65
|
+
`)}estimateTokens(e){return Math.ceil(e.length/4)}},oe=[{name:"Read",description:"Read the contents of a file from the filesystem."},{name:"Write",description:"Create a new file or overwrite an existing file."},{name:"Edit",description:"Make a precise edit to an existing file."},{name:"Bash",description:"Execute a shell command in the terminal."},{name:"Glob",description:"Find files matching a pattern."},{name:"Grep",description:"Search for text within files."},{name:"TodoWrite",description:"Create and manage a todo list."}],se=class{projectMemory="";userMemory="";sessions=[];configDir;projectDir="";constructor(){this.configDir=(0,h.join)((0,A.homedir)(),".cortex")}async initialize(e){this.projectDir=e,await this.ensureConfigDir(),await this.loadProjectMemory(e),await this.loadUserMemory(),await this.loadSessionHistory()}async ensureConfigDir(){try{await(0,p.mkdir)(this.configDir,{recursive:!0})}catch{}}async loadProjectMemory(e){let t=(0,h.join)(e,"CORTEX.md");try{this.projectMemory=await(0,p.readFile)(t,"utf-8")}catch{this.projectMemory=""}}async loadUserMemory(){let e=(0,h.join)(this.configDir,"memory.md");try{this.userMemory=await(0,p.readFile)(e,"utf-8")}catch{this.userMemory=""}}async loadSessionHistory(){let e=(0,h.join)(this.configDir,"sessions.json");try{let t=await(0,p.readFile)(e,"utf-8");this.sessions=JSON.parse(t)}catch{this.sessions=[]}}async saveSession(e){this.sessions.push(e),this.sessions.length>50&&(this.sessions=this.sessions.slice(-50));let t=(0,h.join)(this.configDir,"sessions.json");await(0,p.writeFile)(t,JSON.stringify(this.sessions,null,2))}async updateProjectMemory(e){if(!this.projectDir)return;let t=(0,h.join)(this.projectDir,"CORTEX.md");await(0,p.writeFile)(t,e,"utf-8"),this.projectMemory=e}async updateUserMemory(e){let t=(0,h.join)(this.configDir,"memory.md");await(0,p.writeFile)(t,e,"utf-8"),this.userMemory=e}getProjectMemory(){return this.projectMemory}getUserMemory(){return this.userMemory}getMemoryForContext(){let e=[];if(this.projectMemory&&e.push(`## Project Memory (${this.projectDir})
|
|
66
|
+
${this.projectMemory}`),this.userMemory&&e.push(`## User Memory
|
|
67
|
+
${this.userMemory}`),this.sessions.length>0){let t=this.sessions.slice(-5);e.push(`## Recent Sessions
|
|
68
|
+
${t.map(s=>`- ${new Date(s.endTime).toLocaleDateString()}: ${s.summary}`).join(`
|
|
69
|
+
`)}`)}return e.join(`
|
|
531
70
|
|
|
532
|
-
`);
|
|
533
|
-
}
|
|
534
|
-
getSessions() {
|
|
535
|
-
return this.sessions;
|
|
536
|
-
}
|
|
537
|
-
async createProjectMemoryFile() {
|
|
538
|
-
if (!this.projectDir) return;
|
|
539
|
-
let e = (0, import_path3.join)(this.projectDir, "CORTEX.md");
|
|
540
|
-
try {
|
|
541
|
-
await (0, import_promises3.access)(e);
|
|
542
|
-
} catch {
|
|
543
|
-
let t = `# Cortex Project Memory
|
|
71
|
+
`)}getSessions(){return this.sessions}async createProjectMemoryFile(){if(!this.projectDir)return;let e=(0,h.join)(this.projectDir,"CORTEX.md");try{await(0,p.access)(e)}catch{let t=`# Cortex Project Memory
|
|
544
72
|
|
|
545
73
|
## Project Overview
|
|
546
74
|
<!-- Fill in your project description -->
|
|
@@ -553,231 +81,14 @@ ${t.map((o) => `- ${new Date(o.endTime).toLocaleDateString()}: ${o.summary}`).jo
|
|
|
553
81
|
|
|
554
82
|
## Architecture Notes
|
|
555
83
|
<!-- Document important architectural decisions -->
|
|
556
|
-
`;
|
|
557
|
-
|
|
558
|
-
|
|
559
|
-
}
|
|
560
|
-
};
|
|
561
|
-
U = [{ name: "Read", description: "Read the contents of a file from the filesystem. Use this when you need to see what's in a file. Provide the absolute_path to the file. Use offset and limit to read specific portions of large files.", input_schema: { type: "object", properties: { file_path: { type: "string", description: "Absolute path to the file to read" }, offset: { type: "number", description: "Line number to start reading from (1-indexed)", default: 1 }, limit: { type: "number", description: "Maximum number of lines to read", default: 1e3 } }, required: ["file_path"] } }, { name: "Write", description: "Create a new file or overwrite an existing file with new content. Use this to create new files or make significant changes that warrant full replacement rather than targeted edits.", input_schema: { type: "object", properties: { file_path: { type: "string", description: "Absolute path to the file to write" }, content: { type: "string", description: "Complete file content to write" } }, required: ["file_path", "content"] } }, { name: "Edit", description: "Make a precise edit to an existing file using a search and replace pattern. Use this for targeted modifications. The old_string must match exactly including whitespace and indentation.", input_schema: { type: "object", properties: { file_path: { type: "string", description: "Absolute path to the file to edit" }, old_string: { type: "string", description: "The exact text to find and replace (must match exactly including whitespace)" }, new_string: { type: "string", description: "The replacement text" } }, required: ["file_path", "old_string", "new_string"] } }, { name: "Bash", description: "Execute a shell command in the terminal. Use this to run commands, scripts, git operations, or any other system operations. The working_directory defaults to the current directory.", input_schema: { type: "object", properties: { command: { type: "string", description: "The command to execute" }, description: { type: "string", description: "What this command does (for context)", default: "" } }, required: ["command"] } }, { name: "Glob", description: "Find files matching a pattern using glob syntax. Useful for discovering files in a project without reading their contents.", input_schema: { type: "object", properties: { pattern: { type: "string", description: 'Glob pattern (e.g., "**/*.ts", "src/**/*.js")' }, path: { type: "string", description: "Directory to search in (defaults to current directory)", default: "." } }, required: ["pattern"] } }, { name: "Grep", description: "Search for text within files. Returns matching lines with file paths.", input_schema: { type: "object", properties: { pattern: { type: "string", description: "Regular expression pattern to search for" }, path: { type: "string", description: "Directory or file to search in", default: "." }, include: { type: "string", description: 'File pattern to include (e.g., "*.ts", "*.js")' } }, required: ["pattern"] } }, { name: "TodoWrite", description: "Create and manage a todo list for tracking multi-step tasks. Use this to organize complex tasks into manageable steps.", input_schema: { type: "object", properties: { todos: { type: "array", description: "Array of todo items with content, status, and priority", items: { type: "object", properties: { content: { type: "string", description: "Description of the task" }, status: { type: "string", enum: ["pending", "in_progress", "completed", "cancelled"], default: "pending" }, priority: { type: "string", enum: ["high", "medium", "low"], default: "medium" } }, required: ["content", "status", "priority"] } } }, required: ["todos"] } }];
|
|
562
|
-
P = class extends import_events.EventEmitter {
|
|
563
|
-
servers = /* @__PURE__ */ new Map();
|
|
564
|
-
processes = /* @__PURE__ */ new Map();
|
|
565
|
-
tools = /* @__PURE__ */ new Map();
|
|
566
|
-
requestId = 0;
|
|
567
|
-
pendingRequests = /* @__PURE__ */ new Map();
|
|
568
|
-
initialized = false;
|
|
569
|
-
async addServer(e) {
|
|
570
|
-
this.servers.set(e.name, e), e.type === "stdio" ? await this.connectStdioServer(e) : e.type === "http" && e.url && await this.connectHttpServer(e);
|
|
571
|
-
}
|
|
572
|
-
async removeServer(e) {
|
|
573
|
-
let t = this.servers.get(e);
|
|
574
|
-
if (t) {
|
|
575
|
-
if (t.type === "stdio") {
|
|
576
|
-
let o = this.processes.get(e);
|
|
577
|
-
o && (o.kill(), this.processes.delete(e));
|
|
578
|
-
}
|
|
579
|
-
this.servers.delete(e), this.tools.delete(e);
|
|
580
|
-
}
|
|
581
|
-
}
|
|
582
|
-
async listTools(e) {
|
|
583
|
-
let t = {};
|
|
584
|
-
if (e) {
|
|
585
|
-
let o = this.tools.get(e);
|
|
586
|
-
if (o) for (let s of o) t[`${e}/${s.name}`] = s;
|
|
587
|
-
} else for (let [o, s] of this.tools) for (let r of s) t[`${o}/${r.name}`] = r;
|
|
588
|
-
return t;
|
|
589
|
-
}
|
|
590
|
-
async callTool(e, t, o) {
|
|
591
|
-
let s = this.servers.get(e);
|
|
592
|
-
if (!s) throw new Error(`Unknown MCP server: ${e}`);
|
|
593
|
-
return s.type === "http" && s.url ? this.callHttpTool(s.url, t, o) : this.callStdioTool(e, t, o);
|
|
594
|
-
}
|
|
595
|
-
async connectStdioServer(e) {
|
|
596
|
-
let t = ++this.requestId;
|
|
597
|
-
try {
|
|
598
|
-
let o = (0, import_execa2.execa)(e.command, e.args || [], { env: { ...process.env, ...e.env }, stdio: ["pipe", "pipe", "pipe"] });
|
|
599
|
-
this.processes.set(e.name, o), o.stdout?.on("data", (s) => {
|
|
600
|
-
this.handleStdioMessage(e.name, s.toString());
|
|
601
|
-
}), o.stderr?.on("data", (s) => {
|
|
602
|
-
console.error(`[MCP ${e.name}]`, s.toString());
|
|
603
|
-
}), await this.sendStdioRequest(e.name, "initialize", { protocolVersion: "2024-11-05", capabilities: {}, clientInfo: { name: "cortex", version: "5.0.0" } }), this.tools.set(e.name, []), this.initialized = true;
|
|
604
|
-
} catch (o) {
|
|
605
|
-
console.error(`Failed to connect to MCP server ${e.name}:`, o);
|
|
606
|
-
}
|
|
607
|
-
}
|
|
608
|
-
async connectHttpServer(e) {
|
|
609
|
-
if (e.url) try {
|
|
610
|
-
let t = await fetch(`${e.url}/initialize`, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ protocolVersion: "2024-11-05", capabilities: {}, clientInfo: { name: "cortex", version: "5.0.0" } }) });
|
|
611
|
-
if (t.ok) {
|
|
612
|
-
let o = await t.json();
|
|
613
|
-
this.tools.set(e.name, o.tools || []);
|
|
614
|
-
}
|
|
615
|
-
} catch (t) {
|
|
616
|
-
console.error(`Failed to connect to MCP server ${e.name}:`, t);
|
|
617
|
-
}
|
|
618
|
-
}
|
|
619
|
-
async sendStdioRequest(e, t, o) {
|
|
620
|
-
let s = ++this.requestId, r = this.processes.get(e);
|
|
621
|
-
if (!r || !r.stdin) throw new Error(`MCP server ${e} not connected`);
|
|
622
|
-
return new Promise((n, a) => {
|
|
623
|
-
this.pendingRequests.set(s, { resolve: n, reject: a }), r.stdin?.write(JSON.stringify({ jsonrpc: "2.0", id: s, method: t, params: o }) + `
|
|
624
|
-
`), setTimeout(() => {
|
|
625
|
-
this.pendingRequests.has(s) && (this.pendingRequests.delete(s), a(new Error("MCP request timeout")));
|
|
626
|
-
}, 3e4);
|
|
627
|
-
});
|
|
628
|
-
}
|
|
629
|
-
handleStdioMessage(e, t) {
|
|
630
|
-
try {
|
|
631
|
-
let o = t.split(`
|
|
632
|
-
`).filter(Boolean);
|
|
633
|
-
for (let s of o) {
|
|
634
|
-
let r = JSON.parse(s);
|
|
635
|
-
if (r.id && this.pendingRequests.has(r.id)) {
|
|
636
|
-
let n = this.pendingRequests.get(r.id);
|
|
637
|
-
this.pendingRequests.delete(r.id), r.error ? n.reject(new Error(r.error.message || "MCP error")) : n.resolve(r.result);
|
|
638
|
-
}
|
|
639
|
-
if (r.method === "tools/list") {
|
|
640
|
-
let n = r.params?.tools || [];
|
|
641
|
-
this.tools.set(e, n), this.emit("toolsUpdated", { server: e, tools: n });
|
|
642
|
-
}
|
|
643
|
-
r.method === "notifications/tools_changed" && this.refreshTools(e);
|
|
644
|
-
}
|
|
645
|
-
} catch {
|
|
646
|
-
}
|
|
647
|
-
}
|
|
648
|
-
async callStdioTool(e, t, o) {
|
|
649
|
-
let s = await this.sendStdioRequest(e, "tools/call", { name: t, arguments: o });
|
|
650
|
-
return JSON.stringify(s);
|
|
651
|
-
}
|
|
652
|
-
async callHttpTool(e, t, o) {
|
|
653
|
-
let s = await fetch(`${e}/tools/call`, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ name: t, arguments: o }) });
|
|
654
|
-
if (!s.ok) throw new Error(`MCP tool call failed: ${s.status}`);
|
|
655
|
-
let r = await s.json();
|
|
656
|
-
return JSON.stringify(r.content?.[0]?.text || "");
|
|
657
|
-
}
|
|
658
|
-
async refreshTools(e) {
|
|
659
|
-
let t = this.servers.get(e);
|
|
660
|
-
if (t) if (t.type === "http" && t.url) await this.connectHttpServer(t);
|
|
661
|
-
else try {
|
|
662
|
-
let s = (await this.sendStdioRequest(e, "tools/list", {}))?.tools || [];
|
|
663
|
-
this.tools.set(e, s);
|
|
664
|
-
} catch {
|
|
665
|
-
}
|
|
666
|
-
}
|
|
667
|
-
getToolsForContext() {
|
|
668
|
-
let e = [];
|
|
669
|
-
for (let [t, o] of this.tools) for (let s of o) e.push({ name: `${t}/${s.name}`, description: s.description, input_schema: s.inputSchema });
|
|
670
|
-
return e;
|
|
671
|
-
}
|
|
672
|
-
disconnect() {
|
|
673
|
-
for (let [e, t] of this.processes) t.kill();
|
|
674
|
-
this.processes.clear(), this.servers.clear(), this.tools.clear(), this.initialized = false;
|
|
675
|
-
}
|
|
676
|
-
};
|
|
677
|
-
T = class {
|
|
678
|
-
provider;
|
|
679
|
-
context;
|
|
680
|
-
memory;
|
|
681
|
-
mcp;
|
|
682
|
-
config;
|
|
683
|
-
tools = [...U];
|
|
684
|
-
cwd;
|
|
685
|
-
planMode = false;
|
|
686
|
-
currentPlan = [];
|
|
687
|
-
constructor(e, t) {
|
|
688
|
-
this.config = e, this.cwd = t;
|
|
689
|
-
let o = { type: e.provider, baseUrl: e.providerUrl, apiKey: e.apiKey, defaultModel: e.model };
|
|
690
|
-
this.provider = y(o), this.context = new x(), this.memory = new C(), this.mcp = new P(), this.planMode = e.planMode || false;
|
|
691
|
-
}
|
|
692
|
-
async initialize() {
|
|
693
|
-
console.log(import_chalk2.default.blue("\u{1F9E0} Initializing Cortex...")), await this.memory.initialize(this.cwd), await this.context.loadProjectMemory(this.cwd), await this.context.loadUserMemory(), await this.loadMCPConfig(), console.log(import_chalk2.default.green("\u2705 Cortex ready!"));
|
|
694
|
-
}
|
|
695
|
-
async loadMCPConfig() {
|
|
696
|
-
let e = [(0, import_path.join)(this.cwd, ".cortex", "mcp.json"), (0, import_path.join)((0, import_os.homedir)(), ".cortex", "mcp.json")];
|
|
697
|
-
for (let t of e) try {
|
|
698
|
-
let o = await (0, import_promises.readFile)(t, "utf-8"), s = JSON.parse(o);
|
|
699
|
-
if (s.servers) for (let [r, n] of Object.entries(s.servers)) await this.mcp.addServer({ name: r, command: n.command, args: n.args, env: n.env, type: n.type || "stdio", url: n.url });
|
|
700
|
-
} catch {
|
|
701
|
-
}
|
|
702
|
-
if (this.mcp.listTools()) {
|
|
703
|
-
let t = this.mcp.getToolsForContext();
|
|
704
|
-
this.tools = [...U, ...t];
|
|
705
|
-
}
|
|
706
|
-
}
|
|
707
|
-
async run(e) {
|
|
708
|
-
console.log(import_chalk2.default.cyan(`
|
|
84
|
+
`;await(0,p.writeFile)(e,t),this.projectMemory=t}}},U=[{name:"Read",description:"Read the contents of a file from the filesystem. Use this when you need to see what's in a file. Provide the absolute_path to the file. Use offset and limit to read specific portions of large files.",input_schema:{type:"object",properties:{file_path:{type:"string",description:"Absolute path to the file to read"},offset:{type:"number",description:"Line number to start reading from (1-indexed)",default:1},limit:{type:"number",description:"Maximum number of lines to read",default:1e3}},required:["file_path"]}},{name:"Write",description:"Create a new file or overwrite an existing file with new content. Use this to create new files or make significant changes that warrant full replacement rather than targeted edits.",input_schema:{type:"object",properties:{file_path:{type:"string",description:"Absolute path to the file to write"},content:{type:"string",description:"Complete file content to write"}},required:["file_path","content"]}},{name:"Edit",description:"Make a precise edit to an existing file using a search and replace pattern. Use this for targeted modifications. The old_string must match exactly including whitespace and indentation.",input_schema:{type:"object",properties:{file_path:{type:"string",description:"Absolute path to the file to edit"},old_string:{type:"string",description:"The exact text to find and replace (must match exactly including whitespace)"},new_string:{type:"string",description:"The replacement text"}},required:["file_path","old_string","new_string"]}},{name:"Bash",description:"Execute a shell command in the terminal. Use this to run commands, scripts, git operations, or any other system operations. The working_directory defaults to the current directory.",input_schema:{type:"object",properties:{command:{type:"string",description:"The command to execute"},description:{type:"string",description:"What this command does (for context)",default:""}},required:["command"]}},{name:"Glob",description:"Find files matching a pattern using glob syntax. Useful for discovering files in a project without reading their contents.",input_schema:{type:"object",properties:{pattern:{type:"string",description:'Glob pattern (e.g., "**/*.ts", "src/**/*.js")'},path:{type:"string",description:"Directory to search in (defaults to current directory)",default:"."}},required:["pattern"]}},{name:"Grep",description:"Search for text within files. Returns matching lines with file paths.",input_schema:{type:"object",properties:{pattern:{type:"string",description:"Regular expression pattern to search for"},path:{type:"string",description:"Directory or file to search in",default:"."},include:{type:"string",description:'File pattern to include (e.g., "*.ts", "*.js")'}},required:["pattern"]}},{name:"TodoWrite",description:"Create and manage a todo list for tracking multi-step tasks. Use this to organize complex tasks into manageable steps.",input_schema:{type:"object",properties:{todos:{type:"array",description:"Array of todo items with content, status, and priority",items:{type:"object",properties:{content:{type:"string",description:"Description of the task"},status:{type:"string",enum:["pending","in_progress","completed","cancelled"],default:"pending"},priority:{type:"string",enum:["high","medium","low"],default:"medium"}},required:["content","status","priority"]}}},required:["todos"]}}];ne=class extends z.EventEmitter{servers=new Map;processes=new Map;tools=new Map;requestId=0;pendingRequests=new Map;initialized=!1;async addServer(e){this.servers.set(e.name,e),e.type==="stdio"?await this.connectStdioServer(e):e.type==="http"&&e.url&&await this.connectHttpServer(e)}async removeServer(e){let t=this.servers.get(e);if(t){if(t.type==="stdio"){let s=this.processes.get(e);s&&(s.kill(),this.processes.delete(e))}this.servers.delete(e),this.tools.delete(e)}}async listTools(e){let t={};if(e){let s=this.tools.get(e);if(s)for(let i of s)t[`${e}/${i.name}`]=i}else for(let[s,i]of this.tools)for(let o of i)t[`${s}/${o.name}`]=o;return t}async callTool(e,t,s){let i=this.servers.get(e);if(!i)throw new Error(`Unknown MCP server: ${e}`);return i.type==="http"&&i.url?this.callHttpTool(i.url,t,s):this.callStdioTool(e,t,s)}async connectStdioServer(e){let t=++this.requestId;try{let s=(0,I.execa)(e.command,e.args||[],{env:{...process.env,...e.env},stdio:["pipe","pipe","pipe"]});this.processes.set(e.name,s),s.stdout?.on("data",i=>{this.handleStdioMessage(e.name,i.toString())}),s.stderr?.on("data",i=>{console.error(`[MCP ${e.name}]`,i.toString())}),await this.sendStdioRequest(e.name,"initialize",{protocolVersion:"2024-11-05",capabilities:{},clientInfo:{name:"cortex",version:"5.0.0"}}),this.tools.set(e.name,[]),this.initialized=!0}catch(s){console.error(`Failed to connect to MCP server ${e.name}:`,s)}}async connectHttpServer(e){if(e.url)try{let t=await fetch(`${e.url}/initialize`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({protocolVersion:"2024-11-05",capabilities:{},clientInfo:{name:"cortex",version:"5.0.0"}})});if(t.ok){let s=await t.json();this.tools.set(e.name,s.tools||[])}}catch(t){console.error(`Failed to connect to MCP server ${e.name}:`,t)}}async sendStdioRequest(e,t,s){let i=++this.requestId,o=this.processes.get(e);if(!o||!o.stdin)throw new Error(`MCP server ${e} not connected`);return new Promise((n,r)=>{this.pendingRequests.set(i,{resolve:n,reject:r}),o.stdin?.write(JSON.stringify({jsonrpc:"2.0",id:i,method:t,params:s})+`
|
|
85
|
+
`),setTimeout(()=>{this.pendingRequests.has(i)&&(this.pendingRequests.delete(i),r(new Error("MCP request timeout")))},3e4)})}handleStdioMessage(e,t){try{let s=t.split(`
|
|
86
|
+
`).filter(Boolean);for(let i of s){let o=JSON.parse(i);if(o.id&&this.pendingRequests.has(o.id)){let n=this.pendingRequests.get(o.id);this.pendingRequests.delete(o.id),o.error?n.reject(new Error(o.error.message||"MCP error")):n.resolve(o.result)}if(o.method==="tools/list"){let n=o.params?.tools||[];this.tools.set(e,n),this.emit("toolsUpdated",{server:e,tools:n})}o.method==="notifications/tools_changed"&&this.refreshTools(e)}}catch{}}async callStdioTool(e,t,s){let i=await this.sendStdioRequest(e,"tools/call",{name:t,arguments:s});return JSON.stringify(i)}async callHttpTool(e,t,s){let i=await fetch(`${e}/tools/call`,{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify({name:t,arguments:s})});if(!i.ok)throw new Error(`MCP tool call failed: ${i.status}`);let o=await i.json();return JSON.stringify(o.content?.[0]?.text||"")}async refreshTools(e){let t=this.servers.get(e);if(t)if(t.type==="http"&&t.url)await this.connectHttpServer(t);else try{let s=(await this.sendStdioRequest(e,"tools/list",{}))?.tools||[];this.tools.set(e,s)}catch{}}getToolsForContext(){let e=[];for(let[t,s]of this.tools)for(let i of s)e.push({name:`${t}/${i.name}`,description:i.description,input_schema:i.inputSchema});return e}disconnect(){for(let[e,t]of this.processes)t.kill();this.processes.clear(),this.servers.clear(),this.tools.clear(),this.initialized=!1}},ae=class{provider;context;memory;mcp;config;tools=[...U];cwd;planMode=!1;currentPlan=[];constructor(e,t){this.config=e,this.cwd=t;let s={type:e.provider,baseUrl:e.providerUrl,apiKey:e.apiKey,defaultModel:e.model};this.provider=k(s),this.context=new te,this.memory=new se,this.mcp=new ne,this.planMode=e.planMode||!1}async initialize(){console.log(m.default.blue("\u{1F9E0} Initializing Cortex...")),await this.memory.initialize(this.cwd),await this.context.loadProjectMemory(this.cwd),await this.context.loadUserMemory(),await this.loadMCPConfig(),console.log(m.default.green("\u2705 Cortex ready!"))}async loadMCPConfig(){let e=[(0,T.join)(this.cwd,".cortex","mcp.json"),(0,T.join)((0,R.homedir)(),".cortex","mcp.json")];for(let t of e)try{let s=await(0,O.readFile)(t,"utf-8"),i=JSON.parse(s);if(i.servers)for(let[o,n]of Object.entries(i.servers))await this.mcp.addServer({name:o,command:n.command,args:n.args,env:n.env,type:n.type||"stdio",url:n.url})}catch{}if(this.mcp.listTools()){let t=this.mcp.getToolsForContext();this.tools=[...U,...t]}}async run(e){console.log(m.default.cyan(`
|
|
709
87
|
\u{1F916} Thinking...
|
|
710
|
-
`)),
|
|
711
|
-
|
|
712
|
-
async runDirect(e) {
|
|
713
|
-
let t = 0, o = this.config.maxIterations || 50, s = false;
|
|
714
|
-
for (this.context.addMessage({ role: "user", content: e }); !s && t < o; ) {
|
|
715
|
-
t++, this.context.shouldWarn() && console.log(import_chalk2.default.yellow(`\u26A0\uFE0F Context usage high (${Math.round((1 - this.context.getTokenUsage().available / this.context.getTokenUsage().total) * 100)}%)`));
|
|
716
|
-
try {
|
|
717
|
-
let r = await this.provider.chat(this.context.getMessagesForAPI(), this.tools, this.config.model);
|
|
718
|
-
if (console.log(import_chalk2.default.white(r.content)), r.tool_calls && r.tool_calls.length > 0) {
|
|
719
|
-
let n = { role: "assistant", content: r.content, tool_calls: r.tool_calls };
|
|
720
|
-
this.context.addMessage(n);
|
|
721
|
-
for (let a of r.tool_calls) {
|
|
722
|
-
let c = a.function.name, l = JSON.parse(a.function.arguments);
|
|
723
|
-
console.log(import_chalk2.default.dim(`
|
|
724
|
-
\u{1F527} ${c}: ${JSON.stringify(l)}`));
|
|
725
|
-
let u = await I(c, l, this.cwd), w = { role: "tool", content: u.output, tool_call_id: a.id, name: c };
|
|
726
|
-
this.context.addMessage(w), u.is_error && console.log(import_chalk2.default.red(`\u274C ${u.output}`));
|
|
727
|
-
}
|
|
728
|
-
} else s = true;
|
|
729
|
-
(r.stop_reason === "stop" || r.stop_reason === "end_turn") && (s = true);
|
|
730
|
-
} catch (r) {
|
|
731
|
-
console.error(import_chalk2.default.red(`Error: ${r instanceof Error ? r.message : String(r)}`));
|
|
732
|
-
break;
|
|
733
|
-
}
|
|
734
|
-
this.context.shouldCompact() && (console.log(import_chalk2.default.yellow("\u{1F4E6} Compacting context...")), await this.context.compact());
|
|
735
|
-
}
|
|
736
|
-
t >= o && console.log(import_chalk2.default.yellow(`\u26A0\uFE0F Reached max iterations (${o})`));
|
|
737
|
-
}
|
|
738
|
-
async runWithPlanMode(e) {
|
|
739
|
-
console.log(import_chalk2.default.blue("\u{1F4DD} Running in Plan Mode...")), this.context.addMessage({ role: "user", content: `[PLANNING MODE] ${e}
|
|
88
|
+
`)),this.planMode?await this.runWithPlanMode(e):await this.runDirect(e)}async runDirect(e){let t=0,s=this.config.maxIterations||50,i=!1;for(this.context.addMessage({role:"user",content:e});!i&&t<s;){t++,this.context.shouldWarn()&&console.log(m.default.yellow(`\u26A0\uFE0F Context usage high (${Math.round((1-this.context.getTokenUsage().available/this.context.getTokenUsage().total)*100)}%)`));try{let o=await this.provider.chat(this.context.getMessagesForAPI(),this.tools,this.config.model);if(console.log(m.default.white(o.content)),o.tool_calls&&o.tool_calls.length>0){let n={role:"assistant",content:o.content,tool_calls:o.tool_calls};this.context.addMessage(n);for(let r of o.tool_calls){let l=r.function.name,d=JSON.parse(r.function.arguments);console.log(m.default.dim(`
|
|
89
|
+
\u{1F527} ${l}: ${JSON.stringify(d)}`));let c=await ie(l,d,this.cwd),w={role:"tool",content:c.output,tool_call_id:r.id,name:l};this.context.addMessage(w),c.is_error&&console.log(m.default.red(`\u274C ${c.output}`))}}else i=!0;(o.stop_reason==="stop"||o.stop_reason==="end_turn")&&(i=!0)}catch(o){console.error(m.default.red(`Error: ${o instanceof Error?o.message:String(o)}`));break}this.context.shouldCompact()&&(console.log(m.default.yellow("\u{1F4E6} Compacting context...")),await this.context.compact())}t>=s&&console.log(m.default.yellow(`\u26A0\uFE0F Reached max iterations (${s})`))}async runWithPlanMode(e){console.log(m.default.blue("\u{1F4DD} Running in Plan Mode...")),this.context.addMessage({role:"user",content:`[PLANNING MODE] ${e}
|
|
740
90
|
|
|
741
|
-
Please analyze the task and create a detailed plan before executing anything. List each step with the files that will be modified.`
|
|
742
|
-
|
|
743
|
-
console.log(import_chalk2.default.cyan(`
|
|
744
|
-
\u{1F4CB} Proposed Plan:`)), console.log(import_chalk2.default.white(t.content)), console.log(import_chalk2.default.green(`
|
|
91
|
+
Please analyze the task and create a detailed plan before executing anything. List each step with the files that will be modified.`});let t=await this.provider.chat(this.context.getMessagesForAPI(),void 0,this.config.model);console.log(m.default.cyan(`
|
|
92
|
+
\u{1F4CB} Proposed Plan:`)),console.log(m.default.white(t.content)),console.log(m.default.green(`
|
|
745
93
|
\u2705 Proceeding with execution...
|
|
746
|
-
`)),
|
|
747
|
-
}
|
|
748
|
-
getContextInfo() {
|
|
749
|
-
let e = this.context.getTokenUsage();
|
|
750
|
-
return { tokens: `${e.total - e.available} / ${e.total}`, messages: this.context.getMessages().length };
|
|
751
|
-
}
|
|
752
|
-
async cleanup() {
|
|
753
|
-
this.mcp.disconnect();
|
|
754
|
-
}
|
|
755
|
-
};
|
|
756
|
-
p = new import_commander.Command();
|
|
757
|
-
p.name("cortex").description("The ultimate local AI coding agent - context-aware, memory-powered, MCP-enabled").version("5.0.0-beta.1").option("-p, --provider <name>", "Provider to use: ollama, lmstudio, openrouter").option("-m, --model <name>", "Model to use").option("-u, --url <url>", "Provider URL").option("--plan-mode", "Show plan before executing changes");
|
|
758
|
-
p.command("init").description("Initialize Cortex in the current project").action(re);
|
|
759
|
-
p.command("run [prompt...]").description("Run a coding task").option("-p, --provider <name>", "Provider to use").option("-m, --model <name>", "Model to use").option("-u, --url <url>", "Provider URL").option("--plan-mode", "Show plan before executing").action(async (i, e) => {
|
|
760
|
-
(!i || i.length === 0) && (console.log(import_chalk.default.yellow('Please provide a prompt. Usage: cortex run "your task"')), process.exit(1)), await ie(i.join(" "), e);
|
|
761
|
-
});
|
|
762
|
-
p.command("status").description("Check provider and model status").action(ae);
|
|
763
|
-
p.command("setup").description("Configure Cortex settings").option("-p, --provider <name>", "Provider name").option("-m, --model <name>", "Default model").option("-f, --force", "Overwrite existing config").action(ce);
|
|
764
|
-
p.command("models").description("List available models").option("-p, --provider <name>", "Provider to query").action(le);
|
|
765
|
-
p.command("chat").description("Start interactive chat session").option("-p, --provider <name>", "Provider to use").option("-m, --model <name>", "Model to use").action(async (i) => {
|
|
766
|
-
console.log(import_chalk.default.yellow("Interactive chat coming soon!"));
|
|
767
|
-
});
|
|
768
|
-
process.on("SIGINT", () => v(0));
|
|
769
|
-
process.on("SIGTERM", () => v(0));
|
|
770
|
-
process.on("uncaughtException", (i) => {
|
|
771
|
-
i.message?.includes("ExitPromptError") || i.message?.includes("User force closed") || i.message?.includes("prompt") ? v(0) : (console.error(import_chalk.default.red("Error:"), i.message), process.exit(1));
|
|
772
|
-
});
|
|
773
|
-
process.on("unhandledRejection", (i) => {
|
|
774
|
-
let e = String(i);
|
|
775
|
-
(e.includes("ExitPromptError") || e.includes("User force closed") || e.includes("prompt")) && v(0);
|
|
776
|
-
});
|
|
777
|
-
me = process.argv.slice(2);
|
|
778
|
-
me.length === 0 ? p.help() : p.parse();
|
|
779
|
-
}
|
|
780
|
-
});
|
|
781
|
-
|
|
782
|
-
// bin/index.js
|
|
783
|
-
Promise.resolve().then(() => init_dist());
|
|
94
|
+
`)),this.context.clear(),this.context.addMessage({role:"user",content:e}),await this.runDirect(e)}getContextInfo(){let e=this.context.getTokenUsage();return{tokens:`${e.total-e.available} / ${e.total}`,messages:this.context.getMessages().length}}async cleanup(){this.mcp.disconnect()}},f=new D.Command;f.name("cortex").description("The ultimate local AI coding agent - context-aware, memory-powered, MCP-enabled").version("5.0.0-beta.1").option("-p, --provider <name>","Provider to use: ollama, lmstudio, openrouter").option("-m, --model <name>","Model to use").option("-u, --url <url>","Provider URL").option("--plan-mode","Show plan before executing changes").option("--tui","Start in TUI mode (default when no command provided)").option("--no-tui","Disable TUI mode");f.command("init").description("Initialize Cortex in the current project").action(le);f.command("run [prompt...]").description("Run a coding task").option("-p, --provider <name>","Provider to use").option("-m, --model <name>","Model to use").option("-u, --url <url>","Provider URL").option("--plan-mode","Show plan before executing").action(async(e,t)=>{(!e||e.length===0)&&(console.log(a.default.yellow('Please provide a prompt. Usage: cortex run "your task"')),process.exit(1)),await ce(e.join(" "),t)});f.command("status").description("Check provider and model status").action(de);f.command("setup").description("Configure Cortex settings").option("-p, --provider <name>","Provider name").option("-m, --model <name>","Default model").option("-f, --force","Overwrite existing config").action(me);f.command("models").description("List available models").option("-p, --provider <name>","Provider to query").action(pe);f.command("chat").description("Start interactive chat session").option("-p, --provider <name>","Provider to use").option("-m, --model <name>","Model to use").action(async e=>{console.log(a.default.yellow("Interactive chat coming soon!"))});process.on("SIGINT",()=>_(0));process.on("SIGTERM",()=>_(0));process.on("uncaughtException",e=>{e.message?.includes("ExitPromptError")||e.message?.includes("User force closed")||e.message?.includes("prompt")?_(0):(console.error(a.default.red("Error:"),e.message),process.exit(1))});process.on("unhandledRejection",e=>{let t=String(e);(t.includes("ExitPromptError")||t.includes("User force closed")||t.includes("prompt"))&&_(0)});F=process.argv.slice(2);ue()});Promise.resolve().then(()=>W());
|