@vscode/chat-lib 0.3.0 → 0.3.1-0

package/dist/src/_internal/platform/nesFetch/common/responseStream.js

@@ -0,0 +1,167 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ResponseStream = void 0;
+const errors = __importStar(require("../../../util/common/errors"));
+const result_1 = require("../../../util/common/result");
+const async_1 = require("../../../util/vs/base/common/async");
+const types_1 = require("../../../util/vs/base/common/types");
+class ResponseStream {
+    constructor(fetcherResponse, stream, requestId, headers) {
+        this.fetcherResponse = fetcherResponse;
+        this.requestId = requestId;
+        this.headers = headers;
+        const tokensDeferredPromise = new async_1.DeferredPromise();
+        this.aggregatedStream = tokensDeferredPromise.p;
+        this.response = this.aggregatedStream.then((completions) => {
+            if (completions.isError()) {
+                return completions;
+            }
+            try {
+                return result_1.Result.ok(ResponseStream.aggregateCompletionsStream(completions.val));
+            }
+            catch (err) {
+                return result_1.Result.error(err);
+            }
+        });
+        this.stream = streamWithAggregation(stream, tokensDeferredPromise);
+    }
+    /**
+     * @throws client of the method should handle the error
+     */
+    async destroy() {
+        await this.fetcherResponse.body.destroy();
+    }
+    static aggregateCompletionsStream(stream) {
+        let text = '';
+        let finishReason = null;
+        let aggregatedLogsProbs = null;
+        let aggregatedUsage = undefined;
+        for (const completion of stream) {
+            const choice = completion.choices[0]; // TODO@ulugbekna: we only support choice.index=0
+            text += choice.text ?? '';
+            if (choice.logprobs) {
+                if (aggregatedLogsProbs === null) {
+                    aggregatedLogsProbs = {
+                        tokens: [...choice.logprobs.tokens],
+                        token_logprobs: [...choice.logprobs.token_logprobs],
+                        text_offset: [...choice.logprobs.text_offset],
+                        top_logprobs: [...choice.logprobs.top_logprobs],
+                    };
+                }
+                else {
+                    aggregatedLogsProbs.tokens.push(...choice.logprobs.tokens);
+                    aggregatedLogsProbs.token_logprobs.push(...choice.logprobs.token_logprobs);
+                    aggregatedLogsProbs.text_offset.push(...choice.logprobs.text_offset);
+                    aggregatedLogsProbs.top_logprobs.push(...choice.logprobs.top_logprobs);
+                }
+            }
+            if (completion.usage) {
+                if (aggregatedUsage === undefined) {
+                    aggregatedUsage = {
+                        completion_tokens: completion.usage.completion_tokens,
+                        prompt_tokens: completion.usage.prompt_tokens,
+                        total_tokens: completion.usage.total_tokens,
+                        completion_tokens_details: {
+                            audio_tokens: completion.usage.completion_tokens_details.audio_tokens,
+                            reasoning_tokens: completion.usage.completion_tokens_details.reasoning_tokens,
+                        },
+                        prompt_tokens_details: {
+                            audio_tokens: completion.usage.prompt_tokens_details.audio_tokens,
+                            reasoning_tokens: completion.usage.prompt_tokens_details.reasoning_tokens,
+                        }
+                    };
+                }
+                else {
+                    aggregatedUsage.completion_tokens += completion.usage.completion_tokens;
+                    aggregatedUsage.prompt_tokens += completion.usage.prompt_tokens;
+                    aggregatedUsage.total_tokens += completion.usage.total_tokens;
+                    aggregatedUsage.completion_tokens_details.audio_tokens += completion.usage.completion_tokens_details.audio_tokens;
+                    aggregatedUsage.completion_tokens_details.reasoning_tokens += completion.usage.completion_tokens_details.reasoning_tokens;
+                    aggregatedUsage.prompt_tokens_details.audio_tokens += completion.usage.prompt_tokens_details.audio_tokens;
+                    aggregatedUsage.prompt_tokens_details.reasoning_tokens += completion.usage.prompt_tokens_details.reasoning_tokens;
+                }
+            }
+            if (choice.finish_reason) {
+                (0, types_1.assertType)(finishReason === null, 'cannot already have finishReason if just seeing choice.finish_reason');
+                finishReason = choice.finish_reason;
+            }
+        }
+        if (stream.length === 0) {
+            throw new Error(`Response is empty!`);
+        }
+        const completion = stream[0];
+        const choice = {
+            index: 0,
+            finish_reason: finishReason,
+            logprobs: aggregatedLogsProbs,
+            text,
+        };
+        const aggregatedCompletion = {
+            choices: [choice],
+            system_fingerprint: completion.system_fingerprint,
+            object: completion.object,
+            usage: aggregatedUsage,
+        };
+        return aggregatedCompletion;
+    }
+}
+exports.ResponseStream = ResponseStream;
+/**
+ * Wraps an async iterable stream into an async generator that also collects completions
+ * for aggregation and resolves the deferred promise when done.
+ */
+async function* streamWithAggregation(stream, deferredPromise) {
+    const completions = [];
+    let error;
+    try {
+        for await (const completion of stream) {
+            completions.push(completion);
+            yield completion;
+        }
+    }
+    catch (e) {
+        error = errors.fromUnknown(e);
+        throw error;
+    }
+    finally {
+        deferredPromise.complete(error ? result_1.Result.error(error) : result_1.Result.ok(completions));
+    }
+}
+//# sourceMappingURL=responseStream.js.map

