@dxos/plugin-automation 0.7.5-labs.e27f9b9 → 0.7.5-labs.f5080a1

package/dist/lib/browser/{chunk-5SLV6AUA.mjs → chunk-DL7XA62G.mjs}

@@ -1,32 +1,18 @@
 import {
   AutomationCapabilities
-} from "./chunk-HZ4TA7HY.mjs";
+} from "./chunk-BQHXJZ4K.mjs";
 import {
-  MockServiceRegistry
-} from "./chunk-7XADMUOW.mjs";
+  ServiceType,
+  categoryIcons
+} from "./chunk-43WRHV2L.mjs";
 import {
   AUTOMATION_PLUGIN
-} from "./chunk-HKX3D3ZP.mjs";
+} from "./chunk-DQ7ZSYJJ.mjs";
 
 // packages/plugins/experimental/plugin-automation/src/components/ChatContainer/ChatContainer.tsx
-import React7, { useCallback as useCallback3, useEffect as useEffect6, useMemo as useMemo3, useState as useState7 } from "react";
-import { Capabilities, useCapabilities, useCapability, useIntentDispatcher } from "@dxos/app-framework";
-import { createSystemPrompt } from "@dxos/artifact";
-import { FunctionType as FunctionType2 } from "@dxos/functions";
+import React7, { useCallback as useCallback3 } from "react";
 import { invariant as invariant7 } from "@dxos/invariant";
-import { DXN as DXN2, QueueSubspaceTags } from "@dxos/keys";
-import { useConfig } from "@dxos/react-client";
-import { Filter as Filter2, getSpace, useQuery as useQuery2 } from "@dxos/react-client/echo";
-import { useEdgeClient, useQueue } from "@dxos/react-edge-client";
 import { StackItem } from "@dxos/react-ui-stack";
-import { isNotNullOrUndefined } from "@dxos/util";
-
-// packages/plugins/experimental/plugin-automation/src/capabilities/index.ts
-import { lazy } from "@dxos/app-framework";
-var AiClient = lazy(() => import("./ai-client-AARXEMMJ.mjs"));
-var AppGraphBuilder = lazy(() => import("./app-graph-builder-B4U34VSR.mjs"));
-var IntentResolver = lazy(() => import("./intent-resolver-754MPV7H.mjs"));
-var ReactSurface = lazy(() => import("./react-surface-4QZ6AKBF.mjs"));
 
 // packages/plugins/experimental/plugin-automation/src/hooks/processor.ts
 import { batch, computed, signal } from "@preact/signals-core";
