@augmentcode/auggie-sdk 0.1.6 → 0.1.7
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/auggie/sdk-acp-client.d.ts +0 -102
- package/dist/auggie/sdk-acp-client.d.ts.map +1 -1
- package/dist/auggie/sdk-acp-client.js +1 -32
- package/dist/auggie/sdk-acp-client.js.map +1 -1
- package/dist/context/direct-context.d.ts +287 -0
- package/dist/context/direct-context.d.ts.map +1 -0
- package/dist/context/direct-context.js +624 -0
- package/dist/context/direct-context.js.map +1 -0
- package/dist/context/filesystem-context.d.ts +84 -0
- package/dist/context/filesystem-context.d.ts.map +1 -0
- package/dist/context/filesystem-context.js +198 -0
- package/dist/context/filesystem-context.js.map +1 -0
- package/dist/context/internal/__mocks__/api-client.d.ts +54 -0
- package/dist/context/internal/__mocks__/api-client.d.ts.map +1 -0
- package/dist/context/internal/__mocks__/api-client.js +96 -0
- package/dist/context/internal/__mocks__/api-client.js.map +1 -0
- package/dist/context/internal/api-client.d.ts +75 -0
- package/dist/context/internal/api-client.d.ts.map +1 -0
- package/dist/context/internal/api-client.js +218 -0
- package/dist/context/internal/api-client.js.map +1 -0
- package/dist/context/internal/blob-name-calculator.d.ts +14 -0
- package/dist/context/internal/blob-name-calculator.d.ts.map +1 -0
- package/dist/context/internal/blob-name-calculator.js +44 -0
- package/dist/context/internal/blob-name-calculator.js.map +1 -0
- package/dist/context/internal/retry-utils.d.ts +22 -0
- package/dist/context/internal/retry-utils.d.ts.map +1 -0
- package/dist/context/internal/retry-utils.js +88 -0
- package/dist/context/internal/retry-utils.js.map +1 -0
- package/dist/context/internal/search-utils.d.ts +8 -0
- package/dist/context/internal/search-utils.d.ts.map +1 -0
- package/dist/context/internal/search-utils.js +10 -0
- package/dist/context/internal/search-utils.js.map +1 -0
- package/dist/context/internal/session-reader.d.ts +17 -0
- package/dist/context/internal/session-reader.d.ts.map +1 -0
- package/dist/context/internal/session-reader.js +27 -0
- package/dist/context/internal/session-reader.js.map +1 -0
- package/dist/context/types.d.ts +113 -0
- package/dist/context/types.d.ts.map +1 -0
- package/dist/context/types.js +5 -0
- package/dist/context/types.js.map +1 -0
- package/dist/index.d.ts +6 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +7 -0
- package/dist/index.js.map +1 -1
- package/dist/version.js.map +1 -1
- package/package.json +2 -1
|
@@ -10,68 +10,6 @@ export type PredefinedToolType = "remove-files" | "save-file" | "str-replace-edi
|
|
|
10
10
|
* Tool identifier type - can be a predefined tool or a custom tool name.
|
|
11
11
|
*/
|
|
12
12
|
export type ToolIdentifier = PredefinedToolType | string;
|
|
13
|
-
/**
|
|
14
|
-
* Tool permission types that control how tools are executed.
|
|
15
|
-
*/
|
|
16
|
-
export type ToolPermission = {
|
|
17
|
-
/** Always allow the tool to run without approval */
|
|
18
|
-
type: "allow";
|
|
19
|
-
} | {
|
|
20
|
-
/** Always deny the tool from running */
|
|
21
|
-
type: "deny";
|
|
22
|
-
} | {
|
|
23
|
-
/** Require user approval before running (default behavior) */
|
|
24
|
-
type: "ask-user";
|
|
25
|
-
} | {
|
|
26
|
-
/** Use a webhook to determine if the tool should run */
|
|
27
|
-
type: "webhook-policy";
|
|
28
|
-
/** URL of the webhook endpoint */
|
|
29
|
-
webhookUrl: string;
|
|
30
|
-
} | {
|
|
31
|
-
/** Use a script to determine if the tool should run */
|
|
32
|
-
type: "script-policy";
|
|
33
|
-
/** Path to the script file */
|
|
34
|
-
script: string;
|
|
35
|
-
};
|
|
36
|
-
/**
|
|
37
|
-
* Tool permission rule that specifies permissions for a specific tool.
|
|
38
|
-
*
|
|
39
|
-
* @example
|
|
40
|
-
* ```typescript
|
|
41
|
-
* // Allow a tool to run without approval
|
|
42
|
-
* { toolName: "view", permission: { type: "allow" } }
|
|
43
|
-
*
|
|
44
|
-
* // Deny a tool from running
|
|
45
|
-
* { toolName: "launch-process", permission: { type: "deny" } }
|
|
46
|
-
*
|
|
47
|
-
* // Require user approval
|
|
48
|
-
* { toolName: "remove-files", permission: { type: "ask-user" } }
|
|
49
|
-
*
|
|
50
|
-
* // Use a webhook for dynamic permission checking
|
|
51
|
-
* {
|
|
52
|
-
* toolName: "github-api",
|
|
53
|
-
* permission: {
|
|
54
|
-
* type: "webhook-policy",
|
|
55
|
-
* webhookUrl: "https://example.com/webhook"
|
|
56
|
-
* }
|
|
57
|
-
* }
|
|
58
|
-
*
|
|
59
|
-
* // Use a script for custom permission logic
|
|
60
|
-
* {
|
|
61
|
-
* toolName: "launch-process",
|
|
62
|
-
* permission: {
|
|
63
|
-
* type: "script-policy",
|
|
64
|
-
* script: "/path/to/policy.sh"
|
|
65
|
-
* }
|
|
66
|
-
* }
|
|
67
|
-
* ```
|
|
68
|
-
*/
|
|
69
|
-
export interface ToolPermissionRule {
|
|
70
|
-
/** Name of the tool to apply the permission to */
|
|
71
|
-
toolName: string;
|
|
72
|
-
/** Permission policy for the tool */
|
|
73
|
-
permission: ToolPermission;
|
|
74
|
-
}
|
|
75
13
|
type AuggieOptions = {
|
|
76
14
|
auggiePath?: string;
|
|
77
15
|
workspaceRoot?: string;
|
|
@@ -100,45 +38,6 @@ type AuggieOptions = {
|
|
|
100
38
|
* ```
|
|
101
39
|
*/
|
|
102
40
|
excludedTools?: ToolIdentifier[];
|
|
103
|
-
/**
|
|
104
|
-
* Tool permission rules that control how tools are executed.
|
|
105
|
-
* These rules allow you to automatically allow, deny, or require approval for specific tools.
|
|
106
|
-
*
|
|
107
|
-
* @example
|
|
108
|
-
* ```typescript
|
|
109
|
-
* toolPermissions: [
|
|
110
|
-
* // Auto-allow safe read-only tools
|
|
111
|
-
* { toolName: "view", permission: { type: "allow" } },
|
|
112
|
-
* { toolName: "codebase-retrieval", permission: { type: "allow" } },
|
|
113
|
-
*
|
|
114
|
-
* // Deny dangerous tools
|
|
115
|
-
* { toolName: "launch-process", permission: { type: "deny" } },
|
|
116
|
-
*
|
|
117
|
-
* // Require approval for file modifications
|
|
118
|
-
* { toolName: "save-file", permission: { type: "ask-user" } },
|
|
119
|
-
* { toolName: "remove-files", permission: { type: "ask-user" } },
|
|
120
|
-
*
|
|
121
|
-
* // Use webhook for dynamic permission checking
|
|
122
|
-
* {
|
|
123
|
-
* toolName: "github-api",
|
|
124
|
-
* permission: {
|
|
125
|
-
* type: "webhook-policy",
|
|
126
|
-
* webhookUrl: "https://example.com/webhook"
|
|
127
|
-
* }
|
|
128
|
-
* },
|
|
129
|
-
*
|
|
130
|
-
* // Use script for custom permission logic
|
|
131
|
-
* {
|
|
132
|
-
* toolName: "str-replace-editor",
|
|
133
|
-
* permission: {
|
|
134
|
-
* type: "script-policy",
|
|
135
|
-
* script: "/path/to/policy.sh"
|
|
136
|
-
* }
|
|
137
|
-
* }
|
|
138
|
-
* ]
|
|
139
|
-
* ```
|
|
140
|
-
*/
|
|
141
|
-
toolPermissions?: ToolPermissionRule[];
|
|
142
41
|
};
|
|
143
42
|
/**
|
|
144
43
|
* ACP Client for connecting to Auggie as an ACP server
|
|
@@ -165,7 +64,6 @@ declare class Auggie {
|
|
|
165
64
|
private readonly apiUrl;
|
|
166
65
|
private readonly rules;
|
|
167
66
|
private readonly excludedTools;
|
|
168
|
-
private readonly toolPermissions;
|
|
169
67
|
private constructor();
|
|
170
68
|
/**
|
|
171
69
|
* Create and initialize a new Auggie instance
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"sdk-acp-client.d.ts","sourceRoot":"","sources":["../../src/auggie/sdk-acp-client.ts"],"names":[],"mappings":"AAGA,OAAO,EAUN,KAAK,mBAAmB,EACxB,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAI/B,KAAK,MAAM,GAAG,UAAU,GAAG,WAAW,GAAG,SAAS,GAAG,MAAM,CAAC;
|
|
1
|
+
{"version":3,"file":"sdk-acp-client.d.ts","sourceRoot":"","sources":["../../src/auggie/sdk-acp-client.ts"],"names":[],"mappings":"AAGA,OAAO,EAUN,KAAK,mBAAmB,EACxB,MAAM,0BAA0B,CAAC;AAClC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAI/B,KAAK,MAAM,GAAG,UAAU,GAAG,WAAW,GAAG,SAAS,GAAG,MAAM,CAAC;AAE5D;;;GAGG;AACH,MAAM,MAAM,kBAAkB,GAE3B,cAAc,GACd,WAAW,GACX,oBAAoB,GACpB,MAAM,GAEN,gBAAgB,GAChB,cAAc,GACd,cAAc,GACd,eAAe,GACf,gBAAgB,GAEhB,YAAY,GAEZ,WAAW,GACX,cAAc,GACd,qBAAqB,GACrB,eAAe,CAAC;AAEnB;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,kBAAkB,GAAG,MAAM,CAAC;AAEzD,KAAK,aAAa,GAAG;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC7B,yEAAyE;IACzE,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;IACjB;;;;;;;;;;;;;;;OAeG;IACH,aAAa,CAAC,EAAE,cAAc,EAAE,CAAC;CACjC,CAAC;AAEF;;;;;;;;GAQG;AACH,cAAM,MAAM;IACX,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,gBAAgB,CAAS;IACjD,OAAO,CAAC,UAAU,CAAqC;IACvD,OAAO,CAAC,SAAS,CAAuB;IACxC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAiC;IAC/D,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,qBAAqB,CAEd;IACf,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAkB;IACxC,OAAO,CAAC,SAAS,CAAgC;IACjD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAU;IACxC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;IAC5C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;IAC5C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAuB;IAC7C,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA+B;IAE7D,OAAO;IAoBP;;;OAGG;WACU,MAAM,CAAC,OAAO,GAAE,aAAkB,GAAG,OAAO,CAAC,MAAM,CAAC;IASjE;;OAEG;YACW,OAAO;IAyFrB;;OAEG;IACH,OAAO,CAAC,YAAY;IAqCpB;;OAEG;YACW,oBAAoB;IAoBlC;;OAEG;YACW,aAAa;IA2C3B;;;;;;OAMG;IACG,MAAM,CACX,OAAO,EAAE,MAAM,EACf,EAAE,YAAoB,EAAE,GAAE;QAAE,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,SAEtD,GACC,OAAO,CAAC,MAAM,CAAC;IAoFlB;;OAEG;IACH,eAAe,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,mBAAmB,KAAK,IAAI,GAAG,IAAI;IAItE;;OAEG;IACG,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC;IAc7B;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA0B5B;;OAEG;IACH,OAAO,CAAC,WAAW;CAOnB;AAED,OAAO,EAAE,MAAM,EAAE,CAAC"}
|
|
@@ -4,29 +4,6 @@ import { Readable, Writable } from "node:stream";
|
|
|
4
4
|
import { ClientSideConnection, ndJsonStream, PROTOCOL_VERSION, } from "@agentclientprotocol/sdk";
|
|
5
5
|
import { getSDKVersion } from "../version";
|
|
6
6
|
import { AuggieMCPServer } from "./sdk-mcp-server";
|
|
7
|
-
/**
|
|
8
|
-
* Formats a tool permission rule into a CLI argument string.
|
|
9
|
-
* @param rule - The tool permission rule to format
|
|
10
|
-
* @returns The formatted CLI argument string (e.g., "tool-name:allow")
|
|
11
|
-
*/
|
|
12
|
-
function formatPermissionRule(rule) {
|
|
13
|
-
const { toolName, permission } = rule;
|
|
14
|
-
switch (permission.type) {
|
|
15
|
-
case "allow":
|
|
16
|
-
case "deny":
|
|
17
|
-
case "ask-user":
|
|
18
|
-
return `${toolName}:${permission.type}`;
|
|
19
|
-
case "webhook-policy":
|
|
20
|
-
return `${toolName}:webhook-policy(${permission.webhookUrl})`;
|
|
21
|
-
case "script-policy":
|
|
22
|
-
return `${toolName}:script-policy(${permission.script})`;
|
|
23
|
-
default: {
|
|
24
|
-
// TypeScript exhaustiveness check
|
|
25
|
-
const _exhaustive = permission;
|
|
26
|
-
throw new Error(`Unknown permission type: ${JSON.stringify(_exhaustive)}`);
|
|
27
|
-
}
|
|
28
|
-
}
|
|
29
|
-
}
|
|
30
7
|
/**
|
|
31
8
|
* ACP Client for connecting to Auggie as an ACP server
|
|
32
9
|
*
|
|
@@ -37,7 +14,7 @@ function formatPermissionRule(rule) {
|
|
|
37
14
|
* 4. Send prompts and receive responses
|
|
38
15
|
*/
|
|
39
16
|
class Auggie {
|
|
40
|
-
constructor({ auggiePath = "auggie", workspaceRoot, model = "haiku4.5", allowIndexing = true, apiKey, apiUrl = process.env.AUGMENT_API_URL || "https://api.augmentcode.com", rules, excludedTools,
|
|
17
|
+
constructor({ auggiePath = "auggie", workspaceRoot, model = "haiku4.5", allowIndexing = true, apiKey, apiUrl = process.env.AUGMENT_API_URL || "https://api.augmentcode.com", rules, excludedTools, } = {}) {
|
|
41
18
|
this.connection = null;
|
|
42
19
|
this.sessionId = null;
|
|
43
20
|
this.workspaceRoot = undefined;
|
|
@@ -53,7 +30,6 @@ class Auggie {
|
|
|
53
30
|
this.apiUrl = apiUrl;
|
|
54
31
|
this.rules = rules;
|
|
55
32
|
this.excludedTools = excludedTools;
|
|
56
|
-
this.toolPermissions = toolPermissions;
|
|
57
33
|
}
|
|
58
34
|
/**
|
|
59
35
|
* Create and initialize a new Auggie instance
|
|
@@ -96,13 +72,6 @@ class Auggie {
|
|
|
96
72
|
args.push("--remove-tool", tool);
|
|
97
73
|
}
|
|
98
74
|
}
|
|
99
|
-
if (this.toolPermissions && this.toolPermissions.length > 0) {
|
|
100
|
-
// Add each permission rule as a separate --permission argument
|
|
101
|
-
for (const rule of this.toolPermissions) {
|
|
102
|
-
const formattedRule = formatPermissionRule(rule);
|
|
103
|
-
args.push("--permission", formattedRule);
|
|
104
|
-
}
|
|
105
|
-
}
|
|
106
75
|
// Parse auggiePath to handle cases where it includes arguments (e.g., "node /path/to/file.js")
|
|
107
76
|
const pathParts = this.auggiePath.trim().split(Auggie.PATH_SPLIT_REGEX);
|
|
108
77
|
if (pathParts.length === 0 || !pathParts[0]) {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"sdk-acp-client.js","sourceRoot":"","sources":["../../src/auggie/sdk-acp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAGN,oBAAoB,EAGpB,YAAY,EACZ,gBAAgB,GAIhB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAInD;;;;GAIG;AACH,SAAS,oBAAoB,CAAC,IAAwB;IACrD,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC;IAEtC,QAAQ,UAAU,CAAC,IAAI,EAAE,CAAC;QACzB,KAAK,OAAO,CAAC;QACb,KAAK,MAAM,CAAC;QACZ,KAAK,UAAU;YACd,OAAO,GAAG,QAAQ,IAAI,UAAU,CAAC,IAAI,EAAE,CAAC;QACzC,KAAK,gBAAgB;YACpB,OAAO,GAAG,QAAQ,mBAAmB,UAAU,CAAC,UAAU,GAAG,CAAC;QAC/D,KAAK,eAAe;YACnB,OAAO,GAAG,QAAQ,kBAAkB,UAAU,CAAC,MAAM,GAAG,CAAC;QAC1D,OAAO,CAAC,CAAC,CAAC;YACT,kCAAkC;YAClC,MAAM,WAAW,GAAU,UAAU,CAAC;YACtC,MAAM,IAAI,KAAK,CACd,4BAA4B,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CACzD,CAAC;QACH,CAAC;IACF,CAAC;AACF,CAAC;AAyKD;;;;;;;;GAQG;AACH,MAAM,MAAM;IAoBX,YAAoB,EACnB,UAAU,GAAG,QAAQ,EACrB,aAAa,EACb,KAAK,GAAG,UAAU,EAClB,aAAa,GAAG,IAAI,EACpB,MAAM,EACN,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,6BAA6B,EACrE,KAAK,EACL,aAAa,EACb,eAAe,MACG,EAAE;QA5Bb,eAAU,GAAgC,IAAI,CAAC;QAC/C,cAAS,GAAkB,IAAI,CAAC;QAEvB,kBAAa,GAAuB,SAAS,CAAC;QACvD,kBAAa,GAAwB,IAAI,CAAC;QAC1C,0BAAqB,GAEnB,IAAI,CAAC;QACE,UAAK,GAAY,KAAK,CAAC;QAChC,cAAS,GAA2B,IAAI,CAAC;QAoBhD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IACxC,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAAyB,EAAE;QAC9C,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;QACzB,kEAAkE;QAClE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,QAAQ,CAAC;QAChD,MAAM,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO;QACpB,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,uDAAuD;YACvD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACnC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,EAAE,CAAC,CAAC;gBAC1D,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAChC,CAAC;QACF,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzD,8DAA8D;YAC9D,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACvC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YAClC,CAAC;QACF,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7D,+DAA+D;YAC/D,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzC,MAAM,aAAa,GAAG,oBAAoB,CAAC,IAAI,CAAC,CAAC;gBACjD,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC;YAC1C,CAAC;QACF,CAAC;QAED,+FAA+F;QAC/F,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACxE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACxD,CAAC;QACD,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,WAAW,GAAG,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAErD,qDAAqD;QACrD,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC;QACrC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE;YAChD,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC;YAClC,GAAG;SACH,CAAC,CAAC;QAEH,wBAAwB;QACxB,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACxC,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,IAAI,CAAC,KAAK;gBACb,OAAO,CAAC,GAAG,CACV,mCAAmC,IAAI,eAAe,MAAM,EAAE,CAC9D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,uCAAuC;QACvC,qDAAqD;QACrD,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACpE,CAAC;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAC3B,IAAI,CAAC,aAAa,CAAC,MAAM,CACgB,CAAC;QAC3C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAExD,wCAAwC;QACxC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAE3C,6DAA6D;QAC7D,IAAI,CAAC,UAAU,GAAG,IAAI,oBAAoB,CACzC,CAAC,KAAY,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAC1C,MAAM,CACN,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAC/C,CAAC;QAED,4BAA4B;QAC5B,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAAa;QACjC,OAAO;YACN,wCAAwC;YACxC,aAAa,EAAE,CAAC,MAA2B,EAAiB,EAAE;gBAC7D,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAChC,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBACpC,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;YAED,4CAA4C;YAC5C,iBAAiB,EAAE,CAClB,MAAgC,EACK,EAAE;gBACvC,8DAA8D;gBAC9D,kDAAkD;gBAClD,IAAI,IAAI,CAAC,KAAK;oBAAE,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;gBAE7D,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACjD,OAAO,OAAO,CAAC,OAAO,CAAC;wBACtB,OAAO,EAAE;4BACR,OAAO,EAAE,UAAU;4BACnB,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE;yBACzC;qBACD,CAAC,CAAC;gBACJ,CAAC;gBAED,kCAAkC;gBAClC,OAAO,OAAO,CAAC,OAAO,CAAC;oBACtB,OAAO,EAAE;wBACR,OAAO,EAAE,WAAW;qBACpB;iBACD,CAAC,CAAC;YACJ,CAAC;SACD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB;QACjC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9D,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACrD,eAAe,EAAE,gBAAgB;YACjC,UAAU,EAAE;gBACX,IAAI,EAAE,uBAAuB;gBAC7B,OAAO,EAAE,aAAa,EAAE;aACxB;YACD,kBAAkB,EAAE,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,KAAK;YACb,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CAAC,KAA4B;QACvD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAEvD,0CAA0C;QAC1C,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,wDAAwD;QACxD,IAAI,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;YAE1E,IAAI,CAAC,SAAS,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YAEvD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAE/C,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B,SAAS,EAAE,CAAC,CAAC;YAEnE,kDAAkD;YAClD,UAAU,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,YAAY;gBAClB,GAAG,EAAE,SAAS;gBACd,OAAO,EAAE,EAAE;aACX,CAAC,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACxD,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YAC5D,UAAU;SACV,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC;QAC3C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,eAAe,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAC3E,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,MAAM,CACX,OAAe,EACf,EAAE,YAAY,GAAG,KAAK,KAA6C;QAClE,YAAY,EAAE,KAAK;KACnB;QAED,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,sBAAsB,OAAO,GAAG,CAAC,CAAC;QAE9D,iCAAiC;QACjC,IAAI,eAAe,GAAG,EAAE,CAAC;QACzB,sEAAsE;QACtE,IAAI,iBAAiB,GAAG,EAAE,CAAC;QAC3B,kCAAkC;QAClC,IAAI,eAAe,GAAG,KAAK,CAAC;QAE5B,wCAAwC;QACxC,MAAM,gBAAgB,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACpD,MAAM,mBAAmB,GAAG,CAAC,MAA2B,EAAE,EAAE;YAC3D,kEAAkE;YAClE,IAAI,MAAM,CAAC,MAAM,CAAC,aAAa,KAAK,WAAW,EAAE,CAAC;gBACjD,eAAe,GAAG,IAAI,CAAC;gBACvB,wDAAwD;gBACxD,iBAAiB,GAAG,EAAE,CAAC;YACxB,CAAC;YAED,0BAA0B;YAC1B,IACC,MAAM,CAAC,MAAM,CAAC,aAAa,KAAK,qBAAqB;gBACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EACpC,CAAC;gBACF,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;gBAC7C,eAAe,IAAI,SAAS,CAAC;gBAE7B,gEAAgE;gBAChE,0DAA0D;gBAC1D,IAAI,eAAe,EAAE,CAAC;oBACrB,iBAAiB,IAAI,SAAS,CAAC;gBAChC,CAAC;YACF,CAAC;YAED,+CAA+C;YAC/C,IAAI,gBAAgB,EAAE,CAAC;gBACtB,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC,CAAC;QAEF,mCAAmC;QACnC,IAAI,CAAC,qBAAqB,GAAG,mBAAmB,CAAC;QAEjD,IAAI,CAAC;YACJ,uBAAuB;YACvB,MAAM,OAAO,GAAmB;gBAC/B;oBACC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO;iBACb;aACD,CAAC;YAEF,wCAAwC;YACxC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,OAAO;aACf,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,KAAK;gBACb,OAAO,CAAC,GAAG,CAAC,oCAAoC,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YAExE,4BAA4B;YAC5B,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC;YAE9C,sFAAsF;YACtF,IAAI,YAAY,IAAI,eAAe,EAAE,CAAC;gBACrC,OAAO,iBAAiB,CAAC;YAC1B,CAAC;YACD,OAAO,eAAe,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC7C,2CAA2C;YAC3C,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC;YAC9C,4BAA4B;YAC5B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAA+C;QAC9D,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM;QACX,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9D,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;YAC5B,SAAS,EAAE,IAAI,CAAC,SAAS;SACzB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACV,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QAEtD,gFAAgF;QAChF,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,sEAAsE;QACtE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;YACvD,+DAA+D;YAC/D,MAAM,OAAO,CAAC,IAAI,CAAC;gBAClB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;gBACrB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,4BAA4B;QAC5B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACK,WAAW;QAClB,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;YACtD,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAC1D,wCAAwC;YACxC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;IACF,CAAC;;AApauB,uBAAgB,GAAG,KAAK,AAAR,CAAS;AAualD,OAAO,EAAE,MAAM,EAAE,CAAC"}
|
|
1
|
+
{"version":3,"file":"sdk-acp-client.js","sourceRoot":"","sources":["../../src/auggie/sdk-acp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAqB,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAGN,oBAAoB,EAGpB,YAAY,EACZ,gBAAgB,GAIhB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AA+DnD;;;;;;;;GAQG;AACH,MAAM,MAAM;IAmBX,YAAoB,EACnB,UAAU,GAAG,QAAQ,EACrB,aAAa,EACb,KAAK,GAAG,UAAU,EAClB,aAAa,GAAG,IAAI,EACpB,MAAM,EACN,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,6BAA6B,EACrE,KAAK,EACL,aAAa,MACK,EAAE;QA1Bb,eAAU,GAAgC,IAAI,CAAC;QAC/C,cAAS,GAAkB,IAAI,CAAC;QAEvB,kBAAa,GAAuB,SAAS,CAAC;QACvD,kBAAa,GAAwB,IAAI,CAAC;QAC1C,0BAAqB,GAEnB,IAAI,CAAC;QACE,UAAK,GAAY,KAAK,CAAC;QAChC,cAAS,GAA2B,IAAI,CAAC;QAkBhD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAAyB,EAAE;QAC9C,MAAM,QAAQ,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;QACzB,kEAAkE;QAClE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,QAAQ,CAAC;QAChD,MAAM,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QACpC,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,OAAO;QACpB,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;QAC5C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QACnD,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,uDAAuD;YACvD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACnC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,EAAE,CAAC,CAAC;gBAC1D,CAAC;gBACD,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAChC,CAAC;QACF,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzD,8DAA8D;YAC9D,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACvC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;YAClC,CAAC;QACF,CAAC;QAED,+FAA+F;QAC/F,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QACxE,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACxD,CAAC;QACD,MAAM,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,WAAW,GAAG,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QAErD,qDAAqD;QACrD,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,GAAG,CAAC,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC;QACrC,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE;YAChD,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC;YAClC,GAAG;SACH,CAAC,CAAC;QAEH,wBAAwB;QACxB,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACxC,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,IAAI,CAAC,KAAK;gBACb,OAAO,CAAC,GAAG,CACV,mCAAmC,IAAI,eAAe,MAAM,EAAE,CAC9D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,uCAAuC;QACvC,qDAAqD;QACrD,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9D,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACpE,CAAC;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAC3B,IAAI,CAAC,aAAa,CAAC,MAAM,CACgB,CAAC;QAC3C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAExD,wCAAwC;QACxC,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAE3C,6DAA6D;QAC7D,IAAI,CAAC,UAAU,GAAG,IAAI,oBAAoB,CACzC,CAAC,KAAY,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAC1C,MAAM,CACN,CAAC;QAEF,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAC/C,CAAC;QAED,4BAA4B;QAC5B,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,MAAa;QACjC,OAAO;YACN,wCAAwC;YACxC,aAAa,EAAE,CAAC,MAA2B,EAAiB,EAAE;gBAC7D,IAAI,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAChC,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,CAAC;gBACpC,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;YAED,4CAA4C;YAC5C,iBAAiB,EAAE,CAClB,MAAgC,EACK,EAAE;gBACvC,8DAA8D;gBAC9D,kDAAkD;gBAClD,IAAI,IAAI,CAAC,KAAK;oBAAE,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;gBAE7D,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACjD,OAAO,OAAO,CAAC,OAAO,CAAC;wBACtB,OAAO,EAAE;4BACR,OAAO,EAAE,UAAU;4BACnB,QAAQ,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE;yBACzC;qBACD,CAAC,CAAC;gBACJ,CAAC;gBAED,kCAAkC;gBAClC,OAAO,OAAO,CAAC,OAAO,CAAC;oBACtB,OAAO,EAAE;wBACR,OAAO,EAAE,WAAW;qBACpB;iBACD,CAAC,CAAC;YACJ,CAAC;SACD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB;QACjC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9D,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACrD,eAAe,EAAE,gBAAgB;YACjC,UAAU,EAAE;gBACX,IAAI,EAAE,uBAAuB;gBAC7B,OAAO,EAAE,aAAa,EAAE;aACxB;YACD,kBAAkB,EAAE,EAAE;SACtB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,KAAK;YACb,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa,CAAC,KAA4B;QACvD,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QAEvD,0CAA0C;QAC1C,MAAM,UAAU,GAAgB,EAAE,CAAC;QAEnC,wDAAwD;QACxD,IAAI,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;YAE1E,IAAI,CAAC,SAAS,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YAEvD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YAE/C,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B,SAAS,EAAE,CAAC,CAAC;YAEnE,kDAAkD;YAClD,UAAU,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,YAAY;gBAClB,GAAG,EAAE,SAAS;gBACd,OAAO,EAAE,EAAE;aACX,CAAC,CAAC;QACJ,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC;YACxD,GAAG,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YAC5D,UAAU;SACV,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,eAAe,CAAC,SAAS,CAAC;QAC3C,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,eAAe,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAC3E,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACvB,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,MAAM,CACX,OAAe,EACf,EAAE,YAAY,GAAG,KAAK,KAA6C;QAClE,YAAY,EAAE,KAAK;KACnB;QAED,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACxE,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,sBAAsB,OAAO,GAAG,CAAC,CAAC;QAE9D,iCAAiC;QACjC,IAAI,eAAe,GAAG,EAAE,CAAC;QACzB,sEAAsE;QACtE,IAAI,iBAAiB,GAAG,EAAE,CAAC;QAC3B,kCAAkC;QAClC,IAAI,eAAe,GAAG,KAAK,CAAC;QAE5B,wCAAwC;QACxC,MAAM,gBAAgB,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACpD,MAAM,mBAAmB,GAAG,CAAC,MAA2B,EAAE,EAAE;YAC3D,kEAAkE;YAClE,IAAI,MAAM,CAAC,MAAM,CAAC,aAAa,KAAK,WAAW,EAAE,CAAC;gBACjD,eAAe,GAAG,IAAI,CAAC;gBACvB,wDAAwD;gBACxD,iBAAiB,GAAG,EAAE,CAAC;YACxB,CAAC;YAED,0BAA0B;YAC1B,IACC,MAAM,CAAC,MAAM,CAAC,aAAa,KAAK,qBAAqB;gBACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,EACpC,CAAC;gBACF,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC;gBAC7C,eAAe,IAAI,SAAS,CAAC;gBAE7B,gEAAgE;gBAChE,0DAA0D;gBAC1D,IAAI,eAAe,EAAE,CAAC;oBACrB,iBAAiB,IAAI,SAAS,CAAC;gBAChC,CAAC;YACF,CAAC;YAED,+CAA+C;YAC/C,IAAI,gBAAgB,EAAE,CAAC;gBACtB,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC,CAAC;QAEF,mCAAmC;QACnC,IAAI,CAAC,qBAAqB,GAAG,mBAAmB,CAAC;QAEjD,IAAI,CAAC;YACJ,uBAAuB;YACvB,MAAM,OAAO,GAAmB;gBAC/B;oBACC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO;iBACb;aACD,CAAC;YAEF,wCAAwC;YACxC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;gBAC7C,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,OAAO;aACf,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,KAAK;gBACb,OAAO,CAAC,GAAG,CAAC,oCAAoC,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;YAExE,4BAA4B;YAC5B,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC;YAE9C,sFAAsF;YACtF,IAAI,YAAY,IAAI,eAAe,EAAE,CAAC;gBACrC,OAAO,iBAAiB,CAAC;YAC1B,CAAC;YACD,OAAO,eAAe,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;YAC7C,2CAA2C;YAC3C,IAAI,CAAC,qBAAqB,GAAG,gBAAgB,CAAC;YAC9C,4BAA4B;YAC5B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAA+C;QAC9D,IAAI,CAAC,qBAAqB,GAAG,QAAQ,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM;QACX,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAC;QAE9D,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;YAC5B,SAAS,EAAE,IAAI,CAAC,SAAS;SACzB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACV,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QAEtD,gFAAgF;QAChF,IAAI,CAAC,WAAW,EAAE,CAAC;QAEnB,sEAAsE;QACtE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;YACvD,+DAA+D;YAC/D,MAAM,OAAO,CAAC,IAAI,CAAC;gBAClB,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;gBACrB,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;aACxD,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,4BAA4B;QAC5B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACvB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACvB,CAAC;QAED,IAAI,IAAI,CAAC,KAAK;YAAE,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACK,WAAW;QAClB,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC;YACtD,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;YAC1D,wCAAwC;YACxC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;IACF,CAAC;;AA1ZuB,uBAAgB,GAAG,KAAK,AAAR,CAAS;AA6ZlD,OAAO,EAAE,MAAM,EAAE,CAAC"}
|
|
@@ -0,0 +1,287 @@
|
|
|
1
|
+
import type { DirectContextOptions, DirectContextState, File, IndexingResult } from "./types";
|
|
2
|
+
/**
|
|
3
|
+
* Direct Context - API-based indexing with import/export state
|
|
4
|
+
*
|
|
5
|
+
* ⚠️ **EXPERIMENTAL API** - This API is experimental and may change without notice.
|
|
6
|
+
*
|
|
7
|
+
* @experimental
|
|
8
|
+
*
|
|
9
|
+
* This class provides explicit file indexing via API calls with the ability
|
|
10
|
+
* to export and restore state to avoid re-indexing between sessions.
|
|
11
|
+
*
|
|
12
|
+
* ## Creating a Context
|
|
13
|
+
*
|
|
14
|
+
* **Start fresh:**
|
|
15
|
+
* ```typescript
|
|
16
|
+
* const context = await DirectContext.create();
|
|
17
|
+
* ```
|
|
18
|
+
*
|
|
19
|
+
* **Restore from saved state:**
|
|
20
|
+
* ```typescript
|
|
21
|
+
* const context = await DirectContext.importFromFile("state.json");
|
|
22
|
+
* // or
|
|
23
|
+
* const context = await DirectContext.import(stateObject);
|
|
24
|
+
* ```
|
|
25
|
+
*
|
|
26
|
+
* ## Indexing Flow
|
|
27
|
+
*
|
|
28
|
+
* Files are uploaded to the backend and then indexed asynchronously. You have
|
|
29
|
+
* explicit control over when to wait for indexing:
|
|
30
|
+
*
|
|
31
|
+
* **Option 1: Wait after each upload**
|
|
32
|
+
* ```typescript
|
|
33
|
+
* await context.addToIndex(files); // waits for indexing by default
|
|
34
|
+
* await context.search("query"); // files are already indexed
|
|
35
|
+
* ```
|
|
36
|
+
*
|
|
37
|
+
* **Option 2: Batch uploads then wait**
|
|
38
|
+
* ```typescript
|
|
39
|
+
* await context.addToIndex(files1, { waitForIndexing: false });
|
|
40
|
+
* await context.addToIndex(files2, { waitForIndexing: false });
|
|
41
|
+
* await context.waitForIndexing(); // wait for all files to be indexed
|
|
42
|
+
* await context.search("query"); // files are now indexed
|
|
43
|
+
* ```
|
|
44
|
+
*
|
|
45
|
+
* Note: `search()` and `searchAndAsk()` do not wait for indexing automatically.
|
|
46
|
+
*
|
|
47
|
+
* ## State Persistence
|
|
48
|
+
*
|
|
49
|
+
* **Export state:**
|
|
50
|
+
* ```typescript
|
|
51
|
+
* const state = context.export();
|
|
52
|
+
* // or save to file
|
|
53
|
+
* await context.exportToFile("state.json");
|
|
54
|
+
* ```
|
|
55
|
+
*
|
|
56
|
+
* **Import state:**
|
|
57
|
+
* ```typescript
|
|
58
|
+
* const context = await DirectContext.importFromFile("state.json");
|
|
59
|
+
* ```
|
|
60
|
+
*/
|
|
61
|
+
export declare class DirectContext {
|
|
62
|
+
private static readonly MAX_FILE_SIZE_BYTES;
|
|
63
|
+
private static readonly MAX_BATCH_UPLOAD_SIZE;
|
|
64
|
+
private static readonly MAX_BATCH_CONTENT_BYTES;
|
|
65
|
+
private static readonly MAX_FIND_MISSING_SIZE;
|
|
66
|
+
private static readonly CHECKPOINT_THRESHOLD;
|
|
67
|
+
private readonly apiClient;
|
|
68
|
+
private readonly blobCalculator;
|
|
69
|
+
private readonly debug;
|
|
70
|
+
/**
|
|
71
|
+
* State management:
|
|
72
|
+
*
|
|
73
|
+
* blobMap: Tracks all files that have been uploaded (path -> blobName).
|
|
74
|
+
* This includes both indexed and non-indexed blobs.
|
|
75
|
+
* Also, includes both checkpointed and non-checkpointed blobs.
|
|
76
|
+
*
|
|
77
|
+
* checkpointId: The current checkpoint ID from the backend.
|
|
78
|
+
* Checkpoints are named snapshots of the blob set for performance.
|
|
79
|
+
* Updated after each successful checkpoint operation.
|
|
80
|
+
*
|
|
81
|
+
* pendingAdded: Blob names that have been uploaded but not yet checkpointed.
|
|
82
|
+
* Accumulated until CHECKPOINT_THRESHOLD is reached.
|
|
83
|
+
*
|
|
84
|
+
* pendingDeleted: Blob names that have been removed but not yet checkpointed.
|
|
85
|
+
* Accumulated until CHECKPOINT_THRESHOLD is reached.
|
|
86
|
+
*
|
|
87
|
+
* Note: Checkpointing and indexing are independent operations.
|
|
88
|
+
* A blob can be checkpointed but not indexed, indexed but not checkpointed,
|
|
89
|
+
* both, or neither.
|
|
90
|
+
*/
|
|
91
|
+
private readonly blobMap;
|
|
92
|
+
private checkpointId?;
|
|
93
|
+
private pendingAdded;
|
|
94
|
+
private pendingDeleted;
|
|
95
|
+
private readonly mutex;
|
|
96
|
+
private readonly pollingMutex;
|
|
97
|
+
/**
|
|
98
|
+
* Create and initialize a new DirectContext instance
|
|
99
|
+
*
|
|
100
|
+
* Authentication priority:
|
|
101
|
+
* 1. options.apiKey / options.apiUrl
|
|
102
|
+
* 2. AUGMENT_API_TOKEN / AUGMENT_API_URL environment variables
|
|
103
|
+
* 3. ~/.augment/session.json (created by `auggie login`)
|
|
104
|
+
*
|
|
105
|
+
* @param options Configuration options
|
|
106
|
+
* @returns Promise that resolves to a DirectContext instance
|
|
107
|
+
*/
|
|
108
|
+
static create(options?: DirectContextOptions): Promise<DirectContext>;
|
|
109
|
+
/**
|
|
110
|
+
* Import a DirectContext instance from a saved state object
|
|
111
|
+
*
|
|
112
|
+
* @param state The state object to restore from
|
|
113
|
+
* @param options Configuration options
|
|
114
|
+
* @returns Promise that resolves to a DirectContext instance with restored state
|
|
115
|
+
*/
|
|
116
|
+
static import(state: DirectContextState, options?: DirectContextOptions): Promise<DirectContext>;
|
|
117
|
+
/**
|
|
118
|
+
* Import a DirectContext instance from a saved state file (Node.js only)
|
|
119
|
+
*
|
|
120
|
+
* @param filePath Path to the state file
|
|
121
|
+
* @param options Configuration options
|
|
122
|
+
* @returns Promise that resolves to a DirectContext instance with restored state
|
|
123
|
+
*/
|
|
124
|
+
static importFromFile(filePath: string, options?: DirectContextOptions): Promise<DirectContext>;
|
|
125
|
+
/**
|
|
126
|
+
* Resolve API credentials from options, environment, or session file
|
|
127
|
+
*/
|
|
128
|
+
private static resolveCredentials;
|
|
129
|
+
/**
|
|
130
|
+
* Private constructor - use DirectContext.create() instead
|
|
131
|
+
* @param apiKey API key for authentication
|
|
132
|
+
* @param apiUrl API URL for the tenant
|
|
133
|
+
* @param debug Enable debug logging
|
|
134
|
+
*/
|
|
135
|
+
private constructor();
|
|
136
|
+
/**
|
|
137
|
+
* Log a debug message if debug mode is enabled
|
|
138
|
+
*/
|
|
139
|
+
private log;
|
|
140
|
+
/**
|
|
141
|
+
* Create a checkpoint if threshold is reached
|
|
142
|
+
*/
|
|
143
|
+
private maybeCheckpoint;
|
|
144
|
+
/**
|
|
145
|
+
* Get the list of indexed file paths
|
|
146
|
+
*/
|
|
147
|
+
getIndexedPaths(): string[];
|
|
148
|
+
/**
|
|
149
|
+
* Add files to the index by uploading them to the backend.
|
|
150
|
+
*
|
|
151
|
+
* By default, this method waits for the uploaded files to be fully indexed
|
|
152
|
+
* on the backend before returning. Set `waitForIndexing: false` to return
|
|
153
|
+
* immediately after upload completes (indexing will continue asynchronously).
|
|
154
|
+
*
|
|
155
|
+
* @param files - Array of files to add to the index
|
|
156
|
+
* @param options - Optional configuration
|
|
157
|
+
* @param options.waitForIndexing - If true (default), waits for the newly added files to be indexed before returning
|
|
158
|
+
* @returns Result indicating which files were newly uploaded vs already uploaded
|
|
159
|
+
*/
|
|
160
|
+
addToIndex(files: File[], options?: {
|
|
161
|
+
waitForIndexing?: boolean;
|
|
162
|
+
}): Promise<IndexingResult>;
|
|
163
|
+
/**
|
|
164
|
+
* Check which blobs are missing or not yet indexed on the server
|
|
165
|
+
*
|
|
166
|
+
* @param blobNames - Array of blob names to check
|
|
167
|
+
* @param includeNonIndexed - If true, includes blobs that are uploaded but not yet indexed.
|
|
168
|
+
* If false, only returns blobs that need to be uploaded.
|
|
169
|
+
* @returns Array of blob names that are either missing or not indexed (depending on flag)
|
|
170
|
+
*/
|
|
171
|
+
private findMissingBlobs;
|
|
172
|
+
/**
|
|
173
|
+
* Upload files in batches respecting backend limits
|
|
174
|
+
*/
|
|
175
|
+
private batchUploadFiles;
|
|
176
|
+
/**
|
|
177
|
+
* Validate that all files are within size limits
|
|
178
|
+
*/
|
|
179
|
+
private validateFileSizes;
|
|
180
|
+
/**
|
|
181
|
+
* Calculate blob names and prepare upload data for files
|
|
182
|
+
* Returns files that need uploading, new blob entries, and already uploaded files
|
|
183
|
+
*/
|
|
184
|
+
private prepareBlobsForUpload;
|
|
185
|
+
/**
|
|
186
|
+
* Handle replacing an existing blob (track deletion of old blob)
|
|
187
|
+
*/
|
|
188
|
+
private handleBlobReplacement;
|
|
189
|
+
/**
|
|
190
|
+
* Categorize files into newly uploaded vs already on server based on server response
|
|
191
|
+
*/
|
|
192
|
+
private categorizeFiles;
|
|
193
|
+
/**
|
|
194
|
+
* Internal implementation of addToIndex
|
|
195
|
+
*/
|
|
196
|
+
private doAddToIndex;
|
|
197
|
+
/**
|
|
198
|
+
* Remove paths from the index
|
|
199
|
+
*/
|
|
200
|
+
removeFromIndex(paths: string[]): Promise<void>;
|
|
201
|
+
/**
|
|
202
|
+
* Internal implementation of removeFromIndex
|
|
203
|
+
*/
|
|
204
|
+
private doRemoveFromIndex;
|
|
205
|
+
/**
|
|
206
|
+
* Clear the entire index
|
|
207
|
+
*/
|
|
208
|
+
clearIndex(): Promise<void>;
|
|
209
|
+
/**
|
|
210
|
+
* Internal implementation of clearIndex
|
|
211
|
+
*/
|
|
212
|
+
private doClearIndex;
|
|
213
|
+
/**
|
|
214
|
+
* Wait for specific blobs to be indexed on the backend
|
|
215
|
+
*
|
|
216
|
+
* This method is serialized via pollingMutex to prevent unbounded concurrent
|
|
217
|
+
* polling requests to the backend.
|
|
218
|
+
*/
|
|
219
|
+
private waitForSpecificBlobs;
|
|
220
|
+
/**
|
|
221
|
+
* Wait for all indexed files to be fully indexed on the backend.
|
|
222
|
+
*
|
|
223
|
+
* This method polls the backend until all files that have been added to the index
|
|
224
|
+
* are confirmed to be indexed and searchable.
|
|
225
|
+
*
|
|
226
|
+
* @returns Promise that resolves when all files are indexed
|
|
227
|
+
* @throws Error if indexing times out (default: 10 minutes)
|
|
228
|
+
*/
|
|
229
|
+
waitForIndexing(): Promise<void>;
|
|
230
|
+
/**
|
|
231
|
+
* Search the codebase using natural language and return formatted results.
|
|
232
|
+
*
|
|
233
|
+
* The results are returned as a formatted string designed for use in LLM prompts.
|
|
234
|
+
* The format includes file paths, line numbers, and code content in a structured,
|
|
235
|
+
* readable format that can be passed directly to LLM APIs like `generate()`.
|
|
236
|
+
*
|
|
237
|
+
* Note: This method does not wait for indexing. Ensure files are indexed before
|
|
238
|
+
* searching by either:
|
|
239
|
+
* - Using `addToIndex()` with `waitForIndexing: true` (default)
|
|
240
|
+
* - Calling `waitForIndexing()` explicitly before searching
|
|
241
|
+
*
|
|
242
|
+
* @param query - The search query describing what code you're looking for
|
|
243
|
+
* @param options - Optional search options
|
|
244
|
+
* @param options.maxOutputLength - Maximum character length of the formatted output (default: 20000, max: 80000)
|
|
245
|
+
* @returns A formatted string containing the search results, ready for LLM consumption
|
|
246
|
+
*/
|
|
247
|
+
search(query: string, options?: {
|
|
248
|
+
maxOutputLength?: number;
|
|
249
|
+
}): Promise<string>;
|
|
250
|
+
/**
|
|
251
|
+
* Search the indexed codebase and ask an LLM a question about the results.
|
|
252
|
+
*
|
|
253
|
+
* This is a convenience method that combines search() with an LLM call to answer
|
|
254
|
+
* questions about your codebase.
|
|
255
|
+
*
|
|
256
|
+
* Note: This method does not wait for indexing. Ensure files are indexed before
|
|
257
|
+
* searching by either:
|
|
258
|
+
* - Using `addToIndex()` with `waitForIndexing: true` (default)
|
|
259
|
+
* - Calling `waitForIndexing()` explicitly before searching
|
|
260
|
+
*
|
|
261
|
+
* @param searchQuery - The semantic search query to find relevant code (also used as the prompt if no separate prompt is provided)
|
|
262
|
+
* @param prompt - Optional prompt to ask the LLM about the search results. If not provided, searchQuery is used as the prompt.
|
|
263
|
+
* @returns The LLM's answer to your question
|
|
264
|
+
*
|
|
265
|
+
* @example
|
|
266
|
+
* ```typescript
|
|
267
|
+
* const answer = await context.searchAndAsk(
|
|
268
|
+
* "How does the authentication flow work?"
|
|
269
|
+
* );
|
|
270
|
+
* console.log(answer);
|
|
271
|
+
* ```
|
|
272
|
+
*/
|
|
273
|
+
searchAndAsk(searchQuery: string, prompt?: string): Promise<string>;
|
|
274
|
+
/**
|
|
275
|
+
* Export the current state to a JSON object
|
|
276
|
+
*/
|
|
277
|
+
export(): DirectContextState;
|
|
278
|
+
/**
|
|
279
|
+
* Internal method to import state from a JSON object
|
|
280
|
+
*/
|
|
281
|
+
private doImport;
|
|
282
|
+
/**
|
|
283
|
+
* Export state to a file (Node.js only)
|
|
284
|
+
*/
|
|
285
|
+
exportToFile(filePath: string): Promise<void>;
|
|
286
|
+
}
|
|
287
|
+
//# sourceMappingURL=direct-context.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"direct-context.d.ts","sourceRoot":"","sources":["../../src/context/direct-context.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAEV,oBAAoB,EACpB,kBAAkB,EAClB,IAAI,EACJ,cAAc,EACf,MAAM,SAAS,CAAC;AAEjB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA0DG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,mBAAmB,CAAa;IACxD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAQ;IACrD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,uBAAuB,CAAmB;IAClE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAQ;IACrD,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAQ;IAEpD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAmB;IAC7C,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAqB;IACpD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAU;IAEhC;;;;;;;;;;;;;;;;;;;;OAoBG;IACH,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAkC;IAC1D,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,YAAY,CAA0B;IAC9C,OAAO,CAAC,cAAc,CAA0B;IAGhD,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAe;IAKrC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAE5C;;;;;;;;;;OAUG;WACU,MAAM,CACjB,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,aAAa,CAAC;IAKzB;;;;;;OAMG;WACU,MAAM,CACjB,KAAK,EAAE,kBAAkB,EACzB,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,aAAa,CAAC;IAOzB;;;;;;OAMG;WACU,cAAc,CACzB,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,oBAAyB,GACjC,OAAO,CAAC,aAAa,CAAC;IAMzB;;OAEG;mBACkB,kBAAkB;IAqCvC;;;;;OAKG;IACH,OAAO;IAgBP;;OAEG;IACH,OAAO,CAAC,GAAG;IAMX;;OAEG;YACW,eAAe;IAoC7B;;OAEG;IACH,eAAe,IAAI,MAAM,EAAE;IAI3B;;;;;;;;;;;OAWG;IACG,UAAU,CACd,KAAK,EAAE,IAAI,EAAE,EACb,OAAO,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,OAAO,CAAA;KAAE,GACtC,OAAO,CAAC,cAAc,CAAC;IAmB1B;;;;;;;OAOG;YACW,gBAAgB;IA2B9B;;OAEG;YACW,gBAAgB;IAqC9B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAWzB;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAqC7B;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAW7B;;OAEG;IACH,OAAO,CAAC,eAAe;IAkBvB;;OAEG;YACW,YAAY;IAqD1B;;OAEG;IACG,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAIrD;;OAEG;YACW,iBAAiB;IAgB/B;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAIjC;;OAEG;IACH,OAAO,CAAC,YAAY;IAapB;;;;;OAKG;IACH,OAAO,CAAC,oBAAoB;IAgD5B;;;;;;;;OAQG;IACG,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAKtC;;;;;;;;;;;;;;;;OAgBG;IACG,MAAM,CACV,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,GACrC,OAAO,CAAC,MAAM,CAAC;IAgClB;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,YAAY,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOzE;;OAEG;IACH,MAAM,IAAI,kBAAkB;IAqB5B;;OAEG;YACW,QAAQ;IA8BtB;;OAEG;IACG,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAKpD"}
|