package/dist/src/_internal/platform/nesFetch/common/responseStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"responseStream.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/nesFetch/common/responseStream.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEhG,oEAAsD;AACtD,wDAAqD;AACrD,8DAAqE;AACrE,8DAAgE;AAKhE,MAAa,cAAc;IAwB1B,YAA6B,eAAyB,EAAE,MAAiC,EAAkB,SAAoB,EAAkB,OAAiB;QAArI,oBAAe,GAAf,eAAe,CAAU;QAAqD,cAAS,GAAT,SAAS,CAAW;QAAkB,YAAO,GAAP,OAAO,CAAU;QACjK,MAAM,qBAAqB,GAAG,IAAI,uBAAe,EAA+B,CAAC;QACjF,IAAI,CAAC,gBAAgB,GAAG,qBAAqB,CAAC,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE;YAC1D,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;gBAC3B,OAAO,WAAW,CAAC;YACpB,CAAC;YACD,IAAI,CAAC;gBACJ,OAAO,eAAM,CAAC,EAAE,CAAC,cAAc,CAAC,0BAA0B,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC;YAC9E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACd,OAAO,eAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,GAAG,qBAAqB,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,OAAO;QACnB,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IAC3C,CAAC;IAEO,MAAM,CAAC,0BAA0B,CAAC,MAAoB;QAC7D,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,YAAY,GAAmC,IAAI,CAAC;QACxD,IAAI,mBAAmB,GAA+B,IAAI,CAAC;QAC3D,IAAI,eAAe,GAAiC,SAAS,CAAC;QAE9D,KAAK,MAAM,UAAU,IAAI,MAAM,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,iDAAiD;YACvF,IAAI,IAAI,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YAC1B,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACrB,IAAI,mBAAmB,KAAK,IAAI,EAAE,CAAC;oBAClC,mBAAmB,GAAG;wBACrB,MAAM,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;wBACnC,cAAc,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC;wBACnD,WAAW,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC;wBAC7C,YAAY,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC;qBAC/C,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACP,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;oBAC3D,mBAAmB,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;oBAC3E,mBAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;oBACrE,mBAAmB,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBACxE,CAAC;YACF,CAAC;YACD,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACtB,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;oBACnC,eAAe,GAAG;wBACjB,iBAAiB,EAAE,UAAU,CAAC,KAAK,CAAC,iBAAiB;wBACrD,aAAa,EAAE,UAAU,CAAC,KAAK,CAAC,aAAa;wBAC7C,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,YAAY;wBAC3C,yBAAyB,EAAE;4BAC1B,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,yBAAyB,CAAC,YAAY;4BACrE,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,yBAAyB,CAAC,gBAAgB;yBAC7E;wBACD,qBAAqB,EAAE;4BACtB,YAAY,EAAE,UAAU,CAAC,KAAK,CAAC,qBAAqB,CAAC,YAAY;4BACjE,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,qBAAqB,CAAC,gBAAgB;yBACzE;qBACD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACP,eAAe,CAAC,iBAAiB,IAAI,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC;oBACxE,eAAe,CAAC,aAAa,IAAI,UAAU,CAAC,KAAK,CAAC,aAAa,CAAC;oBAChE,eAAe,CAAC,YAAY,IAAI,UAAU,CAAC,KAAK,CAAC,YAAY,CAAC;oBAC9D,eAAe,CAAC,yBAAyB,CAAC,YAAY,IAAI,UAAU,CAAC,KAAK,CAAC,yBAAyB,CAAC,YAAY,CAAC;oBAClH,eAAe,CAAC,yBAAyB,CAAC,gBAAgB,IAAI,UAAU,CAAC,KAAK,CAAC,yBAAyB,CAAC,gBAAgB,CAAC;oBAC1H,eAAe,CAAC,qBAAqB,CAAC,YAAY,IAAI,UAAU,CAAC,KAAK,CAAC,qBAAqB,CAAC,YAAY,CAAC;oBAC1G,eAAe,CAAC,qBAAqB,CAAC,gBAAgB,IAAI,UAAU,CAAC,KAAK,CAAC,qBAAqB,CAAC,gBAAgB,CAAC;gBACnH,CAAC;YACF,CAAC;YACD,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC1B,IAAA,kBAAU,EACT,YAAY,KAAK,IAAI,EACrB,sEAAsE,CACtE,CAAC;gBACF,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;YACrC,CAAC;QACF,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QAE7B,MAAM,MAAM,GAAsB;YACjC,KAAK,EAAE,CAAC;YACR,aAAa,EAAE,YAAY;YAC3B,QAAQ,EAAE,mBAAmB;YAC7B,IAAI;SACJ,CAAC;QAEF,MAAM,oBAAoB,GAAe;YACxC,OAAO,EAAE,CAAC,MAAM,CAAC;YACjB,kBAAkB,EAAE,UAAU,CAAC,kBAAkB;YACjD,MAAM,EAAE,UAAU,CAAC,MAAM;YACzB,KAAK,EAAE,eAAe;SACtB,CAAC;QAEF,OAAO,oBAAoB,CAAC;IAC7B,CAAC;CACD;AAhID,wCAgIC;AAED;;;GAGG;AACH,KAAK,SAAS,CAAC,CAAC,qBAAqB,CACpC,MAAiC,EACjC,eAA6D;IAE7D,MAAM,WAAW,GAAiB,EAAE,CAAC;IACrC,IAAI,KAAwB,CAAC;IAC7B,IAAI,CAAC;QACJ,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,MAAM,EAAE,CAAC;YACvC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC7B,MAAM,UAAU,CAAC;QAClB,CAAC;IACF,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACrB,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,KAAK,CAAC;IACb,CAAC;YAAS,CAAC;QACV,eAAe,CAAC,QAAQ,CACvB,KAAK,CAAC,CAAC,CAAC,eAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,eAAM,CAAC,EAAE,CAAC,WAAW,CAAC,CACpD,CAAC;IACH,CAAC;AACF,CAAC"}

