@letta-ai/letta-agent-sdk 0.8.34 → 0.8.36

package/dist/index.js

@@ -17791,20 +17791,21 @@ function buildLocalAppServerProcess(cliPath, options = {}, confineMemory = creat
     process.execPath,
     ...buildLocalAppServerArgs(cliPath, options)
   ];
-  if (options.filesystemConfinement !== "memory") {
-    return {
-      command: launcher[0],
-      args: launcher.slice(1),
-      env
-    };
-  }
-  const confined = confineMemory({
-    launcher,
-    env: withDefaultMemoryDirectory(env, options)
-  });
+  const confined = options.filesystemConfinement === "memory" ? confineMemory({ launcher, env: withDefaultMemoryDirectory(env, options) }) : { launcher, env };
+  const slice = env.LETTA_WORKLOAD_SYSTEMD_SLICE?.trim();
+  const managedLauncher = process.platform === "linux" && slice ? [
+    "systemd-run",
+    "--scope",
+    "--quiet",
+    "--collect",
+    "--property=TimeoutStopSec=5s",
+    `--slice=${slice}`,
+    "--",
+    ...confined.launcher
+  ] : confined.launcher;
   return {
-    command: confined.launcher[0],
-    args: confined.launcher.slice(1),
+    command: managedLauncher[0],
+    args: managedLauncher.slice(1),
     env: confined.env
   };
 }
@@ -17851,6 +17852,7 @@ function startLocalAppServer(options = {}) {
       cleanup();
       resolve3({
         url,
+        isRunning: () => child.exitCode === null && child.signalCode === null,
         close: () => terminateProcess(child)
       });
     };
@@ -17883,12 +17885,14 @@ ${output.trim()}` : ""}`));
 }
 
 // src/local-app-server-session.ts
