@rxap/plugin-llm 20.0.0-dev.1

package/src/lib/assert-token-limit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"assert-token-limit.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/assert-token-limit.ts"],"names":[],"mappings":";;AAMA,4CAUC;AAhBD,iDAA6C;AAC7C,mCAGiB;AAEjB,SAAgB,gBAAgB,CAAC,KAAY,EAAE,GAAG,MAAgB;IAEhE,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,IAAA,0BAAW,EAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;IAE9E,IAAI,GAAG,GAAG,mBAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,kBAAkB,GAAG,yCAAyC,mBAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAC/G,CAAC;IAED,OAAO,GAAG,CAAC;AAEb,CAAC"}

package/src/lib/cleanup-js-doc.d.ts

@@ -0,0 +1 @@
+export declare function cleanupJsDoc(jsDoc: string): string;

package/src/lib/cleanup-js-doc.js

@@ -0,0 +1,26 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.cleanupJsDoc = cleanupJsDoc;
+function cleanupJsDoc(jsDoc) {
+    if (!(jsDoc.includes('/**') || jsDoc.includes('*/'))) {
+        return jsDoc;
+    }
+    const cleanJsDocArray = jsDoc
+        .split('\n')
+        .map(line => line
+        .trim()
+        .replace(/^\/\*\*/, '')
+        .replace(/^\*\//, '')
+        .replace(/^\*\s/, '')
+        .replace(/^\*$/, ''));
+    // cleanJsDocArray.pop();
+    // cleanJsDocArray.shift();
+    if (cleanJsDocArray[0].trim() === '') {
+        cleanJsDocArray.shift();
+    }
+    if (cleanJsDocArray[cleanJsDocArray.length - 1].trim() === '') {
+        cleanJsDocArray.pop();
+    }
+    return cleanJsDocArray.join('\n');
+}
+//# sourceMappingURL=cleanup-js-doc.js.map

package/src/lib/cleanup-js-doc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cleanup-js-doc.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/cleanup-js-doc.ts"],"names":[],"mappings":";;AAAA,oCA+BC;AA/BD,SAAgB,YAAY,CAAC,KAAa;IAExC,IAAI,CAAC,CACH,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAC9C,EAAE,CAAC;QACF,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,eAAe,GAAG,KAAK;SAC1B,KAAK,CAAC,IAAI,CAAC;SACX,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI;SACd,IAAI,EAAE;SACN,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;SACpB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;SACpB,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CACrB,CAAC;IAEJ,yBAAyB;IACzB,2BAA2B;IAE3B,IAAI,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACrC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAED,IAAI,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QAC9D,eAAe,CAAC,GAAG,EAAE,CAAC;IACxB,CAAC;IAED,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAEpC,CAAC"}

package/src/lib/clear-all-js-docs.d.ts

@@ -0,0 +1,2 @@
+import { SourceFile } from 'ts-morph';
+export declare function clearAllJsDocs(sourceFile: SourceFile, inplace?: boolean): SourceFile;

package/src/lib/clear-all-js-docs.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.clearAllJsDocs = clearAllJsDocs;
+const ts_morph_1 = require("ts-morph");
+const clear_js_doc_1 = require("./clear-js-doc");
+function clearAllJsDocs(sourceFile, inplace = false) {
+    const p = new ts_morph_1.Project({ useInMemoryFileSystem: true });
+    const clone = inplace ? sourceFile : p.createSourceFile('clean.ts', sourceFile.getFullText());
+    clone.getClasses().forEach(classD => {
+        (0, clear_js_doc_1.clearJsDoc)(classD);
+        classD.getProperties().filter(property => !property.getScope() || property.getScope() === ts_morph_1.Scope.Public).forEach(prop => (0, clear_js_doc_1.clearJsDoc)(prop));
+        classD.getMethods().filter(method => !method.getScope() || method.getScope() === ts_morph_1.Scope.Public).forEach(method => (0, clear_js_doc_1.clearJsDoc)(method));
+    });
+    clone.getInterfaces().forEach(interfaceD => {
+        (0, clear_js_doc_1.clearJsDoc)(interfaceD);
+        interfaceD.getProperties().forEach(prop => (0, clear_js_doc_1.clearJsDoc)(prop));
+        interfaceD.getMethods().forEach(method => (0, clear_js_doc_1.clearJsDoc)(method));
+    });
+    clone.getTypeAliases().forEach(alias => (0, clear_js_doc_1.clearJsDoc)(alias));
+    clone.getFunctions().forEach(funcD => (0, clear_js_doc_1.clearJsDoc)(funcD));
+    return clone;
+}
+//# sourceMappingURL=clear-all-js-docs.js.map

package/src/lib/clear-all-js-docs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"clear-all-js-docs.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/clear-all-js-docs.ts"],"names":[],"mappings":";;AAGA,wCAkBC;AArBD,uCAAsD;AACtD,iDAA4C;AAE5C,SAAgB,cAAc,CAAC,UAAsB,EAAE,OAAO,GAAG,KAAK;IACpE,MAAM,CAAC,GAAG,IAAI,kBAAO,CAAC,EAAE,qBAAqB,EAAE,IAAI,EAAE,CAAC,CAAC;IACvD,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,UAAU,EAAE,UAAU,CAAC,WAAW,EAAE,CAAC,CAAC;IAC9F,KAAK,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;QAClC,IAAA,yBAAU,EAAC,MAAM,CAAC,CAAC;QACnB,MAAM,CAAC,aAAa,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,QAAQ,CAAC,QAAQ,EAAE,KAAK,gBAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAC7G,IAAI,CAAC,EAAE,CAAC,IAAA,yBAAU,EAAC,IAAI,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,MAAM,CAAC,QAAQ,EAAE,KAAK,gBAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CACpG,MAAM,CAAC,EAAE,CAAC,IAAA,yBAAU,EAAC,MAAM,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;QACzC,IAAA,yBAAU,EAAC,UAAU,CAAC,CAAC;QACvB,UAAU,CAAC,aAAa,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAA,yBAAU,EAAC,IAAI,CAAC,CAAC,CAAC;QAC7D,UAAU,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,IAAA,yBAAU,EAAC,MAAM,CAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,cAAc,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,IAAA,yBAAU,EAAC,KAAK,CAAC,CAAC,CAAC;IAC3D,KAAK,CAAC,YAAY,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,IAAA,yBAAU,EAAC,KAAK,CAAC,CAAC,CAAC;IACzD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/src/lib/clear-js-doc.d.ts

