@copilotkitnext/core 0.0.11 → 0.0.13-alpha.0

package/dist/index.js

@@ -20,18 +20,22 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AgentRegistry: () => AgentRegistry,
+  ContextStore: () => ContextStore,
   CopilotKitCore: () => CopilotKitCore,
   CopilotKitCoreErrorCode: () => CopilotKitCoreErrorCode,
   CopilotKitCoreRuntimeConnectionStatus: () => CopilotKitCoreRuntimeConnectionStatus,
   ProxiedCopilotRuntimeAgent: () => ProxiedCopilotRuntimeAgent,
+  RunHandler: () => RunHandler,
+  SuggestionEngine: () => SuggestionEngine,
   ToolCallStatus: () => ToolCallStatus,
   completePartialMarkdown: () => completePartialMarkdown
 });
 module.exports = __toCommonJS(index_exports);
 
-// src/core.ts
-var import_shared = require("@copilotkitnext/shared");
+// src/core/agent-registry.ts
 var import_client2 = require("@ag-ui/client");
+var import_shared = require("@copilotkitnext/shared");
 
 // src/agent.ts
 var import_client = require("@ag-ui/client");
@@ -53,179 +57,133 @@ var ProxiedCopilotRuntimeAgent = class extends import_client.HttpAgent {
   }
 };
 