package/dist/src/_internal/platform/nesFetch/node/completionsFetchServiceImpl.d.ts

@@ -0,0 +1,28 @@
+import { Result } from '../../../util/common/result';
+import { CancellationToken } from '../../../util/vs/base/common/cancellation';
+import { IAuthenticationService } from '../../authentication/common/authentication';
+import { RequestId } from '../../networking/common/fetch';
+import { IFetcherService, IHeaders, Response } from '../../networking/common/fetcherService';
+import { Completions, ICompletionsFetchService } from '../common/completionsFetchService';
+import { ResponseStream } from '../common/responseStream';
+export type FetchResponse = {
+    status: number;
+    statusText: string;
+    headers: IHeaders;
+    body: AsyncIterable<string>;
+    requestId: RequestId;
+    response: Response;
+};
+export interface IFetchRequestParams extends Completions.ModelParams {
+}
+export declare class CompletionsFetchService implements ICompletionsFetchService {
+    private authService;
+    private fetcherService;
+    readonly _serviceBrand: undefined;
+    constructor(authService: IAuthenticationService, fetcherService: IFetcherService);
+    disconnectAll(): Promise<unknown>;
+    fetch(url: string, secretKey: string, params: IFetchRequestParams, requestId: string, ct: CancellationToken, headerOverrides?: Record<string, string>): Promise<Result<ResponseStream, Completions.CompletionsFetchFailure>>;
+    protected _fetchFromUrl(url: string, options: Completions.Internal.FetchOptions, ct: CancellationToken): Promise<Result<FetchResponse, Completions.CompletionsFetchFailure>>;
+    private getHeaders;
+}
+//# sourceMappingURL=completionsFetchServiceImpl.d.ts.map

package/dist/src/_internal/platform/nesFetch/node/completionsFetchServiceImpl.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"completionsFetchServiceImpl.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/nesFetch/node/completionsFetchServiceImpl.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,MAAM,EAAE,MAAM,6BAA6B,CAAC;AACrD,OAAO,EAAE,iBAAiB,EAAE,MAAM,2CAA2C,CAAC;AAE9E,OAAO,EAAE,sBAAsB,EAAE,MAAM,4CAA4C,CAAC;AACpF,OAAO,EAAgB,SAAS,EAAE,MAAM,+BAA+B,CAAC;AACxE,OAAO,EAAgB,eAAe,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,wCAAwC,CAAC;AAC3G,OAAO,EAAE,WAAW,EAAE,wBAAwB,EAAE,MAAM,mCAAmC,CAAC;AAC1F,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAG1D,MAAM,MAAM,aAAa,GAAG;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,QAAQ,CAAC;IAClB,IAAI,EAAE,aAAa,CAAC,MAAM,CAAC,CAAC;IAC5B,SAAS,EAAE,SAAS,CAAC;IACrB,QAAQ,EAAE,QAAQ,CAAC;CACnB,CAAC;AAEF,MAAM,WAAW,mBAAoB,SAAQ,WAAW,CAAC,WAAW;CAAI;AAExE,qBAAa,uBAAwB,YAAW,wBAAwB;IAI9C,OAAO,CAAC,WAAW;IAC1B,OAAO,CAAC,cAAc;IAJxC,QAAQ,CAAC,aAAa,EAAE,SAAS,CAAC;gBAGD,WAAW,EAAE,sBAAsB,EAC1C,cAAc,EAAE,eAAe;IAIlD,aAAa,IAAI,OAAO,CAAC,OAAO,CAAC;IAI3B,KAAK,CACjB,GAAG,EAAE,MAAM,EACX,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,mBAAmB,EAC3B,SAAS,EAAE,MAAM,EACjB,EAAE,EAAE,iBAAiB,EACrB,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACtC,OAAO,CAAC,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,uBAAuB,CAAC,CAAC;cA0CvD,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC,aAAa,EAAE,WAAW,CAAC,uBAAuB,CAAC,CAAC;IA8DlL,OAAO,CAAC,UAAU;CAgBlB"}

