agentlang 0.9.6 → 0.9.8

package/out/runtime/module.js

@@ -1,6 +1,6 @@
 import chalk from 'chalk';
-import { isRelNodes, isLiteral, } from '../language/generated/ast.js';
-import { Path, nameToPath, isString, isNumber, isBoolean, isFqName, makeFqName, DefaultModuleName, DefaultModules, joinStatements, isMinusZero, now, findMetaSchema, findAllPrePostTriggerSchema, asCrudType, isPath, findUqCompositeAttributes, escapeFqName, encryptPassword, splitFqName, splitRefs, forceAsFqName, validateIdFormat, nameContainsSepEscape, registerInitFunction, ScratchModuleName, } from './util.js';
+import { isLiteral, isRelNodes, } from '../language/generated/ast.js';
+import { asCrudType, DefaultModuleName, DefaultModules, encryptPassword, escapeFqName, escapeSpecialChars, findAllPrePostTriggerSchema, findMetaSchema, findUqCompositeAttributes, forceAsFqName, isBoolean, isFqName, isMinusZero, isNumber, isPath, isString, joinStatements, makeFqName, nameContainsSepEscape, nameToPath, now, Path, registerInitFunction, ScratchModuleName, splitFqName, splitRefs, validateIdFormat, } from './util.js';
 import { parseStatement } from '../language/parser.js';
 import { AdminSession } from './auth/defs.js';
 import { FetchModuleFn, getUserTenantId, PathAttributeName, TenantAttributeName, } from './defs.js';
@@ -393,7 +393,10 @@ export class Record extends ModuleEntry {
         }
     }
     getFullTextSearchAttributes() {
-        const fts = this.getMeta('fullTextSearch');
+        let fts = this.getMeta('fullTextSearchAttributes');
+        if (!fts) {
+            fts = this.getMeta('fullTextSearch');
+        }
         if (fts) {
             if (fts instanceof Array) {
                 return fts;
@@ -409,6 +412,13 @@ export class Record extends ModuleEntry {
             return undefined;
         }
     }
+    getEmbeddingConfig() {
+        const config = this.getMeta('embeddingConfig');
+        if (config && typeof config === 'object') {
+            return config;
+        }
+        return undefined;
+    }
     resetUserAttrs() {
         this.userAttrNames = undefined;
         this.userAttrsSchema = undefined;
@@ -954,7 +964,17 @@ export class Agent extends Record {
             if (!skip && value !== null && value !== undefined) {
                 let v = value;
                 const isf = key == 'flows';
-                if (isf || key == 'tools') {
+                const isDocs = key == 'documents';
+                if (isDocs) {
+                    const raw = isString(v) ? v : `${v}`;
+                    const items = raw
+                        .split(',')
+                        .map((entry) => entry.trim())
+                        .filter((entry) => entry.length > 0)
+                        .map((entry) => entry.replace(/\\/g, '\\\\').replace(/"/g, '\\"'));
+                    v = `[${items.map((entry) => `"${entry}"`).join(', ')}]`;
+                }
+                else if (isf || key == 'tools') {
                     if (isf || v.indexOf(',') > 0 || v.indexOf('/') > 0)
                         v = `[${v}]`;
                     else
@@ -1588,6 +1608,29 @@ export class GlossaryEntry extends ModuleEntry {
         return `glossaryEntry ${this.name} \n{\n${ss.join(',\n')}\n}`;
     }
 }
+export class DocumentEntry extends ModuleEntry {
+    constructor(title, moduleName, url) {
+        super(title, moduleName);
+        this.doc = { title, url };
+    }
+    getUrl() {
+        return this.doc.url;
+    }
+    getTitle() {
+        return this.doc.title;
+    }
+    toString() {
+        const escapedTitle = escapeSpecialChars(this.doc.title);
+        const escapedUrl = escapeSpecialChars(this.doc.url);
+        return `{${DefaultModuleName}.ai/doc {\n    title "${escapedTitle}",\n    url "${escapedUrl}"\n}}`;
+    }
+    get url() {
+        return this.doc.url;
+    }
+    set url(value) {
+        this.doc.url = value;
+    }
+}
 function statementLabel(stmt) {
     var _a, _b;
     if (stmt === undefined)
@@ -2095,6 +2138,43 @@ export class Module {
         }
         return this;
     }
+    addDocument(title, url) {
+        if (this.hasEntry(title)) {
+            const existing = this.getEntry(title);
+            if (existing instanceof DocumentEntry) {
+                const doc = existing;
+                doc.url = url;
+                return this;
+            }
+        }
+        const docEntry = new DocumentEntry(title, this.name, url);
+        this.addEntry(docEntry);
+        return this;
+    }
+    getDocument(title) {
+        if (this.hasEntry(title)) {
+            const entry = this.getEntry(title);
+            if (entry instanceof DocumentEntry) {
+                return entry.url;
+            }
+        }
+        return undefined;
+    }
+    getAllDocuments() {
+        return this.entries.filter((e) => {
+            return e instanceof DocumentEntry;
+        });
+    }
+    removeDocument(title) {
+        for (let i = 0; i < this.entries.length; ++i) {
+            const entry = this.entries[i];
+            if (entry.name === title && entry instanceof DocumentEntry) {
+                this.entries.splice(i, 1);
+                break;
+            }
+        }
+        return this;
+    }
     addRetry(retry) {
         this.addEntry(retry);
         return this;
@@ -2125,7 +2205,7 @@ export class Module {
     }
     getEntryIndex(entryName) {
         return this.entries.findIndex((v) => {
-            return v.name == entryName;
+            return v.name === entryName;
         });
     }
     hasEntry(entryName) {
@@ -2324,6 +2404,15 @@ export class Module {
     }
 }
 let GlobalRetries = undefined;
+const DocumentAliasMap = new Map();
+export function registerDocumentAlias(name, title) {
+    if (!name || !title)
+        return;
+    DocumentAliasMap.set(name, title);
+}
+export function resolveDocumentAliases(names) {
+    return names.map((name) => { var _a; return (_a = DocumentAliasMap.get(name)) !== null && _a !== void 0 ? _a : name; });
+}
 export function addGlobalRetry(r) {
     if (GlobalRetries === undefined) {
         GlobalRetries = new Array();
@@ -2372,8 +2461,6 @@ export function removeModule(name) {
     }
     return false;
 }
-addModule(DefaultModuleName);
-addRecord('env', DefaultModuleName);
 addModule(ScratchModuleName);
 export function getModuleNames() {
     const ks = getModuleDb().keys();