@@ -38,7 +24,7 @@ import { log } from "@dxos/log";
 var __dxlog_file = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-automation/src/hooks/processor.ts";
 var defaultOptions = {
   model: "@anthropic/claude-3-5-sonnet-20241022",
-  systemPrompt: ""
+  systemPrompt: "you are a helpful assistant"
 };
 var ChatProcessor = class {
   constructor(_client, _tools, _extensions, _options = defaultOptions) {
@@ -59,7 +45,7 @@ var ChatProcessor = class {
         const current = messages.pop();
         invariant(current, void 0, {
           F: __dxlog_file,
-          L: 76,
+          L: 78,
           S: this,
           A: [
             "current",
@@ -94,6 +80,12 @@ var ChatProcessor = class {
     });
   }
   /**
+  * Update tools.
+  */
+  setTools(tools) {
+    this._tools = tools;
+  }
+  /**
   * Make GPT request.
   */
   async request(message, options = {}) {
@@ -123,7 +115,7 @@ var ChatProcessor = class {
   async cancel() {
     log.info("cancelling...", void 0, {
       F: __dxlog_file,
-      L: 132,
+      L: 141,
       S: this,
       C: (f, a) => f(...a)
     });
@@ -152,7 +144,7 @@ var ChatProcessor = class {
           messages: this._pending.value.length
         }, {
           F: __dxlog_file,
-          L: 156,
+          L: 165,
           S: this,
           C: (f, a) => f(...a)
         });
@@ -171,7 +163,7 @@ var ChatProcessor = class {
           messages: this._pending.value
         }, {
           F: __dxlog_file,
-          L: 169,
+          L: 178,
           S: this,
           C: (f, a) => f(...a)
         });
@@ -179,7 +171,7 @@ var ChatProcessor = class {
         const message = this._pending.value.at(-1);
         invariant(message, void 0, {
           F: __dxlog_file,
-          L: 174,
+          L: 183,
           S: this,
           A: [
             "message",
@@ -189,7 +181,7 @@ var ChatProcessor = class {
         if (isToolUse(message)) {
           log.info("tool request...", void 0, {
             F: __dxlog_file,
-            L: 176,
+            L: 185,
             S: this,
             C: (f, a) => f(...a)
           });
@@ -202,7 +194,7 @@ var ChatProcessor = class {
             response
           }, {
             F: __dxlog_file,
-            L: 183,
+            L: 192,
             S: this,
             C: (f, a) => f(...a)
           });
@@ -223,14 +215,14 @@ var ChatProcessor = class {
         err
       }, {
         F: __dxlog_file,
-        L: 195,
+        L: 204,
         S: this,
         C: (f, a) => f(...a)
       });
     } finally {
       log.info("done", void 0, {
         F: __dxlog_file,
-        L: 197,
+        L: 206,
         S: this,
         C: (f, a) => f(...a)
       });
@@ -239,51 +231,27 @@ var ChatProcessor = class {
   }
 };
 
-// packages/plugins/experimental/plugin-automation/src/hooks/useServiceRegistry.ts
+// packages/plugins/experimental/plugin-automation/src/hooks/useChatProcessor.tsx
 import { useEffect, useMemo, useState } from "react";
-var useServiceRegistry = (query) => {
-  const registry = useMemo(() => new MockServiceRegistry(), []);
-  const [services, setServices] = useState([]);
-  useEffect(() => {
-    const t = setTimeout(async () => {
-      const services2 = await registry.queryServices(query);
-      setServices(services2);
-    });
-    return () => clearTimeout(t);
-  }, [
-    query,
-    registry
-  ]);
-  return services;
-};
-
-// packages/plugins/experimental/plugin-automation/src/hooks/useLocalTriggerManager.ts
-import { useEffect as useEffect2, useState as useState2 } from "react";
-import { Mutex } from "@dxos/async";
-import { Context } from "@dxos/context";
-import { createSubscriptionTrigger } from "@dxos/functions";
-import { FunctionTrigger } from "@dxos/functions";
-import { invariant as invariant3 } from "@dxos/invariant";
-import { log as log3 } from "@dxos/log";
-import { useClient } from "@dxos/react-client";
-import { Filter, useQuery } from "@dxos/react-client/echo";
-
-// packages/plugins/experimental/plugin-automation/src/hooks/invocation-handler.ts
-import { sleep } from "@dxos/async";
-import { getObjectCore, ResultFormat } from "@dxos/echo-db";
+import { Capabilities, useCapabilities, useCapability, useIntentDispatcher } from "@dxos/app-framework";
+import { createSystemPrompt } from "@dxos/artifact";
 import { FunctionType } from "@dxos/functions";
-import { invariant as invariant2 } from "@dxos/invariant";
-import { DXN, LOCAL_SPACE_TAG } from "@dxos/keys";
-import { log as log2 } from "@dxos/log";
+import { useConfig } from "@dxos/react-client";
+import { Filter, getSpace, useQuery } from "@dxos/react-client/echo";
+import { isNotNullOrUndefined } from "@dxos/util";
 
-// packages/plugins/experimental/plugin-automation/src/hooks/useLocalTriggerManager.ts
-var registerTriggersMutex = new Mutex();
+// packages/plugins/experimental/plugin-automation/src/capabilities/index.ts
+import { lazy } from "@dxos/app-framework";
+var AiClient = lazy(() => import("./ai-client-UJLNYP7B.mjs"));
+var AppGraphBuilder = lazy(() => import("./app-graph-builder-3H5TCRG4.mjs"));
+var IntentResolver = lazy(() => import("./intent-resolver-5YVZJFS3.mjs"));
+var ReactSurface = lazy(() => import("./react-surface-WRHRCEV5.mjs"));
 
 // packages/plugins/experimental/plugin-automation/src/tools/function.ts
 import { defineTool, ToolResult } from "@dxos/artifact";
 import { toEffectSchema } from "@dxos/echo-schema";
 import { getInvocationUrl, getUserFunctionUrlInMetadata } from "@dxos/functions";
-import { log as log4 } from "@dxos/log";
+import { log as log2 } from "@dxos/log";
 import { getMeta } from "@dxos/react-client/echo";
 var __dxlog_file2 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-automation/src/tools/function.ts";
 var covertFunctionToTool = (fn, edgeUrl, spaceId) => {
@@ -302,7 +270,7 @@ var covertFunctionToTool = (fn, edgeUrl, spaceId) => {
     description: fn.description,
     schema: toEffectSchema(fn.inputSchema),
     execute: async (input) => {
-      log4.info("execute function tool", {
+      log2.info("execute function tool", {
         name: fn.name,
         url,
         input
@@ -328,12 +296,12 @@ var covertFunctionToTool = (fn, edgeUrl, spaceId) => {
 import jsonpointer from "jsonpointer";
 import { ToolResult as ToolResult2 } from "@dxos/artifact";
 import { JsonSchemaType, normalizeSchema, S, toEffectSchema as toEffectSchema2 } from "@dxos/echo-schema";
-import { invariant as invariant4 } from "@dxos/invariant";
-import { log as log5 } from "@dxos/log";
+import { invariant as invariant2 } from "@dxos/invariant";
+import { log as log3 } from "@dxos/log";
 import { deepMapValues } from "@dxos/util";
 var __dxlog_file3 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-automation/src/tools/openapi.ts";
 var createToolsFromService = async (service) => {
-  invariant4(service.interfaces?.length === 1 && service.interfaces[0].kind === "api", void 0, {
+  invariant2(service.interfaces?.length === 1 && service.interfaces[0].kind === "api", void 0, {
     F: __dxlog_file3,
     L: 22,
     S: void 0,
@@ -343,7 +311,7 @@ var createToolsFromService = async (service) => {
     ]
   });
   const iface = service.interfaces[0];
-  invariant4(iface.schemaUrl, void 0, {
+  invariant2(iface.schemaUrl, void 0, {
     F: __dxlog_file3,
     L: 24,
     S: void 0,
@@ -352,7 +320,7 @@ var createToolsFromService = async (service) => {
       ""
     ]
   });
-  invariant4(iface.schemaUrl, void 0, {
+  invariant2(iface.schemaUrl, void 0, {
     F: __dxlog_file3,
     L: 25,
     S: void 0,
@@ -368,7 +336,7 @@ var createToolsFromService = async (service) => {
 var createToolsFromApi = async (url, options) => {
   const res = await fetch(url);
   const spec = await res.json();
-  log5("spec", {
+  log3("spec", {
     spec
   }, {
     F: __dxlog_file3,
@@ -384,7 +352,7 @@ var createToolsFromApi = async (url, options) => {
     const { ...methods } = pathItem;
     for (const [method, m] of Object.entries(methods)) {
       const methodItem = m;
-      log5("methodItem", {
+      log3("methodItem", {
         path,
         method,
         methodItem
@@ -404,7 +372,7 @@ var createToolsFromApi = async (url, options) => {
       };
       const endpointParameters = [];
       for (const parameter of parametersResolved) {
-        log5("parameter", {
+        log3("parameter", {
           parameter
         }, {
           F: __dxlog_file3,
@@ -427,7 +395,7 @@ var createToolsFromApi = async (url, options) => {
           }
         }
       }
-      log5("inputSchema", {
+      log3("inputSchema", {
         inputSchema
       }, {
         F: __dxlog_file3,
@@ -438,7 +406,7 @@ var createToolsFromApi = async (url, options) => {
       S.validateSync(JsonSchemaType)(inputSchema);
       const description = methodItem.description ?? methodItem.summary;
       if (!description) {
-        log5.warn("no description", {
+        log3.warn("no description", {
           path,
           method
         }, {
@@ -513,7 +481,7 @@ var GENERIC_WORDS = [
   "endpoint"
 ];
 var callApiEndpoint = async (endpoint, input) => {
-  log5.info("endpoint", {
+  log3.info("endpoint", {
     method: endpoint.method,
     name: endpoint.path,
     input
@@ -573,7 +541,7 @@ var callApiEndpoint = async (endpoint, input) => {
     request.body = JSON.stringify(body);
     request.headers["Content-Type"] = "application/json";
   }
-  log5.info("request", {
+  log3.info("request", {
     url,
     request
   }, {
@@ -583,7 +551,7 @@ var callApiEndpoint = async (endpoint, input) => {
     C: (f, a) => f(...a)
   });
   const response = await fetch(url, request);
-  log5.info("response", {
+  log3.info("response", {
     ok: response.ok,
     status: response.status,
     statusText: response.statusText
@@ -609,7 +577,7 @@ var callApiEndpoint = async (endpoint, input) => {
       } catch {
         error = responseBody;
       }
-      log5.error("error", {
+      log3.error("error", {
         error
       }, {
         F: __dxlog_file3,
@@ -620,7 +588,7 @@ var callApiEndpoint = async (endpoint, input) => {
       throw new Error(error.message);
     } else {
       const error = await response.text();
-      log5.error("error", {
+      log3.error("error", {
         error
       }, {
         F: __dxlog_file3,
@@ -637,7 +605,7 @@ var getEndpointUrl = (endpoint) => {
   if (isV3_1(endpoint.document) && endpoint.document.servers && endpoint.document.servers.length > 0) {
     url = endpoint.document.servers[0].url;
   } else {
-    invariant4(!isV3_1(endpoint.document), void 0, {
+    invariant2(!isV3_1(endpoint.document), void 0, {
       F: __dxlog_file3,
       L: 272,
       S: void 0,
@@ -657,7 +625,7 @@ var getEndpointUrl = (endpoint) => {
 var resolveAuthorization = async (authorization) => {
   switch (authorization.type) {
     case "api-key": {
-      invariant4(authorization.placement.type === "authorization-header", void 0, {
+      invariant2(authorization.placement.type === "authorization-header", void 0, {
         F: __dxlog_file3,
         L: 288,
         S: void 0,
@@ -692,7 +660,7 @@ var resolveJsonSchema = (schema, base) => {
         if (resolved) {
           return recurse(resolved);
         } else {
-          log5.warn("unresolved", {
+          log3.warn("unresolved", {
             ref: value.$ref,
             base
           }, {
@@ -711,14 +679,248 @@ var isV3_1 = (document) => {
   return document.openapi === "3.0.1";
 };
 
+// packages/plugins/experimental/plugin-automation/src/hooks/useChatProcessor.tsx
+var useChatProcessor = (chat) => {
+  const aiClient = useCapability(AutomationCapabilities.AiClient);
+  const globalTools = useCapabilities(Capabilities.Tools);
+  const artifactDefinitions = useCapabilities(Capabilities.ArtifactDefinition);
+  const { dispatchPromise: dispatch } = useIntentDispatcher();
+  const space = getSpace(chat);
+  const services = useQuery(space, Filter.schema(ServiceType));
+  const [serviceTools, setServiceTools] = useState([]);
+  useEffect(() => {
+    queueMicrotask(async () => {
+      const tools2 = await Promise.all(services.map((service) => createToolsFromService(service)));
+      setServiceTools(tools2.flat());
+    });
+  }, [
+    services
+  ]);
+  const config = useConfig();
+  const functions2 = useQuery(space, Filter.schema(FunctionType));
+  const tools = useMemo(() => [
+    ...globalTools.flat(),
+    ...artifactDefinitions.flatMap((definition) => definition.tools),
+    ...serviceTools,
+    ...functions2.map((fn) => covertFunctionToTool(fn, config.values.runtime?.services?.edge?.url ?? "", space?.id)).filter(isNotNullOrUndefined)
+  ], [
+    globalTools,
+    artifactDefinitions,
+    serviceTools,
+    functions2,
+    space?.id
+  ]);
+  const systemPrompt = useMemo(() => createSystemPrompt({
+    artifacts: artifactDefinitions.map((definition) => definition.instructions)
+  }), [
+    artifactDefinitions
+  ]);
+  const processor = useMemo(() => new ChatProcessor(aiClient, tools, {
+    space,
+    dispatch
+  }, {
+    model: "@anthropic/claude-3-5-sonnet-20241022",
+    systemPrompt
+  }), [
+    aiClient,
+    tools,
+    space,
+    dispatch,
+    systemPrompt
+  ]);
+  return processor;
+};
+
+// packages/plugins/experimental/plugin-automation/src/hooks/useLocalTriggerManager.ts
+import { useEffect as useEffect2, useState as useState2 } from "react";
+import { Mutex } from "@dxos/async";
+import { Context } from "@dxos/context";
+import { createSubscriptionTrigger } from "@dxos/functions";
+import { FunctionTrigger } from "@dxos/functions";
+import { invariant as invariant4 } from "@dxos/invariant";
+import { log as log5 } from "@dxos/log";
+import { useClient } from "@dxos/react-client";
+import { Filter as Filter2, useQuery as useQuery2 } from "@dxos/react-client/echo";
+
+// packages/plugins/experimental/plugin-automation/src/hooks/invocation-handler.ts
+import { sleep } from "@dxos/async";
+import { getObjectCore, ResultFormat } from "@dxos/echo-db";
+import { FunctionType as FunctionType2 } from "@dxos/functions";
+import { invariant as invariant3 } from "@dxos/invariant";
+import { DXN, LOCAL_SPACE_TAG } from "@dxos/keys";
+import { log as log4 } from "@dxos/log";
+
+// packages/plugins/experimental/plugin-automation/src/hooks/useLocalTriggerManager.ts
+var registerTriggersMutex = new Mutex();
+
+// packages/plugins/experimental/plugin-automation/src/hooks/useMessageQueue.ts
+import { useMemo as useMemo2 } from "react";
+import { DXN as DXN2, QueueSubspaceTags } from "@dxos/keys";
+import { getSpace as getSpace2 } from "@dxos/react-client/echo";
+import { useEdgeClient, useQueue } from "@dxos/react-edge-client";
+var useMessageQueue = (chat) => {
+  const edgeClient = useEdgeClient();
+  const space = getSpace2(chat);
+  const queueDxn = useMemo2(() => new DXN2(DXN2.kind.QUEUE, [
+    QueueSubspaceTags.DATA,
+    space.id,
+    chat.queue.dxn.parts.at(-1)
+  ]), [
+    chat.queue.dxn
+  ]);
+  return useQueue(edgeClient, queueDxn);
+};
+
+// packages/plugins/experimental/plugin-automation/src/hooks/useServices.ts
+import { useEffect as useEffect3, useMemo as useMemo3, useState as useState3 } from "react";
+
+// packages/plugins/experimental/plugin-automation/src/testing/test-functions.ts
+import { createSystemPrompt as createSystemPrompt2 } from "@dxos/artifact";
+import { AST, S as S2, toJsonSchema } from "@dxos/echo-schema";
+var functions = [
+  {
+    name: "example.com/function/chess",
+    version: "0.1.0",
+    inputSchema: toJsonSchema(S2.Struct({
+      level: S2.Number.annotations({
+        [AST.TitleAnnotationId]: "Level"
+      })
+    }))
+  },
+  {
+    name: "example.com/function/forex",
+    version: "0.1.0",
+    binding: "FOREX",
+    inputSchema: toJsonSchema(S2.Struct({
+      from: S2.String.annotations({
+        [AST.TitleAnnotationId]: "Currency from"
+      }),
+      to: S2.String.annotations({
+        [AST.TitleAnnotationId]: "Currency to"
+      })
+    }))
+  }
+];
+
+// packages/plugins/experimental/plugin-automation/src/testing/test-services.ts
+import { createStatic as createStatic2 } from "@dxos/echo-schema";
+var MockServiceRegistry = class {
+  async queryServices(query) {
+    return TEST_SERVICES;
+  }
+};
+var AMADEUS_AUTH = {
+  type: "oauth",
+  clientId: "BOEnpLd1sMyKjAPGKYeAPFFy60u53QEG",
+  clientSecret: "n4qldSN7usvD57gm",
+  tokenUrl: "https://test.api.amadeus.com/v1/security/oauth2/token",
+  grantType: "client_credentials"
+};
+var VISUAL_CROSSING_CREDENTIALS = {
+  type: "api-key",
+  key: "FDPRVS953KB4GQQLD25GRT975",
+  placement: {
+    type: "query",
+    name: "key"
+  }
+};
+var TEST_SERVICES = [
+  createStatic2(ServiceType, {
+    serviceId: "amadeus.com/service/FlightSearch",
+    name: "Amadeus Flight Search",
+    description: "Search for local and international flights.",
+    category: "travel",
+    interfaces: [
+      {
+        kind: "api",
+        schemaUrl: "https://api.apis.guru/v2/specs/amadeus.com/amadeus-flight-availabilities-search/1.0.2/swagger.json",
+        authorization: AMADEUS_AUTH
+      }
+    ]
+  }),
+  createStatic2(ServiceType, {
+    serviceId: "amadeus.com/service/HotelSearch",
+    name: "Amadeus Hotel Search",
+    description: "Search for local and international hotels.",
+    category: "travel",
+    interfaces: [
+      {
+        kind: "api",
+        schemaUrl: "https://api.apis.guru/v2/specs/amadeus.com/amadeus-hotel-search/3.0.8/swagger.json",
+        authorization: AMADEUS_AUTH
+      }
+    ]
+  }),
+  createStatic2(ServiceType, {
+    serviceId: "visualcrossing.com/service/Weather",
+    name: "Visual Crossing Weather",
+    description: "Search for global weather forecasts.",
+    category: "weather",
+    interfaces: [
+      {
+        kind: "api",
+        schemaUrl: "https://api.apis.guru/v2/specs/visualcrossing.com/weather/4.6/openapi.json",
+        authorization: VISUAL_CROSSING_CREDENTIALS
+      }
+    ]
+  }),
+  // TODO(burdon): Needs auth.
+  createStatic2(ServiceType, {
+    serviceId: "abstractapi.com/service/GeoLocation",
+    name: "Abstract GeoLocation",
+    description: "Get the location of any IP address.",
+    category: "geolocation",
+    interfaces: [
+      {
+        kind: "api",
+        schemaUrl: "https://api.apis.guru/v2/specs/abstractapi.com/geolocation/1.0.0/openapi.json"
+      }
+    ]
+  }),
+  //
+  // Testing
+  //
+  ...Array.from({
+    length: 20
+  }, (_, i) => createStatic2(ServiceType, {
+    serviceId: `example.com/service/test-${i}`,
+    name: `Test ${i}`,
+    description: `Test ${i}`,
+    category: Object.keys(categoryIcons)[Math.floor(Math.random() * Object.keys(categoryIcons).length)],
+    interfaces: [
+      {
+        kind: "api",
+        schemaUrl: "https://petstore.swagger.io/v2/swagger.json"
+      }
+    ]
+  }))
+];
+
+// packages/plugins/experimental/plugin-automation/src/hooks/useServices.ts
+var useServices = (space, query) => {
+  const registry = useMemo3(() => new MockServiceRegistry(), []);
+  const [services, setServices] = useState3([]);
+  useEffect3(() => {
+    const t = setTimeout(async () => {
+      const services2 = await registry.queryServices(query);
+      setServices(services2);
+    });
+    return () => clearTimeout(t);
+  }, [
+    query,
+    registry
+  ]);
+  return services;
+};
+
 // packages/plugins/experimental/plugin-automation/src/components/Thread/Thread.tsx
-import React6, { useCallback as useCallback2, useMemo as useMemo2, useRef as useRef3, useState as useState6 } from "react";
+import React6, { useCallback as useCallback2, useMemo as useMemo4, useRef as useRef3, useState as useState7 } from "react";
 import { IconButton, Input, useTranslation } from "@dxos/react-ui";
 import { Spinner } from "@dxos/react-ui-sfx";
 import { mx as mx6 } from "@dxos/react-ui-theme";
 
 // packages/plugins/experimental/plugin-automation/src/components/Thread/ScrollContainer.tsx
-import React, { forwardRef, useCallback, useEffect as useEffect3, useImperativeHandle, useRef, useState as useState3 } from "react";
+import React, { forwardRef, useCallback, useEffect as useEffect4, useImperativeHandle, useRef, useState as useState4 } from "react";
 import { invariant as invariant5 } from "@dxos/invariant";
 import { mx } from "@dxos/react-ui-theme";
 var __dxlog_file4 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-automation/src/components/Thread/ScrollContainer.tsx";
@@ -743,8 +945,8 @@ var ScrollContainer = /* @__PURE__ */ forwardRef(({ children, classNames }, forw
       autoScrollRef.current = false;
     }
   }), []);
-  const [sticky, setSticky] = useState3(true);
-  useEffect3(() => {
+  const [sticky, setSticky] = useState4(true);
+  useEffect4(() => {
     if (!sticky || !containerRef.current) {
       return;
     }
@@ -756,7 +958,7 @@ var ScrollContainer = /* @__PURE__ */ forwardRef(({ children, classNames }, forw
   }, [
     children
   ]);
-  useEffect3(() => {
+  useEffect4(() => {
     invariant5(containerRef.current, void 0, {
       F: __dxlog_file4,
       L: 62,
@@ -790,30 +992,31 @@ var ScrollContainer = /* @__PURE__ */ forwardRef(({ children, classNames }, forw
 // packages/plugins/experimental/plugin-automation/src/components/Thread/ThreadMessage.tsx
 import React5 from "react";
 import { invariant as invariant6 } from "@dxos/invariant";
+import { log as log6 } from "@dxos/log";
 import { Button, ButtonGroup, Icon as Icon2 } from "@dxos/react-ui";
 import { Json } from "@dxos/react-ui-syntax-highlighter";
 import { mx as mx5 } from "@dxos/react-ui-theme";
 import { safeParseJson } from "@dxos/util";
 
 // packages/plugins/experimental/plugin-automation/src/components/Thread/StatusLine.tsx
-import React2, { useEffect as useEffect4, useRef as useRef2, useState as useState4 } from "react";
+import React2, { useEffect as useEffect5, useRef as useRef2, useState as useState5 } from "react";
 import { mx as mx2 } from "@dxos/react-ui-theme";
 var emptyLines = [];
 var StatusLine = ({ classNames, line = -1, lines = emptyLines, transition = 300, advance = 1e3, autoAdvance }) => {
   const containerRef = useRef2(null);
-  const [currentLine, setCurrentLine] = useState4(line);
-  useEffect4(() => {
+  const [currentLine, setCurrentLine] = useState5(line);
+  useEffect5(() => {
     setCurrentLine(line);
   }, [
     line
   ]);
-  useEffect4(() => {
+  useEffect5(() => {
     if (!autoAdvance) {
       return;
     }
     const next = () => {
       setCurrentLine((prev) => {
-        if (prev === lines.length - 1) {
+        if (prev >= lines.length - 1) {
           clearInterval(interval);
           return prev;
         }
@@ -828,7 +1031,7 @@ var StatusLine = ({ classNames, line = -1, lines = emptyLines, transition = 300,
     autoAdvance,
     advance
   ]);
-  useEffect4(() => {
+  useEffect5(() => {
     if (containerRef.current) {
       containerRef.current.style.transition = `transform ${transition}ms ease-in-out`;
       containerRef.current.style.transform = `translateY(-${currentLine * 24}px)`;
@@ -852,14 +1055,14 @@ var StatusLine = ({ classNames, line = -1, lines = emptyLines, transition = 300,
 };
 
 // packages/plugins/experimental/plugin-automation/src/components/Thread/ToggleContainer.tsx
-import React3, { useEffect as useEffect5, useState as useState5 } from "react";
+import React3, { useEffect as useEffect6, useState as useState6 } from "react";
 import { Icon } from "@dxos/react-ui";
 import { mx as mx3 } from "@dxos/react-ui-theme";
 var ToggleContainer = ({ title, icon, toggle, defaultOpen, duration = 400, shrinkX = false, children, classNames }) => {
-  const [expand, setExpand] = useState5(defaultOpen || !toggle);
-  const [expandX, setExpandX] = useState5(shrinkX ? expand : true);
-  const [expandY, setExpandY] = useState5(expand);
-  useEffect5(() => {
+  const [expand, setExpand] = useState6(defaultOpen || !toggle);
+  const [expandX, setExpandX] = useState6(shrinkX ? expand : true);
+  const [expandY, setExpandY] = useState6(expand);
+  useEffect6(() => {
     let t;
     if (expand) {
       if (shrinkX) {
@@ -977,29 +1180,42 @@ var ThreadMessage = ({ classNames, message, collapse, debug, onSuggest, onDelete
     }, message);
   }
   const { role, content = [] } = message;
-  const tools = content.filter((block) => block.type === "tool_use" || block.type === "tool_result");
-  if (collapse && tools.length > 0) {
+  const toolBlocks = content.filter((block) => block.type === "tool_use" || block.type === "tool_result");
+  if (collapse && toolBlocks.length > 0) {
     let request;
-    const lines = tools.map((tool) => {
-      switch (tool.type) {
+    const json = [];
+    const lines = toolBlocks.map((block) => {
+      switch (block.type) {
         case "tool_use": {
-          request = tool;
-          return `Calling ${tool.name}...`;
+          request = block;
+          json.push(block);
+          return `Calling ${block.name}...`;
         }
         case "tool_result": {
           if (!request) {
+            log6.warn("unexpected message", {
+              tool: block
+            }, {
+              F: __dxlog_file5,
+              L: 57,
+              S: void 0,
+              C: (f, a) => f(...a)
+            });
             return "Error";
           }
+          json.push(block);
           return `Processed ${request.name}`;
         }
-        default:
+        default: {
+          request = void 0;
           return "Error";
+        }
       }
     });
     return /* @__PURE__ */ React5.createElement("div", {
       className: mx5("flex", classNames)
     }, /* @__PURE__ */ React5.createElement("div", {
-      className: "w-full p-1 px-2 overflow-hidden rounded-md bg-base"
+      className: "w-full p-1 px-2 overflow-hidden rounded-md bg-baseSurface"
     }, /* @__PURE__ */ React5.createElement(ToggleContainer, {
       title: /* @__PURE__ */ React5.createElement(StatusLine, {
         lines,
@@ -1007,7 +1223,7 @@ var ThreadMessage = ({ classNames, message, collapse, debug, onSuggest, onDelete
       }),
       toggle: true
     }, /* @__PURE__ */ React5.createElement(Json, {
-      data: content[content.length - 1],
+      data: json,
       classNames: "!p-1 text-xs"
     }))));
   }
@@ -1031,7 +1247,7 @@ var ThreadMessage = ({ classNames, message, collapse, debug, onSuggest, onDelete
 var Block = ({ block, role, onSuggest }) => {
   const Component = componentMap[block.type] ?? componentMap.default;
   return /* @__PURE__ */ React5.createElement("div", {
-    className: mx5("p-1 px-2 overflow-hidden rounded-md", (block.type !== "text" || block.disposition) && "w-full bg-base", block.type === "text" && role === "user" && "bg-blue-200 dark:bg-blue-800")
+    className: mx5("p-1 px-2 overflow-hidden rounded-md", (block.type !== "text" || block.disposition) && "w-full bg-baseSurface", block.type === "text" && role === "user" && "bg-primary-200 dark:bg-primary-500")
   }, /* @__PURE__ */ React5.createElement(Component, {
     block,
     onSuggest
@@ -1039,16 +1255,15 @@ var Block = ({ block, role, onSuggest }) => {
 };
 var titles = {
   ["cot"]: "Chain of thought",
-  // TODO(burdon): Only show if debugging.
+  ["artifact"]: "Artifact",
   ["tool_use"]: "Tool request",
-  ["tool_result"]: "Tool result",
-  ["artifact"]: "Artifact"
+  ["tool_result"]: "Tool result"
 };
 var componentMap = {
   text: ({ block }) => {
     invariant6(block.type === "text", void 0, {
       F: __dxlog_file5,
-      L: 128,
+      L: 134,
       S: void 0,
       A: [
         "block.type === 'text'",
@@ -1083,7 +1298,7 @@ var componentMap = {
   json: ({ block, onSuggest }) => {
     invariant6(block.type === "json", void 0, {
       F: __dxlog_file5,
-      L: 153,
+      L: 159,
       S: void 0,
       A: [
         "block.type === 'json'",
@@ -1135,7 +1350,7 @@ var componentMap = {
 var Thread = ({ messages, streaming, collapse, debug, onSubmit, onStop, onSuggest, onDelete }) => {
   const { t } = useTranslation(AUTOMATION_PLUGIN);
   const scroller = useRef3(null);
-  const [text, setText] = useState6("");
+  const [text, setText] = useState7("");
   const handleKeyDown = useCallback2((ev) => {
     switch (ev.key) {
       case "Escape": {
@@ -1155,7 +1370,7 @@ var Thread = ({ messages, streaming, collapse, debug, onSubmit, onStop, onSugges
   }, [
     text
   ]);
-  const { messages: lines = [] } = useMemo2(() => {
+  const { messages: lines = [] } = useMemo4(() => {
     if (!collapse) {
       return {
         messages: messages ?? []
@@ -1232,7 +1447,7 @@ var Thread = ({ messages, streaming, collapse, debug, onSubmit, onStop, onSugges
     active: streaming
   }), /* @__PURE__ */ React6.createElement(Input.Root, null, /* @__PURE__ */ React6.createElement(Input.TextInput, {
     autoFocus: true,
-    classNames: "px-2 bg-base rounded",
+    classNames: "px-2 baseSurface rounded",
     placeholder: t("chat input placeholder"),
     value: text,
     onChange: (ev) => setText(ev.target.value),
@@ -1252,79 +1467,26 @@ var Thread = ({ messages, streaming, collapse, debug, onSubmit, onStop, onSugges
 // packages/plugins/experimental/plugin-automation/src/components/ChatContainer/ChatContainer.tsx
 var __dxlog_file6 = "/home/runner/work/dxos/dxos/packages/plugins/experimental/plugin-automation/src/components/ChatContainer/ChatContainer.tsx";
 var ChatContainer = ({ chat, role }) => {
-  const config = useConfig();
-  const space = getSpace(chat);
-  const aiClient = useCapability(AutomationCapabilities.AiClient);
-  const artifactDefinitions = useCapabilities(Capabilities.ArtifactDefinition);
-  const globalTools = useCapabilities(Capabilities.Tools);
-  const functions = useQuery2(space, Filter2.schema(FunctionType2));
-  const serviceRegistry = useMemo3(() => new MockServiceRegistry(), []);
-  const [serviceTools, setServiceTools] = useState7([]);
-  useEffect6(() => {
-    queueMicrotask(async () => {
-      const services = await serviceRegistry.queryServices({});
-      const tools2 = await Promise.all(services.map((service) => createToolsFromService(service)));
-      setServiceTools(tools2.flat());
-    });
-  }, []);
-  const tools = useMemo3(() => [
-    ...globalTools.flat(),
-    ...artifactDefinitions.flatMap((definition) => definition.tools),
-    ...functions.map((fn) => covertFunctionToTool(fn, config.values.runtime?.services?.edge?.url ?? "", space?.id)).filter(isNotNullOrUndefined),
-    ...serviceTools
-  ], [
-    globalTools,
-    artifactDefinitions,
-    functions,
-    serviceTools,
-    space?.id
-  ]);
-  const systemPrompt = useMemo3(() => createSystemPrompt({
-    artifacts: artifactDefinitions.map((definition) => definition.instructions)
-  }), [
-    artifactDefinitions
-  ]);
-  const { dispatchPromise: dispatch } = useIntentDispatcher();
-  const processor = useMemo3(() => new ChatProcessor(aiClient, tools, {
-    space,
-    dispatch
-  }, {
-    model: "@anthropic/claude-3-5-sonnet-20241022",
-    systemPrompt
-  }), [
-    aiClient,
-    tools,
-    space,
-    dispatch,
-    systemPrompt
-  ]);
-  const queueDxn = useMemo3(() => new DXN2(DXN2.kind.QUEUE, [
-    QueueSubspaceTags.DATA,
-    space.id,
-    chat.queue.dxn.parts.at(-1)
-  ]), [
-    chat.queue.dxn
-  ]);
-  const edgeClient = useEdgeClient();
-  const messageQueue = useQueue(edgeClient, queueDxn);
+  const processor = useChatProcessor(chat);
+  const messageQueue = useMessageQueue(chat);
   const messages = [
     ...messageQueue?.items ?? [],
     ...processor.messages.value
   ];
-  const handleSubmit = useCallback3(async (message) => {
+  const handleSubmit = useCallback3(async (text) => {
     if (processor.streaming.value) {
       await processor.cancel();
     }
     invariant7(messageQueue, void 0, {
       F: __dxlog_file6,
-      L: 95,
+      L: 25,
       S: void 0,
       A: [
         "messageQueue",
         ""
       ]
     });
-    await processor.request(message, {
+    await processor.request(text, {
       history: messageQueue.items,
       onComplete: (messages2) => messageQueue.append(messages2)
     });
@@ -1345,7 +1507,9 @@ var ChatContainer = ({ chat, role }) => {
   }, /* @__PURE__ */ React7.createElement(Thread, {
     messages,
     streaming: processor.streaming.value,
+    collapse: true,
     onSubmit: handleSubmit,
+    onSuggest: handleSubmit,
     onStop: handleStop
   }));
 };
@@ -1354,7 +1518,7 @@ var ChatContainer = ({ chat, role }) => {
 var ChatContainer_default = ChatContainer;
 
 export {
-  useServiceRegistry,
+  useServices,
   MarkdownViewer,
   Thread,
   ChatContainer,
@@ -1364,4 +1528,4 @@ export {
   IntentResolver,
   ReactSurface
 };
-//# sourceMappingURL=chunk-5SLV6AUA.mjs.map
+//# sourceMappingURL=chunk-DL7XA62G.mjs.map