-function createLocalAppServerSession(options, mode) {
+function createLocalAppServerSession(options, mode, sharedServer) {
   const appServer = options ?? {};
+  const session = mode.options;
+  const needsOwnProcess = session.filesystemConfinement === "memory" || Object.keys(session.env ?? {}).length > 0;
   const sessionOptions = {
     ...mode.kind === "conversation" && (appServer.harnessBackend ?? "local") === "local" ? { requireLocalConversationCapability: true } : {},
     ...appServer.url !== undefined ? { url: appServer.url } : {
-      connect: (sessionEnv) => startLocalAppServer({
+      connect: (sessionEnv) => sharedServer && !needsOwnProcess ? sharedServer.connect() : startLocalAppServer({
         listen: appServer.listen,
         backend: appServer.harnessBackend ?? "local",
         startupTimeoutMs: appServer.startupTimeoutMs,
@@ -17904,6 +17908,54 @@ function createLocalAppServerSession(options, mode) {
   return new AppServerSession(sessionOptions, mode);
 }
 
+// src/shared-local-app-server.ts
+class SharedLocalAppServer {
+  options;
+  starting = null;
+  server = null;
+  closed = false;
+  closing = null;
+  constructor(options) {
+    this.options = options;
+  }
+  async connect() {
+    if (this.closed)
+      throw new Error("Local App Server owner is closed");
+    if (this.server && !this.server.isRunning()) {
+      this.server = null;
+      this.starting = null;
+    }
+    if (!this.starting) {
+      this.starting = startLocalAppServer(this.options).then((server2) => {
+        if (this.closed) {
+          server2.close();
+          throw new Error("Local App Server owner closed during startup");
+        }
+        this.server = server2;
+        return server2;
+      }, (error) => {
+        this.starting = null;
+        throw error;
+      });
+    }
+    const server = await this.starting;
+    if (this.closed)
+      throw new Error("Local App Server owner is closed");
+    return {
+      url: server.url,
+      isRunning: server.isRunning,
+      close() {}
+    };
+  }
+  close() {
+    if (this.closing)
+      return this.closing;
+    this.closed = true;
+    this.closing = this.starting ? this.starting.then((server) => server.close(), () => {}) : Promise.resolve();
+    return this.closing;
+  }
+}
+
 // src/skill-node.ts
 import { readFile, readdir, stat, mkdtemp, rm, mkdir, writeFile } from "node:fs/promises";
 import { tmpdir } from "node:os";
@@ -17992,18 +18044,36 @@ async function pushSkillSupportFiles(target, skills) {
 
 // src/client.ts
 class LettaAgentClient extends LettaAgentClientBase {
+  localServer = null;
+  localClosePromise = null;
+  sharedLocalServer() {
+    if (this.localClosePromise)
+      throw new Error("LettaAgentClient is closed");
+    const options = this.options.appServer;
+    return this.localServer ??= new SharedLocalAppServer({
+      listen: options?.listen,
+      backend: options?.harnessBackend ?? "local",
+      startupTimeoutMs: options?.startupTimeoutMs
+    });
+  }
+  close() {
+    if (!this.localClosePromise) {
+      this.localClosePromise = Promise.all([
+        super.close(),
+        this.localServer?.close()
+      ]).then(() => {});
+    }
+    return this.localClosePromise;
+  }
   skillNodeSupport() {
     return { loadSkillDirectory, pushSkillSupportFiles };
   }
   createLocalManagementTransport() {
     const localOptions = this.options.appServer;
+    const sharedServer = this.sharedLocalServer();
     return new AppServerManagementTransport({
       ...localOptions?.url !== undefined ? { url: localOptions.url } : {
-        connect: () => startLocalAppServer({
-          listen: localOptions?.listen,
-          backend: localOptions?.harnessBackend ?? "local",
-          startupTimeoutMs: localOptions?.startupTimeoutMs
-        })
+        connect: () => sharedServer.connect()
       },
       ...localOptions?.WebSocket !== undefined ? { WebSocket: localOptions.WebSocket } : {},
       ...localOptions?.requestTimeoutMs !== undefined ? { requestTimeoutMs: localOptions.requestTimeoutMs } : {}
@@ -18014,13 +18084,16 @@ class LettaAgentClient extends LettaAgentClientBase {
     const session = createLocalAppServerSession(localOptions.appServer, {
       kind: "create-agent",
       options
-    });
-    const initMsg = await session.initialize();
-    session.close();
-    if (!initMsg.agentId) {
-      throw new Error("Local App Server agent creation did not return an agent id.");
+    }, this.sharedLocalServer());
+    try {
+      const initMsg = await session.initialize();
+      if (!initMsg.agentId) {
+        throw new Error("Local App Server agent creation did not return an agent id.");
+      }
+      return initMsg.agentId;
+    } finally {
+      session.close();
     }
-    return initMsg.agentId;
   }
   createLocalSession(agentId, options) {
     const localOptions = this.options;
@@ -18029,7 +18102,7 @@ class LettaAgentClient extends LettaAgentClientBase {
       agentId,
       newConversation: true,
       options
-    });
+    }, this.sharedLocalServer());
   }
   createLocalQuerySession(queryOptions, sessionOptions) {
     const localOptions = this.options;
@@ -18048,7 +18121,7 @@ class LettaAgentClient extends LettaAgentClientBase {
       },
       options: sessionOptions,
       ...queryOptions.disableMemoryGuard === true ? { disableMemoryGuard: true } : {}
-    });
+    }, this.sharedLocalServer());
   }
   resumeLocalSession(id, options) {
     const localOptions = this.options;
@@ -18057,14 +18130,14 @@ class LettaAgentClient extends LettaAgentClientBase {
         kind: "session",
         conversationId: id,
         options
-      });
+      }, this.sharedLocalServer());
     }
     return createLocalAppServerSession(localOptions.appServer, {
       kind: "session",
       agentId: id,
       defaultConversation: true,
       options
-    });
+    }, this.sharedLocalServer());
   }
 }
 function looksLikeConversationId2(id) {
@@ -35589,15 +35662,41 @@ import { readFileSync } from "node:fs";
 registerMcpConnector(connectMcpServers2);
 async function createAgent(options = {}) {
   validateCreateAgentOptions(options);
-  return new LettaAgentClient().createAgent(options);
+  const client = new LettaAgentClient;
+  try {
+    return await client.createAgent(options);
+  } finally {
+    await client.close();
+  }
+}
+function ownSession(client, session) {
+  const close = session.close.bind(session);
+  const dispose = session[Symbol.asyncDispose].bind(session);
+  session.close = () => {
+    try {
+      close();
+    } finally {
+      client.close();
+    }
+  };
+  session[Symbol.asyncDispose] = async () => {
+    try {
+      await dispose();
+    } finally {
+      await client.close();
+    }
+  };
+  return session;
 }
 function createSession(agentId, options = {}) {
   validateCreateSessionOptions(options);
-  return new LettaAgentClient().createSession(agentId, options);
+  const client = new LettaAgentClient;
+  return ownSession(client, client.createSession(agentId, options));
 }
 function resumeSession(id, options = {}) {
   validateCreateSessionOptions(options);
-  return new LettaAgentClient().resumeSession(id, options);
+  const client = new LettaAgentClient;
+  return ownSession(client, client.resumeSession(id, options));
 }
 async function prompt(message, agentId, options = {}) {
   const session = createSession(agentId, options);
@@ -35608,17 +35707,36 @@ async function prompt(message, agentId, options = {}) {
   }
 }
 function query(params) {
-  return new LettaAgentClient({
+  const client = new LettaAgentClient({
     backend: "local",
     appServer: { harnessBackend: "api" }
-  }).query(params);
+  });
+  const inner = client.query(params);
+  const result = async function* () {
+    try {
+      yield* inner;
+    } finally {
+      await client.close();
+    }
+  }();
+  return Object.defineProperties(Object.assign(result, {
+    interrupt: () => inner.interrupt(),
+    close() {
+      inner.close();
+      client.close();
+      result.return();
+    }
+  }), {
+    conversationId: { get: () => inner.conversationId, enumerable: true },
+    agentId: { get: () => inner.agentId, enumerable: true }
+  });
 }
 async function listMessagesDirect(agentId, options = {}) {
-  const session = new LettaAgentClient().resumeSession(agentId, {
+  const session = resumeSession(agentId, {
     permissionMode: "unrestricted"
   });
-  await session.initialize();
   try {
+    await session.initialize();
     return await session.listMessages(options);
   } finally {
     session.close();
@@ -35681,4 +35799,4 @@ export {
   CloudManagedSandboxExpiredError
 };
 
-//# debugId=521C138A96EC8F7F64756E2164756E21
+//# debugId=449C4232C4B789D064756E2164756E21