@axiom-lattice/protocols 4.6.1 → 4.7.1

package/dist/index.js

@@ -20,6 +20,9 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AFFINITY_ECHO_HEADER: () => AFFINITY_ECHO_HEADER,
+  AFFINITY_MISS_CODE: () => AFFINITY_MISS_CODE,
+  AGENT_AFFINITY_HEADER: () => AGENT_AFFINITY_HEADER,
   AgentType: () => AgentType,
   ChannelBindingMigrationConflictError: () => ChannelBindingMigrationConflictError,
   DuplicateChannelBindingSubjectError: () => DuplicateChannelBindingSubjectError,
@@ -40,12 +43,15 @@ __export(index_exports, {
   ProjectRoomCursorError: () => ProjectRoomCursorError,
   ProjectTaskStoreUnsupportedError: () => ProjectTaskStoreUnsupportedError,
   QueueType: () => QueueType,
+  REPLICA_HEADER: () => REPLICA_HEADER,
   ScheduleExecutionType: () => ScheduleExecutionType,
   ScheduleType: () => ScheduleType,
   ScheduledTaskStatus: () => ScheduledTaskStatus,
   UIComponentType: () => UIComponentType,
   WORKSPACE_PUBLIC_ROOM_PROJECT_PREFIX: () => WORKSPACE_PUBLIC_ROOM_PROJECT_PREFIX,
+  affinityEchoHeaders: () => affinityEchoHeaders,
   assertGenericProjectConfig: () => assertGenericProjectConfig,
+  buildAffinityMissChunk: () => buildAffinityMissChunk,
   createAgentWebAppStreamProjectionContext: () => createAgentWebAppStreamProjectionContext,
   descriptorDataValue: () => descriptorDataValue,
   effectiveGrants: () => effectiveGrants,
@@ -81,6 +87,7 @@ __export(index_exports, {
   snapshotProjectRoomMembershipRealtime: () => snapshotProjectRoomMembershipRealtime,
   snapshotProjectRoomMessageRealtime: () => snapshotProjectRoomMessageRealtime,
   taskBeliefStatesEqual: () => taskBeliefStatesEqual,
+  threadAffinityHeaders: () => threadAffinityHeaders,
   toModelKey: () => toModelKey,
   toProjectRoomPublicBotMembership: () => toProjectRoomPublicBotMembership,
   toProjectRoomPublicMembership: () => toProjectRoomPublicMembership,
@@ -1027,6 +1034,31 @@ function resolveRunScope(runConfig) {
   };
 }
 
+// src/AgentAffinity.ts
+var AGENT_AFFINITY_HEADER = "x-axiom-affinity";
+var REPLICA_HEADER = "x-axiom-replica";
+var AFFINITY_ECHO_HEADER = "x-axiom-affinity-echo";
+function affinityEchoHeaders(value) {
+  const single = Array.isArray(value) ? value[0] : value;
+  return typeof single === "string" && single.length > 0 ? { [AFFINITY_ECHO_HEADER]: single } : {};
+}
+var AFFINITY_MISS_CODE = "AFFINITY_MISS";
+function threadAffinityHeaders(threadId) {
+  return threadId ? { [AGENT_AFFINITY_HEADER]: threadId } : {};
+}
+function buildAffinityMissChunk(owner, threadId) {
+  return {
+    type: "error",
+    data: {
+      id: "",
+      content: "Run is owned by another replica",
+      code: AFFINITY_MISS_CODE,
+      owner,
+      thread_id: threadId
+    }
+  };
+}
+
 // src/OpenProtocol.ts
 function effectiveGrants(record) {
   if (record.grants && record.grants.length > 0) return record.grants;
@@ -1063,6 +1095,9 @@ function parseMcpUiRef(value) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AFFINITY_ECHO_HEADER,
+  AFFINITY_MISS_CODE,
+  AGENT_AFFINITY_HEADER,
   AgentType,
   ChannelBindingMigrationConflictError,
   DuplicateChannelBindingSubjectError,
@@ -1083,12 +1118,15 @@ function parseMcpUiRef(value) {
   ProjectRoomCursorError,
   ProjectTaskStoreUnsupportedError,
   QueueType,
+  REPLICA_HEADER,
   ScheduleExecutionType,
   ScheduleType,
   ScheduledTaskStatus,
   UIComponentType,
   WORKSPACE_PUBLIC_ROOM_PROJECT_PREFIX,
+  affinityEchoHeaders,
   assertGenericProjectConfig,
+  buildAffinityMissChunk,
   createAgentWebAppStreamProjectionContext,
   descriptorDataValue,
   effectiveGrants,
@@ -1124,6 +1162,7 @@ function parseMcpUiRef(value) {
   snapshotProjectRoomMembershipRealtime,
   snapshotProjectRoomMessageRealtime,
   taskBeliefStatesEqual,
+  threadAffinityHeaders,
   toModelKey,
   toProjectRoomPublicBotMembership,
   toProjectRoomPublicMembership,