@promptbook/cli 0.92.0-4 → 0.92.0-6

package/esm/index.es.js

@@ -46,7 +46,7 @@ const BOOK_LANGUAGE_VERSION = '1.0.0';
  * @generated
  * @see https://github.com/webgptorg/promptbook
  */
-const PROMPTBOOK_ENGINE_VERSION = '0.92.0-4';
+const PROMPTBOOK_ENGINE_VERSION = '0.92.0-6';
 /**
  * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
  * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -1004,6 +1004,45 @@ function $sideEffect(...sideEffectSubjects) {
     keepUnused(...sideEffectSubjects);
 }
 
+/**
+ * Converts a JavaScript Object Notation (JSON) string into an object.
+ *
+ * Note: This is wrapper around `JSON.parse()` with better error and type handling
+ *
+ * @public exported from `@promptbook/utils`
+ */
+function jsonParse(value) {
+    if (value === undefined) {
+        throw new Error(`Can not parse JSON from undefined value.`);
+    }
+    else if (typeof value !== 'string') {
+        console.error('Can not parse JSON from non-string value.', { text: value });
+        throw new Error(spaceTrim(`
+                    Can not parse JSON from non-string value.
+
+                    The value type: ${typeof value}
+                    See more in console.
+                `));
+    }
+    try {
+        return JSON.parse(value);
+    }
+    catch (error) {
+        if (!(error instanceof Error)) {
+            throw error;
+        }
+        throw new Error(spaceTrim((block) => `
+                    ${block(error.message)}
+            
+                    The JSON text:
+                    ${block(value)}
+                `));
+    }
+}
+/**
+ * TODO: !!!! Use in Promptbook.studio
+ */
+
 /**
  * Convert identification to Promptbook token
  *
@@ -2085,7 +2124,7 @@ class FileCacheStorage {
             return null;
         }
         const fileContent = await readFile(filename, 'utf-8');
-        const value = JSON.parse(fileContent);
+        const value = jsonParse(fileContent);
         // TODO: [🌗]
         return value;
     }
@@ -3420,7 +3459,7 @@ async function $provideLlmToolsForCli(options) {
                         password,
                     }),
                 });
-                const { isSuccess, message, error, identification } = (await response.json());
+                const { isSuccess, message, error, identification } = jsonParse(await response.text());
                 if (message) {
                     if (isSuccess) {
                         console.log(colors.green(message));
@@ -4451,7 +4490,7 @@ async function loadArchive(filePath, fs) {
     if (!indexFile) {
         throw new UnexpectedError(`Archive does not contain 'index.book.json' file`);
     }
-    const collectionJson = JSON.parse(await indexFile.async('text'));
+    const collectionJson = jsonParse(await indexFile.async('text'));
     for (const pipeline of collectionJson) {
         validatePipeline(pipeline);
     }
@@ -4922,7 +4961,7 @@ function jsonStringsToJsons(object) {
     const newObject = { ...object };
     for (const [key, value] of Object.entries(object)) {
         if (typeof value === 'string' && isValidJsonString(value)) {
-            newObject[key] = JSON.parse(value);
+            newObject[key] = jsonParse(value);
         }
         else {
             newObject[key] = jsonStringsToJsons(value);
@@ -6477,13 +6516,79 @@ async function getExamplesForTask(task) {
 /**
  * @@@
  *
+ * Here is the place where RAG (retrieval-augmented generation) happens
+ *
  * @private internal utility of `createPipelineExecutor`
  */
 async function getKnowledgeForTask(options) {
-    const { preparedPipeline, task } = options;
-    return preparedPipeline.knowledgePieces.map(({ content }) => `- ${content}`).join('\n');
+    const { tools, preparedPipeline, task } = options;
+    const firstKnowlegePiece = preparedPipeline.knowledgePieces[0];
+    const firstKnowlegeIndex = firstKnowlegePiece === null || firstKnowlegePiece === void 0 ? void 0 : firstKnowlegePiece.index[0];
+    // <- TODO: Do not use just first knowledge piece and first index to determine embedding model, use also keyword search
+    if (firstKnowlegePiece === undefined || firstKnowlegeIndex === undefined) {
+        return 'No knowledge pieces found';
+    }
+    // TODO: [🚐] Make arrayable LLMs -> single LLM DRY
+    const _llms = arrayableToArray(tools.llm);
+    const llmTools = _llms.length === 1 ? _llms[0] : joinLlmExecutionTools(..._llms);
+    const taskEmbeddingPrompt = {
+        title: 'Knowledge Search',
+        modelRequirements: {
+            modelVariant: 'EMBEDDING',
+            modelName: firstKnowlegeIndex.modelName,
+        },
+        content: task.content,
+        parameters: {
+        /* !!!!!!!! */
+        },
+    };
+    const taskEmbeddingResult = await llmTools.callEmbeddingModel(taskEmbeddingPrompt);
+    const knowledgePiecesWithRelevance = preparedPipeline.knowledgePieces.map((knowledgePiece) => {
+        const { index } = knowledgePiece;
+        const knowledgePieceIndex = index.find((i) => i.modelName === firstKnowlegeIndex.modelName);
+        // <- TODO: Do not use just first knowledge piece and first index to determine embedding model
+        if (knowledgePieceIndex === undefined) {
+            return {
+                content: knowledgePiece.content,
+                relevance: 0,
+            };
+        }
+        const relevance = computeCosineSimilarity(knowledgePieceIndex.position, taskEmbeddingResult.content);
+        return {
+            content: knowledgePiece.content,
+            relevance,
+        };
+    });
+    const knowledgePiecesSorted = knowledgePiecesWithRelevance.sort((a, b) => a.relevance - b.relevance);
+    const knowledgePiecesLimited = knowledgePiecesSorted.slice(0, 5);
+    console.log('!!! Embedding', {
+        task,
+        taskEmbeddingPrompt,
+        taskEmbeddingResult,
+        firstKnowlegePiece,
+        firstKnowlegeIndex,
+        knowledgePiecesWithRelevance,
+        knowledgePiecesSorted,
+        knowledgePiecesLimited,
+    });
+    return knowledgePiecesLimited.map(({ content }) => `- ${content}`).join('\n');
     //                                                      <- TODO: [🧠] Some smart aggregation of knowledge pieces, single-line vs multi-line vs mixed
 }
