@dxos/plugin-assistant 0.10.0 → 0.11.0

package/dist/lib/neutral/hooks/index.mjs

@@ -1,877 +0,0 @@
-import {
-  useChatContext
-} from "../chunk-VTK5R4H3.mjs";
-import {
-  AiChatProcessor,
-  defaultsKeyForProvider,
-  presetsForProvider,
-  resolveProvider
-} from "../chunk-BXI2R4OO.mjs";
-import "../chunk-J5LGTIGS.mjs";
-
-// src/hooks/useSkillRegistry.ts
-import { useCallback, useEffect, useMemo, useState } from "react";
-import { Skill } from "@dxos/compute";
-import { Filter, Obj, Ref } from "@dxos/echo";
-import { useQuery } from "@dxos/react-client/echo";
-import { distinctBy } from "@dxos/util";
-var useSkills = ({ registry, db }) => {
-  const [registrySkills, setRegistrySkills] = useState(() => registry?.query(Filter.type(Skill.Skill)).runSync() ?? []);
-  useEffect(() => {
-    if (!registry) {
-      setRegistrySkills([]);
-      return;
-    }
-    setRegistrySkills(registry.query(Filter.type(Skill.Skill)).runSync());
-    return registry.changed.on(() => {
-      setRegistrySkills(registry.query(Filter.type(Skill.Skill)).runSync());
-    });
-  }, [
-    registry
-  ]);
-  const spaceSkills = useQuery(db, Filter.type(Skill.Skill));
-  return useMemo(() => {
-    const skills = distinctBy([
-      ...registrySkills,
-      ...spaceSkills
-    ], (b) => Obj.getMeta(b).key);
-    skills.sort(({ name: a }, { name: b }) => a.localeCompare(b));
-    return skills;
-  }, [
-    registrySkills,
-    spaceSkills
-  ]);
-};
-var useActiveSkills = ({ context }) => {
-  const [active, setActive] = useState(/* @__PURE__ */ new Map());
-  useEffect(() => {
-    if (!context) {
-      setActive(/* @__PURE__ */ new Map());
-      return;
-    }
-    const updateActive = () => {
-      const skills = context.getSkills();
-      setActive(new Map(skills.map((skill) => [
-        Obj.getMeta(skill).key,
-        skill
-      ]).filter((entry) => entry[0] !== void 0)));
-    };
-    updateActive();
-    return context.subscribeSkills(updateActive);
-  }, [
-    context
-  ]);
-  return active;
-};
-var useSkillHandlers = ({ db, context, registry }) => {
-  const onUpdateSkill = useCallback(async (key, checked) => {
-    if (!context) {
-      return;
-    }
-    if (checked) {
-      const registrySkill = registry?.query(Filter.type(Skill.Skill)).runSync().find((b) => Obj.getMeta(b).key === key);
-      if (registrySkill) {
-        await context.bind({
-          skills: [
-            Ref.fromURI(Skill.registryURI(key))
-          ]
-        });
-      } else {
-        const objects = await db.query(Filter.type(Skill.Skill)).run();
-        const storedSkill = objects.find((skill) => Obj.getMeta(skill).key === key);
-        if (storedSkill) {
-          await context.bind({
-            skills: [
-              Ref.make(storedSkill)
-            ]
-          });
-        }
-      }
-    } else {
-      const registrySkill = registry?.query(Filter.type(Skill.Skill)).runSync().find((b) => Obj.getMeta(b).key === key);
-      if (registrySkill) {
-        await context.unbind({
-          skills: [
-            Ref.fromURI(Skill.registryURI(key))
-          ]
-        });
-      } else {
-        const objects = await db.query(Filter.type(Skill.Skill)).run();
-        const storedSkill = objects.find((skill) => Obj.getMeta(skill).key === key);
-        if (storedSkill) {
-          await context.unbind({
-            skills: [
-              Ref.make(storedSkill)
-            ]
-          });
-        }
-      }
-    }
-  }, [
-    db,
-    context,
-    registry
-  ]);
-  return {
-    onUpdateSkill
-  };
-};
-
-// src/hooks/useChatKeymap.ts
-import { Prec } from "@codemirror/state";
-import { keymap } from "@codemirror/view";
-import { useMemo as useMemo2 } from "react";
-import { isTruthy } from "@dxos/util";
-var useChatKeymapExtensions = ({ event }) => {
-  return useMemo2(() => {
-    return [
-      Prec.highest(keymap.of([
-        {
-          key: "Mod-ArrowUp",
-          preventDefault: true,
-          run: () => {
-            event.emit({
-              type: "nav-previous"
-            });
-            return true;
-          },
-          shift: () => {
-            event.emit({
-              type: "thread-open"
-            });
-            return true;
-          }
-        },
-        {
-          key: "Mod-ArrowDown",
-          preventDefault: true,
-          run: () => {
-            event.emit({
-              type: "nav-next"
-            });
-            return true;
-          },
-          shift: () => {
-            event.emit({
-              type: "thread-close"
-            });
-            return true;
-          }
-        }
-      ]))
-    ].filter(isTruthy);
-  }, [
-    event
-  ]);
-};
-
-// src/hooks/useChatProcessor.ts
-import { RegistryContext } from "@effect-atom/atom-react";
-import * as Effect from "effect/Effect";
-import * as Layer from "effect/Layer";
-import { useContext, useMemo as useMemo3, useState as useState2 } from "react";
-import { AiService, OpaqueToolkit } from "@dxos/ai";
-import { Capabilities } from "@dxos/app-framework";
-import { useCapability } from "@dxos/app-framework/ui";
-import { AiSession } from "@dxos/assistant";
-import { AgentService, Credential, ServiceResolver } from "@dxos/compute";
-import { Database, Obj as Obj2, Ref as Ref2, Registry } from "@dxos/echo";
-import { EffectEx } from "@dxos/effect";
-import { log } from "@dxos/log";
-import { useObject } from "@dxos/react-client/echo";
-import { useAsyncEffect } from "@dxos/react-ui";
-var __dxlog_file = "/__w/dxos/dxos/packages/plugins/plugin-assistant/src/hooks/useChatProcessor.ts";
-var useChatProcessor = ({ space, chat, preset, runtime: runtime3, registry, settings }) => {
-  const observableRegistry = useContext(RegistryContext);
-  const [feedSnapshot] = useObject(chat?.feed);
-  const feed = Obj2.getReactiveOrUndefined(feedSnapshot);
-  const [session, setSession] = useState2();
-  useAsyncEffect(async () => {
-    if (!space || !chat || !feed) {
-      return;
-    }
-    const runtime4 = await EffectEx.runAndForwardErrors(Effect.runtime().pipe(Effect.provide(Database.layer(space.db))));
-    const session2 = new AiSession.Session({
-      feed,
-      runtime: runtime4,
-      registry: observableRegistry
-    });
-    await session2.open();
-    setSession(session2);
-    return () => {
-      void session2.close();
-      setSession(void 0);
-    };
-  }, [
-    space,
-    chat,
-    feed
-  ]);
-  const serviceResolver = useCapability(Capabilities.ServiceResolver);
-  const processor = useMemo3(() => {
-    if (!runtime3 || !session || !chat || !feed || !space) {
-      return void 0;
-    }
-    const spaceLayer = ServiceResolver.provide({
-      space: space.id
-    }, Database.Service, Credential.CredentialsService, AiService.AiService, AgentService.AgentService, Registry.Service, OpaqueToolkit.OpaqueToolkitProvider).pipe(Layer.provide(Layer.succeed(ServiceResolver.ServiceResolver, serviceResolver)));
-    log("creating processor", {
-      preset,
-      model: preset?.model,
-      settings
-    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 56, S: void 0 });
-    return new AiChatProcessor(session, runtime3, feed, spaceLayer, {
-      chat: chat ? Ref2.make(chat) : void 0,
-      observableRegistry,
-      registry,
-      model: preset?.model,
-      provider: preset?.provider
-    });
-  }, [
-    runtime3,
-    session,
-    registry,
-    preset,
-    chat,
-    feed,
-    space?.id
-  ]);
-  return processor;
-};
-
-// src/hooks/useChatServices.ts
-import { Capabilities as Capabilities2 } from "@dxos/app-framework";
-import { useCapability as useCapability2 } from "@dxos/app-framework/ui";
-import { AppSpace } from "@dxos/app-toolkit";
-import { useClient } from "@dxos/react-client";
-var useChatServices = ({ id }) => {
-  const client = useClient();
-  id ??= AppSpace.getPersonalSpace(client)?.id;
-  const runtime3 = useCapability2(Capabilities2.ProcessManagerRuntime);
-  return id ? runtime3 : void 0;
-};
-
-// src/hooks/useChatToolbarActions.ts
-import * as Effect2 from "effect/Effect";
-import { useOperationInvoker } from "@dxos/app-framework/ui";
-import { Chat } from "@dxos/assistant-toolkit";
-import { Filter as Filter2, Obj as Obj3, Query, Type } from "@dxos/echo";
-import { EffectEx as EffectEx2 } from "@dxos/effect";
-import { invariant } from "@dxos/invariant";
-import { MenuBuilder, useMenuBuilder } from "@dxos/react-ui-menu";
-import { meta } from "#meta";
-import { AssistantOperation } from "#types";
-var __dxlog_file2 = "/__w/dxos/dxos/packages/plugins/plugin-assistant/src/hooks/useChatToolbarActions.ts";
-var useChatToolbarActions = ({ chat, companionTo }) => {
-  const { invoke } = useOperationInvoker();
-  const { db } = useChatContext("useChatToolbarActions");
-  return useMenuBuilder((get) => {
-    const chats = companionTo && db ? get(db.query(Query.select(Filter2.id(companionTo.id)).targetOf(Chat.CompanionTo).source()).atom) : [];
-    const builder = MenuBuilder.make().root({
-      label: [
-        "chat-toolbar.title",
-        {
-          ns: meta.profile.key
-        }
-      ]
-    }).action("new", {
-      label: [
-        "new-thread.button",
-        {
-          ns: meta.profile.key
-        }
-      ],
-      icon: "ph--plus--regular",
-      type: "new",
-      disabled: !companionTo
-    }, () => {
-      invariant(companionTo, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file2, L: 38, S: void 0, A: ["companionTo", ""] });
-      return invoke(AssistantOperation.SetCurrentChat, {
-        companionTo,
-        chat: void 0
-      }).pipe(EffectEx2.runAndForwardErrors);
-    }).action("rename", {
-      label: [
-        "rename-thread.button",
-        {
-          ns: meta.profile.key
-        }
-      ],
-      icon: "ph--magic-wand--regular",
-      type: "rename",
-      disabled: !chat
-    }, () => Effect2.gen(function* () {
-      invariant(chat, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file2, L: 54, S: this, A: ["chat", ""] });
-      yield* invoke(AssistantOperation.UpdateChatName, {
-        chat
-      }, {
-        spaceId: db?.spaceId
-      });
-    }).pipe(EffectEx2.runAndForwardErrors)).action("branch", {
-      label: [
-        "branch-thread.menu",
-        {
-          ns: meta.profile.key
-        }
-      ],
-      icon: "ph--git-branch--regular",
-      type: "branch",
-      disabled: !chat || !db
-    }, () => Effect2.gen(function* () {
-      invariant(chat, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file2, L: 71, S: this, A: ["chat", ""] });
-      yield* invoke(AssistantOperation.ForkChat, {
-        chat,
-        companionTo
-      }, {
-        spaceId: db?.spaceId
-      });
-    }).pipe(EffectEx2.runAndForwardErrors));
-    if (chats.length > 0) {
-      builder.group("chats", {
-        label: [
-          "chat-history.label",
-          {
-            ns: meta.profile.key
-          }
-        ],
-        icon: "ph--clock-counter-clockwise--regular",
-        selectCardinality: "single",
-        variant: "dropdownMenu"
-      }, (builder2) => {
-        chats.toSorted((a, b) => a.id.localeCompare(b.id)).forEach((chat2) => {
-          get(Obj3.atomProperty(chat2, "name"));
-          builder2.action(chat2.id, {
-            label: Obj3.getLabel(chat2) ?? [
-              "object-name.placeholder",
-              {
-                ns: Type.getTypename(Chat.Chat)
-              }
-            ]
-          }, () => Effect2.gen(function* () {
-            invariant(companionTo, void 0, { "~LogMeta": "~LogMeta", F: __dxlog_file2, L: 102, S: this, A: ["companionTo", ""] });
-            yield* invoke(AssistantOperation.SetCurrentChat, {
-              companionTo,
-              chat: chat2
-            });
-          }).pipe(EffectEx2.runAndForwardErrors));
-        });
-      });
-    }
-    return builder.build();
-  }, [
-    db?.spaceId,
-    companionTo?.id,
-    chat?.id,
-    invoke
-  ]);
-};
-
-// src/hooks/useContextBinder.ts
-import { RegistryContext as RegistryContext2 } from "@effect-atom/atom-react";
-import * as Effect3 from "effect/Effect";
-import { useContext as useContext2, useState as useState3 } from "react";
-import { AiContext } from "@dxos/assistant";
-import { Database as Database2 } from "@dxos/echo";
-import { EffectEx as EffectEx3 } from "@dxos/effect";
-import { useAsyncEffect as useAsyncEffect2 } from "@dxos/react-ui";
-var useContextBinder = (space, feed) => {
-  const registry = useContext2(RegistryContext2);
-  const [binder, setBinder] = useState3();
-  useAsyncEffect2(async () => {
-    setBinder(void 0);
-    if (!space || !feed) {
-      return;
-    }
-    const runtime3 = await EffectEx3.runAndForwardErrors(Effect3.runtime().pipe(Effect3.provide(Database2.layer(space.db))));
-    const binder2 = new AiContext.Binder({
-      feed,
-      runtime: runtime3,
-      registry
-    });
-    await binder2.open();
-    setBinder(binder2);
-    return () => {
-      void binder2.close();
-    };
-  }, [
-    space,
-    feed
-  ]);
-  return binder;
-};
-
-// src/hooks/useContextObjects.ts
-import { Atom, useAtomValue } from "@effect-atom/atom-react";
-import { useCallback as useCallback2 } from "react";
-var emptyObjectsAtom = Atom.make([]);
-var useContextObjects = ({ db, context }) => {
-  const objects = useAtomValue(context?.objects ?? emptyObjectsAtom);
-  const handleUpdateObject = useCallback2(async (dxn, checked) => {
-    if (!db || !context) {
-      return;
-    }
-    const ref = db.makeRef(dxn);
-    await ref.load();
-    if (checked) {
-      await context.bind({
-        objects: [
-          ref
-        ]
-      });
-    } else {
-      await context.unbind({
-        objects: [
-          ref
-        ]
-      });
-    }
-  }, [
-    db,
-    context
-  ]);
-  return {
-    objects,
-    onUpdateObject: handleUpdateObject
-  };
-};
-
-// src/hooks/useDebug.ts
-import { useCallback as useCallback3 } from "react";
-import { trim } from "@dxos/util";
-var useDebug = ({ processor }) => {
-  return useCallback3(async () => {
-    const objects = processor.context.getObjects();
-    const skills = processor.context.getSkills();
-    const system = await processor.getSystemPrompt();
-    const tools = await processor.getTools() ?? {};
-    console.group("Chat", {
-      objects,
-      skills
-    });
-    try {
-      console.log(trim`
-        System Prompt:
-        ${system}
-      `);
-      console.log(trim`
-        Tools:
-        ${Object.values(tools).map((tool) => JSON.stringify(tool, null, 2)).join("\n")}
-      `);
-    } finally {
-      console.groupEnd();
-    }
-  }, [
-    processor
-  ]);
-};
-
-// src/hooks/useFilteredTypes.ts
-import * as Option from "effect/Option";
-import { useEffect as useEffect2, useState as useState4 } from "react";
-import { Type as Type2 } from "@dxos/echo";
-import { HiddenAnnotation, getTypeAnnotation } from "@dxos/echo/Annotation";
-import { Kind as EntityKind } from "@dxos/echo/Entity";
-var getFilteredTypes = (db) => Array.from(new Set(db.graph.registry.list().filter(Type2.isType).filter((schema) => getTypeAnnotation(Type2.getSchema(schema))?.kind !== EntityKind.Relation).filter((schema) => !HiddenAnnotation.get(Type2.getSchema(schema)).pipe(Option.getOrElse(() => false)))));
-var useFilteredTypes = (db) => {
-  const [types, setTypes] = useState4([]);
-  useEffect2(() => {
-    if (!db) {
-      return;
-    }
-    setTypes(getFilteredTypes(db));
-    return db.graph.registry.changed.on(() => {
-      setTypes(getFilteredTypes(db));
-    });
-  }, [
-    db
-  ]);
-  return types;
-};
-
-// src/hooks/useFlush.ts
-import { useCallback as useCallback4, useRef, useState as useState5 } from "react";
-var useFlush = (space) => {
-  const [state, setState] = useState5("idle");
-  const resetTimer = useRef(null);
-  const handleFlush = useCallback4(() => {
-    if (!space) {
-      return;
-    }
-    queueMicrotask(async () => {
-      if (resetTimer.current) {
-        clearTimeout(resetTimer.current);
-      }
-      setState("flushing");
-      await space.db.flush();
-      setState("flushed");
-      resetTimer.current = setTimeout(() => {
-        setState("idle");
-        resetTimer.current = null;
-      }, 1e3);
-    });
-  }, [
-    space
-  ]);
-  return {
-    state,
-    handleFlush
-  };
-};
-
-// src/hooks/usePresets.ts
-import { useAtomValue as useAtomValue2 } from "@effect-atom/atom-react";
-import { useCallback as useCallback5, useEffect as useEffect3, useMemo as useMemo4, useState as useState6 } from "react";
-import { Provider } from "@dxos/ai";
-import { useOptionalCapability } from "@dxos/app-framework/ui";
-import { EffectEx as EffectEx4 } from "@dxos/effect";
-import { AssistantCapabilities, Ollama } from "#types";
-var usePresets = (settings) => {
-  const [preset, setPreset] = useState6();
-  const ollamaManager = useOptionalCapability(AssistantCapabilities.OllamaManager);
-  const provider = resolveProvider(settings.modelProvider, !!ollamaManager);
-  const defaultModel = settings.modelDefaults?.[defaultsKeyForProvider(provider)];
-  const localModels = useAtomValue2(ollamaManager?.state ?? Ollama.emptyState);
-  useEffect3(() => {
-    if (provider === Provider.builtIn.id && ollamaManager && localModels.kind === "idle") {
-      void EffectEx4.runPromise(ollamaManager.refresh);
-    }
-  }, [
-    provider,
-    ollamaManager,
-    localModels.kind
-  ]);
-  const lmStudioProbeUrl = provider === Provider.lmStudio.id && ollamaManager ? `${Provider.lmStudio.endpoint}/v1/models` : void 0;
-  const lmStudioReachable = useEndpointReachable(lmStudioProbeUrl);
-  const presets = useMemo4(() => {
-    const base = presetsForProvider(provider);
-    if (provider === Provider.lmStudio.id && ollamaManager) {
-      return lmStudioReachable ? base : [];
-    }
-    if (provider === Provider.builtIn.id && ollamaManager) {
-      const installed = new Set(localModels.models.map((model) => model.name));
-      return base.filter((preset2) => installed.has(preset2.backend));
-    }
-    return base;
-  }, [
-    provider,
-    ollamaManager,
-    lmStudioReachable,
-    localModels
-  ]);
-  const presetOptions = useMemo4(() => presets.map(({ id, model, label }) => ({
-    id,
-    label: label ?? model
-  })), [
-    presets
-  ]);
-  useEffect3(() => {
-    const configured = defaultModel ? presets.find((preset2) => preset2.model === defaultModel) : void 0;
-    setPreset(configured ?? presets[0]);
-  }, [
-    presets,
-    defaultModel
-  ]);
-  const handlePresetChange = useCallback5((id) => {
-    const preset2 = presets.find((preset3) => preset3.id === id);
-    if (preset2) {
-      setPreset(preset2);
-    }
-  }, [
-    presets
-  ]);
-  return {
-    preset,
-    presets: presetOptions,
-    onPresetChange: handlePresetChange
-  };
-};
-var useEndpointReachable = (url) => {
-  const [reachable, setReachable] = useState6(false);
-  useEffect3(() => {
-    if (!url) {
-      setReachable(false);
-      return;
-    }
-    let active = true;
-    const controller = new AbortController();
-    const timer = setTimeout(() => controller.abort(), 1500);
-    fetch(url, {
-      signal: controller.signal
-    }).then((response) => active && setReachable(response.ok)).catch(() => active && setReachable(false)).finally(() => clearTimeout(timer));
-    return () => {
-      active = false;
-      controller.abort();
-      clearTimeout(timer);
-    };
-  }, [
-    url
-  ]);
-  return reachable;
-};
-
-// src/hooks/useReferencesProvider.ts
-import { useMemo as useMemo5 } from "react";
-import { Filter as Filter3, Obj as Obj4 } from "@dxos/echo";
-var useReferencesProvider = (space) => {
-  return useMemo5(() => {
-    if (!space) {
-      return void 0;
-    }
-    return {
-      getReferences: async ({ query }) => {
-        const objects = await space.db.query(Filter3.everything()).run();
-        return objects.filter(Obj4.isObject).filter((object) => stringMatch(query, Obj4.getLabel(object) ?? "")).map((object) => ({
-          uri: Obj4.getURI(object),
-          label: Obj4.getLabel(object) ?? ""
-        }));
-      },
-      resolveReference: async ({ uri }) => {
-        const object = await space.db.query(Filter3.id(uri)).first();
-        return {
-          uri,
-          label: Obj4.getLabel(object) ?? ""
-        };
-      }
-    };
-  }, [
-    space
-  ]);
-};
-var stringMatch = (query, label) => label.toLowerCase().startsWith(query.toLowerCase());
-
-// src/hooks/useTraceMessages.ts
-import { Atom as Atom2 } from "@effect-atom/atom";
-import { useAtomValue as useAtomValue3 } from "@effect-atom/atom-react";
-import { pipe } from "effect/Function";
-import { useMemo as useMemo6 } from "react";
-import { Trace } from "@dxos/compute";
-import { Filter as Filter4, Query as Query2 } from "@dxos/echo";
-import { FeedTraceSink } from "@dxos/functions-runtime";
-var atomEmpty = Atom2.make(() => []);
-var getTraceMessagesAtom = (space) => pipe(space.db.query(FeedTraceSink.query).atom, Atom2.map((feeds) => (
-  // TODO(dmaretskyi): Single query with limit(1) and feed traversal when query supports it.
-  space.db.query(feeds.length > 0 ? Query2.type(Trace.Message).from(feeds[0]) : Query2.select(Filter4.nothing())).atom
-)), (atom) => Atom2.make((get) => get(get(atom))));
-var useTraceMessages = (space) => {
-  const atom = useMemo6(() => space ? getTraceMessagesAtom(space) : atomEmpty, [
-    space
-  ]);
-  return useAtomValue3(atom);
-};
-
-// src/hooks/useHomeSuggestions.ts
-import { useMemo as useMemo7, useState as useState7 } from "react";
-import { useOperationInvoker as useOperationInvoker2 } from "@dxos/app-framework/ui";
-import { useAsyncEffect as useAsyncEffect3, useTranslation } from "@dxos/react-ui";
-import { meta as meta2 } from "#meta";
-import { AssistantOperation as AssistantOperation2 } from "#types";
-var FALLBACK_SUGGESTION_KEYS = [
-  "space-home.suggestion-magazine.label",
-  "space-home.suggestion-spreadsheet.label",
-  "space-home.suggestion-kanban.label"
-];
-var useHomeSuggestions = (space) => {
-  const { t } = useTranslation(meta2.profile.key);
-  const { invokePromise } = useOperationInvoker2();
-  const fallbacks = useMemo7(() => FALLBACK_SUGGESTION_KEYS.map((key) => t(key, {
-    year: (/* @__PURE__ */ new Date()).getFullYear()
-  })), [
-    t
-  ]);
-  const [suggestions, setSuggestions] = useState7(void 0);
-  useAsyncEffect3(async (controller) => {
-    setSuggestions(void 0);
-    if (!space) {
-      return;
-    }
-    const result = await invokePromise(AssistantOperation2.GenerateHomeSuggestions, {
-      db: space.db
-    }, {
-      spaceId: space.db.spaceId
-    });
-    if (controller.signal.aborted) {
-      return;
-    }
-    const prompts = result.data?.prompts;
-    setSuggestions(prompts && prompts.length > 0 ? prompts : fallbacks);
-  }, [
-    space,
-    invokePromise
-  ]);
-  return suggestions;
-};
-
-// src/hooks/useProcessEphemeralStatus.ts
-import { Atom as Atom3 } from "@effect-atom/atom";
-import { useAtomValue as useAtomValue4 } from "@effect-atom/atom-react";
-import * as Effect4 from "effect/Effect";
-import * as Fiber from "effect/Fiber";
-import * as Stream from "effect/Stream";
-import { useEffect as useEffect4, useMemo as useMemo8, useRef as useRef2, useState as useState8 } from "react";
-import { Capabilities as Capabilities3 } from "@dxos/app-framework";
-import { useOptionalCapability as useOptionalCapability2 } from "@dxos/app-framework/ui";
-import { Process, ServiceResolver as ServiceResolver2 } from "@dxos/compute";
-import { ProcessManager } from "@dxos/compute-runtime";
-import { resolveEphemeralStatusUpdate } from "#execution-graph";
-var atomEmpty2 = Atom3.make(() => []);
-var ACTIVE_PROCESS_STATES = /* @__PURE__ */ new Set([
-  Process.State.RUNNING,
-  Process.State.HYBERNATING
-]);
-var isTerminalActivityLine = (line) => line.endsWith(" - Success") || line.endsWith(" - Error") || line === "Agent completed request";
-var collectDescendantPids = (processes, rootPid) => {
-  const pids = /* @__PURE__ */ new Set([
-    rootPid
-  ]);
-  let expanded = true;
-  while (expanded) {
-    expanded = false;
-    for (const process of processes) {
-      const pid = String(process.pid);
-      const parentPid = process.parentPid ? String(process.parentPid) : void 0;
-      if (parentPid && pids.has(parentPid) && !pids.has(pid)) {
-        pids.add(pid);
-        expanded = true;
-      }
-    }
-  }
-  return pids;
-};
-var resolveSubscribePids = (agentPid, processes) => {
-  const rootPid = String(agentPid);
-  const descendants = collectDescendantPids(processes, rootPid);
-  const activePids = processes.filter((process) => descendants.has(String(process.pid)) && ACTIVE_PROCESS_STATES.has(process.state)).map((process) => process.pid);
-  if (activePids.length > 0) {
-    return activePids;
-  }
-  return [
-    agentPid
-  ];
-};
-var attachActiveHandle = (processManager, pid) => Effect4.gen(function* () {
-  const maxAttempts = 15;
-  for (let attempt = 0; attempt < maxAttempts; attempt++) {
-    const handle = yield* processManager.attach(pid).pipe(Effect4.catchAll(() => Effect4.succeed(void 0)));
-    if (handle && ACTIVE_PROCESS_STATES.has(handle.status.state)) {
-      return handle;
-    }
-    if (handle) {
-      return void 0;
-    }
-    yield* Effect4.sleep(100);
-  }
-  return void 0;
-});
-var useProcessEphemeralStatus = (agentPid, space) => {
-  const runtime3 = useOptionalCapability2(Capabilities3.ProcessManagerRuntime);
-  const monitor = useOptionalCapability2(Capabilities3.ProcessMonitor);
-  const processes = useAtomValue4(monitor?.processTreeAtom ?? atomEmpty2);
-  const [status, setStatus] = useState8();
-  const fibersRef = useRef2([]);
-  const subscribePidsKey = useMemo8(() => {
-    if (!agentPid) {
-      return "";
-    }
-    return resolveSubscribePids(agentPid, processes).map((pid) => String(pid)).sort().join("|");
-  }, [
-    agentPid,
-    processes
-  ]);
-  useEffect4(() => {
-    if (!agentPid || !space?.id || !runtime3 || !subscribePidsKey) {
-      setStatus(void 0);
-      return;
-    }
-    const pids = subscribePidsKey.split("|").map((pid) => Process.ID.make(pid));
-    let disposed = false;
-    let replaying = true;
-    let replayStatus;
-    const endReplay = () => {
-      replaying = false;
-      if (!disposed) {
-        setStatus(replayStatus);
-      }
-    };
-    queueMicrotask(endReplay);
-    fibersRef.current = [];
-    const handleEphemeralMessage = (message) => {
-      const update = resolveEphemeralStatusUpdate(message);
-      if (update === "unchanged") {
-        return;
-      }
-      if (replaying) {
-        replayStatus = update.line;
-        return;
-      }
-      setStatus(update.line);
-    };
-    const layer = ServiceResolver2.provide({
-      space: space.id
-    }, ProcessManager.ProcessManagerService);
-    const subscribe = (pid) => Effect4.gen(function* () {
-      if (disposed) {
-        return;
-      }
-      const processManager = yield* ProcessManager.ProcessManagerService;
-      const handle = yield* attachActiveHandle(processManager, pid);
-      if (!handle || disposed) {
-        return;
-      }
-      const fiber = yield* handle.subscribeEphemeral().pipe(Stream.runForEach((message) => Effect4.sync(() => {
-        if (disposed) {
-          return;
-        }
-        handleEphemeralMessage(message);
-      })), Effect4.forkDaemon);
-      if (disposed) {
-        yield* Fiber.interrupt(fiber);
-        return;
-      }
-      fibersRef.current.push(fiber);
-    });
-    void runtime3.runPromise(Effect4.forEach(pids, subscribe, {
-      concurrency: "unbounded",
-      discard: true
-    }).pipe(Effect4.provide(layer))).catch(() => {
-      if (!disposed) {
-        setStatus(void 0);
-      }
-    });
-    return () => {
-      disposed = true;
-      const fibers = fibersRef.current;
-      fibersRef.current = [];
-      for (const fiber of fibers) {
-        void runtime3.runPromise(Fiber.interrupt(fiber).pipe(Effect4.provide(layer)));
-      }
-    };
-  }, [
-    agentPid,
-    runtime3,
-    space?.id,
-    subscribePidsKey
-  ]);
-  return status;
-};
-export {
-  getTraceMessagesAtom,
-  isTerminalActivityLine,
-  useActiveSkills,
-  useChatKeymapExtensions,
-  useChatProcessor,
-  useChatServices,
-  useChatToolbarActions,
-  useContextBinder,
-  useContextObjects,
-  useDebug,
-  useFilteredTypes,
-  useFlush,
-  useHomeSuggestions,
-  usePresets,
-  useProcessEphemeralStatus,
-  useReferencesProvider,
-  useSkillHandlers,
-  useSkills,
-  useTraceMessages
-};
-//# sourceMappingURL=index.mjs.map