n8n-nodes-vercel-ai-sdk-universal-temp 0.3.9 → 0.3.10

package/dist/nodes/shared/letta/client.js

@@ -15,7 +15,7 @@ class LettaClientWrapper {
             return await this.client.agents.retrieve(agentId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             if (message.includes('401') || message.includes('Unauthorized')) {
                 throw new Error(`Authentication failed: Invalid or missing API key for Letta instance`);
             }
@@ -27,12 +27,13 @@ class LettaClientWrapper {
     }
     async listAgents(tags) {
         try {
-            return await this.client.agents.list({
+            const result = this.client.agents.list({
                 tags: tags || [],
             });
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             if (message.includes('401') || message.includes('Unauthorized')) {
                 throw new Error(`Authentication failed: Invalid or missing API key for Letta instance`);
             }
@@ -44,7 +45,7 @@ class LettaClientWrapper {
             return await this.client.agents.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create agent: ${message}`);
         }
     }
@@ -53,25 +54,74 @@ class LettaClientWrapper {
             return await this.client.agents.delete(agentId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete agent: ${message}`);
         }
     }
     async updateAgent(agentId, updates) {
         try {
-            return await this.client.agents.modify(agentId, updates);
+            return await this.client.agents.update(agentId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update agent: ${message}`);
         }
     }
+    async listAgentBlocks(agentId, query) {
+        try {
+            const result = this.client.agents.blocks.list(agentId, query);
+            return await this.collectResult(result);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to list agent blocks: ${message}`);
+        }
+    }
+    async getAgentBlock(agentId, blockLabel) {
+        try {
+            return await this.client.agents.blocks.retrieve(blockLabel, { agent_id: agentId });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to get agent block: ${message}`);
+        }
+    }
+    async updateAgentBlock(agentId, blockLabel, updates) {
+        try {
+            return await this.client.agents.blocks.update(blockLabel, {
+                agent_id: agentId,
+                ...updates,
+            });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to update agent block: ${message}`);
+        }
+    }
+    async attachAgentBlock(agentId, blockId) {
+        try {
+            return await this.client.agents.blocks.attach(blockId, { agent_id: agentId });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to attach agent block: ${message}`);
+        }
+    }
+    async detachAgentBlock(agentId, blockId) {
+        try {
+            return await this.client.agents.blocks.detach(blockId, { agent_id: agentId });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to detach agent block: ${message}`);
+        }
+    }
     async getIdentity(identityId) {
         try {
             return await this.client.identities.retrieve(identityId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             if (message.includes('404') || message.includes('not found')) {
                 throw new Error(`Identity not found: ${identityId}`);
             }
@@ -80,10 +130,11 @@ class LettaClientWrapper {
     }
     async listIdentities() {
         try {
-            return await this.client.identities.list();
+            const result = this.client.identities.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list identities: ${message}`);
         }
     }
@@ -92,7 +143,7 @@ class LettaClientWrapper {
             return await this.client.identities.delete(identityId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete identity: ${message}`);
         }
     }
@@ -101,22 +152,23 @@ class LettaClientWrapper {
             return await this.client.identities.update(identityId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update identity: ${message}`);
         }
     }
     async getIdentityByIdentifierKey(identifierKey) {
         try {
-            const identities = await this.client.identities.list({
-                identifierKey,
+            const result = this.client.identities.list({
+                identifier_key: identifierKey,
+                limit: 1,
             });
-            if (identities && identities.length > 0) {
+            const identities = await this.collectResult(result, 1);
+            if (identities.length > 0)
                 return identities[0];
-            }
             return null;
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             if (message.includes('404') || message.includes('not found')) {
                 return null;
             }
@@ -137,15 +189,15 @@ class LettaClientWrapper {
             })
                 .filter((p) => p !== null);
             return await this.client.identities.create({
-                identifierKey,
+                identifier_key: identifierKey,
                 name,
-                identityType,
-                agentIds,
+                identity_type: identityType,
+                agent_ids: agentIds,
                 properties: propertiesToUpsert,
             });
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create identity: ${message}`);
         }
     }
@@ -212,10 +264,10 @@ class LettaClientWrapper {
                 return existingIdentity;
             }
             return await this.client.identities.upsert({
-                identifierKey,
+                identifier_key: identifierKey,
                 name: finalName,
-                identityType: finalIdentityType,
-                agentIds: finalAgentIds,
+                identity_type: finalIdentityType,
+                agent_ids: finalAgentIds,
                 properties: finalProperties,
             });
         }
@@ -295,7 +347,7 @@ class LettaClientWrapper {
             messages: [{ role: 'user', content: message }],
         };
         if (options.maxSteps) {
-            messagePayload.maxSteps = options.maxSteps;
+            messagePayload.max_steps = options.maxSteps;
         }
         if (identityId) {
             messagePayload.messages[0].sender_id = identityId;
@@ -303,14 +355,35 @@ class LettaClientWrapper {
         if (options.background) {
             messagePayload.background = options.background;
         }
+        if (options.streaming) {
+            messagePayload.streaming = options.streaming;
+        }
         if (options.streamTokens) {
             messagePayload.stream_tokens = options.streamTokens;
         }
+        if (options.includePings) {
+            messagePayload.include_pings = options.includePings;
+        }
+        if (options.enableThinking && options.enableThinking !== 'auto') {
+            messagePayload.enable_thinking = options.enableThinking;
+        }
+        if (options.includeReturnMessageTypes) {
+            messagePayload.include_return_message_types = options.includeReturnMessageTypes;
+        }
+        if (options.useAssistantMessage) {
+            messagePayload.use_assistant_message = options.useAssistantMessage;
+            if (options.assistantMessageToolName) {
+                messagePayload.assistant_message_tool_name = options.assistantMessageToolName;
+            }
+            if (options.assistantMessageToolKwarg) {
+                messagePayload.assistant_message_tool_kwarg = options.assistantMessageToolKwarg;
+            }
+        }
         try {
             return await this.client.agents.messages.create(agentId, messagePayload);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to send message to agent ${agentId}: ${message}`);
         }
     }
@@ -335,49 +408,30 @@ class LettaClientWrapper {
     async listLlmModels() {
         try {
             const response = await this.client.models.list();
-            if (Array.isArray(response)) {
-                return response;
-            }
-            if (response && Array.isArray(response.data)) {
-                return response.data;
-            }
-            if (response && Array.isArray(response.models)) {
-                return response.models;
-            }
-            console.warn('Unexpected response structure from models.list():', response);
-            return [];
+            return await this.collectResult(response);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list LLM models: ${message}`);
         }
     }
     async listEmbeddingModels() {
         try {
             const response = await this.client.models.embeddings.list();
-            if (Array.isArray(response)) {
-                return response;
-            }
-            if (response && Array.isArray(response.data)) {
-                return response.data;
-            }
-            if (response && Array.isArray(response.models)) {
-                return response.models;
-            }
-            console.warn('Unexpected response structure from models.embeddings.list():', response);
-            return [];
+            return await this.collectResult(response);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list embedding models: ${message}`);
         }
     }
     async listTools() {
         try {
-            return await this.client.tools.list();
+            const result = this.client.tools.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list tools: ${message}`);
         }
     }
@@ -386,7 +440,7 @@ class LettaClientWrapper {
             return await this.client.tools.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create tool: ${message}`);
         }
     }
@@ -395,7 +449,7 @@ class LettaClientWrapper {
             return await this.client.tools.retrieve(toolId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get tool: ${message}`);
         }
     }