@@ -0,0 +1,2 @@
+import { JSDocableNode } from 'ts-morph';
+export declare function clearJsDoc(node: JSDocableNode): void;

package/src/lib/clear-js-doc.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.clearJsDoc = clearJsDoc;
+function clearJsDoc(node) {
+    node.getJsDocs().forEach(item => item.remove());
+}
+//# sourceMappingURL=clear-js-doc.js.map

package/src/lib/clear-js-doc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"clear-js-doc.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/clear-js-doc.ts"],"names":[],"mappings":";;AAEA,gCAEC;AAFD,SAAgB,UAAU,CAAC,IAAmB;IAC5C,IAAI,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAClD,CAAC"}

package/src/lib/completion.d.ts

@@ -0,0 +1,11 @@
+import { Model } from './model';
+import 'colors';
+export interface CompletionOptions {
+    model?: Model;
+    maxTokens?: number;
+    baseUrl?: string;
+    apiKey?: string;
+    orgId?: string;
+    projectId?: string;
+}
+export declare function completion(systemPrompt: string, prompt: string, options?: CompletionOptions): Promise<string>;

package/src/lib/completion.js

@@ -0,0 +1,49 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.completion = completion;
+const tslib_1 = require("tslib");
+const openai_1 = require("openai");
+const assert_token_limit_1 = require("./assert-token-limit");
+const model_1 = require("./model");
+require("colors");
+function completion(systemPrompt_1, prompt_1) {
+    return tslib_1.__awaiter(this, arguments, void 0, function* (systemPrompt, prompt, options = {}) {
+        var _a;
+        const { model = 'openai/gpt-4o-mini', maxTokens = model_1.tokenLimits[model][1] } = options;
+        const count = (0, assert_token_limit_1.assertTokenLimit)(model, systemPrompt, prompt);
+        const input = {
+            model,
+            messages: [
+                {
+                    'role': 'system',
+                    'content': systemPrompt,
+                },
+                {
+                    'role': 'user',
+                    'content': prompt,
+                },
+            ],
+            temperature: 0,
+        };
+        if (model.startsWith('o')) {
+            input.max_completion_tokens = maxTokens;
+        }
+        else {
+            input.max_tokens = maxTokens;
+        }
+        console.log('TOKENS'.grey + ' ' + count);
+        const response = yield new openai_1.default({
+            apiKey: options.apiKey,
+            project: options.projectId,
+            organization: options.orgId,
+            baseURL: options.baseUrl,
+        }).chat.completions.create(input);
+        const content = (_a = response.choices[0].message) === null || _a === void 0 ? void 0 : _a.content;
+        if (!content) {
+            console.log('No content in response'.red);
+            throw new Error(`No content in response`);
+        }
+        return content;
+    });
+}
+//# sourceMappingURL=completion.js.map