package/dist/src/_internal/platform/nesFetch/node/completionsFetchServiceImpl.js

@@ -0,0 +1,184 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompletionsFetchService = void 0;
+const errors = __importStar(require("../../../util/common/errors"));
+const result_1 = require("../../../util/common/result");
+const authentication_1 = require("../../authentication/common/authentication");
+const fetch_1 = require("../../networking/common/fetch");
+const fetcherService_1 = require("../../networking/common/fetcherService");
+const completionsFetchService_1 = require("../common/completionsFetchService");
+const responseStream_1 = require("../common/responseStream");
+const streamTransformer_1 = require("./streamTransformer");
+let CompletionsFetchService = class CompletionsFetchService {
+    constructor(authService, fetcherService) {
+        this.authService = authService;
+        this.fetcherService = fetcherService;
+    }
+    disconnectAll() {
+        return this.fetcherService.disconnectAll();
+    }
+    async fetch(url, secretKey, params, requestId, ct, headerOverrides) {
+        if (ct.isCancellationRequested) {
+            return result_1.Result.error(new completionsFetchService_1.Completions.RequestCancelled());
+        }
+        const options = {
+            requestId,
+            headers: this.getHeaders(requestId, secretKey, headerOverrides),
+            body: JSON.stringify({
+                ...params,
+                stream: true,
+            })
+        };
+        const fetchResponse = await this._fetchFromUrl(url, options, ct);
+        if (fetchResponse.isError()) {
+            return fetchResponse;
+        }
+        if (fetchResponse.val.status === 200) {
+            const jsonlStream = (0, streamTransformer_1.streamToLines)(fetchResponse.val.body);
+            const completionsStream = (0, streamTransformer_1.jsonlStreamToCompletions)(jsonlStream);
+            const response = new responseStream_1.ResponseStream(fetchResponse.val.response, completionsStream, fetchResponse.val.requestId, fetchResponse.val.headers);
+            return result_1.Result.ok(response);
+        }
+        else {
+            const error = new completionsFetchService_1.Completions.UnsuccessfulResponse(fetchResponse.val.status, fetchResponse.val.statusText, fetchResponse.val.headers, () => collectAsyncIterableToString(fetchResponse.val.body).catch(() => ''));
+            return result_1.Result.error(error);
+        }
+    }
+    async _fetchFromUrl(url, options, ct) {
+        const fetchAbortCtl = this.fetcherService.makeAbortController();
+        const onCancellationDisposable = ct.onCancellationRequested(() => {
+            fetchAbortCtl.abort();
+        });
+        try {
+            const request = {
+                headers: options.headers,
+                body: options.body,
+                signal: fetchAbortCtl.signal,
+                method: 'POST',
+            };
+            const response = await this.fetcherService.fetch(url, request);
+            if (response.status === 200 && this.authService.copilotToken?.isFreeUser && this.authService.copilotToken?.isChatQuotaExceeded) {
+                this.authService.resetCopilotToken();
+            }
+            if (response.status !== 200) {
+                if (response.status === 402) {
+                    // When we receive a 402, we have exceed the free tier quota
+                    // This is stored on the token so let's refresh it
+                    if (!this.authService.copilotToken?.isCompletionsQuotaExceeded) {
+                        this.authService.resetCopilotToken(response.status);
+                        await this.authService.getCopilotToken();
+                    }
+                }
+                return result_1.Result.error(new completionsFetchService_1.Completions.UnsuccessfulResponse(response.status, response.statusText, response.headers, () => response.text().catch(() => '')));
+            }
+            const body = response.body.pipeThrough(new TextDecoderStream());
+            const responseStream = streamWithCleanup(body, onCancellationDisposable);
+            return result_1.Result.ok({
+                status: response.status,
+                statusText: response.statusText,
+                headers: response.headers,
+                body: responseStream,
+                requestId: (0, fetch_1.getRequestId)(response.headers),
+                response,
+            });
+        }
+        catch (reason) {
+            onCancellationDisposable.dispose();
+            if (reason instanceof Error && reason.message === 'This operation was aborted') {
+                return result_1.Result.error(new completionsFetchService_1.Completions.RequestCancelled());
+            }
+            const error = errors.fromUnknown(reason);
+            return result_1.Result.error(new completionsFetchService_1.Completions.Unexpected(error));
+        }
+    }
+    getHeaders(requestId, secretKey, headerOverrides = {}) {
+        const headers = {
+            'Content-Type': 'application/json',
+            'x-policy-id': 'nil',
+            Authorization: 'Bearer ' + secretKey,
+            'X-Request-Id': requestId,
+            'X-GitHub-Api-Version': '2025-04-01',
+            ...headerOverrides,
+        };
+        return headers;
+    }
+};
+exports.CompletionsFetchService = CompletionsFetchService;
+exports.CompletionsFetchService = CompletionsFetchService = __decorate([
+    __param(0, authentication_1.IAuthenticationService),
+    __param(1, fetcherService_1.IFetcherService)
+], CompletionsFetchService);
+/**
+ * Wraps an async iterable stream and disposes the cleanup disposable when the stream completes or errors.
+ */
+async function* streamWithCleanup(stream, cleanupDisposable) {
+    try {
+        for await (const str of stream) {
+            yield str;
+        }
+    }
+    catch (err) {
+        const error = errors.fromUnknown(err);
+        throw error;
+    }
+    finally {
+        cleanupDisposable.dispose();
+    }
+}
+/**
+ * Collects all strings from an async iterable and joins them into a single string.
+ */
+async function collectAsyncIterableToString(iterable) {
+    const parts = [];
+    for await (const part of iterable) {
+        parts.push(part);
+    }
+    return parts.join('');
+}
+//# sourceMappingURL=completionsFetchServiceImpl.js.map