@@ -404,7 +458,7 @@ class LettaClientWrapper {
             return await this.client.tools.update(toolId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update tool: ${message}`);
         }
     }
@@ -413,7 +467,7 @@ class LettaClientWrapper {
             return await this.client.tools.delete(toolId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete tool: ${message}`);
         }
     }
@@ -422,25 +476,26 @@ class LettaClientWrapper {
             return await this.client.tools.upsert(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to upsert tool: ${message}`);
         }
     }
-    async upsertBaseTools() {
+    async searchTools(params) {
         try {
-            return await this.client.tools.upsertBaseTools();
+            return await this.client.tools.search(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
-            throw new Error(`Failed to upsert base tools: ${message}`);
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to search tools: ${message}`);
         }
     }
     async listBlocks(limit) {
         try {
-            return await this.client.blocks.list({ limit });
+            const result = this.client.blocks.list({ limit });
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list blocks: ${message}`);
         }
     }
@@ -449,7 +504,7 @@ class LettaClientWrapper {
             return await this.client.blocks.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create block: ${message}`);
         }
     }
@@ -458,7 +513,7 @@ class LettaClientWrapper {
             return await this.client.blocks.retrieve(blockId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get block: ${message}`);
         }
     }
@@ -467,7 +522,7 @@ class LettaClientWrapper {
             return await this.client.blocks.update(blockId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update block: ${message}`);
         }
     }
@@ -476,16 +531,17 @@ class LettaClientWrapper {
             return await this.client.blocks.delete(blockId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete block: ${message}`);
         }
     }
     async listFolders() {
         try {
-            return await this.client.folders.list();
+            const result = this.client.folders.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list folders: ${message}`);
         }
     }
@@ -494,7 +550,7 @@ class LettaClientWrapper {
             return await this.client.folders.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create folder: ${message}`);
         }
     }
@@ -503,7 +559,7 @@ class LettaClientWrapper {
             return await this.client.folders.retrieve(folderId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get folder: ${message}`);
         }
     }