package/src/lib/completion.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"completion.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/completion.ts"],"names":[],"mappings":";;AAmBA,gCAgDC;;AAlED,mCAA4B;AAE5B,6DAAwD;AACxD,mCAGiB;AACjB,kBAAgB;AAWhB,SAAsB,UAAU;iEAC9B,YAAoB,EACpB,MAAc,EACd,UAA6B,EAAE;;QAG/B,MAAM,EAAE,KAAK,GAAG,oBAAoB,EAAE,SAAS,GAAG,mBAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC;QAEpF,MAAM,KAAK,GAAG,IAAA,qCAAgB,EAAC,KAAK,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAE5D,MAAM,KAAK,GAA2C;YACpD,KAAK;YACL,QAAQ,EAAE;gBACR;oBACE,MAAM,EAAE,QAAQ;oBAChB,SAAS,EAAE,YAAY;iBACxB;gBACD;oBACE,MAAM,EAAE,MAAM;oBACd,SAAS,EAAE,MAAM;iBAClB;aACF;YACD,WAAW,EAAE,CAAC;SACf,CAAC;QAEF,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,KAAK,CAAC,qBAAqB,GAAG,SAAS,CAAC;QAC1C,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,UAAU,GAAG,SAAS,CAAC;QAC/B,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,GAAG,GAAG,GAAG,KAAK,CAAC,CAAC;QACzC,MAAM,QAAQ,GAAG,MAAM,IAAI,gBAAM,CAAC;YAChC,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,OAAO,EAAE,OAAO,CAAC,SAAS;YAC1B,YAAY,EAAE,OAAO,CAAC,KAAK;YAC3B,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAElC,MAAM,OAAO,GAAG,MAAA,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,0CAAE,OAAO,CAAC;QAErD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAC;YAC1C,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAC5C,CAAC;QAED,OAAO,OAAO,CAAC;IAEjB,CAAC;CAAA"}

package/src/lib/compose-context.d.ts

@@ -0,0 +1,9 @@
+import { SourceFile } from 'ts-morph';
+export interface ContextSourceFile {
+    $: {
+        path: string;
+    };
+    _: string;
+}
+export declare function sourceFileToContextSourceFile(sourceFile: SourceFile): ContextSourceFile;
+export declare function composeContext(sourceFile: SourceFile): string;