package/dist/src/_internal/platform/nesFetch/node/completionsFetchServiceImpl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"completionsFetchServiceImpl.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/nesFetch/node/completionsFetchServiceImpl.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEhG,oEAAsD;AACtD,wDAAqD;AAGrD,+EAAoF;AACpF,yDAAwE;AACxE,2EAA2G;AAC3G,+EAA0F;AAC1F,6DAA0D;AAC1D,2DAA8E;AAavE,IAAM,uBAAuB,GAA7B,MAAM,uBAAuB;IAGnC,YACiC,WAAmC,EAC1C,cAA+B;QADxB,gBAAW,GAAX,WAAW,CAAwB;QAC1C,mBAAc,GAAd,cAAc,CAAiB;IAEzD,CAAC;IAEM,aAAa;QACnB,OAAO,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,CAAC;IAC5C,CAAC;IAEM,KAAK,CAAC,KAAK,CACjB,GAAW,EACX,SAAiB,EACjB,MAA2B,EAC3B,SAAiB,EACjB,EAAqB,EACrB,eAAwC;QAGxC,IAAI,EAAE,CAAC,uBAAuB,EAAE,CAAC;YAChC,OAAO,eAAM,CAAC,KAAK,CAAC,IAAI,qCAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,OAAO,GAAG;YACf,SAAS;YACT,OAAO,EAAE,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,SAAS,EAAE,eAAe,CAAC;YAC/D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACpB,GAAG,MAAM;gBACT,MAAM,EAAE,IAAI;aACZ,CAAC;SACF,CAAC;QAEF,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;QAEjE,IAAI,aAAa,CAAC,OAAO,EAAE,EAAE,CAAC;YAC7B,OAAO,aAAa,CAAC;QACtB,CAAC;QAED,IAAI,aAAa,CAAC,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAEtC,MAAM,WAAW,GAAG,IAAA,iCAAa,EAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YAC1D,MAAM,iBAAiB,GAAG,IAAA,4CAAwB,EAAC,WAAW,CAAC,CAAC;YAEhE,MAAM,QAAQ,GAAG,IAAI,+BAAc,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAE3I,OAAO,eAAM,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC;QAE5B,CAAC;aAAM,CAAC;YACP,MAAM,KAAK,GAAwC,IAAI,qCAAW,CAAC,oBAAoB,CACtF,aAAa,CAAC,GAAG,CAAC,MAAM,EACxB,aAAa,CAAC,GAAG,CAAC,UAAU,EAC5B,aAAa,CAAC,GAAG,CAAC,OAAO,EACzB,GAAG,EAAE,CAAC,4BAA4B,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAC1E,CAAC;YAEF,OAAO,eAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;IACF,CAAC;IAES,KAAK,CAAC,aAAa,CAAC,GAAW,EAAE,OAA0C,EAAE,EAAqB;QAE3G,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,mBAAmB,EAAE,CAAC;QAEhE,MAAM,wBAAwB,GAAG,EAAE,CAAC,uBAAuB,CAAC,GAAG,EAAE;YAChE,aAAa,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YAEJ,MAAM,OAAO,GAAiB;gBAC7B,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,MAAM,EAAE,aAAa,CAAC,MAAM;gBAC5B,MAAM,EAAE,MAAM;aACd,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAE/D,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,UAAU,IAAI,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,mBAAmB,EAAE,CAAC;gBAChI,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,CAAC;YACtC,CAAC;YAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC7B,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;oBAC7B,4DAA4D;oBAC5D,kDAAkD;oBAClD,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,0BAA0B,EAAE,CAAC;wBAChE,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;wBACpD,MAAM,IAAI,CAAC,WAAW,CAAC,eAAe,EAAE,CAAC;oBAC1C,CAAC;gBACF,CAAC;gBAED,OAAO,eAAM,CAAC,KAAK,CAAC,IAAI,qCAAW,CAAC,oBAAoB,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC1J,CAAC;YAED,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC,CAAC;YAEhE,MAAM,cAAc,GAAG,iBAAiB,CAAC,IAAI,EAAE,wBAAwB,CAAC,CAAC;YAEzE,OAAO,eAAM,CAAC,EAAE,CAAC;gBAChB,MAAM,EAAE,QAAQ,CAAC,MAAM;gBACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,IAAI,EAAE,cAAc;gBACpB,SAAS,EAAE,IAAA,oBAAY,EAAC,QAAQ,CAAC,OAAO,CAAC;gBACzC,QAAQ;aACR,CAAC,CAAC;QAEJ,CAAC;QAAC,OAAO,MAAe,EAAE,CAAC;YAE1B,wBAAwB,CAAC,OAAO,EAAE,CAAC;YAEnC,IAAI,MAAM,YAAY,KAAK,IAAI,MAAM,CAAC,OAAO,KAAK,4BAA4B,EAAE,CAAC;gBAChF,OAAO,eAAM,CAAC,KAAK,CAAC,IAAI,qCAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;YACzD,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YACzC,OAAO,eAAM,CAAC,KAAK,CAAC,IAAI,qCAAW,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;QACxD,CAAC;IACF,CAAC;IAEO,UAAU,CACjB,SAAiB,EACjB,SAAiB,EACjB,kBAA0C,EAAE;QAE5C,MAAM,OAAO,GAA2B;YACvC,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,KAAK;YACpB,aAAa,EAAE,SAAS,GAAG,SAAS;YACpC,cAAc,EAAE,SAAS;YACzB,sBAAsB,EAAE,YAAY;YACpC,GAAG,eAAe;SAClB,CAAC;QAEF,OAAO,OAAO,CAAC;IAChB,CAAC;CACD,CAAA;AA5IY,0DAAuB;kCAAvB,uBAAuB;IAIjC,WAAA,uCAAsB,CAAA;IACtB,WAAA,gCAAe,CAAA;GALL,uBAAuB,CA4InC;AAED;;GAEG;AACH,KAAK,SAAS,CAAC,CAAC,iBAAiB,CAChC,MAA6B,EAC7B,iBAA8B;IAE9B,IAAI,CAAC;QACJ,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAChC,MAAM,GAAG,CAAC;QACX,CAAC;IACF,CAAC;IAAC,OAAO,GAAY,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,KAAK,CAAC;IACb,CAAC;YAAS,CAAC;QACV,iBAAiB,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC;AACF,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,4BAA4B,CAAC,QAA+B;IAC1E,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClB,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACvB,CAAC"}

