@agent-os-lab/agent-game-sdk 0.1.8 → 0.1.10

package/src/runtime-agent-list.ts

@@ -0,0 +1,101 @@
+import {
+  mergeAgentPresence,
+  type AgentGameRuntimeBrowserSubscribeOptions,
+  type AgentPresence,
+  type AgentPresenceStatus,
+  type GameRuntimePatchMessage,
+  type GameRuntimeServerMessage,
+  type GameRuntimeSnapshotMessage,
+  type GameRuntimeSubscription,
+} from "./runtime-client";
+
+export const AGENT_PRESENCE_STATUS_LABELS = {
+  working: "Working",
+  thinking: "Thinking",
+  meeting: "Meeting",
+  resting: "Resting",
+  entertaining: "Entertainment",
+  idle: "Idle",
+  offline: "Offline",
+} satisfies Record<AgentPresenceStatus, string>;
+
+export type AgentPresenceListSubscription = GameRuntimeSubscription & {
+  getAgents(): AgentPresence[];
+};
+
+export type AgentPresenceListClient = {
+  subscribe(options?: AgentGameRuntimeBrowserSubscribeOptions): Promise<GameRuntimeSubscription>;
+};
+
+export type SubscribeAgentPresenceListOptions = AgentGameRuntimeBrowserSubscribeOptions & {
+  onAgentsChange?: (
+    agents: AgentPresence[],
+    message: GameRuntimeSnapshotMessage | GameRuntimePatchMessage,
+  ) => void;
+};
+
+export function reduceAgentPresenceList(
+  current: AgentPresence[],
+  message: GameRuntimeServerMessage,
+): AgentPresence[] {
+  if (message.type === "snapshot") {
+    return mergeAgentPresence([], message.agents);
+  }
+  return mergeAgentPresence(current, message.agents);
+}
+
+export function formatAgentPresenceStatus(status: AgentPresenceStatus): string {
+  return AGENT_PRESENCE_STATUS_LABELS[status];
+}
+
+export function formatAgentPresenceActivitySummary(agent: AgentPresence): string | undefined {
+  if (!isActiveAgentPresenceStatus(agent.status)) {
+    return undefined;
+  }
+  if (agent.activity?.kind === "completed") {
+    return undefined;
+  }
+  const summary = agent.activity?.summary.trim();
+  return summary || undefined;
+}
+
+export async function subscribeAgentPresenceList(
+  client: AgentPresenceListClient,
+  options: SubscribeAgentPresenceListOptions = {},
+): Promise<AgentPresenceListSubscription> {
+  let agents: AgentPresence[] = [];
+  const {
+    onAgentsChange,
+    onPatch,
+    onSnapshot,
+    ...subscribeOptions
+  } = options;
+
+  function applyMessage(message: GameRuntimeSnapshotMessage | GameRuntimePatchMessage) {
+    agents = reduceAgentPresenceList(agents, message);
+    onAgentsChange?.([...agents], message);
+  }
+
+  const subscription = await client.subscribe({
+    ...subscribeOptions,
+    onSnapshot: (message) => {
+      applyMessage(message);
+      onSnapshot?.(message);
+    },
+    onPatch: (message) => {
+      applyMessage(message);
+      onPatch?.(message);
+    },
+  });
+
+  return {
+    ...subscription,
+    getAgents() {
+      return [...agents];
+    },
+  };
+}
+
+function isActiveAgentPresenceStatus(status: AgentPresenceStatus): boolean {
+  return status === "working" || status === "thinking";
+}