+// TODO: !!!!!! Annotate + to new file
+function computeCosineSimilarity(embeddingVector1, embeddingVector2) {
+    if (embeddingVector1.length !== embeddingVector2.length) {
+        throw new TypeError('Embedding vectors must have the same length');
+    }
+    const dotProduct = embeddingVector1.reduce((sum, value, index) => sum + value * embeddingVector2[index], 0);
+    const magnitude1 = Math.sqrt(embeddingVector1.reduce((sum, value) => sum + value * value, 0));
+    const magnitude2 = Math.sqrt(embeddingVector2.reduce((sum, value) => sum + value * value, 0));
+    return 1 - dotProduct / (magnitude1 * magnitude2);
+}
+/**
+ * TODO: !!!! Verify if this is working
+ * TODO: [♨] Implement Better - use keyword search
+ * TODO: [♨] Examples of values
+ */
 
 /**
  * @@@
@@ -6491,9 +6596,9 @@ async function getKnowledgeForTask(options) {
  * @private internal utility of `createPipelineExecutor`
  */
 async function getReservedParametersForTask(options) {
-    const { preparedPipeline, task, pipelineIdentification } = options;
+    const { tools, preparedPipeline, task, pipelineIdentification } = options;
     const context = await getContextForTask(); // <- [🏍]
-    const knowledge = await getKnowledgeForTask({ preparedPipeline, task });
+    const knowledge = await getKnowledgeForTask({ tools, preparedPipeline, task });
     const examples = await getExamplesForTask();
     const currentDate = new Date().toISOString(); // <- TODO: [🧠][💩] Better
     const modelName = RESERVED_PARAMETER_MISSING_VALUE;
@@ -6555,6 +6660,7 @@ async function executeTask(options) {
     }
     const definedParameters = Object.freeze({
         ...(await getReservedParametersForTask({
+            tools,
             preparedPipeline,
             task: currentTask,
             pipelineIdentification,
@@ -7055,7 +7161,7 @@ async function preparePersona(personaDescription, tools, options) {
     }).asPromise();
     const { outputParameters } = result;
     const { modelsRequirements: modelsRequirementsJson } = outputParameters;
-    const modelsRequirementsUnchecked = JSON.parse(modelsRequirementsJson);
+    const modelsRequirementsUnchecked = jsonParse(modelsRequirementsJson);
     if (isVerbose) {
         console.info(`PERSONA ${personaDescription}`, modelsRequirementsUnchecked);
     }
@@ -7242,7 +7348,7 @@ async function makeKnowledgeSourceHandler(knowledgeSource, tools, options) {
                 > },
             */
             async asJson() {
-                return JSON.parse(await tools.fs.readFile(filename, 'utf-8'));
+                return jsonParse(await tools.fs.readFile(filename, 'utf-8'));
             },
             async asText() {
                 return await tools.fs.readFile(filename, 'utf-8');
@@ -12947,7 +13053,7 @@ function $initializeRunCommand(program) {
         }
         let inputParameters = {};
         if (json) {
-            inputParameters = JSON.parse(json);
+            inputParameters = jsonParse(json);
             //                <- TODO: Maybe check shape of passed JSON and if its valid parameters Record
         }
         // TODO: DRY [◽]
@@ -13178,7 +13284,7 @@ function $initializeRunCommand(program) {
 const openapiJson = {
     openapi: '3.0.0',
     info: {
-        title: 'Promptbook Remote Server API (!!!! From TS)',
+        title: 'Promptbook Remote Server API (!!!! From YML)',
         version: '1.0.0',
         description: 'API documentation for the Promptbook Remote Server',
     },
@@ -13190,6 +13296,13 @@ const openapiJson = {
                 responses: {
                     '200': {
                         description: 'Server details in markdown format.',
+                        content: {
+                            'text/markdown': {
+                                schema: {
+                                    type: 'string',
+                                },
+                            },
+                        },
                     },
                 },
             },
@@ -13220,13 +13333,22 @@ const openapiJson = {
                     },
                 },
                 responses: {
-                    '200': {
+                    '201': {
                         description: 'Successful login',
                         content: {
                             'application/json': {
                                 schema: {
                                     type: 'object',
                                     properties: {
+                                        isSuccess: {
+                                            type: 'boolean',
+                                        },
+                                        message: {
+                                            type: 'string',
+                                        },
+                                        error: {
+                                            type: 'object',
+                                        },
                                         identification: {
                                             type: 'object',
                                         },
@@ -13235,6 +13357,43 @@ const openapiJson = {
                             },
                         },
                     },
+                    '400': {
+                        description: 'Bad request or login failed',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                    properties: {
+                                        error: {
+                                            type: 'object',
+                                        },
+                                    },
+                                },
+                            },
+                        },
+                    },
+                    '401': {
+                        description: 'Authentication error',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                    properties: {
+                                        isSuccess: {
+                                            type: 'boolean',
+                                            enum: [false],
+                                        },
+                                        message: {
+                                            type: 'string',
+                                        },
+                                        error: {
+                                            type: 'object',
+                                        },
+                                    },
+                                },
+                            },
+                        },
+                    },
                 },
             },
         },
@@ -13256,6 +13415,16 @@ const openapiJson = {
                             },
                         },
                     },
+                    '500': {
+                        description: 'No collection available',
+                        content: {
+                            'text/plain': {
+                                schema: {
+                                    type: 'string',
+                                },
+                            },
+                        },
+                    },
                 },
             },
         },