package/dist/src/_internal/platform/nesFetch/node/streamTransformer.d.ts

@@ -0,0 +1,9 @@
+import { Completion } from '../common/completionsAPI';
+/**
+ * Transforms a stream of strings into a stream of lines.
+ *
+ * Listener should handle the errors coming from the input stream.
+ */
+export declare function streamToLines(stream: AsyncIterable<string>): AsyncGenerator<string>;
+export declare function jsonlStreamToCompletions(jsonlStream: AsyncIterable<string>): AsyncGenerator<Completion>;
+//# sourceMappingURL=streamTransformer.d.ts.map

package/dist/src/_internal/platform/nesFetch/node/streamTransformer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamTransformer.d.ts","sourceRoot":"","sources":["../../../../../../src/_internal/platform/nesFetch/node/streamTransformer.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,UAAU,EAAE,MAAM,0BAA0B,CAAC;AAEtD;;;;GAIG;AACH,wBAAuB,aAAa,CAAC,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,cAAc,CAAC,MAAM,CAAC,CAkB1F;AAED,wBAAuB,wBAAwB,CAAC,WAAW,EAAE,aAAa,CAAC,MAAM,CAAC,GAAG,cAAc,CAAC,UAAU,CAAC,CAgB9G"}

package/dist/src/_internal/platform/nesFetch/node/streamTransformer.js

@@ -0,0 +1,71 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.streamToLines = streamToLines;
+exports.jsonlStreamToCompletions = jsonlStreamToCompletions;
+/**
+ * Transforms a stream of strings into a stream of lines.
+ *
+ * Listener should handle the errors coming from the input stream.
+ */
+async function* streamToLines(stream) {
+    let buffer = '';
+    for await (const str of stream) {
+        buffer += str;
+        let newlineIndex;
+        while ((newlineIndex = buffer.indexOf('\n')) !== -1) {
+            // take the first line
+            const line = buffer.substring(0, newlineIndex);
+            buffer = buffer.substring(newlineIndex + 1);
+            yield line;
+        }
+    }
+    if (buffer.length > 0) {
+        // last line which doesn't end with \n
+        yield buffer;
+    }
+}
+async function* jsonlStreamToCompletions(jsonlStream) {
+    for await (const line of jsonlStream) {
+        if (line.trim() === 'data: [DONE]') {
+            continue;
+        }
+        if (line.startsWith('data: ')) {
+            const message = JSON.parse(line.substring('data: '.length));
+            if (message.error) {
+                throw new Error(message.error.message);
+            }
+            yield message;
+        }
+    }
+}
+// function replaceBytes(s: string): string {
+// 	if (!s.startsWith('bytes:')) {
+// 		return s;
+// 	}
+// 	const bytes: number[] = [];
+// 	let i = 'bytes:'.length;
+// 	const textEncoder = new TextEncoder();
+// 	while (i < s.length) {
+// 		if (s.slice(i, i + 3) === '\\\\x') {
+// 			bytes.push(parseInt(s.slice(i + 3, i + 5), 16));
+// 			i += 5;
+// 		} else if (s.slice(i, i + 2) === '\\x') {
+// 			bytes.push(parseInt(s.slice(i + 2, i + 4), 16));
+// 			i += 4;
+// 		} else {
+// 			const encoded = textEncoder.encode(s.slice(i, i + 1));
+// 			for (const b of encoded) {
+// 				bytes.push(b);
+// 			}
+// 			i += 1;
+// 		}
+// 	}
+// 	return new TextDecoder('utf8', { fatal: false }).decode(
+// 		new Uint8Array(bytes)
+// 	);
+// }
+//# sourceMappingURL=streamTransformer.js.map