-// src/core.ts
-var import_zod_to_json_schema = require("zod-to-json-schema");
-var CopilotKitCoreErrorCode = /* @__PURE__ */ ((CopilotKitCoreErrorCode2) => {
-  CopilotKitCoreErrorCode2["RUNTIME_INFO_FETCH_FAILED"] = "runtime_info_fetch_failed";
-  CopilotKitCoreErrorCode2["AGENT_CONNECT_FAILED"] = "agent_connect_failed";
-  CopilotKitCoreErrorCode2["AGENT_RUN_FAILED"] = "agent_run_failed";
-  CopilotKitCoreErrorCode2["AGENT_RUN_FAILED_EVENT"] = "agent_run_failed_event";
-  CopilotKitCoreErrorCode2["AGENT_RUN_ERROR_EVENT"] = "agent_run_error_event";
-  CopilotKitCoreErrorCode2["TOOL_ARGUMENT_PARSE_FAILED"] = "tool_argument_parse_failed";
-  CopilotKitCoreErrorCode2["TOOL_HANDLER_FAILED"] = "tool_handler_failed";
-  return CopilotKitCoreErrorCode2;
-})(CopilotKitCoreErrorCode || {});
-var CopilotKitCoreRuntimeConnectionStatus = /* @__PURE__ */ ((CopilotKitCoreRuntimeConnectionStatus2) => {
-  CopilotKitCoreRuntimeConnectionStatus2["Disconnected"] = "disconnected";
-  CopilotKitCoreRuntimeConnectionStatus2["Connected"] = "connected";
-  CopilotKitCoreRuntimeConnectionStatus2["Connecting"] = "connecting";
-  CopilotKitCoreRuntimeConnectionStatus2["Error"] = "error";
-  return CopilotKitCoreRuntimeConnectionStatus2;
-})(CopilotKitCoreRuntimeConnectionStatus || {});
-var CopilotKitCore = class {
-  headers;
-  properties;
-  _context = {};
+// src/core/agent-registry.ts
+var AgentRegistry = class {
+  constructor(core) {
+    this.core = core;
+  }
   _agents = {};
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  _tools = [];
   localAgents = {};
   remoteAgents = {};
-  subscribers = /* @__PURE__ */ new Set();
   _runtimeUrl;
   _runtimeVersion;
   _runtimeConnectionStatus = "disconnected" /* Disconnected */;
-  constructor({
-    runtimeUrl,
-    headers = {},
-    properties = {},
-    agents = {},
-    tools = []
-  }) {
-    this.headers = headers;
-    this.properties = properties;
+  /**
+   * Get all agents as a readonly record
+   */
+  get agents() {
+    return this._agents;
+  }
+  get runtimeUrl() {
+    return this._runtimeUrl;
+  }
+  get runtimeVersion() {
+    return this._runtimeVersion;
+  }
+  get runtimeConnectionStatus() {
+    return this._runtimeConnectionStatus;
+  }
+  /**
+   * Initialize agents from configuration
+   */
+  initialize(agents) {
     this.localAgents = this.assignAgentIds(agents);
     this.applyHeadersToAgents(this.localAgents);
     this._agents = this.localAgents;
-    this._tools = tools;
-    this.setRuntimeUrl(runtimeUrl);
   }
-  applyHeadersToAgent(agent) {
-    if (agent instanceof import_client2.HttpAgent) {
-      agent.headers = { ...this.headers };
+  /**
+   * Set the runtime URL and update connection
+   */
+  setRuntimeUrl(runtimeUrl) {
+    const normalizedRuntimeUrl = runtimeUrl ? runtimeUrl.replace(/\/$/, "") : void 0;
+    if (this._runtimeUrl === normalizedRuntimeUrl) {
+      return;
     }
+    this._runtimeUrl = normalizedRuntimeUrl;
+    void this.updateRuntimeConnection();
   }
-  applyHeadersToAgents(agents) {
-    Object.values(agents).forEach((agent) => {
-      this.applyHeadersToAgent(agent);
-    });
-  }
-  assignAgentIds(agents) {
+  /**
+   * Set all agents at once (for development use)
+   */
+  setAgents__unsafe_dev_only(agents) {
     Object.entries(agents).forEach(([id, agent]) => {
-      if (agent && !agent.agentId) {
-        agent.agentId = id;
+      if (agent) {
+        this.validateAndAssignAgentId(id, agent);
       }
     });
-    return agents;
+    this.localAgents = agents;
+    this._agents = { ...this.localAgents, ...this.remoteAgents };
+    this.applyHeadersToAgents(this._agents);
+    void this.notifyAgentsChanged();
   }
-  async notifySubscribers(handler, errorMessage) {
-    await Promise.all(
-      Array.from(this.subscribers).map(async (subscriber) => {
-        try {
-          await handler(subscriber);
-        } catch (error) {
-          import_shared.logger.error(errorMessage, error);
-        }
-      })
-    );
+  /**
+   * Add a single agent (for development use)
+   */
+  addAgent__unsafe_dev_only({ id, agent }) {
+    this.validateAndAssignAgentId(id, agent);
+    this.localAgents[id] = agent;
+    this.applyHeadersToAgent(agent);
+    this._agents = { ...this.localAgents, ...this.remoteAgents };
+    void this.notifyAgentsChanged();
   }
-  async emitError({
-    error,
-    code,
-    context = {}
-  }) {
-    await this.notifySubscribers(
-      (subscriber) => subscriber.onError?.({
-        copilotkit: this,
-        error,
-        code,
-        context
-      }),
-      "Subscriber onError error:"
-    );
+  /**
+   * Remove an agent by ID (for development use)
+   */
+  removeAgent__unsafe_dev_only(id) {
+    delete this.localAgents[id];
+    this._agents = { ...this.localAgents, ...this.remoteAgents };
+    void this.notifyAgentsChanged();
   }
-  resolveAgentId(agent, providedAgentId) {
-    if (providedAgentId) {
-      return providedAgentId;
-    }
-    if (agent.agentId) {
-      return agent.agentId;
+  /**
+   * Get an agent by ID
+   */
+  getAgent(id) {
+    if (id in this._agents) {
+      return this._agents[id];
     }
-    const found = Object.entries(this._agents).find(([, storedAgent]) => {
-      return storedAgent === agent;
-    });
-    if (found) {
-      agent.agentId = found[0];
-      return found[0];
+    if (this.runtimeUrl !== void 0 && (this.runtimeConnectionStatus === "disconnected" /* Disconnected */ || this.runtimeConnectionStatus === "connecting" /* Connecting */)) {
+      return void 0;
     }
-    agent.agentId = import_shared.DEFAULT_AGENT_ID;
-    return import_shared.DEFAULT_AGENT_ID;
+    console.warn(`Agent ${id} not found`);
+    return void 0;
   }
   /**
-   * Snapshot accessors
+   * Apply current headers to an agent
    */
-  get context() {
-    return this._context;
-  }
-  get agents() {
-    return this._agents;
-  }
-  get tools() {
-    return this._tools;
-  }
-  get runtimeUrl() {
-    return this._runtimeUrl;
-  }
-  setRuntimeUrl(runtimeUrl) {
-    const normalizedRuntimeUrl = runtimeUrl ? runtimeUrl.replace(/\/$/, "") : void 0;
-    if (this._runtimeUrl === normalizedRuntimeUrl) {
-      return;
+  applyHeadersToAgent(agent) {
+    if (agent instanceof import_client2.HttpAgent) {
+      agent.headers = { ...this.core.headers };
     }
-    this._runtimeUrl = normalizedRuntimeUrl;
-    void this.updateRuntimeConnection();
-  }
-  get runtimeVersion() {
-    return this._runtimeVersion;
   }
-  get runtimeConnectionStatus() {
-    return this._runtimeConnectionStatus;
+  /**
+   * Apply current headers to all agents
+   */
+  applyHeadersToAgents(agents) {
+    Object.values(agents).forEach((agent) => {
+      this.applyHeadersToAgent(agent);
+    });
   }
   /**
-   * Runtime connection
+   * Update runtime connection and fetch remote agents
    */
   async updateRuntimeConnection() {
+    if (typeof window === "undefined") {
+      return;
+    }
     if (!this.runtimeUrl) {
       this._runtimeConnectionStatus = "disconnected" /* Disconnected */;
       this._runtimeVersion = void 0;
       this.remoteAgents = {};
       this._agents = this.localAgents;
-      await this.notifySubscribers(
-        (subscriber) => subscriber.onRuntimeConnectionStatusChanged?.({
-          copilotkit: this,
-          status: "disconnected" /* Disconnected */
-        }),
-        "Error in CopilotKitCore subscriber (onRuntimeConnectionStatusChanged):"
-      );
-      await this.notifySubscribers(
-        (subscriber) => subscriber.onAgentsChanged?.({
-          copilotkit: this,
-          agents: this._agents
-        }),
-        "Subscriber onAgentsChanged error:"
-      );
+      await this.notifyRuntimeStatusChanged("disconnected" /* Disconnected */);
+      await this.notifyAgentsChanged();
       return;
     }
     this._runtimeConnectionStatus = "connecting" /* Connecting */;
-    await this.notifySubscribers(
-      (subscriber) => subscriber.onRuntimeConnectionStatusChanged?.({
-        copilotkit: this,
-        status: "connecting" /* Connecting */
-      }),
-      "Error in CopilotKitCore subscriber (onRuntimeConnectionStatusChanged):"
-    );
+    await this.notifyRuntimeStatusChanged("connecting" /* Connecting */);
     try {
       const response = await fetch(`${this.runtimeUrl}/info`, {
-        headers: this.headers
+        headers: this.core.headers
       });
       const {
         version,
@@ -236,6 +194,7 @@ var CopilotKitCore = class {
           const agent = new ProxiedCopilotRuntimeAgent({
             runtimeUrl: this.runtimeUrl,
             agentId: id,
+            // Runtime agents always have their ID set correctly
             description
           });
           this.applyHeadersToAgent(agent);
@@ -246,45 +205,19 @@ var CopilotKitCore = class {
       this._agents = { ...this.localAgents, ...this.remoteAgents };
       this._runtimeConnectionStatus = "connected" /* Connected */;
       this._runtimeVersion = version;
-      await this.notifySubscribers(
-        (subscriber) => subscriber.onRuntimeConnectionStatusChanged?.({
-          copilotkit: this,
-          status: "connected" /* Connected */
-        }),
-        "Error in CopilotKitCore subscriber (onRuntimeConnectionStatusChanged):"
-      );
-      await this.notifySubscribers(
-        (subscriber) => subscriber.onAgentsChanged?.({
-          copilotkit: this,
-          agents: this._agents
-        }),
-        "Subscriber onAgentsChanged error:"
-      );
+      await this.notifyRuntimeStatusChanged("connected" /* Connected */);
+      await this.notifyAgentsChanged();
     } catch (error) {
       this._runtimeConnectionStatus = "error" /* Error */;
       this._runtimeVersion = void 0;
       this.remoteAgents = {};
       this._agents = this.localAgents;
-      await this.notifySubscribers(
-        (subscriber) => subscriber.onRuntimeConnectionStatusChanged?.({
-          copilotkit: this,
-          status: "error" /* Error */
-        }),
-        "Error in CopilotKitCore subscriber (onRuntimeConnectionStatusChanged):"
-      );
-      await this.notifySubscribers(
-        (subscriber) => subscriber.onAgentsChanged?.({
-          copilotkit: this,
-          agents: this._agents
-        }),
-        "Subscriber onAgentsChanged error:"
-      );
+      await this.notifyRuntimeStatusChanged("error" /* Error */);
+      await this.notifyAgentsChanged();
       const message = error instanceof Error ? error.message : JSON.stringify(error);
-      import_shared.logger.warn(
-        `Failed to load runtime info (${this.runtimeUrl}/info): ${message}`
-      );
+      import_shared.logger.warn(`Failed to load runtime info (${this.runtimeUrl}/info): ${message}`);
       const runtimeError = error instanceof Error ? error : new Error(String(error));
-      await this.emitError({
+      await this.core.emitError({
         error: runtimeError,
         code: "runtime_info_fetch_failed" /* RUNTIME_INFO_FETCH_FAILED */,
         context: {
@@ -294,118 +227,479 @@ var CopilotKitCore = class {
     }
   }
   /**
-   * Configuration updates
+   * Assign agent IDs to a record of agents
    */
-  setHeaders(headers) {
-    this.headers = headers;
-    this.applyHeadersToAgents(this._agents);
-    void this.notifySubscribers(
-      (subscriber) => subscriber.onHeadersChanged?.({
-        copilotkit: this,
-        headers: this.headers
-      }),
-      "Subscriber onHeadersChanged error:"
-    );
-  }
-  setProperties(properties) {
-    this.properties = properties;
-    void this.notifySubscribers(
-      (subscriber) => subscriber.onPropertiesChanged?.({
-        copilotkit: this,
-        properties: this.properties
-      }),
-      "Subscriber onPropertiesChanged error:"
-    );
-  }
-  setAgents(agents) {
-    this.localAgents = this.assignAgentIds(agents);
-    this._agents = { ...this.localAgents, ...this.remoteAgents };
-    this.applyHeadersToAgents(this._agents);
-    void this.notifySubscribers(
-      (subscriber) => subscriber.onAgentsChanged?.({
-        copilotkit: this,
-        agents: this._agents
-      }),
-      "Subscriber onAgentsChanged error:"
-    );
+  assignAgentIds(agents) {
+    Object.entries(agents).forEach(([id, agent]) => {
+      if (agent) {
+        this.validateAndAssignAgentId(id, agent);
+      }
+    });
+    return agents;
   }
-  addAgent({ id, agent }) {
-    this.localAgents[id] = agent;
+  /**
+   * Validate and assign an agent ID
+   */
+  validateAndAssignAgentId(registrationId, agent) {
+    if (agent.agentId && agent.agentId !== registrationId) {
+      throw new Error(
+        `Agent registration mismatch: Agent with ID "${agent.agentId}" cannot be registered under key "${registrationId}". The agent ID must match the registration key or be undefined.`
+      );
+    }
     if (!agent.agentId) {
-      agent.agentId = id;
+      agent.agentId = registrationId;
     }
-    this.applyHeadersToAgent(agent);
-    this._agents = { ...this.localAgents, ...this.remoteAgents };
-    void this.notifySubscribers(
-      (subscriber) => subscriber.onAgentsChanged?.({
-        copilotkit: this,
-        agents: this._agents
+  }
+  /**
+   * Notify subscribers of runtime status changes
+   */
+  async notifyRuntimeStatusChanged(status) {
+    await this.core.notifySubscribers(
+      (subscriber) => subscriber.onRuntimeConnectionStatusChanged?.({
+        copilotkit: this.core,
+        status
       }),
-      "Subscriber onAgentsChanged error:"
+      "Error in CopilotKitCore subscriber (onRuntimeConnectionStatusChanged):"
     );
   }
-  removeAgent(id) {
-    delete this.localAgents[id];
-    this._agents = { ...this.localAgents, ...this.remoteAgents };
-    void this.notifySubscribers(
+  /**
+   * Notify subscribers of agent changes
+   */
+  async notifyAgentsChanged() {
+    await this.core.notifySubscribers(
       (subscriber) => subscriber.onAgentsChanged?.({
-        copilotkit: this,
+        copilotkit: this.core,
         agents: this._agents
       }),
       "Subscriber onAgentsChanged error:"
     );
   }
-  getAgent(id) {
-    if (id in this._agents) {
-      return this._agents[id];
-    }
-    if (this.runtimeUrl !== void 0 && (this.runtimeConnectionStatus === "disconnected" /* Disconnected */ || this.runtimeConnectionStatus === "connecting" /* Connecting */)) {
-      return void 0;
-    } else {
-      console.warn(`Agent ${id} not found`);
-      return void 0;
-    }
+};
+
+// src/core/context-store.ts
+var import_shared2 = require("@copilotkitnext/shared");
+var ContextStore = class {
+  constructor(core) {
+    this.core = core;
+  }
+  _context = {};
+  /**
+   * Get all context entries as a readonly record
+   */
+  get context() {
+    return this._context;
   }
   /**
-   * Context management
+   * Add a new context entry
+   * @returns The ID of the created context entry
    */
   addContext({ description, value }) {
-    const id = (0, import_shared.randomUUID)();
+    const id = (0, import_shared2.randomUUID)();
     this._context[id] = { description, value };
-    void this.notifySubscribers(
-      (subscriber) => subscriber.onContextChanged?.({
-        copilotkit: this,
-        context: this._context
-      }),
-      "Subscriber onContextChanged error:"
-    );
+    void this.notifySubscribers();
     return id;
   }
+  /**
+   * Remove a context entry by ID
+   */
   removeContext(id) {
     delete this._context[id];
-    void this.notifySubscribers(
+    void this.notifySubscribers();
+  }
+  /**
+   * Notify all subscribers of context changes
+   */
+  async notifySubscribers() {
+    await this.core.notifySubscribers(
       (subscriber) => subscriber.onContextChanged?.({
-        copilotkit: this,
+        copilotkit: this.core,
         context: this._context
       }),
       "Subscriber onContextChanged error:"
     );
   }
+};
+
+// src/core/suggestion-engine.ts
+var import_shared3 = require("@copilotkitnext/shared");
+var SuggestionEngine = class {
+  constructor(core) {
+    this.core = core;
+  }
+  _suggestionsConfig = {};
+  _suggestions = {};
+  _runningSuggestions = {};
+  /**
+   * Initialize with suggestion configs
+   */
+  initialize(suggestionsConfig) {
+    for (const config of suggestionsConfig) {
+      this._suggestionsConfig[(0, import_shared3.randomUUID)()] = config;
+    }
+  }
+  /**
+   * Add a suggestion configuration
+   * @returns The ID of the created config
+   */
+  addSuggestionsConfig(config) {
+    const id = (0, import_shared3.randomUUID)();
+    this._suggestionsConfig[id] = config;
+    void this.notifySuggestionsConfigChanged();
+    return id;
+  }
+  /**
+   * Remove a suggestion configuration by ID
+   */
+  removeSuggestionsConfig(id) {
+    delete this._suggestionsConfig[id];
+    void this.notifySuggestionsConfigChanged();
+  }
+  /**
+   * Reload suggestions for a specific agent
+   * This triggers generation of new suggestions based on current configs
+   */
+  reloadSuggestions(agentId) {
+    this.clearSuggestions(agentId);
+    const agent = this.core.getAgent(agentId);
+    if (!agent) {
+      return;
+    }
+    const messageCount = agent.messages?.length ?? 0;
+    let hasAnySuggestions = false;
+    for (const config of Object.values(this._suggestionsConfig)) {
+      if (config.consumerAgentId !== void 0 && config.consumerAgentId !== "*" && config.consumerAgentId !== agentId) {
+        continue;
+      }
+      if (!this.shouldShowSuggestions(config, messageCount)) {
+        continue;
+      }
+      const suggestionId = (0, import_shared3.randomUUID)();
+      if (isDynamicSuggestionsConfig(config)) {
+        if (!hasAnySuggestions) {
+          hasAnySuggestions = true;
+          void this.notifySuggestionsStartedLoading(agentId);
+        }
+        void this.generateSuggestions(suggestionId, config, agentId);
+      } else if (isStaticSuggestionsConfig(config)) {
+        this.addStaticSuggestions(suggestionId, config, agentId);
+      }
+    }
+  }
+  /**
+   * Clear all suggestions for a specific agent
+   */
+  clearSuggestions(agentId) {
+    const runningAgents = this._runningSuggestions[agentId];
+    if (runningAgents) {
+      for (const agent of runningAgents) {
+        agent.abortRun();
+      }
+      delete this._runningSuggestions[agentId];
+    }
+    this._suggestions[agentId] = {};
+    void this.notifySuggestionsChanged(agentId, []);
+  }
+  /**
+   * Get current suggestions for an agent
+   */
+  getSuggestions(agentId) {
+    const suggestions = Object.values(this._suggestions[agentId] ?? {}).flat();
+    const isLoading = (this._runningSuggestions[agentId]?.length ?? 0) > 0;
+    return { suggestions, isLoading };
+  }
+  /**
+   * Generate suggestions using a provider agent
+   */
+  async generateSuggestions(suggestionId, config, consumerAgentId) {
+    let agent = void 0;
+    try {
+      const suggestionsProviderAgent = this.core.getAgent(config.providerAgentId ?? "default");
+      if (!suggestionsProviderAgent) {
+        throw new Error(`Suggestions provider agent not found: ${config.providerAgentId}`);
+      }
+      const suggestionsConsumerAgent = this.core.getAgent(consumerAgentId);
+      if (!suggestionsConsumerAgent) {
+        throw new Error(`Suggestions consumer agent not found: ${consumerAgentId}`);
+      }
+      const clonedAgent = suggestionsProviderAgent.clone();
+      agent = clonedAgent;
+      agent.agentId = suggestionId;
+      agent.threadId = suggestionId;
+      agent.messages = JSON.parse(JSON.stringify(suggestionsConsumerAgent.messages));
+      agent.state = JSON.parse(JSON.stringify(suggestionsConsumerAgent.state));
+      this._suggestions[consumerAgentId] = {
+        ...this._suggestions[consumerAgentId] ?? {},
+        [suggestionId]: []
+      };
+      this._runningSuggestions[consumerAgentId] = [...this._runningSuggestions[consumerAgentId] ?? [], agent];
+      agent.addMessage({
+        id: suggestionId,
+        role: "user",
+        content: [
+          `Suggest what the user could say next. Provide clear, highly relevant suggestions by calling the \`copilotkitSuggest\` tool.`,
+          `Provide at least ${config.minSuggestions ?? 1} and at most ${config.maxSuggestions ?? 3} suggestions.`,
+          `The user has the following tools available: ${JSON.stringify(this.core.buildFrontendTools(consumerAgentId))}.`,
+          ` ${config.instructions}`
+        ].join("\n")
+      });
+      await agent.runAgent(
+        {
+          context: Object.values(this.core.context),
+          forwardedProps: {
+            ...this.core.properties,
+            toolChoice: { type: "function", function: { name: "copilotkitSuggest" } }
+          },
+          tools: [SUGGEST_TOOL]
+        },
+        {
+          onMessagesChanged: ({ messages }) => {
+            this.extractSuggestions(messages, suggestionId, consumerAgentId, true);
+          }
+        }
+      );
+    } catch (error) {
+      console.warn("Error generating suggestions:", error);
+    } finally {
+      this.finalizeSuggestions(suggestionId, consumerAgentId);
+      const runningAgents = this._runningSuggestions[consumerAgentId];
+      if (agent && runningAgents) {
+        const filteredAgents = runningAgents.filter((a) => a !== agent);
+        this._runningSuggestions[consumerAgentId] = filteredAgents;
+        if (filteredAgents.length === 0) {
+          delete this._runningSuggestions[consumerAgentId];
+          await this.notifySuggestionsFinishedLoading(consumerAgentId);
+        }
+      }
+    }
+  }
+  /**
+   * Finalize suggestions by marking them as no longer loading
+   */
+  finalizeSuggestions(suggestionId, consumerAgentId) {
+    const agentSuggestions = this._suggestions[consumerAgentId];
+    const currentSuggestions = agentSuggestions?.[suggestionId];
+    if (agentSuggestions && currentSuggestions && currentSuggestions.length > 0) {
+      const finalizedSuggestions = currentSuggestions.filter((suggestion) => suggestion.title !== "" || suggestion.message !== "").map((suggestion) => ({
+        ...suggestion,
+        isLoading: false
+      }));
+      if (finalizedSuggestions.length > 0) {
+        agentSuggestions[suggestionId] = finalizedSuggestions;
+      } else {
+        delete agentSuggestions[suggestionId];
+      }
+      const allSuggestions = Object.values(this._suggestions[consumerAgentId] ?? {}).flat();
+      void this.notifySuggestionsChanged(consumerAgentId, allSuggestions, "finalized");
+    }
+  }
+  /**
+   * Extract suggestions from messages (called during streaming)
+   */
+  extractSuggestions(messages, suggestionId, consumerAgentId, isRunning) {
+    const idx = messages.findIndex((message) => message.id === suggestionId);
+    if (idx == -1) {
+      return;
+    }
+    const suggestions = [];
+    const newMessages = messages.slice(idx + 1);
+    for (const message of newMessages) {
+      if (message.role === "assistant" && message.toolCalls) {
+        for (const toolCall of message.toolCalls) {
+          if (toolCall.function.name === "copilotkitSuggest") {
+            const fullArgs = Array.isArray(toolCall.function.arguments) ? toolCall.function.arguments.join("") : toolCall.function.arguments;
+            const parsed = (0, import_shared3.partialJSONParse)(fullArgs);
+            if (parsed && typeof parsed === "object" && "suggestions" in parsed) {
+              const parsedSuggestions = parsed.suggestions;
+              if (Array.isArray(parsedSuggestions)) {
+                for (const item of parsedSuggestions) {
+                  if (item && typeof item === "object" && "title" in item) {
+                    suggestions.push({
+                      title: item.title ?? "",
+                      message: item.message ?? "",
+                      isLoading: false
+                      // Will be set correctly below
+                    });
+                  }
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+    if (isRunning && suggestions.length > 0) {
+      suggestions[suggestions.length - 1].isLoading = true;
+    }
+    const agentSuggestions = this._suggestions[consumerAgentId];
+    if (agentSuggestions) {
+      agentSuggestions[suggestionId] = suggestions;
+      const allSuggestions = Object.values(this._suggestions[consumerAgentId] ?? {}).flat();
+      void this.notifySuggestionsChanged(consumerAgentId, allSuggestions, "suggestions changed");
+    }
+  }
+  /**
+   * Notify subscribers of suggestions config changes
+   */
+  async notifySuggestionsConfigChanged() {
+    await this.core.notifySubscribers(
+      (subscriber) => subscriber.onSuggestionsConfigChanged?.({
+        copilotkit: this.core,
+        suggestionsConfig: this._suggestionsConfig
+      }),
+      "Subscriber onSuggestionsConfigChanged error:"
+    );
+  }
+  /**
+   * Notify subscribers of suggestions changes
+   */
+  async notifySuggestionsChanged(agentId, suggestions, context = "") {
+    await this.core.notifySubscribers(
+      (subscriber) => subscriber.onSuggestionsChanged?.({
+        copilotkit: this.core,
+        agentId,
+        suggestions
+      }),
+      `Subscriber onSuggestionsChanged error: ${context}`
+    );
+  }
+  /**
+   * Notify subscribers that suggestions started loading
+   */
+  async notifySuggestionsStartedLoading(agentId) {
+    await this.core.notifySubscribers(
+      (subscriber) => subscriber.onSuggestionsStartedLoading?.({
+        copilotkit: this.core,
+        agentId
+      }),
+      "Subscriber onSuggestionsStartedLoading error:"
+    );
+  }
+  /**
+   * Notify subscribers that suggestions finished loading
+   */
+  async notifySuggestionsFinishedLoading(agentId) {
+    await this.core.notifySubscribers(
+      (subscriber) => subscriber.onSuggestionsFinishedLoading?.({
+        copilotkit: this.core,
+        agentId
+      }),
+      "Subscriber onSuggestionsFinishedLoading error:"
+    );
+  }
+  /**
+   * Check if suggestions should be shown based on availability and message count
+   */
+  shouldShowSuggestions(config, messageCount) {
+    const availability = config.available;
+    if (!availability) {
+      if (isDynamicSuggestionsConfig(config)) {
+        return messageCount > 0;
+      } else {
+        return messageCount === 0;
+      }
+    }
+    switch (availability) {
+      case "disabled":
+        return false;
+      case "before-first-message":
+        return messageCount === 0;
+      case "after-first-message":
+        return messageCount > 0;
+      case "always":
+        return true;
+      default:
+        return false;
+    }
+  }
+  /**
+   * Add static suggestions directly without AI generation
+   */
+  addStaticSuggestions(suggestionId, config, consumerAgentId) {
+    const suggestions = config.suggestions.map((s) => ({
+      ...s,
+      isLoading: false
+    }));
+    this._suggestions[consumerAgentId] = {
+      ...this._suggestions[consumerAgentId] ?? {},
+      [suggestionId]: suggestions
+    };
+    const allSuggestions = Object.values(this._suggestions[consumerAgentId] ?? {}).flat();
+    void this.notifySuggestionsChanged(consumerAgentId, allSuggestions, "static suggestions added");
+  }
+};
+function isDynamicSuggestionsConfig(config) {
+  return "instructions" in config;
+}
+function isStaticSuggestionsConfig(config) {
+  return "suggestions" in config;
+}
+var SUGGEST_TOOL = {
+  name: "copilotkitSuggest",
+  description: "Suggest what the user could say next",
+  parameters: {
+    type: "object",
+    properties: {
+      suggestions: {
+        type: "array",
+        description: "List of suggestions shown to the user as buttons.",
+        items: {
+          type: "object",
+          properties: {
+            title: {
+              type: "string",
+              description: "The title of the suggestion. This is shown as a button and should be short."
+            },
+            message: {
+              type: "string",
+              description: "The message to send when the suggestion is clicked. This should be a clear, complete sentence and will be sent as an instruction to the AI."
+            }
+          },
+          required: ["title", "message"],
+          additionalProperties: false
+        }
+      }
+    },
+    required: ["suggestions"],
+    additionalProperties: false
+  }
+};
+
+// src/core/run-handler.ts
+var import_client3 = require("@ag-ui/client");
+var import_shared4 = require("@copilotkitnext/shared");
+var import_zod_to_json_schema = require("zod-to-json-schema");
+var RunHandler = class {
+  constructor(core) {
+    this.core = core;
+  }
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  _tools = [];
+  /**
+   * Get all tools as a readonly array
+   */
+  get tools() {
+    return this._tools;
+  }
+  /**
+   * Initialize with tools
+   */
+  initialize(tools) {
+    this._tools = tools;
+  }
   /**
-   * Tool management
+   * Add a tool to the registry
    */
   addTool(tool) {
-    const existingToolIndex = this._tools.findIndex(
-      (t) => t.name === tool.name && t.agentId === tool.agentId
-    );
+    const existingToolIndex = this._tools.findIndex((t) => t.name === tool.name && t.agentId === tool.agentId);
     if (existingToolIndex !== -1) {
-      import_shared.logger.warn(
-        `Tool already exists: '${tool.name}' for agent '${tool.agentId || "global"}', skipping.`
-      );
+      import_shared4.logger.warn(`Tool already exists: '${tool.name}' for agent '${tool.agentId || "global"}', skipping.`);
       return;
     }
     this._tools.push(tool);
   }
+  /**
+   * Remove a tool by name and optionally by agentId
+   */
   removeTool(id, agentId) {
     this._tools = this._tools.filter((tool) => {
       if (agentId !== void 0) {
@@ -422,9 +716,7 @@ var CopilotKitCore = class {
   getTool(params) {
     const { toolName, agentId } = params;
     if (agentId) {
-      const agentTool = this._tools.find(
-        (tool) => tool.name === toolName && tool.agentId === agentId
-      );
+      const agentTool = this._tools.find((tool) => tool.name === toolName && tool.agentId === agentId);
       if (agentTool) {
         return agentTool;
       }
@@ -438,43 +730,28 @@ var CopilotKitCore = class {
     this._tools = [...tools];
   }
   /**
-   * Subscription lifecycle
-   */
-  subscribe(subscriber) {
-    this.subscribers.add(subscriber);
-    return () => {
-      this.unsubscribe(subscriber);
-    };
-  }
-  unsubscribe(subscriber) {
-    this.subscribers.delete(subscriber);
-  }
-  /**
-   * Agent connectivity
+   * Connect an agent (establish initial connection)
    */
-  async connectAgent({
-    agent,
-    agentId
-  }) {
+  async connectAgent({ agent }) {
     try {
-      if (agent instanceof import_client2.HttpAgent) {
-        agent.headers = { ...this.headers };
+      if (agent instanceof import_client3.HttpAgent) {
+        agent.headers = { ...this.core.headers };
       }
       const runAgentResult = await agent.connectAgent(
         {
-          forwardedProps: this.properties,
-          tools: this.buildFrontendTools(agentId)
+          forwardedProps: this.core.properties,
+          tools: this.buildFrontendTools(agent.agentId)
         },
-        this.createAgentErrorSubscriber(agent, agentId)
+        this.createAgentErrorSubscriber(agent)
       );
-      return this.processAgentResult({ runAgentResult, agent, agentId });
+      return this.processAgentResult({ runAgentResult, agent });
     } catch (error) {
       const connectError = error instanceof Error ? error : new Error(String(error));
       const context = {};
-      if (agentId ?? agent.agentId) {
-        context.agentId = agentId ?? agent.agentId;
+      if (agent.agentId) {
+        context.agentId = agent.agentId;
       }
-      await this.emitError({
+      await this.core.emitError({
         error: connectError,
         code: "agent_connect_failed" /* AGENT_CONNECT_FAILED */,
         context
@@ -482,13 +759,13 @@ var CopilotKitCore = class {
       throw error;
     }
   }
-  async runAgent({
-    agent,
-    withMessages,
-    agentId
-  }) {
-    if (agent instanceof import_client2.HttpAgent) {
-      agent.headers = { ...this.headers };
+  /**
+   * Run an agent
+   */
+  async runAgent({ agent, withMessages }) {
+    void this.core.suggestionEngine.clearSuggestions(agent.agentId);
+    if (agent instanceof import_client3.HttpAgent) {
+      agent.headers = { ...this.core.headers };
     }
     if (withMessages) {
       agent.addMessages(withMessages);
@@ -496,22 +773,22 @@ var CopilotKitCore = class {
     try {
       const runAgentResult = await agent.runAgent(
         {
-          forwardedProps: this.properties,
-          tools: this.buildFrontendTools(agentId)
+          forwardedProps: this.core.properties,
+          tools: this.buildFrontendTools(agent.agentId)
         },
-        this.createAgentErrorSubscriber(agent, agentId)
+        this.createAgentErrorSubscriber(agent)
       );
-      return this.processAgentResult({ runAgentResult, agent, agentId });
+      return this.processAgentResult({ runAgentResult, agent });
     } catch (error) {
       const runError = error instanceof Error ? error : new Error(String(error));
       const context = {};
-      if (agentId ?? agent.agentId) {
-        context.agentId = agentId ?? agent.agentId;
+      if (agent.agentId) {
+        context.agentId = agent.agentId;
       }
       if (withMessages) {
         context.messageCount = withMessages.length;
       }
-      await this.emitError({
+      await this.core.emitError({
         error: runError,
         code: "agent_run_failed" /* AGENT_RUN_FAILED */,
         context
@@ -519,231 +796,35 @@ var CopilotKitCore = class {
       throw error;
     }
   }
+  /**
+   * Process agent result and execute tools
+   */
   async processAgentResult({
     runAgentResult,
-    agent,
-    agentId
+    agent
   }) {
     const { newMessages } = runAgentResult;
-    const effectiveAgentId = this.resolveAgentId(agent, agentId);
+    const agentId = agent.agentId;
     let needsFollowUp = false;
     for (const message of newMessages) {
       if (message.role === "assistant") {
         for (const toolCall of message.toolCalls || []) {
-          if (newMessages.findIndex(
-            (m) => m.role === "tool" && m.toolCallId === toolCall.id
-          ) === -1) {
+          if (newMessages.findIndex((m) => m.role === "tool" && m.toolCallId === toolCall.id) === -1) {
             const tool = this.getTool({
               toolName: toolCall.function.name,
-              agentId
+              agentId: agent.agentId
             });
             if (tool) {
-              if (tool?.agentId && tool.agentId !== agentId) {
-                continue;
-              }
-              let toolCallResult = "";
-              let errorMessage;
-              let isArgumentError = false;
-              if (tool?.handler) {
-                let parsedArgs;
-                try {
-                  parsedArgs = JSON.parse(toolCall.function.arguments);
-                } catch (error) {
-                  const parseError = error instanceof Error ? error : new Error(String(error));
-                  errorMessage = parseError.message;
-                  isArgumentError = true;
-                  await this.emitError({
-                    error: parseError,
-                    code: "tool_argument_parse_failed" /* TOOL_ARGUMENT_PARSE_FAILED */,
-                    context: {
-                      agentId: effectiveAgentId,
-                      toolCallId: toolCall.id,
-                      toolName: toolCall.function.name,
-                      rawArguments: toolCall.function.arguments,
-                      toolType: "specific",
-                      messageId: message.id
-                    }
-                  });
-                }
-                await this.notifySubscribers(
-                  (subscriber) => subscriber.onToolExecutionStart?.({
-                    copilotkit: this,
-                    toolCallId: toolCall.id,
-                    agentId: effectiveAgentId,
-                    toolName: toolCall.function.name,
-                    args: parsedArgs
-                  }),
-                  "Subscriber onToolExecutionStart error:"
-                );
-                if (!errorMessage) {
-                  try {
-                    const result = await tool.handler(
-                      parsedArgs,
-                      toolCall
-                    );
-                    if (result === void 0 || result === null) {
-                      toolCallResult = "";
-                    } else if (typeof result === "string") {
-                      toolCallResult = result;
-                    } else {
-                      toolCallResult = JSON.stringify(result);
-                    }
-                  } catch (error) {
-                    const handlerError = error instanceof Error ? error : new Error(String(error));
-                    errorMessage = handlerError.message;
-                    await this.emitError({
-                      error: handlerError,
-                      code: "tool_handler_failed" /* TOOL_HANDLER_FAILED */,
-                      context: {
-                        agentId: effectiveAgentId,
-                        toolCallId: toolCall.id,
-                        toolName: toolCall.function.name,
-                        parsedArgs,
-                        toolType: "specific",
-                        messageId: message.id
-                      }
-                    });
-                  }
-                }
-                if (errorMessage) {
-                  toolCallResult = `Error: ${errorMessage}`;
-                }
-                await this.notifySubscribers(
-                  (subscriber) => subscriber.onToolExecutionEnd?.({
-                    copilotkit: this,
-                    toolCallId: toolCall.id,
-                    agentId: effectiveAgentId,
-                    toolName: toolCall.function.name,
-                    result: errorMessage ? "" : toolCallResult,
-                    error: errorMessage
-                  }),
-                  "Subscriber onToolExecutionEnd error:"
-                );
-                if (isArgumentError) {
-                  throw new Error(errorMessage ?? "Tool execution failed");
-                }
-              }
-              if (!errorMessage || !isArgumentError) {
-                const messageIndex = agent.messages.findIndex(
-                  (m) => m.id === message.id
-                );
-                const toolMessage = {
-                  id: (0, import_shared.randomUUID)(),
-                  role: "tool",
-                  toolCallId: toolCall.id,
-                  content: toolCallResult
-                };
-                agent.messages.splice(messageIndex + 1, 0, toolMessage);
-                if (!errorMessage && tool?.followUp !== false) {
-                  needsFollowUp = true;
-                }
+              const followUp = await this.executeSpecificTool(tool, toolCall, message, agent, agentId);
+              if (followUp) {
+                needsFollowUp = true;
               }
             } else {
-              const wildcardTool = this.getTool({ toolName: "*", agentId });
+              const wildcardTool = this.getTool({ toolName: "*", agentId: agent.agentId });
               if (wildcardTool) {
-                if (wildcardTool?.agentId && wildcardTool.agentId !== agentId) {
-                  continue;
-                }
-                let toolCallResult = "";
-                let errorMessage;
-                let isArgumentError = false;
-                if (wildcardTool?.handler) {
-                  let parsedArgs;
-                  try {
-                    parsedArgs = JSON.parse(toolCall.function.arguments);
-                  } catch (error) {
-                    const parseError = error instanceof Error ? error : new Error(String(error));
-                    errorMessage = parseError.message;
-                    isArgumentError = true;
-                    await this.emitError({
-                      error: parseError,
-                      code: "tool_argument_parse_failed" /* TOOL_ARGUMENT_PARSE_FAILED */,
-                      context: {
-                        agentId: effectiveAgentId,
-                        toolCallId: toolCall.id,
-                        toolName: toolCall.function.name,
-                        rawArguments: toolCall.function.arguments,
-                        toolType: "wildcard",
-                        messageId: message.id
-                      }
-                    });
-                  }
-                  const wildcardArgs = {
-                    toolName: toolCall.function.name,
-                    args: parsedArgs
-                  };
-                  await this.notifySubscribers(
-                    (subscriber) => subscriber.onToolExecutionStart?.({
-                      copilotkit: this,
-                      toolCallId: toolCall.id,
-                      agentId: effectiveAgentId,
-                      toolName: toolCall.function.name,
-                      args: wildcardArgs
-                    }),
-                    "Subscriber onToolExecutionStart error:"
-                  );
-                  if (!errorMessage) {
-                    try {
-                      const result = await wildcardTool.handler(
-                        wildcardArgs,
-                        toolCall
-                      );
-                      if (result === void 0 || result === null) {
-                        toolCallResult = "";
-                      } else if (typeof result === "string") {
-                        toolCallResult = result;
-                      } else {
-                        toolCallResult = JSON.stringify(result);
-                      }
-                    } catch (error) {
-                      const handlerError = error instanceof Error ? error : new Error(String(error));
-                      errorMessage = handlerError.message;
-                      await this.emitError({
-                        error: handlerError,
-                        code: "tool_handler_failed" /* TOOL_HANDLER_FAILED */,
-                        context: {
-                          agentId: effectiveAgentId,
-                          toolCallId: toolCall.id,
-                          toolName: toolCall.function.name,
-                          parsedArgs: wildcardArgs,
-                          toolType: "wildcard",
-                          messageId: message.id
-                        }
-                      });
-                    }
-                  }
-                  if (errorMessage) {
-                    toolCallResult = `Error: ${errorMessage}`;
-                  }
-                  await this.notifySubscribers(
-                    (subscriber) => subscriber.onToolExecutionEnd?.({
-                      copilotkit: this,
-                      toolCallId: toolCall.id,
-                      agentId: effectiveAgentId,
-                      toolName: toolCall.function.name,
-                      result: errorMessage ? "" : toolCallResult,
-                      error: errorMessage
-                    }),
-                    "Subscriber onToolExecutionEnd error:"
-                  );
-                  if (isArgumentError) {
-                    throw new Error(errorMessage ?? "Tool execution failed");
-                  }
-                }
-                if (!errorMessage || !isArgumentError) {
-                  const messageIndex = agent.messages.findIndex(
-                    (m) => m.id === message.id
-                  );
-                  const toolMessage = {
-                    id: (0, import_shared.randomUUID)(),
-                    role: "tool",
-                    toolCallId: toolCall.id,
-                    content: toolCallResult
-                  };
-                  agent.messages.splice(messageIndex + 1, 0, toolMessage);
-                  if (!errorMessage && wildcardTool?.followUp !== false) {
-                    needsFollowUp = true;
-                  }
+                const followUp = await this.executeWildcardTool(wildcardTool, toolCall, message, agent, agentId);
+                if (followUp) {
+                  needsFollowUp = true;
                 }
               }
             }
@@ -752,10 +833,220 @@ var CopilotKitCore = class {
       }
     }
     if (needsFollowUp) {
-      return await this.runAgent({ agent, agentId });
+      return await this.runAgent({ agent });
     }
+    void this.core.suggestionEngine.reloadSuggestions(agentId);
     return runAgentResult;
   }
+  /**
+   * Execute a specific tool
+   */
+  async executeSpecificTool(tool, toolCall, message, agent, agentId) {
+    if (tool?.agentId && tool.agentId !== agent.agentId) {
+      return false;
+    }
+    let toolCallResult = "";
+    let errorMessage;
+    let isArgumentError = false;
+    if (tool?.handler) {
+      let parsedArgs;
+      try {
+        parsedArgs = JSON.parse(toolCall.function.arguments);
+      } catch (error) {
+        const parseError = error instanceof Error ? error : new Error(String(error));
+        errorMessage = parseError.message;
+        isArgumentError = true;
+        await this.core.emitError({
+          error: parseError,
+          code: "tool_argument_parse_failed" /* TOOL_ARGUMENT_PARSE_FAILED */,
+          context: {
+            agentId,
+            toolCallId: toolCall.id,
+            toolName: toolCall.function.name,
+            rawArguments: toolCall.function.arguments,
+            toolType: "specific",
+            messageId: message.id
+          }
+        });
+      }
+      await this.core.notifySubscribers(
+        (subscriber) => subscriber.onToolExecutionStart?.({
+          copilotkit: this.core,
+          toolCallId: toolCall.id,
+          agentId,
+          toolName: toolCall.function.name,
+          args: parsedArgs
+        }),
+        "Subscriber onToolExecutionStart error:"
+      );
+      if (!errorMessage) {
+        try {
+          const result = await tool.handler(parsedArgs, toolCall);
+          if (result === void 0 || result === null) {
+            toolCallResult = "";
+          } else if (typeof result === "string") {
+            toolCallResult = result;
+          } else {
+            toolCallResult = JSON.stringify(result);
+          }
+        } catch (error) {
+          const handlerError = error instanceof Error ? error : new Error(String(error));
+          errorMessage = handlerError.message;
+          await this.core.emitError({
+            error: handlerError,
+            code: "tool_handler_failed" /* TOOL_HANDLER_FAILED */,
+            context: {
+              agentId,
+              toolCallId: toolCall.id,
+              toolName: toolCall.function.name,
+              parsedArgs,
+              toolType: "specific",
+              messageId: message.id
+            }
+          });
+        }
+      }
+      if (errorMessage) {
+        toolCallResult = `Error: ${errorMessage}`;
+      }
+      await this.core.notifySubscribers(
+        (subscriber) => subscriber.onToolExecutionEnd?.({
+          copilotkit: this.core,
+          toolCallId: toolCall.id,
+          agentId,
+          toolName: toolCall.function.name,
+          result: errorMessage ? "" : toolCallResult,
+          error: errorMessage
+        }),
+        "Subscriber onToolExecutionEnd error:"
+      );
+      if (isArgumentError) {
+        throw new Error(errorMessage ?? "Tool execution failed");
+      }
+    }
+    if (!errorMessage || !isArgumentError) {
+      const messageIndex = agent.messages.findIndex((m) => m.id === message.id);
+      const toolMessage = {
+        id: (0, import_shared4.randomUUID)(),
+        role: "tool",
+        toolCallId: toolCall.id,
+        content: toolCallResult
+      };
+      agent.messages.splice(messageIndex + 1, 0, toolMessage);
+      if (!errorMessage && tool?.followUp !== false) {
+        return true;
+      }
+    }
+    return false;
+  }
+  /**
+   * Execute a wildcard tool
+   */
+  async executeWildcardTool(wildcardTool, toolCall, message, agent, agentId) {
+    if (wildcardTool?.agentId && wildcardTool.agentId !== agent.agentId) {
+      return false;
+    }
+    let toolCallResult = "";
+    let errorMessage;
+    let isArgumentError = false;
+    if (wildcardTool?.handler) {
+      let parsedArgs;
+      try {
+        parsedArgs = JSON.parse(toolCall.function.arguments);
+      } catch (error) {
+        const parseError = error instanceof Error ? error : new Error(String(error));
+        errorMessage = parseError.message;
+        isArgumentError = true;
+        await this.core.emitError({
+          error: parseError,
+          code: "tool_argument_parse_failed" /* TOOL_ARGUMENT_PARSE_FAILED */,
+          context: {
+            agentId,
+            toolCallId: toolCall.id,
+            toolName: toolCall.function.name,
+            rawArguments: toolCall.function.arguments,
+            toolType: "wildcard",
+            messageId: message.id
+          }
+        });
+      }
+      const wildcardArgs = {
+        toolName: toolCall.function.name,
+        args: parsedArgs
+      };
+      await this.core.notifySubscribers(
+        (subscriber) => subscriber.onToolExecutionStart?.({
+          copilotkit: this.core,
+          toolCallId: toolCall.id,
+          agentId,
+          toolName: toolCall.function.name,
+          args: wildcardArgs
+        }),
+        "Subscriber onToolExecutionStart error:"
+      );
+      if (!errorMessage) {
+        try {
+          const result = await wildcardTool.handler(wildcardArgs, toolCall);
+          if (result === void 0 || result === null) {
+            toolCallResult = "";
+          } else if (typeof result === "string") {
+            toolCallResult = result;
+          } else {
+            toolCallResult = JSON.stringify(result);
+          }
+        } catch (error) {
+          const handlerError = error instanceof Error ? error : new Error(String(error));
+          errorMessage = handlerError.message;
+          await this.core.emitError({
+            error: handlerError,
+            code: "tool_handler_failed" /* TOOL_HANDLER_FAILED */,
+            context: {
+              agentId,
+              toolCallId: toolCall.id,
+              toolName: toolCall.function.name,
+              parsedArgs: wildcardArgs,
+              toolType: "wildcard",
+              messageId: message.id
+            }
+          });
+        }
+      }
+      if (errorMessage) {
+        toolCallResult = `Error: ${errorMessage}`;
+      }
+      await this.core.notifySubscribers(
+        (subscriber) => subscriber.onToolExecutionEnd?.({
+          copilotkit: this.core,
+          toolCallId: toolCall.id,
+          agentId,
+          toolName: toolCall.function.name,
+          result: errorMessage ? "" : toolCallResult,
+          error: errorMessage
+        }),
+        "Subscriber onToolExecutionEnd error:"
+      );
+      if (isArgumentError) {
+        throw new Error(errorMessage ?? "Tool execution failed");
+      }
+    }
+    if (!errorMessage || !isArgumentError) {
+      const messageIndex = agent.messages.findIndex((m) => m.id === message.id);
+      const toolMessage = {
+        id: (0, import_shared4.randomUUID)(),
+        role: "tool",
+        toolCallId: toolCall.id,
+        content: toolCallResult
+      };
+      agent.messages.splice(messageIndex + 1, 0, toolMessage);
+      if (!errorMessage && wildcardTool?.followUp !== false) {
+        return true;
+      }
+    }
+    return false;
+  }
+  /**
+   * Build frontend tools for an agent
+   */
   buildFrontendTools(agentId) {
     return this._tools.filter((tool) => !tool.agentId || tool.agentId === agentId).map((tool) => ({
       name: tool.name,
@@ -763,13 +1054,16 @@ var CopilotKitCore = class {
       parameters: createToolSchema(tool)
     }));
   }
-  createAgentErrorSubscriber(agent, agentId) {
+  /**
+   * Create an agent error subscriber
+   */
+  createAgentErrorSubscriber(agent) {
     const emitAgentError = async (error, code, extraContext = {}) => {
       const context = { ...extraContext };
-      if (agentId ?? agent.agentId) {
-        context.agentId = agentId ?? agent.agentId;
+      if (agent.agentId) {
+        context.agentId = agent.agentId;
       }
-      await this.emitError({
+      await this.core.emitError({
         error,
         code,
         context
@@ -777,13 +1071,9 @@ var CopilotKitCore = class {
     };
     return {
       onRunFailed: async ({ error }) => {
-        await emitAgentError(
-          error,
-          "agent_run_failed_event" /* AGENT_RUN_FAILED_EVENT */,
-          {
-            source: "onRunFailed"
-          }
-        );
+        await emitAgentError(error, "agent_run_failed_event" /* AGENT_RUN_FAILED_EVENT */, {
+          source: "onRunFailed"
+        });
       },
       onRunErrorEvent: async ({ event }) => {
         const eventError = event?.rawEvent instanceof Error ? event.rawEvent : event?.rawEvent?.error instanceof Error ? event.rawEvent.error : void 0;
@@ -792,15 +1082,11 @@ var CopilotKitCore = class {
         if (event?.code && !rawError.code) {
           rawError.code = event.code;
         }
-        await emitAgentError(
-          rawError,
-          "agent_run_error_event" /* AGENT_RUN_ERROR_EVENT */,
-          {
-            source: "onRunErrorEvent",
-            event,
-            runtimeErrorCode: event?.code
-          }
-        );
+        await emitAgentError(rawError, "agent_run_error_event" /* AGENT_RUN_ERROR_EVENT */, {
+          source: "onRunErrorEvent",
+          event,
+          runtimeErrorCode: event?.code
+        });
       }
     };
   }
@@ -833,6 +1119,224 @@ function createToolSchema(tool) {
   return schema;
 }
 
+// src/core/core.ts
+var CopilotKitCoreErrorCode = /* @__PURE__ */ ((CopilotKitCoreErrorCode2) => {
+  CopilotKitCoreErrorCode2["RUNTIME_INFO_FETCH_FAILED"] = "runtime_info_fetch_failed";
+  CopilotKitCoreErrorCode2["AGENT_CONNECT_FAILED"] = "agent_connect_failed";
+  CopilotKitCoreErrorCode2["AGENT_RUN_FAILED"] = "agent_run_failed";
+  CopilotKitCoreErrorCode2["AGENT_RUN_FAILED_EVENT"] = "agent_run_failed_event";
+  CopilotKitCoreErrorCode2["AGENT_RUN_ERROR_EVENT"] = "agent_run_error_event";
+  CopilotKitCoreErrorCode2["TOOL_ARGUMENT_PARSE_FAILED"] = "tool_argument_parse_failed";
+  CopilotKitCoreErrorCode2["TOOL_HANDLER_FAILED"] = "tool_handler_failed";
+  return CopilotKitCoreErrorCode2;
+})(CopilotKitCoreErrorCode || {});
+var CopilotKitCoreRuntimeConnectionStatus = /* @__PURE__ */ ((CopilotKitCoreRuntimeConnectionStatus2) => {
+  CopilotKitCoreRuntimeConnectionStatus2["Disconnected"] = "disconnected";
+  CopilotKitCoreRuntimeConnectionStatus2["Connected"] = "connected";
+  CopilotKitCoreRuntimeConnectionStatus2["Connecting"] = "connecting";
+  CopilotKitCoreRuntimeConnectionStatus2["Error"] = "error";
+  return CopilotKitCoreRuntimeConnectionStatus2;
+})(CopilotKitCoreRuntimeConnectionStatus || {});
+var CopilotKitCore = class {
+  _headers;
+  _properties;
+  subscribers = /* @__PURE__ */ new Set();
+  // Delegate classes
+  agentRegistry;
+  contextStore;
+  suggestionEngine;
+  runHandler;
+  constructor({
+    runtimeUrl,
+    headers = {},
+    properties = {},
+    agents__unsafe_dev_only = {},
+    tools = [],
+    suggestionsConfig = []
+  }) {
+    this._headers = headers;
+    this._properties = properties;
+    this.agentRegistry = new AgentRegistry(this);
+    this.contextStore = new ContextStore(this);
+    this.suggestionEngine = new SuggestionEngine(this);
+    this.runHandler = new RunHandler(this);
+    this.agentRegistry.initialize(agents__unsafe_dev_only);
+    this.runHandler.initialize(tools);
+    this.suggestionEngine.initialize(suggestionsConfig);
+    this.agentRegistry.setRuntimeUrl(runtimeUrl);
+  }
+  /**
+   * Internal method used by delegate classes to notify subscribers
+   */
+  async notifySubscribers(handler, errorMessage) {
+    await Promise.all(
+      Array.from(this.subscribers).map(async (subscriber) => {
+        try {
+          await handler(subscriber);
+        } catch (error) {
+          console.error(errorMessage, error);
+        }
+      })
+    );
+  }
+  /**
+   * Internal method used by delegate classes to emit errors
+   */
+  async emitError({
+    error,
+    code,
+    context = {}
+  }) {
+    await this.notifySubscribers(
+      (subscriber) => subscriber.onError?.({
+        copilotkit: this,
+        error,
+        code,
+        context
+      }),
+      "Subscriber onError error:"
+    );
+  }
+  /**
+   * Snapshot accessors
+   */
+  get context() {
+    return this.contextStore.context;
+  }
+  get agents() {
+    return this.agentRegistry.agents;
+  }
+  get tools() {
+    return this.runHandler.tools;
+  }
+  get runtimeUrl() {
+    return this.agentRegistry.runtimeUrl;
+  }
+  setRuntimeUrl(runtimeUrl) {
+    this.agentRegistry.setRuntimeUrl(runtimeUrl);
+  }
+  get runtimeVersion() {
+    return this.agentRegistry.runtimeVersion;
+  }
+  get headers() {
+    return this._headers;
+  }
+  get properties() {
+    return this._properties;
+  }
+  get runtimeConnectionStatus() {
+    return this.agentRegistry.runtimeConnectionStatus;
+  }
+  /**
+   * Configuration updates
+   */
+  setHeaders(headers) {
+    this._headers = headers;
+    this.agentRegistry.applyHeadersToAgents(this.agentRegistry.agents);
+    void this.notifySubscribers(
+      (subscriber) => subscriber.onHeadersChanged?.({
+        copilotkit: this,
+        headers: this.headers
+      }),
+      "Subscriber onHeadersChanged error:"
+    );
+  }
+  setProperties(properties) {
+    this._properties = properties;
+    void this.notifySubscribers(
+      (subscriber) => subscriber.onPropertiesChanged?.({
+        copilotkit: this,
+        properties: this.properties
+      }),
+      "Subscriber onPropertiesChanged error:"
+    );
+  }
+  /**
+   * Agent management (delegated to AgentRegistry)
+   */
+  setAgents__unsafe_dev_only(agents) {
+    this.agentRegistry.setAgents__unsafe_dev_only(agents);
+  }
+  addAgent__unsafe_dev_only(params) {
+    this.agentRegistry.addAgent__unsafe_dev_only(params);
+  }
+  removeAgent__unsafe_dev_only(id) {
+    this.agentRegistry.removeAgent__unsafe_dev_only(id);
+  }
+  getAgent(id) {
+    return this.agentRegistry.getAgent(id);
+  }
+  /**
+   * Context management (delegated to ContextStore)
+   */
+  addContext(context) {
+    return this.contextStore.addContext(context);
+  }
+  removeContext(id) {
+    this.contextStore.removeContext(id);
+  }
+  /**
+   * Suggestions management (delegated to SuggestionEngine)
+   */
+  addSuggestionsConfig(config) {
+    return this.suggestionEngine.addSuggestionsConfig(config);
+  }
+  removeSuggestionsConfig(id) {
+    this.suggestionEngine.removeSuggestionsConfig(id);
+  }
+  reloadSuggestions(agentId) {
+    this.suggestionEngine.reloadSuggestions(agentId);
+  }
+  clearSuggestions(agentId) {
+    this.suggestionEngine.clearSuggestions(agentId);
+  }
+  getSuggestions(agentId) {
+    return this.suggestionEngine.getSuggestions(agentId);
+  }
+  /**
+   * Tool management (delegated to RunHandler)
+   */
+  addTool(tool) {
+    this.runHandler.addTool(tool);
+  }
+  removeTool(id, agentId) {
+    this.runHandler.removeTool(id, agentId);
+  }
+  getTool(params) {
+    return this.runHandler.getTool(params);
+  }
+  setTools(tools) {
+    this.runHandler.setTools(tools);
+  }
+  /**
+   * Subscription lifecycle
+   */
+  subscribe(subscriber) {
+    this.subscribers.add(subscriber);
+    return () => {
+      this.unsubscribe(subscriber);
+    };
+  }
+  unsubscribe(subscriber) {
+    this.subscribers.delete(subscriber);
+  }
+  /**
+   * Agent connectivity (delegated to RunHandler)
+   */
+  async connectAgent(params) {
+    return this.runHandler.connectAgent(params);
+  }
+  async runAgent(params) {
+    return this.runHandler.runAgent(params);
+  }
+  /**
+   * Internal method used by RunHandler to build frontend tools
+   */
+  buildFrontendTools(agentId) {
+    return this.runHandler.buildFrontendTools(agentId);
+  }
+};
+
 // src/types.ts
 var ToolCallStatus = /* @__PURE__ */ ((ToolCallStatus2) => {
   ToolCallStatus2["InProgress"] = "inProgress";
@@ -1058,10 +1562,14 @@ ${indent}${fence}`;
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AgentRegistry,
+  ContextStore,
   CopilotKitCore,
   CopilotKitCoreErrorCode,
   CopilotKitCoreRuntimeConnectionStatus,
   ProxiedCopilotRuntimeAgent,
+  RunHandler,
+  SuggestionEngine,
   ToolCallStatus,
   completePartialMarkdown
 });