@mate-academy/llm-gateway 6.1.0 → 6.2.1

package/dist/services/fileStorage/InMemoryFileStorage.d.ts

@@ -0,0 +1,20 @@
+import { type LLMUploadFileMimeTypes } from '../../LLMService.typedefs';
+interface StoredFile {
+    fileId: string;
+    name: string;
+    path: string;
+    mimeType: LLMUploadFileMimeTypes;
+    content: string;
+}
+export declare class InMemoryFileStorage {
+    private files;
+    private filesByPath;
+    private storages;
+    addFile(path: string, name: string, mimeType: LLMUploadFileMimeTypes): StoredFile;
+    getFileByPath(path: string): StoredFile | null;
+    removeFile(fileId: string): void;
+    createStorage(fileIds: string[]): string;
+    removeStorage(storageId: string): void;
+    getStorageFiles(storageId: string): StoredFile[];
+}
+export {};

package/dist/services/fileStorage/InMemoryFileStorage.js

@@ -0,0 +1,67 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryFileStorage = void 0;
+const fs_1 = require("fs");
+const uuid_1 = require("uuid");
+class InMemoryFileStorage {
+    files = new Map();
+    filesByPath = new Map();
+    storages = new Map();
+    addFile(path, name, mimeType) {
+        const existing = this.filesByPath.get(path);
+        if (existing) {
+            return existing;
+        }
+        const content = (0, fs_1.readFileSync)(path, 'utf-8');
+        const fileId = (0, uuid_1.v4)();
+        const storedFile = {
+            fileId,
+            name,
+            path,
+            mimeType,
+            content,
+        };
+        this.files.set(fileId, storedFile);
+        this.filesByPath.set(path, storedFile);
+        return storedFile;
+    }
+    getFileByPath(path) {
+        return this.filesByPath.get(path) ?? null;
+    }
+    removeFile(fileId) {
+        const file = this.files.get(fileId);
+        if (!file) {
+            return;
+        }
+        this.filesByPath.delete(file.path);
+        this.files.delete(fileId);
+    }
+    createStorage(fileIds) {
+        const storageId = (0, uuid_1.v4)();
+        this.storages.set(storageId, {
+            storageId,
+            fileIds,
+        });
+        return storageId;
+    }
+    removeStorage(storageId) {
+        const group = this.storages.get(storageId);
+        if (group) {
+            for (const fileId of group.fileIds) {
+                this.removeFile(fileId);
+            }
+        }
+        this.storages.delete(storageId);
+    }
+    getStorageFiles(storageId) {
+        const group = this.storages.get(storageId);
+        if (!group) {
+            return [];
+        }
+        return group.fileIds
+            .map((fileId) => this.files.get(fileId))
+            .filter((file) => file !== undefined);
+    }
+}
+exports.InMemoryFileStorage = InMemoryFileStorage;
+//# sourceMappingURL=InMemoryFileStorage.js.map

package/dist/services/fileStorage/InMemoryFileStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"InMemoryFileStorage.js","sourceRoot":"","sources":["../../../src/services/fileStorage/InMemoryFileStorage.ts"],"names":[],"mappings":";;;AAAA,2BAAkC;AAClC,+BAAoC;AAgBpC,MAAa,mBAAmB;IACtB,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IACtC,WAAW,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC5C,QAAQ,GAAG,IAAI,GAAG,EAA4B,CAAC;IAEvD,OAAO,CACL,IAAY,EACZ,IAAY,EACZ,QAAgC;QAEhC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAE5C,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,OAAO,GAAG,IAAA,iBAAY,EAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAA,SAAM,GAAE,CAAC;QAExB,MAAM,UAAU,GAAe;YAC7B,MAAM;YACN,IAAI;YACJ,IAAI;YACJ,QAAQ;YACR,OAAO;SACR,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACnC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAEvC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,aAAa,CAAC,IAAY;QACxB,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC;IAC5C,CAAC;IAED,UAAU,CAAC,MAAc;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAEpC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAED,aAAa,CAAC,OAAiB;QAC7B,MAAM,SAAS,GAAG,IAAA,SAAM,GAAE,CAAC;QAE3B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE;YAC3B,SAAS;YACT,OAAO;SACR,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAE3C,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;gBACnC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,eAAe,CAAC,SAAiB;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAE3C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,KAAK,CAAC,OAAO;aACjB,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;aACvC,MAAM,CAAC,CAAC,IAAI,EAAsB,EAAE,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;IAC9D,CAAC;CACF;AAlFD,kDAkFC"}