@@ -512,7 +568,7 @@ class LettaClientWrapper {
             return await this.client.folders.update(folderId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update folder: ${message}`);
         }
     }
@@ -521,16 +577,17 @@ class LettaClientWrapper {
             return await this.client.folders.delete(folderId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete folder: ${message}`);
         }
     }
     async listMcpServers() {
         try {
-            return await this.client.mcpServers.list();
+            const result = this.client.mcpServers.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list MCP servers: ${message}`);
         }
     }
@@ -539,7 +596,7 @@ class LettaClientWrapper {
             return await this.client.mcpServers.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create MCP server: ${message}`);
         }
     }
@@ -548,7 +605,7 @@ class LettaClientWrapper {
             return await this.client.mcpServers.retrieve(serverId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get MCP server: ${message}`);
         }
     }
@@ -557,7 +614,7 @@ class LettaClientWrapper {
             return await this.client.mcpServers.update(serverId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update MCP server: ${message}`);
         }
     }
@@ -566,25 +623,49 @@ class LettaClientWrapper {
             return await this.client.mcpServers.delete(serverId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete MCP server: ${message}`);
         }
     }
-    async refreshMcpServer(serverId) {
+    async refreshMcpServer(serverId, params) {
         try {
+            if (params && Object.keys(params).length) {
+                return await this.client.mcpServers.refresh(serverId, params);
+            }
             return await this.client.mcpServers.refresh(serverId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to refresh MCP server: ${message}`);
         }
     }
+    async listMcpServerTools(serverId) {
+        try {
+            return await this.client.mcpServers.tools.list(serverId);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to list MCP server tools: ${message}`);
+        }
+    }
+    async getMcpServerTool(serverId, toolId) {
+        try {
+            return await this.client.mcpServers.tools.retrieve(toolId, {
+                mcp_server_id: serverId,
+            });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to retrieve MCP server tool: ${message}`);
+        }
+    }
     async listArchives() {
         try {
-            return await this.client.archives.list();
+            const result = this.client.archives.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list archives: ${message}`);
         }
     }
@@ -593,7 +674,7 @@ class LettaClientWrapper {
             return await this.client.archives.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create archive: ${message}`);
         }
     }
@@ -602,7 +683,7 @@ class LettaClientWrapper {
             return await this.client.archives.retrieve(archiveId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get archive: ${message}`);
         }
     }
@@ -611,7 +692,7 @@ class LettaClientWrapper {
             return await this.client.archives.update(archiveId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update archive: ${message}`);
         }
     }
@@ -620,16 +701,17 @@ class LettaClientWrapper {
             return await this.client.archives.delete(archiveId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete archive: ${message}`);
         }
     }
     async listBatches() {
         try {
-            return await this.client.batches.list();
+            const result = this.client.batches.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list batches: ${message}`);
         }
     }
@@ -638,7 +720,7 @@ class LettaClientWrapper {
             return await this.client.batches.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create batch: ${message}`);
         }
     }
@@ -647,7 +729,7 @@ class LettaClientWrapper {
             return await this.client.batches.retrieve(batchId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get batch: ${message}`);
         }
     }
@@ -656,16 +738,17 @@ class LettaClientWrapper {
             return await this.client.batches.delete(batchId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete batch: ${message}`);
         }
     }
     async listGroups() {
         try {
-            return await this.client.groups.list();
+            const result = this.client.groups.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list groups: ${message}`);
         }
     }
@@ -674,7 +757,7 @@ class LettaClientWrapper {
             return await this.client.groups.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create group: ${message}`);
         }
     }
@@ -683,7 +766,7 @@ class LettaClientWrapper {
             return await this.client.groups.retrieve(groupId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get group: ${message}`);
         }
     }
@@ -692,7 +775,7 @@ class LettaClientWrapper {
             return await this.client.groups.update(groupId, updates);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to update group: ${message}`);
         }
     }
@@ -701,16 +784,17 @@ class LettaClientWrapper {
             return await this.client.groups.delete(groupId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete group: ${message}`);
         }
     }
     async listRuns() {
         try {
-            return await this.client.runs.list();
+            const result = this.client.runs.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list runs: ${message}`);
         }
     }
@@ -719,7 +803,7 @@ class LettaClientWrapper {
             return await this.client.runs.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create run: ${message}`);
         }
     }
