agentxjs 0.0.1

package/dist/runtime/sse/index.js

@@ -0,0 +1,621 @@
+import { createLogger, setLoggerFactory } from '@agentxjs/common';
+import { LogLevel, STREAM_EVENT_TYPE_NAMES } from '@agentxjs/types';
+import ky from 'ky';
+
+var __defProp = Object.defineProperty;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
+var PersistentSSEConnection = class {
+  constructor(serverUrl, agentId, sseParams = {}) {
+    this.serverUrl = serverUrl;
+    this.agentId = agentId;
+    this.sseParams = sseParams;
+    __publicField(this, "eventSource", null);
+    __publicField(this, "messageQueue", []);
+    __publicField(this, "activeIterators", /* @__PURE__ */ new Set());
+    __publicField(this, "isDone", false);
+  }
+  /**
+   * Initialize SSE connection
+   */
+  connect() {
+    if (this.eventSource) {
+      return;
+    }
+    let sseUrl = `${this.serverUrl}/agents/${this.agentId}/sse`;
+    if (Object.keys(this.sseParams).length > 0) {
+      const params = new URLSearchParams(this.sseParams);
+      sseUrl += `?${params.toString()}`;
+    }
+    this.eventSource = new EventSource(sseUrl);
+    const handleEvent = (event) => {
+      try {
+        const data = JSON.parse(event.data);
+        if (this.activeIterators.size > 0) {
+          const iterator = this.activeIterators.values().next().value;
+          if (iterator) {
+            this.activeIterators.delete(iterator);
+            iterator.resolve({ value: data, done: false });
+          }
+        } else {
+          this.messageQueue.push(data);
+        }
+      } catch {
+      }
+    };
+    const handleError = () => {
+      this.isDone = true;
+      this.eventSource?.close();
+      this.eventSource = null;
+      for (const iterator of this.activeIterators) {
+        iterator.reject(new Error("SSE connection error"));
+      }
+      this.activeIterators.clear();
+    };
+    for (const eventType of STREAM_EVENT_TYPE_NAMES) {
+      this.eventSource.addEventListener(eventType, handleEvent);
+    }
+    this.eventSource.addEventListener("error", handleEvent);
+    this.eventSource.onmessage = handleEvent;
+    this.eventSource.onerror = handleError;
+  }
+  /**
+   * Create an async iterable for a single receive() call
+   *
+   * This iterator continues until a final message_stop is received (stopReason !== "tool_use").
+   * For tool calls, this means it will span multiple message_start/message_stop cycles.
+   * The SSE connection itself remains open for future receive() calls.
+   */
+  createIterator() {
+    const connection = this;
+    return {
+      [Symbol.asyncIterator]() {
+        let turnComplete = false;
+        return {
+          async next() {
+            if (connection.messageQueue.length > 0) {
+              const event = connection.messageQueue.shift();
+              if (event.type === "message_stop") {
+                const stopReason = event.data.stopReason;
+                if (stopReason !== "tool_use") {
+                  turnComplete = true;
+                }
+              }
+              return { value: event, done: false };
+            }
+            if (turnComplete) {
+              return { done: true, value: void 0 };
+            }
+            if (connection.isDone) {
+              return { done: true, value: void 0 };
+            }
+            return new Promise((resolve, reject) => {
+              const wrappedResolve = (result) => {
+                if (!result.done) {
+                  if (result.value.type === "message_stop") {
+                    const stopReason = result.value.data.stopReason;
+                    if (stopReason !== "tool_use") {
+                      turnComplete = true;
+                    }
+                  }
+                }
+                resolve(result);
+              };
+              const iterator = { resolve: wrappedResolve, reject };
+              connection.activeIterators.add(iterator);
+            });
+          },
+          async return() {
+            return { done: true, value: void 0 };
+          }
+        };
+      }
+    };
+  }
+  /**
+   * Close the connection
+   */
+  close() {
+    this.isDone = true;
+    if (this.eventSource) {
+      this.eventSource.close();
+      this.eventSource = null;
+    }
+    for (const iterator of this.activeIterators) {
+      iterator.reject(new Error("Connection closed"));
+    }
+    this.activeIterators.clear();
+    this.messageQueue = [];
+  }
+};
+function createSSEDriver(config) {
+  const { serverUrl, agentId, headers = {}, sseParams = {} } = config;
+  let connection = null;
+  return {
+    name: "SSEDriver",
+    async *receive(message) {
+      if (!connection) {
+        connection = new PersistentSSEConnection(serverUrl, agentId, sseParams);
+        connection.connect();
+      }
+      const messageUrl = `${serverUrl}/agents/${agentId}/messages`;
+      const response = await fetch(messageUrl, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+          ...headers
+        },
+        body: JSON.stringify({
+          content: typeof message.content === "string" ? message.content : message.content
+        })
+      });
+      if (!response.ok) {
+        const errorBody = await response.json().catch(() => ({}));
+        throw new Error(errorBody.error?.message || `HTTP ${response.status}`);
+      }
+      yield* connection.createIterator();
+    },
+    interrupt() {
+      const interruptUrl = `${serverUrl}/agents/${agentId}/interrupt`;
+      fetch(interruptUrl, {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+          ...headers
+        }
+      }).catch(() => {
+      });
+    }
+  };
+}
+var ApiError = class extends Error {
+  constructor(code, message, details) {
+    super(message);
+    this.code = code;
+    this.details = details;
+    this.name = "ApiError";
+  }
+};
+function createHttpClient(options) {
+  return ky.create({
+    prefixUrl: options.baseUrl.replace(/\/+$/, ""),
+    headers: options.headers,
+    timeout: options.timeout || 3e4,
+    hooks: {
+      afterResponse: [
+        async (_request, _options, response) => {
+          if (!response.ok) {
+            const data = await response.json().catch(() => ({}));
+            throw new ApiError(
+              data.error?.code || "UNKNOWN_ERROR",
+              data.error?.message || `Request failed: ${response.status}`,
+              data.error?.details
+            );
+          }
+          return response;
+        }
+      ]
+    }
+  });
+}
+
+// src/runtime/sse/repository/RemoteRepository.ts
+var logger = createLogger("agentx/RemoteRepository");
+var RemoteRepository = class {
+  constructor(options) {
+    __publicField(this, "client");
+    this.client = createHttpClient({
+      baseUrl: options.serverUrl,
+      headers: options.headers
+    });
+    logger.debug("RemoteRepository created", { serverUrl: options.serverUrl });
+  }
+  // ==================== Definition ====================
+  async saveDefinition(record) {
+    await this.client.put(`definitions/${record.name}`, { json: record });
+  }
+  async findDefinitionByName(name) {
+    try {
+      const result = await this.client.get(`definitions/${name}`).json();
+      return this.parseDefinitionRecord(result);
+    } catch (error) {
+      if (this.isNotFound(error)) return null;
+      throw error;
+    }
+  }
+  async findAllDefinitions() {
+    const result = await this.client.get("definitions").json();
+    return result.map((r) => this.parseDefinitionRecord(r));
+  }
+  async deleteDefinition(name) {
+    await this.client.delete(`definitions/${name}`);
+  }
+  async definitionExists(name) {
+    try {
+      await this.client.head(`definitions/${name}`);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  // ==================== Image ====================
+  async saveImage(record) {
+    await this.client.put(`images/${record.imageId}`, { json: record });
+  }
+  async findImageById(imageId) {
+    try {
+      const result = await this.client.get(`images/${imageId}`).json();
+      return this.parseImageRecord(result);
+    } catch (error) {
+      if (this.isNotFound(error)) return null;
+      throw error;
+    }
+  }
+  async findAllImages() {
+    const result = await this.client.get("images").json();
+    return result.map((r) => this.parseImageRecord(r));
+  }
+  async deleteImage(imageId) {
+    await this.client.delete(`images/${imageId}`);
+  }
+  async imageExists(imageId) {
+    try {
+      await this.client.head(`images/${imageId}`);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  // ==================== Session ====================
+  async saveSession(record) {
+    await this.client.put(`sessions/${record.sessionId}`, { json: record });
+  }
+  async findSessionById(sessionId) {
+    try {
+      const result = await this.client.get(`sessions/${sessionId}`).json();
+      return this.parseSessionRecord(result);
+    } catch (error) {
+      if (this.isNotFound(error)) return null;
+      throw error;
+    }
+  }
+  async findSessionsByImageId(imageId) {
+    const result = await this.client.get(`images/${imageId}/sessions`).json();
+    return result.map((r) => this.parseSessionRecord(r));
+  }
+  async findSessionsByUserId(userId) {
+    const result = await this.client.get(`users/${userId}/sessions`).json();
+    return result.map((r) => this.parseSessionRecord(r));
+  }
+  async findAllSessions() {
+    const result = await this.client.get("sessions").json();
+    return result.map((r) => this.parseSessionRecord(r));
+  }
+  async deleteSession(sessionId) {
+    await this.client.delete(`sessions/${sessionId}`);
+  }
+  async deleteSessionsByImageId(imageId) {
+    await this.client.delete(`images/${imageId}/sessions`);
+  }
+  async sessionExists(sessionId) {
+    try {
+      await this.client.head(`sessions/${sessionId}`);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  // ==================== Message ====================
+  /**
+   * Save message - noop in browser
+   *
+   * Messages are persisted by server-side SessionCollector.
+   * Browser-side calls this but it does nothing to avoid duplicate saves.
+   */
+  async saveMessage(_record) {
+    logger.debug("saveMessage called (noop in browser)");
+  }
+  async findMessageById(messageId) {
+    try {
+      const result = await this.client.get(`messages/${messageId}`).json();
+      return this.parseMessageRecord(result);
+    } catch (error) {
+      if (this.isNotFound(error)) return null;
+      throw error;
+    }
+  }
+  async findMessagesBySessionId(sessionId) {
+    const result = await this.client.get(`sessions/${sessionId}/messages`).json();
+    return result.map((r) => this.parseMessageRecord(r));
+  }
+  async deleteMessage(messageId) {
+    await this.client.delete(`messages/${messageId}`);
+  }
+  async deleteMessagesBySessionId(sessionId) {
+    await this.client.delete(`sessions/${sessionId}/messages`);
+  }
+  async countMessagesBySessionId(sessionId) {
+    const result = await this.client.get(`sessions/${sessionId}/messages/count`).json();
+    return result.count;
+  }
+  // ==================== Container ====================
+  async saveContainer(record) {
+    await this.client.put(`containers/${record.containerId}`, { json: record });
+  }
+  async findContainerById(containerId) {
+    try {
+      const result = await this.client.get(`containers/${containerId}`).json();
+      return this.parseContainerRecord(result);
+    } catch (error) {
+      if (this.isNotFound(error)) return null;
+      throw error;
+    }
+  }
+  async findAllContainers() {
+    const result = await this.client.get("containers").json();
+    return result.map((r) => this.parseContainerRecord(r));
+  }
+  async deleteContainer(containerId) {
+    await this.client.delete(`containers/${containerId}`);
+  }
+  async containerExists(containerId) {
+    try {
+      await this.client.head(`containers/${containerId}`);
+      return true;
+    } catch {
+      return false;
+    }
+  }
+  // ==================== Helpers ====================
+  isNotFound(error) {
+    return error?.response?.status === 404;
+  }
+  parseDefinitionRecord(raw) {
+    return {
+      ...raw,
+      createdAt: new Date(raw.createdAt),
+      updatedAt: new Date(raw.updatedAt)
+    };
+  }
+  parseImageRecord(raw) {
+    return {
+      ...raw,
+      createdAt: new Date(raw.createdAt)
+    };
+  }
+  parseSessionRecord(raw) {
+    return {
+      ...raw,
+      createdAt: new Date(raw.createdAt),
+      updatedAt: new Date(raw.updatedAt)
+    };
+  }
+  parseMessageRecord(raw) {
+    return {
+      ...raw,
+      createdAt: new Date(raw.createdAt)
+    };
+  }
+  parseContainerRecord(raw) {
+    return {
+      ...raw
+      // ContainerRecord uses number timestamps, no conversion needed
+    };
+  }
+};
+var _BrowserLogger = class _BrowserLogger {
+  constructor(name, options = {}) {
+    __publicField(this, "name");
+    __publicField(this, "level");
+    __publicField(this, "collapsed");
+    this.name = name;
+    this.level = options.level ?? LogLevel.INFO;
+    this.collapsed = options.collapsed ?? true;
+  }
+  debug(message, context) {
+    if (this.isDebugEnabled()) {
+      this.log("DEBUG", message, context);
+    }
+  }
+  info(message, context) {
+    if (this.isInfoEnabled()) {
+      this.log("INFO", message, context);
+    }
+  }
+  warn(message, context) {
+    if (this.isWarnEnabled()) {
+      this.log("WARN", message, context);
+    }
+  }
+  error(message, context) {
+    if (this.isErrorEnabled()) {
+      if (message instanceof Error) {
+        this.log("ERROR", message.message, { ...context, stack: message.stack });
+      } else {
+        this.log("ERROR", message, context);
+      }
+    }
+  }
+  isDebugEnabled() {
+    return this.level <= LogLevel.DEBUG;
+  }
+  isInfoEnabled() {
+    return this.level <= LogLevel.INFO;
+  }
+  isWarnEnabled() {
+    return this.level <= LogLevel.WARN;
+  }
+  isErrorEnabled() {
+    return this.level <= LogLevel.ERROR;
+  }
+  log(level, message, context) {
+    const timestamp = (/* @__PURE__ */ new Date()).toLocaleTimeString();
+    const consoleMethod = this.getConsoleMethod(level);
+    const badgeStyle = _BrowserLogger.BADGE_STYLES[level];
+    const format = `%c${timestamp} %c${level}%c %c[${this.name}]%c ${message}`;
+    const styles = [
+      _BrowserLogger.STYLES.TIMESTAMP,
+      badgeStyle,
+      "",
+      // reset after badge
+      _BrowserLogger.STYLES.NAME,
+      _BrowserLogger.STYLES.MESSAGE
+    ];
+    if (context && Object.keys(context).length > 0) {
+      const groupMethod = this.collapsed ? console.groupCollapsed : console.group;
+      groupMethod.call(console, format, ...styles);
+      console.log("Context:", context);
+      console.groupEnd();
+    } else {
+      consoleMethod(format, ...styles);
+    }
+  }
+  getConsoleMethod(level) {
+    switch (level) {
+      case "DEBUG":
+        return console.debug.bind(console);
+      case "INFO":
+        return console.info.bind(console);
+      case "WARN":
+        return console.warn.bind(console);
+      case "ERROR":
+        return console.error.bind(console);
+      default:
+        return console.log.bind(console);
+    }
+  }
+};
+// CSS styles for different log levels
+__publicField(_BrowserLogger, "STYLES", {
+  DEBUG: "color: #6B7280; font-weight: normal;",
+  // gray
+  INFO: "color: #10B981; font-weight: normal;",
+  // green
+  WARN: "color: #F59E0B; font-weight: bold;",
+  // amber
+  ERROR: "color: #EF4444; font-weight: bold;",
+  // red
+  TIMESTAMP: "color: #9CA3AF; font-weight: normal;",
+  // light gray
+  NAME: "color: #8B5CF6; font-weight: normal;",
+  // purple
+  MESSAGE: "color: inherit; font-weight: normal;"
+});
+// Badge styles for level indicators
+__publicField(_BrowserLogger, "BADGE_STYLES", {
+  DEBUG: "background: #E5E7EB; color: #374151; padding: 2px 6px; border-radius: 3px; font-size: 11px;",
+  INFO: "background: #D1FAE5; color: #065F46; padding: 2px 6px; border-radius: 3px; font-size: 11px;",
+  WARN: "background: #FEF3C7; color: #92400E; padding: 2px 6px; border-radius: 3px; font-size: 11px;",
+  ERROR: "background: #FEE2E2; color: #991B1B; padding: 2px 6px; border-radius: 3px; font-size: 11px;"
+});
+var BrowserLogger = _BrowserLogger;
+
+// src/runtime/sse/logger/BrowserLoggerFactory.ts
+var BrowserLoggerFactory = class {
+  constructor(options = {}) {
+    __publicField(this, "options");
+    __publicField(this, "loggers", /* @__PURE__ */ new Map());
+    this.options = options;
+  }
+  getLogger(name) {
+    if (this.loggers.has(name)) {
+      return this.loggers.get(name);
+    }
+    const logger3 = new BrowserLogger(name, this.options);
+    this.loggers.set(name, logger3);
+    return logger3;
+  }
+};
+var logger2 = createLogger("agentx/RemoteAgentIdResolver");
+var RemoteAgentIdResolver = class {
+  constructor(options) {
+    __publicField(this, "client");
+    this.client = createHttpClient({
+      baseUrl: options.serverUrl,
+      headers: options.headers
+    });
+  }
+  async resolveForRun(imageId, containerId) {
+    logger2.debug("Resolving agent ID for run", { imageId, containerId });
+    const response = await this.client.post(`images/${imageId}/run`, {
+      json: { containerId }
+    }).json();
+    logger2.info("Agent ID resolved for run", { imageId, agentId: response.agentId });
+    return response.agentId;
+  }
+  async resolveForResume(sessionId, containerId) {
+    logger2.debug("Resolving agent ID for resume", { sessionId, containerId });
+    const response = await this.client.post(`sessions/${sessionId}/resume`, {
+      json: { containerId }
+    }).json();
+    logger2.info("Agent ID resolved for resume", { sessionId, agentId: response.agentId });
+    return response.agentId;
+  }
+};
+
+// src/runtime/sse/SSERuntime.ts
+var noopSandbox = {
+  name: "browser-noop",
+  workspace: {
+    id: "noop",
+    name: "noop",
+    path: ""
+    // Browser has no local workspace
+  },
+  llm: {
+    name: "noop",
+    provide: () => ({})
+  }
+};
+var SSERuntime = class {
+  constructor(config) {
+    __publicField(this, "name", "sse");
+    __publicField(this, "repository");
+    __publicField(this, "loggerFactory");
+    __publicField(this, "agentIdResolver");
+    __publicField(this, "serverUrl");
+    __publicField(this, "headers");
+    __publicField(this, "sseParams");
+    this.serverUrl = config.serverUrl.replace(/\/+$/, "");
+    this.headers = config.headers ?? {};
+    this.sseParams = config.sseParams ?? {};
+    this.loggerFactory = new BrowserLoggerFactory({
+      collapsed: true
+    });
+    setLoggerFactory(this.loggerFactory);
+    this.repository = new RemoteRepository({
+      serverUrl: this.serverUrl,
+      headers: this.headers
+    });
+    this.agentIdResolver = new RemoteAgentIdResolver({
+      serverUrl: this.serverUrl,
+      headers: this.headers
+    });
+  }
+  createSandbox(_containerId) {
+    return noopSandbox;
+  }
+  createDriver(_definition, context, _sandbox) {
+    const driver = createSSEDriver({
+      serverUrl: this.serverUrl,
+      agentId: context.agentId,
+      headers: this.headers,
+      sseParams: this.sseParams
+    });
+    return {
+      ...driver,
+      sandbox: noopSandbox
+    };
+  }
+  createLogger(name) {
+    return this.loggerFactory.getLogger(name);
+  }
+};
+function sseRuntime(config) {
+  return new SSERuntime(config);
+}
+var createSSERuntime = sseRuntime;
+
+export { BrowserLogger, BrowserLoggerFactory, RemoteRepository, SSERuntime, createSSEDriver, createSSERuntime, sseRuntime };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map