package/src/lib/compose-context.js

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sourceFileToContextSourceFile = sourceFileToContextSourceFile;
+exports.composeContext = composeContext;
+const xml2js_1 = require("xml2js");
+const clear_all_js_docs_1 = require("./clear-all-js-docs");
+function sourceFileToContextSourceFile(sourceFile) {
+    return {
+        $: {
+            path: sourceFile.getFilePath(),
+        },
+        _: '\n' + (0, clear_all_js_docs_1.clearAllJsDocs)(sourceFile).getFullText()
+    };
+}
+function getReferenced(sourceFile) {
+    return [
+        ...sourceFile.getReferencedSourceFiles(),
+        ...sourceFile.getReferencedSourceFiles().map(sf => getReferenced(sf)).flat()
+    ];
+}
+function composeContext(sourceFile) {
+    const builder = new xml2js_1.Builder();
+    const context = {
+        referenced: { sourceFile: [] },
+        used: { sourceFile: [] },
+    };
+    sourceFile.getReferencingSourceFiles().forEach(sf => {
+        context.used.sourceFile.push(sourceFileToContextSourceFile(sf));
+    });
+    getReferenced(sourceFile).forEach(sf => {
+        context.referenced.sourceFile.push(sourceFileToContextSourceFile(sf));
+    });
+    return builder.buildObject(context).replace(/root>/g, 'context>') + '\n';
+}
+//# sourceMappingURL=compose-context.js.map

package/src/lib/compose-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compose-context.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/compose-context.ts"],"names":[],"mappings":";;AAoBA,sEAOC;AASD,wCAmBC;AAtDD,mCAAiC;AACjC,2DAAqD;AAkBrD,SAAgB,6BAA6B,CAAC,UAAsB;IAClE,OAAO;QACL,CAAC,EAAE;YACD,IAAI,EAAE,UAAU,CAAC,WAAW,EAAE;SAC/B;QACD,CAAC,EAAE,IAAI,GAAG,IAAA,kCAAc,EAAC,UAAU,CAAC,CAAC,WAAW,EAAE;KACnD,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,UAAsB;IAC3C,OAAO;QACL,GAAG,UAAU,CAAC,wBAAwB,EAAE;QACxC,GAAG,UAAU,CAAC,wBAAwB,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;KAC7E,CAAC;AACJ,CAAC;AAED,SAAgB,cAAc,CAAC,UAAsB;IAEnD,MAAM,OAAO,GAAG,IAAI,gBAAO,EAAE,CAAC;IAE9B,MAAM,OAAO,GAAY;QACvB,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;QAC9B,IAAI,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE;KACzB,CAAC;IAEF,UAAU,CAAC,yBAAyB,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;QAClD,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,aAAa,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE;QACrC,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,UAAU,CAAC,GAAG,IAAI,CAAC;AAE3E,CAAC"}

package/src/lib/count-tokens.d.ts

@@ -0,0 +1,2 @@
+import { Model } from './model';
+export declare function countTokens(model: Model, prompt: string): number;

package/src/lib/count-tokens.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.countTokens = countTokens;
+const tiktoken_1 = require("tiktoken");
+function countTokens(model, prompt) {
+    let normalizedModel;
+    if (model.startsWith('openai/')) {
+        normalizedModel = model.replace(/openai\//, '');
+    }
+    else {
+        normalizedModel = 'gpt-4o';
+    }
+    const enc = (0, tiktoken_1.encoding_for_model)(normalizedModel);
+    const count = enc.encode(prompt).length;
+    enc.free();
+    return count;
+}
+//# sourceMappingURL=count-tokens.js.map

package/src/lib/count-tokens.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"count-tokens.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/count-tokens.ts"],"names":[],"mappings":";;AAMA,kCAcC;AApBD,uCAGkB;AAGlB,SAAgB,WAAW,CAAC,KAAY,EAAE,MAAc;IAEtD,IAAI,eAA8B,CAAC;IACnC,IAAI,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAChC,eAAe,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,EAAE,EAAE,CAAkB,CAAC;IACnE,CAAC;SAAM,CAAC;QACN,eAAe,GAAG,QAAQ,CAAC;IAC7B,CAAC;IACD,MAAM,GAAG,GAAG,IAAA,6BAAkB,EAAC,eAAe,CAAC,CAAC;IAChD,MAAM,KAAK,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;IACxC,GAAG,CAAC,IAAI,EAAE,CAAC;IAEX,OAAO,KAAK,CAAC;AAEf,CAAC"}