@@ -13287,6 +13456,28 @@ const openapiJson = {
                     },
                     '404': {
                         description: 'Book not found.',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                    properties: {
+                                        error: {
+                                            type: 'object',
+                                        },
+                                    },
+                                },
+                            },
+                        },
+                    },
+                    '500': {
+                        description: 'No collection available',
+                        content: {
+                            'text/plain': {
+                                schema: {
+                                    type: 'string',
+                                },
+                            },
+                        },
                     },
                 },
             },
@@ -13304,11 +13495,174 @@ const openapiJson = {
                                     type: 'array',
                                     items: {
                                         type: 'object',
+                                        properties: {
+                                            nonce: {
+                                                type: 'string',
+                                            },
+                                            taskId: {
+                                                type: 'string',
+                                            },
+                                            taskType: {
+                                                type: 'string',
+                                            },
+                                            status: {
+                                                type: 'string',
+                                            },
+                                            createdAt: {
+                                                type: 'string',
+                                                format: 'date-time',
+                                            },
+                                            updatedAt: {
+                                                type: 'string',
+                                                format: 'date-time',
+                                            },
+                                        },
+                                    },
+                                },
+                            },
+                        },
+                    },
+                },
+            },
+        },
+        '/executions/last': {
+            get: {
+                summary: 'Get the last execution',
+                description: 'Returns details of the last execution task.',
+                responses: {
+                    '200': {
+                        description: 'The last execution task with full details.',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                    properties: {
+                                        nonce: {
+                                            type: 'string',
+                                        },
+                                        taskId: {
+                                            type: 'string',
+                                        },
+                                        taskType: {
+                                            type: 'string',
+                                        },
+                                        status: {
+                                            type: 'string',
+                                        },
+                                        errors: {
+                                            type: 'array',
+                                            items: {
+                                                type: 'object',
+                                            },
+                                        },
+                                        warnings: {
+                                            type: 'array',
+                                            items: {
+                                                type: 'object',
+                                            },
+                                        },
+                                        createdAt: {
+                                            type: 'string',
+                                            format: 'date-time',
+                                        },
+                                        updatedAt: {
+                                            type: 'string',
+                                            format: 'date-time',
+                                        },
+                                        currentValue: {
+                                            type: 'object',
+                                        },
+                                    },
+                                },
+                            },
+                        },
+                    },
+                    '404': {
+                        description: 'No execution tasks found.',
+                        content: {
+                            'text/plain': {
+                                schema: {
+                                    type: 'string',
+                                },
+                            },
+                        },
+                    },
+                },
+            },
+        },
+        '/executions/{taskId}': {
+            get: {
+                summary: 'Get specific execution',
+                description: 'Returns details of a specific execution task.',
+                parameters: [
+                    {
+                        in: 'path',
+                        name: 'taskId',
+                        required: true,
+                        schema: {
+                            type: 'string',
+                        },
+                        description: 'The ID of the execution task to retrieve.',
+                    },
+                ],
+                responses: {
+                    '200': {
+                        description: 'The execution task with full details.',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                    properties: {
+                                        nonce: {
+                                            type: 'string',
+                                        },
+                                        taskId: {
+                                            type: 'string',
+                                        },
+                                        taskType: {
+                                            type: 'string',
+                                        },
+                                        status: {
+                                            type: 'string',
+                                        },
+                                        errors: {
+                                            type: 'array',
+                                            items: {
+                                                type: 'object',
+                                            },
+                                        },
+                                        warnings: {
+                                            type: 'array',
+                                            items: {
+                                                type: 'object',
+                                            },
+                                        },
+                                        createdAt: {
+                                            type: 'string',
+                                            format: 'date-time',
+                                        },
+                                        updatedAt: {
+                                            type: 'string',
+                                            format: 'date-time',
+                                        },
+                                        currentValue: {
+                                            type: 'object',
+                                        },
                                     },
                                 },
                             },
                         },
                     },