package/dist/src/_internal/platform/nesFetch/node/streamTransformer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamTransformer.js","sourceRoot":"","sources":["../../../../../../src/_internal/platform/nesFetch/node/streamTransformer.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;AAShG,sCAkBC;AAED,4DAgBC;AAzCD;;;;GAIG;AACI,KAAK,SAAS,CAAC,CAAC,aAAa,CAAC,MAA6B;IACjE,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,CAAC;QACd,IAAI,YAAoB,CAAC;QACzB,OAAO,CAAC,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACrD,sBAAsB;YACtB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;YAC/C,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC;YAC5C,MAAM,IAAI,CAAC;QACZ,CAAC;IACF,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,sCAAsC;QACtC,MAAM,MAAM,CAAC;IACd,CAAC;AACF,CAAC;AAEM,KAAK,SAAS,CAAC,CAAC,wBAAwB,CAAC,WAAkC;IACjF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QACtC,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,cAAc,EAAE,CAAC;YACpC,SAAS;QACV,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAiD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;YAE1G,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,OAAO,CAAC;QACf,CAAC;IACF,CAAC;AACF,CAAC;AAED,6CAA6C;AAC7C,kCAAkC;AAClC,cAAc;AACd,KAAK;AACL,+BAA+B;AAC/B,4BAA4B;AAC5B,0CAA0C;AAC1C,0BAA0B;AAC1B,yCAAyC;AACzC,sDAAsD;AACtD,aAAa;AACb,8CAA8C;AAC9C,sDAAsD;AACtD,aAAa;AACb,aAAa;AACb,4DAA4D;AAC5D,gCAAgC;AAChC,qBAAqB;AACrB,OAAO;AACP,aAAa;AACb,MAAM;AACN,KAAK;AACL,4DAA4D;AAC5D,0BAA0B;AAC1B,MAAM;AACN,IAAI"}

package/dist/src/_internal/platform/networking/common/anthropic.d.ts