package/dist/services/fileStorage/fileStorageTools.d.ts

@@ -0,0 +1,26 @@
+import { z } from 'zod';
+import type { ChatCompletionTool } from 'openai/resources/chat';
+export declare enum FileStorageToolName {
+    ListFiles = "list_files",
+    SearchFiles = "search_files",
+    ReadFile = "read_file",
+    ReadFileLines = "read_file_lines"
+}
+declare const searchFilesParameters: z.ZodObject<{
+    query: z.ZodString;
+    path: z.ZodOptional<z.ZodString>;
+    context_lines: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
+declare const readFileParameters: z.ZodObject<{
+    path: z.ZodString;
+}, z.core.$strip>;
+declare const readFileLinesParameters: z.ZodObject<{
+    path: z.ZodString;
+    start_line: z.ZodOptional<z.ZodNumber>;
+    end_line: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
+export type SearchFilesArgs = z.infer<typeof searchFilesParameters>;
+export type ReadFileArgs = z.infer<typeof readFileParameters>;
+export type ReadFileLinesArgs = z.infer<typeof readFileLinesParameters>;
+export declare function getFileStorageToolsForChatCompletions(): ChatCompletionTool[];
+export {};

package/dist/services/fileStorage/fileStorageTools.js

@@ -0,0 +1,68 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FileStorageToolName = void 0;
+exports.getFileStorageToolsForChatCompletions = getFileStorageToolsForChatCompletions;
+const zod_1 = require("zod");
+var FileStorageToolName;
+(function (FileStorageToolName) {
+    FileStorageToolName["ListFiles"] = "list_files";
+    FileStorageToolName["SearchFiles"] = "search_files";
+    FileStorageToolName["ReadFile"] = "read_file";
+    FileStorageToolName["ReadFileLines"] = "read_file_lines";
+})(FileStorageToolName || (exports.FileStorageToolName = FileStorageToolName = {}));
+const listFilesParameters = zod_1.z.object({});
+const searchFilesParameters = zod_1.z.object({
+    query: zod_1.z.string().describe('The search query string to find in file contents'),
+    path: zod_1.z.string().optional().describe('Optional file path filter to narrow the search scope'),
+    context_lines: zod_1.z.number().optional().describe('Number of context lines to show around each match (default: 3)'),
+});
+const readFileParameters = zod_1.z.object({
+    path: zod_1.z.string().describe('The path of the file to read'),
+});
+const readFileLinesParameters = zod_1.z.object({
+    path: zod_1.z.string().describe('The path of the file to read'),
+    start_line: zod_1.z.number().optional().describe('The 1-based starting line number to read from'),
+    end_line: zod_1.z.number().optional().describe('The 1-based ending line number to read to (inclusive). '
+        + 'Negative values read from end of file.'),
+});
+const FILE_STORAGE_TOOLS = [
+    {
+        name: FileStorageToolName.ListFiles,
+        description: 'List all files available in the file storage with their '
+            + 'names, paths, and line counts.',
+        parametersSchema: listFilesParameters,
+    },
+    {
+        name: FileStorageToolName.SearchFiles,
+        description: 'Search for a query string across all files in the storage. '
+            + 'Returns matching lines with surrounding context.',
+        parametersSchema: searchFilesParameters,
+    },
+    {
+        name: FileStorageToolName.ReadFile,
+        description: 'Read the entire contents of a file from the storage.',
+        parametersSchema: readFileParameters,
+    },
+    {
+        name: FileStorageToolName.ReadFileLines,
+        description: 'Read specific lines from a file in the storage. '
+            + 'Supports line ranges and reading from end of file.',
+        parametersSchema: readFileLinesParameters,
+    },
+];
+const FILE_STORAGE_CHAT_COMPLETION_TOOLS = FILE_STORAGE_TOOLS
+    .map((tool) => ({
+    type: 'function',
+    function: {
+        name: tool.name,
+        description: tool.description,
+        parameters: zod_1.z.toJSONSchema(tool.parametersSchema, {
+            target: 'draft-7',
+            reused: 'inline',
+        }),
+    },
+}));
+function getFileStorageToolsForChatCompletions() {
+    return FILE_STORAGE_CHAT_COMPLETION_TOOLS;
+}
+//# sourceMappingURL=fileStorageTools.js.map

package/dist/services/fileStorage/fileStorageTools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fileStorageTools.js","sourceRoot":"","sources":["../../../src/services/fileStorage/fileStorageTools.ts"],"names":[],"mappings":";;;AA0FA,sFAGC;AA7FD,6BAAwB;AAGxB,IAAY,mBAKX;AALD,WAAY,mBAAmB;IAC7B,+CAAwB,CAAA;IACxB,mDAA4B,CAAA;IAC5B,6CAAsB,CAAA;IACtB,wDAAiC,CAAA;AACnC,CAAC,EALW,mBAAmB,mCAAnB,mBAAmB,QAK9B;AAED,MAAM,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;AAEzC,MAAM,qBAAqB,GAAG,OAAC,CAAC,MAAM,CAAC;IACrC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CACxB,kDAAkD,CACnD;IACD,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAClC,sDAAsD,CACvD;IACD,aAAa,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAC3C,gEAAgE,CACjE;CACF,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC;IAClC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CACvB,8BAA8B,CAC/B;CACF,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CACvB,8BAA8B,CAC/B;IACD,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CACxC,+CAA+C,CAChD;IACD,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CACtC,yDAAyD;UACvD,wCAAwC,CAC3C;CACF,CAAC,CAAC;AAMH,MAAM,kBAAkB,GAAG;IACzB;QACE,IAAI,EAAE,mBAAmB,CAAC,SAAS;QACnC,WAAW,EAAE,0DAA0D;cACnE,gCAAgC;QACpC,gBAAgB,EAAE,mBAAmB;KACtC;IACD;QACE,IAAI,EAAE,mBAAmB,CAAC,WAAW;QACrC,WAAW,EAAE,6DAA6D;cACtE,kDAAkD;QACtD,gBAAgB,EAAE,qBAAqB;KACxC;IACD;QACE,IAAI,EAAE,mBAAmB,CAAC,QAAQ;QAClC,WAAW,EAAE,sDAAsD;QACnE,gBAAgB,EAAE,kBAAkB;KACrC;IACD;QACE,IAAI,EAAE,mBAAmB,CAAC,aAAa;QACvC,WAAW,EAAE,kDAAkD;cAC3D,oDAAoD;QACxD,gBAAgB,EAAE,uBAAuB;KAC1C;CAKA,CAAC;AAEJ,MAAM,kCAAkC,GAAyB,kBAAkB;KAChF,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACd,IAAI,EAAE,UAAmB;IACzB,QAAQ,EAAE;QACR,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU,EAAE,OAAC,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAChD,MAAM,EAAE,SAAS;YACjB,MAAM,EAAE,QAAQ;SACjB,CAAC;KACH;CACF,CAAC,CAAC,CAAC;AAEN,SAAgB,qCAAqC;IAEnD,OAAO,kCAAkC,CAAC;AAC5C,CAAC"}

package/dist/services/fileStorage/index.d.ts

@@ -0,0 +1,4 @@
+export { InMemoryFileStorage } from './InMemoryFileStorage';
+export { FileStorageToolExecutor } from './FileStorageToolExecutor';
+export { getFileStorageToolsForChatCompletions, FileStorageToolName, } from './fileStorageTools';
+export type { SearchFilesArgs, ReadFileArgs, ReadFileLinesArgs, } from './fileStorageTools';

package/dist/services/fileStorage/index.js

@@ -0,0 +1,11 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FileStorageToolName = exports.getFileStorageToolsForChatCompletions = exports.FileStorageToolExecutor = exports.InMemoryFileStorage = void 0;
+var InMemoryFileStorage_1 = require("./InMemoryFileStorage");
+Object.defineProperty(exports, "InMemoryFileStorage", { enumerable: true, get: function () { return InMemoryFileStorage_1.InMemoryFileStorage; } });
+var FileStorageToolExecutor_1 = require("./FileStorageToolExecutor");
+Object.defineProperty(exports, "FileStorageToolExecutor", { enumerable: true, get: function () { return FileStorageToolExecutor_1.FileStorageToolExecutor; } });
+var fileStorageTools_1 = require("./fileStorageTools");
+Object.defineProperty(exports, "getFileStorageToolsForChatCompletions", { enumerable: true, get: function () { return fileStorageTools_1.getFileStorageToolsForChatCompletions; } });
+Object.defineProperty(exports, "FileStorageToolName", { enumerable: true, get: function () { return fileStorageTools_1.FileStorageToolName; } });
+//# sourceMappingURL=index.js.map

package/dist/services/fileStorage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/services/fileStorage/index.ts"],"names":[],"mappings":";;;AAAA,6DAA4D;AAAnD,0HAAA,mBAAmB,OAAA;AAC5B,qEAAoE;AAA3D,kIAAA,uBAAuB,OAAA;AAChC,uDAG4B;AAF1B,yIAAA,qCAAqC,OAAA;AACrC,uHAAA,mBAAmB,OAAA"}

package/dist/utilities/index.d.ts

@@ -5,3 +5,4 @@ export * from './logger';
 export * from './reporter';
 export * from './schema';
 export * from './serializeError';
+export * from './tools';

package/dist/utilities/index.js

@@ -21,4 +21,5 @@ __exportStar(require("./logger"), exports);
 __exportStar(require("./reporter"), exports);
 __exportStar(require("./schema"), exports);
 __exportStar(require("./serializeError"), exports);
+__exportStar(require("./tools"), exports);
 //# sourceMappingURL=index.js.map

package/dist/utilities/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/utilities/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,kDAAgC;AAChC,mDAAiC;AACjC,2CAAyB;AACzB,6CAA2B;AAC3B,2CAAyB;AACzB,mDAAiC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/utilities/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,kDAAgC;AAChC,mDAAiC;AACjC,2CAAyB;AACzB,6CAA2B;AAC3B,2CAAyB;AACzB,mDAAiC;AACjC,0CAAwB"}

package/dist/utilities/tools/LLMTool.d.ts

@@ -0,0 +1,10 @@
+import { type LLMToolDefinition } from '../../LLMService.typedefs';
+import { type LLMSchemaInterface, type InferSchema } from '../../utilities/schema';
+export declare class LLMTool {
+    static create<Schema extends LLMSchemaInterface>(config: {
+        name: string;
+        description: string;
+        parameters: Schema;
+        execute: (args: InferSchema<Schema>) => Promise<string> | string;
+    }): LLMToolDefinition<Schema>;
+}

package/dist/utilities/tools/LLMTool.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LLMTool = void 0;
+class LLMTool {
+    static create(config) {
+        return config;
+    }
+}
+exports.LLMTool = LLMTool;
+//# sourceMappingURL=LLMTool.js.map

package/dist/utilities/tools/LLMTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"LLMTool.js","sourceRoot":"","sources":["../../../src/utilities/tools/LLMTool.ts"],"names":[],"mappings":";;;AAQA,MAAa,OAAO;IAClB,MAAM,CAAC,MAAM,CAAoC,MAKhD;QACC,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AATD,0BASC"}

package/dist/utilities/tools/index.d.ts

@@ -0,0 +1 @@
+export { LLMTool } from './LLMTool';

package/dist/utilities/tools/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LLMTool = void 0;
+var LLMTool_1 = require("./LLMTool");
+Object.defineProperty(exports, "LLMTool", { enumerable: true, get: function () { return LLMTool_1.LLMTool; } });
+//# sourceMappingURL=index.js.map

package/dist/utilities/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/utilities/tools/index.ts"],"names":[],"mappings":";;;AAAA,qCAAoC;AAA3B,kGAAA,OAAO,OAAA"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mate-academy/llm-gateway",
-  "version": "6.1.0",
+  "version": "6.2.1",
   "description": "A gateway package for LLM services.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",