+                    '404': {
+                        description: 'Execution task not found.',
+                        content: {
+                            'text/plain': {
+                                schema: {
+                                    type: 'string',
+                                },
+                            },
+                        },
+                    },
                 },
             },
         },
@@ -13325,12 +13679,19 @@ const openapiJson = {
                                 properties: {
                                     pipelineUrl: {
                                         type: 'string',
+                                        description: 'URL of the pipeline to execute',
+                                    },
+                                    book: {
+                                        type: 'string',
+                                        description: 'Alternative field for pipelineUrl',
                                     },
                                     inputParameters: {
                                         type: 'object',
+                                        description: 'Parameters for pipeline execution',
                                     },
                                     identification: {
                                         type: 'object',
+                                        description: 'User identification data',
                                     },
                                 },
                             },
@@ -13350,13 +13711,164 @@ const openapiJson = {
                     },
                     '400': {
                         description: 'Invalid input.',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                    properties: {
+                                        error: {
+                                            type: 'object',
+                                        },
+                                    },
+                                },
+                            },
+                        },
+                    },
+                    '404': {
+                        description: 'Pipeline not found.',
+                        content: {
+                            'text/plain': {
+                                schema: {
+                                    type: 'string',
+                                },
+                            },
+                        },
+                    },
+                },
+            },
+        },
+        '/api-docs': {
+            get: {
+                summary: 'API documentation UI',
+                description: 'Swagger UI for API documentation',
+                responses: {
+                    '200': {
+                        description: 'HTML Swagger UI',
+                    },
+                },
+            },
+        },
+        '/swagger': {
+            get: {
+                summary: 'API documentation UI (alternative path)',
+                description: 'Swagger UI for API documentation',
+                responses: {
+                    '200': {
+                        description: 'HTML Swagger UI',
+                    },
+                },
+            },
+        },
+        '/openapi': {
+            get: {
+                summary: 'OpenAPI specification',
+                description: 'Returns the OpenAPI JSON specification',
+                responses: {
+                    '200': {
+                        description: 'OpenAPI specification',
+                        content: {
+                            'application/json': {
+                                schema: {
+                                    type: 'object',
+                                },
+                            },
+                        },
                     },
                 },
             },
         },
     },