@@ -0,0 +1,166 @@
+import { IConfigurationService } from '../../configuration/common/configurationService';
+import { IExperimentationService } from '../../telemetry/common/nullExperimentationService';
+import { IChatEndpoint } from './networking';
+/**
+ * Types for Anthropic Messages API
+ * Based on https://platform.claude.com/docs/en/api/messages
+ *
+ * This interface supports both regular tools and server tools (web search, tool search):
+ * - Regular tools: require name, description, and input_schema
+ * - Tool search tools: require only type and name
+ */
+export interface AnthropicMessagesTool {
+    name: string;
+    type?: string;
+    description?: string;
+    input_schema?: {
+        type: 'object';
+        properties?: Record<string, unknown>;
+        required?: string[];
+    };
+    defer_loading?: boolean;
+}
+export interface ToolReference {
+    type: 'tool_reference';
+    tool_name: string;
+}
+export interface ToolSearchToolSearchResult {
+    type: 'tool_search_tool_search_result';
+    tool_references: ToolReference[];
+}
+export interface ToolSearchToolResultError {
+    type: 'tool_search_tool_result_error';
+    error_code: 'too_many_requests' | 'invalid_pattern' | 'pattern_too_long' | 'unavailable';
+}
+export interface ServerToolUse {
+    type: 'server_tool_use';
+    id: string;
+    name: string;
+    input: {
+        query: string;
+    };
+}
+export interface ToolSearchToolResult {
+    type: 'tool_search_tool_result';
+    tool_use_id: string;
+    content: ToolSearchToolSearchResult | ToolSearchToolResultError;
+}
+export interface ToolSearchUsage {
+    tool_search_requests: number;
+}
+/**
+ * Tools that should not use deferred loading when tool search is enabled.
+ * These are frequently used tools that benefit from being immediately available.
+ *
+ * TODO: @bhavyaus Replace these hardcoded strings with constants from ToolName enum
+ */
+export declare const nonDeferredToolNames: Set<string>;
+/**
+ * Context management types for Anthropic Messages API
+ * Based on https://platform.claude.com/docs/en/build-with-claude/context-editing
+ */
+export type ContextManagementTrigger = {
+    type: 'input_tokens';
+    value: number;
+} | {
+    type: 'tool_uses';
+    value: number;
+};
+export type ContextManagementKeep = {
+    type: 'tool_uses';
+    value: number;
+} | {
+    type: 'thinking_turns';
+    value: number;
+} | 'all';
+export type ContextManagementClearAtLeast = {
+    type: 'input_tokens';
+    value: number;
+};
+export interface ClearToolUsesEdit {
+    type: 'clear_tool_uses_20250919';
+    trigger?: ContextManagementTrigger;
+    keep?: ContextManagementKeep;
+    clear_at_least?: ContextManagementClearAtLeast;
+    exclude_tools?: string[];
+    clear_tool_inputs?: boolean;
+}
+export interface ClearThinkingEdit {
+    type: 'clear_thinking_20251015';
+    keep?: ContextManagementKeep;
+}
+export type ContextManagementEdit = ClearToolUsesEdit | ClearThinkingEdit;
+export interface ContextManagement {
+    edits: ContextManagementEdit[];
+}
+export interface AppliedContextEdit {
+    type: 'clear_thinking_20251015' | 'clear_tool_uses_20250919';
+    cleared_thinking_turns?: number;
+    cleared_tool_uses?: number;
+    cleared_input_tokens?: number;
+}
+export interface ContextManagementResponse {
+    applied_edits: AppliedContextEdit[];
+}
+/**
+ * Context editing is supported by:
+ * - Claude Haiku 4.5 (claude-haiku-4-5-* or claude-haiku-4.5-*)
+ * - Claude Sonnet 4.5 (claude-sonnet-4-5-* or claude-sonnet-4.5-*)
+ * - Claude Sonnet 4 (claude-sonnet-4-*)
+ * - Claude Opus 4.5 (claude-opus-4-5-* or claude-opus-4.5-*)
+ * - Claude Opus 4.1 (claude-opus-4-1-* or claude-opus-4.1-*)
+ * - Claude Opus 4 (claude-opus-4-*)
+ * @param modelId The model ID to check
+ * @returns true if the model supports context editing
+ */
+export declare function modelSupportsContextEditing(modelId: string): boolean;
+/**
+ * Tool search is only supported by:
+ * - Claude Opus 4.5 (claude-opus-4-5-* or claude-opus-4.5-*)
+ * @param modelId The model ID to check
+ * @returns true if the model supports tool search
+ */
+export declare function modelSupportsToolSearch(modelId: string): boolean;
+/**
+ * Memory is supported by:
+ * - Claude Haiku 4.5 (claude-haiku-4-5-* or claude-haiku-4.5-*)
+ * - Claude Sonnet 4.5 (claude-sonnet-4-5-* or claude-sonnet-4.5-*)
+ * - Claude Sonnet 4 (claude-sonnet-4-*)
+ * - Claude Opus 4.5 (claude-opus-4-5-* or claude-opus-4.5-*)
+ * - Claude Opus 4.1 (claude-opus-4-1-* or claude-opus-4.1-*)
+ * - Claude Opus 4 (claude-opus-4-*)
+ * @param modelId The model ID to check
+ * @returns true if the model supports memory
+ */
+export declare function modelSupportsMemory(modelId: string): boolean;
+export declare function isAnthropicToolSearchEnabled(endpoint: IChatEndpoint | string, configurationService: IConfigurationService, experimentationService: IExperimentationService): boolean;
+export declare function isAnthropicContextEditingEnabled(endpoint: IChatEndpoint | string, configurationService: IConfigurationService, experimentationService: IExperimentationService): boolean;
+export interface ContextEditingConfig {
+    triggerType: 'input_tokens' | 'tool_uses';
+    triggerValue: number;
+    keepCount: number;
+    clearAtLeastTokens: number | undefined;
+    excludeTools: string[];
+    clearInputs: boolean;
+    thinkingKeepTurns: number;
+}
+/**
+ * Builds the context_management configuration object for the Messages API request.
+ * @param config The context editing configuration from individual settings
+ * @param thinkingEnabled Whether extended thinking is enabled
+ * @returns The context_management object to include in the request, or undefined if no edits
+ */
+export declare function buildContextManagement(config: ContextEditingConfig, thinkingEnabled: boolean): ContextManagement | undefined;
+/**
+ * Default values for context editing configuration.
+ */
+export declare const CONTEXT_EDITING_DEFAULTS: ContextEditingConfig;
+/**
+ * Reads context editing configuration from settings and builds the context_management object.
+ * This is a convenience function that combines reading configuration with buildContextManagement.
+ * @param configurationService The configuration service to read settings from
+ * @param thinkingEnabled Whether extended thinking is enabled
+ * @returns The context_management object to include in the request, or undefined if disabled
+ */
+export declare function getContextManagementFromConfig(configurationService: IConfigurationService, thinkingEnabled: boolean): ContextManagement | undefined;
+//# sourceMappingURL=anthropic.d.ts.map