@@ -728,7 +812,7 @@ class LettaClientWrapper {
             return await this.client.runs.retrieve(runId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get run: ${message}`);
         }
     }
@@ -737,16 +821,17 @@ class LettaClientWrapper {
             return await this.client.runs.delete(runId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete run: ${message}`);
         }
     }
     async listSteps(params) {
         try {
-            return await this.client.steps.list(params);
+            const result = this.client.steps.list(params);
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list steps: ${message}`);
         }
     }
@@ -755,16 +840,17 @@ class LettaClientWrapper {
             return await this.client.steps.retrieve(stepId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get step: ${message}`);
         }
     }
     async listTags() {
         try {
-            return await this.client.tags.list();
+            const result = this.client.tags.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list tags: ${message}`);
         }
     }
@@ -773,16 +859,17 @@ class LettaClientWrapper {
             return await this.client.tags.retrieve(tagId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get tag: ${message}`);
         }
     }
     async listTemplates() {
         try {
-            return await this.client.templates.list();
+            const result = this.client.templates.list();
+            return await this.collectResult(result);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to list templates: ${message}`);
         }
     }
@@ -791,7 +878,7 @@ class LettaClientWrapper {
             return await this.client.templates.create(params);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to create template: ${message}`);
         }
     }
@@ -800,7 +887,7 @@ class LettaClientWrapper {
             return await this.client.templates.retrieve(templateId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to get template: ${message}`);
         }
     }
@@ -809,10 +896,162 @@ class LettaClientWrapper {
             return await this.client.templates.delete(templateId);
         }
         catch (error) {
-            const message = error instanceof Error ? error.message : String(error);
+            const message = this.getErrorMessage(error);
             throw new Error(`Failed to delete template: ${message}`);
         }
     }
+    async updateTemplate(templateId, updates) {
+        try {
+            return await this.client.templates.update(templateId, updates);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to update template: ${message}`);
+        }
+    }
+    async createAgentFromTemplate(templateVersion, params) {
+        try {
+            return await this.client.templates.agents.create(templateVersion, params);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to create agent from template: ${message}`);
+        }
+    }
+    async createAccessToken(params) {
+        try {
+            return await this.client.accessTokens.create(params);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to create access token: ${message}`);
+        }
+    }
+    async listAccessTokens() {
+        try {
+            const result = this.client.accessTokens.list();
+            return await this.collectResult(result);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to list access tokens: ${message}`);
+        }
+    }
+    async deleteAccessToken(tokenId) {
+        try {
+            return await this.client.accessTokens.delete(tokenId);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to delete access token: ${message}`);
+        }
+    }
+    async searchMessages(params) {
+        try {
+            return await this.client.messages.search(params);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to search messages: ${message}`);
+        }
+    }
+    async listMessages(params) {
+        try {
+            const result = this.client.messages.list(params);
+            return await this.collectResult(result);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to list messages: ${message}`);
+        }
+    }
+    async listAgentMessages(agentId, params) {
+        try {
+            const result = this.client.agents.messages.list(agentId, params);
+            return await this.collectResult(result);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to list agent messages: ${message}`);
+        }
+    }
+    async searchPassages(params) {
+        try {
+            return await this.client.passages.search(params);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to search passages: ${message}`);
+        }
+    }
+    async listAgentPassages(agentId, params) {
+        try {
+            const result = this.client.agents.passages.list(agentId, params);
+            return await this.collectResult(result);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to list agent passages: ${message}`);
+        }
+    }
+    async createAgentPassage(agentId, params) {
+        try {
+            return await this.client.agents.passages.create(agentId, params);
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to create agent passage: ${message}`);
+        }
+    }
+    async deleteAgentPassage(agentId, passageId) {
+        try {
+            return await this.client.agents.passages.delete(passageId, { agent_id: agentId });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to delete agent passage: ${message}`);
+        }
+    }
+    async runAgentTool(agentId, toolName, args) {
+        try {
+            return await this.client.agents.tools.run(toolName, {
+                agent_id: agentId,
+                args,
+            });
+        }
+        catch (error) {
+            const message = this.getErrorMessage(error);
+            throw new Error(`Failed to run agent tool: ${message}`);
+        }
+    }
+    getErrorMessage(error) {
+        return error instanceof Error ? error.message : String(error);
+    }
+    async collectResult(result, limit) {
+        if (!result)
+            return [];
+        if (Array.isArray(result))
+            return result;
+        if (typeof result === 'object' && result !== null) {
+            const asAny = result;
+            if (Array.isArray(asAny.data))
+                return asAny.data;
+            if (Array.isArray(asAny.models))
+                return asAny.models;
+        }
+        const items = [];
+        const maybeIterable = result;
+        const isIterable = typeof maybeIterable?.[Symbol.asyncIterator] === 'function' ||
+            typeof maybeIterable?.[Symbol.iterator] === 'function';
+        if (!isIterable)
+            return [];
+        for await (const item of maybeIterable) {
+            items.push(item);
+            if (limit !== undefined && items.length >= limit)
+                break;
+        }
+        return items;
+    }
 }
 exports.LettaClientWrapper = LettaClientWrapper;
 //# sourceMappingURL=client.js.map