-    components: {},
-    tags: [],
+    components: {
+        schemas: {
+            Error: {
+                type: 'object',
+                properties: {
+                    error: {
+                        type: 'object',
+                    },
+                },
+            },
+            ExecutionTaskSummary: {
+                type: 'object',
+                properties: {
+                    nonce: {
+                        type: 'string',
+                    },
+                    taskId: {
+                        type: 'string',
+                    },
+                    taskType: {
+                        type: 'string',
+                    },
+                    status: {
+                        type: 'string',
+                    },
+                    createdAt: {
+                        type: 'string',
+                        format: 'date-time',
+                    },
+                    updatedAt: {
+                        type: 'string',
+                        format: 'date-time',
+                    },
+                },
+            },
+            ExecutionTaskFull: {
+                type: 'object',
+                properties: {
+                    nonce: {
+                        type: 'string',
+                    },
+                    taskId: {
+                        type: 'string',
+                    },
+                    taskType: {
+                        type: 'string',
+                    },
+                    status: {
+                        type: 'string',
+                    },
+                    errors: {
+                        type: 'array',
+                        items: {
+                            type: 'object',
+                        },
+                    },
+                    warnings: {
+                        type: 'array',
+                        items: {
+                            type: 'object',
+                        },
+                    },
+                    createdAt: {
+                        type: 'string',
+                        format: 'date-time',
+                    },
+                    updatedAt: {
+                        type: 'string',
+                        format: 'date-time',
+                    },
+                    currentValue: {
+                        type: 'object',
+                    },
+                },
+            },
+        },
+    },
+    tags: [
+        {
+            name: 'Books',
+            description: 'Operations related to books and pipelines',
+        },
+        {
+            name: 'Executions',
+            description: 'Operations related to execution tasks',
+        },
+        {
+            name: 'Authentication',
+            description: 'Authentication operations',
+        },
+    ],
 };
 /**
  * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -14001,7 +14513,7 @@ function $initializeTestCommand(program) {
                     }
                 }
                 if (filename.endsWith('.bookc')) {
-                    pipeline = JSON.parse(await readFile(filename, 'utf-8'));
+                    pipeline = jsonParse(await readFile(filename, 'utf-8'));
                 }
                 else {
                     if (isVerbose) {
@@ -14110,6 +14622,37 @@ const _CLI = {
  * Note: [🟡] Code in this file should never be published outside of `@promptbook/cli`
  */
 
