@alia.onl/sdk 15.8.0 → 19.0.0

package/src/runtime/use-agent-activity.ts

@@ -0,0 +1,80 @@
+import { useOxy } from '@oxy.so/services/ui/client';
+import { useCallback, useEffect, useState } from 'react';
+import { io } from 'socket.io-client';
+import { useAliaApi } from './api';
+import { useAliaRuntime } from './config';
+import { AGENT_ROUTES } from './routes';
+import {
+  type AgentActivityEvent,
+  type AgentActivityState,
+  type AgentSessionStatus,
+  applyActivityEvent,
+  hydrateActivity,
+  INITIAL_ACTIVITY,
+} from './thread/agent-activity';
+
+/** What a run has done so far, live. */
+export type UseAgentActivityResult = AgentActivityState;
+
+/**
+ * Follow one agent run live: the session's room over Socket.IO, seeded with what the run had
+ * already done (`GET` on its status, best-effort — the socket still carries live progress).
+ *
+ * @param sessionId - The agent session to follow; `null` follows nothing.
+ * @param agentId - The agent's own room too, for the runs that announce themselves there.
+ */
+export function useAgentActivity(
+  sessionId: string | null,
+  agentId?: string | null,
+): UseAgentActivityResult {
+  const [state, setState] = useState<AgentActivityState>(INITIAL_ACTIVITY);
+  const api = useAliaApi();
+  const { apiUrl } = useAliaRuntime();
+  const { oxyServices } = useOxy();
+
+  const handleEvent = useCallback((event: AgentActivityEvent) => {
+    setState((prev) => applyActivityEvent(prev, event));
+  }, []);
+
+  useEffect(() => {
+    if (!sessionId) return;
+
+    setState(INITIAL_ACTIVITY);
+    let cancelled = false;
+
+    api
+      .get<AgentSessionStatus>(AGENT_ROUTES.sessionStatus(sessionId))
+      .then(({ data }) => {
+        if (!cancelled) setState((prev) => hydrateActivity(prev, data, sessionId));
+      })
+      .catch(() => {
+        // Socket updates still provide live progress; hydration is best-effort.
+      });
+
+    const socket = io(apiUrl, {
+      transports: ['websocket'],
+      // The function form reads a fresh token on every (re)connect, so it survives rotation.
+      auth: (cb) => cb({ token: oxyServices.session.accessToken }),
+      reconnection: true,
+      reconnectionAttempts: Infinity,
+      reconnectionDelay: 1000,
+      reconnectionDelayMax: 10000,
+    });
+
+    socket.on('connect', () => {
+      socket.emit('subscribe-agent-session', sessionId);
+      if (agentId) socket.emit('subscribe-agent', agentId);
+    });
+
+    socket.on('agent-activity', (event: AgentActivityEvent) => {
+      if (event.sessionId === sessionId) handleEvent(event);
+    });
+
+    return () => {
+      cancelled = true;
+      socket.disconnect();
+    };
+  }, [sessionId, agentId, handleEvent, api, apiUrl, oxyServices]);
+
+  return state;
+}