package/src/lib/has-js-doc.d.ts

@@ -0,0 +1,2 @@
+import { JSDocableNode } from 'ts-morph';
+export declare function hasJsDoc(node: JSDocableNode): boolean;

package/src/lib/has-js-doc.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.hasJsDoc = hasJsDoc;
+function hasJsDoc(node) {
+    return node.getJsDocs().length > 0;
+}
+//# sourceMappingURL=has-js-doc.js.map

package/src/lib/has-js-doc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"has-js-doc.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/has-js-doc.ts"],"names":[],"mappings":";;AAEA,4BAEC;AAFD,SAAgB,QAAQ,CAAC,IAAmB;IAC1C,OAAO,IAAI,CAAC,SAAS,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;AACrC,CAAC"}

package/src/lib/model.d.ts

@@ -0,0 +1,16 @@
+/**
+ * Model Context size https://platform.openai.com/docs/models
+ */
+export declare const tokenLimits: {
+    'anthropic/claude-3-5-sonnet': number[];
+    'anthropic/claude-3-5-haiku': number[];
+    'gemini/2.0-flash': number[];
+    'gemini/1.5-flash': number[];
+    'gemini/1.5-pro': number[];
+    'openai/gpt-4o': number[];
+    'openai/gpt-4o-mini': number[];
+    'openai/o1': number[];
+    'openai/o1-mini': number[];
+    'openai/o3-mini': number[];
+};
+export type Model = keyof typeof tokenLimits;

package/src/lib/model.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.tokenLimits = void 0;
+/**
+ * Model Context size https://platform.openai.com/docs/models
+ */
+exports.tokenLimits = {
+    'anthropic/claude-3-5-sonnet': [200000, 8192],
+    'anthropic/claude-3-5-haiku': [200000, 8192],
+    'gemini/2.0-flash': [1000000, 1000000], // $0.10 $0.40
+    'gemini/1.5-flash': [128000, 128000], // $0.075 $0.30
+    'gemini/1.5-pro': [128000, 128000], // $1.25 $5.00
+    'openai/gpt-4o': [128000], // $2.50 $10.00
+    'openai/gpt-4o-mini': [128000], // $0.15 $0.60
+    'openai/o1': [200000], // $15.00 $60.00
+    'openai/o1-mini': [128000], // $1.10 $4.40
+    'openai/o3-mini': [200000], // $1.10 $4.40
+};
+//# sourceMappingURL=model.js.map

package/src/lib/model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.js","sourceRoot":"","sources":["../../../../../../packages/plugin/llm/src/lib/model.ts"],"names":[],"mappings":";;;AAAA;;GAEG;AACU,QAAA,WAAW,GAAG;IAEzB,6BAA6B,EAAE,CAAC,MAAO,EAAE,IAAI,CAAC;IAC9C,4BAA4B,EAAE,CAAC,MAAO,EAAE,IAAI,CAAC;IAE7C,kBAAkB,EAAE,CAAC,OAAS,EAAE,OAAS,CAAC,EAAE,cAAc;IAC1D,kBAAkB,EAAE,CAAC,MAAO,EAAC,MAAO,CAAC,EAAE,eAAe;IACtD,gBAAgB,EAAE,CAAC,MAAO,EAAC,MAAO,CAAC,EAAE,cAAc;IAEnD,eAAe,EAAE,CAAC,MAAO,CAAC,EAAE,eAAe;IAC3C,oBAAoB,EAAE,CAAC,MAAO,CAAC,EAAE,cAAc;IAC/C,WAAW,EAAE,CAAC,MAAO,CAAC,EAAE,gBAAgB;IACxC,gBAAgB,EAAE,CAAC,MAAO,CAAC,EAAE,cAAc;IAC3C,gBAAgB,EAAE,CAAC,MAAO,CAAC,EAAE,cAAc;CAE5C,CAAC"}