+/**
+ * How is the model provider trusted?
+ *
+ * @public exported from `@promptbook/core`
+ */
+// <- TODO: Maybe do better levels of trust
+/**
+ * How is the model provider important?
+ *
+ * @public exported from `@promptbook/core`
+ */
+const MODEL_ORDER = {
+    /**
+     * Top-tier models, e.g. OpenAI, Anthropic,...
+     */
+    TOP_TIER: 333,
+    /**
+     * Mid-tier models, e.g. Llama, Mistral, etc.
+     */
+    NORMAL: 100,
+    /**
+     * Low-tier models, e.g. Phi, Tiny, etc.
+     */
+    LOW_TIER: 0,
+};
+/**
+ * TODO: Add configuration schema and maybe some documentation link
+ * TODO: Maybe constrain LlmToolsConfiguration[number] by generic to ensure that `createConfigurationFromEnv` and `getBoilerplateConfiguration` always create same `packageName` and `className`
+ * TODO: [®] DRY Register logic
+ */
+
 /**
  * Registration of LLM provider metadata
  *
@@ -14124,9 +14667,11 @@ const _AnthropicClaudeMetadataRegistration = $llmToolsMetadataRegister.register(
     packageName: '@promptbook/anthropic-claude',
     className: 'AnthropicClaudeExecutionTools',
     envVariables: ['ANTHROPIC_CLAUDE_API_KEY'],
+    trustLevel: 'CLOSED',
+    order: MODEL_ORDER.TOP_TIER,
     getBoilerplateConfiguration() {
         return {
-            title: 'Anthropic Claude (boilerplate)',
+            title: 'Anthropic Claude',
             packageName: '@promptbook/anthropic-claude',
             className: 'AnthropicClaudeExecutionTools',
             options: {
@@ -14662,9 +15207,11 @@ const _AzureOpenAiMetadataRegistration = $llmToolsMetadataRegister.register({
     packageName: '@promptbook/azure-openai',
     className: 'AzureOpenAiExecutionTools',
     envVariables: ['AZUREOPENAI_RESOURCE_NAME', 'AZUREOPENAI_DEPLOYMENT_NAME', 'AZUREOPENAI_API_KEY'],
+    trustLevel: 'CLOSED_BUSINESS',
+    order: MODEL_ORDER.NORMAL,
     getBoilerplateConfiguration() {
         return {
-            title: 'Azure Open AI (boilerplate)',
+            title: 'Azure Open AI',
             packageName: '@promptbook/azure-openai',
             className: 'AzureOpenAiExecutionTools',
             options: {
@@ -15512,9 +16059,11 @@ const _DeepseekMetadataRegistration = $llmToolsMetadataRegister.register({
     packageName: '@promptbook/deepseek',
     className: 'DeepseekExecutionTools',
     envVariables: ['DEEPSEEK_GENERATIVE_AI_API_KEY'],
+    trustLevel: 'UNTRUSTED',
+    order: MODEL_ORDER.NORMAL,
     getBoilerplateConfiguration() {
         return {
-            title: 'Deepseek (boilerplate)',
+            title: 'Deepseek',
             packageName: '@promptbook/deepseek',
             className: 'DeepseekExecutionTools',
             options: {
@@ -15833,9 +16382,11 @@ const _GoogleMetadataRegistration = $llmToolsMetadataRegister.register({
     packageName: '@promptbook/google',
     className: 'GoogleExecutionTools',
     envVariables: ['GOOGLE_GENERATIVE_AI_API_KEY'],
+    trustLevel: 'CLOSED',
+    order: MODEL_ORDER.NORMAL,
     getBoilerplateConfiguration() {
         return {
-            title: 'Google Gemini (boilerplate)',
+            title: 'Google Gemini',
             packageName: '@promptbook/google',
             className: 'GoogleExecutionTools',
             options: {
@@ -16096,9 +16647,11 @@ const _OpenAiMetadataRegistration = $llmToolsMetadataRegister.register({
     packageName: '@promptbook/openai',
     className: 'OpenAiExecutionTools',
     envVariables: ['OPENAI_API_KEY'],
+    trustLevel: 'CLOSED',
+    order: MODEL_ORDER.TOP_TIER,
     getBoilerplateConfiguration() {
         return {
-            title: 'Open AI (boilerplate)',
+            title: 'Open AI',
             packageName: '@promptbook/openai',
             className: 'OpenAiExecutionTools',
             options: {
@@ -16136,9 +16689,11 @@ const _OpenAiAssistantMetadataRegistration = $llmToolsMetadataRegister.register(
     className: 'OpenAiAssistantExecutionTools',
     envVariables: null,
     //            <- TODO: ['OPENAI_API_KEY', 'OPENAI_ASSISTANT_ID']
+    trustLevel: 'CLOSED',
+    order: MODEL_ORDER.NORMAL,
     getBoilerplateConfiguration() {
         return {
-            title: 'Open AI Assistant (boilerplate)',
+            title: 'Open AI Assistant',
             packageName: '@promptbook/openai',
             className: 'OpenAiAssistantExecutionTools',
             options: {