@memberjunction/server 6.2.0-edge.1 → 6.2.0-edge.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (93) hide show
  1. package/dist/.tsbuildinfo +1 -1
  2. package/dist/generated/graphql-schemas/__mj.d.ts +1010 -0
  3. package/dist/generated/graphql-schemas/__mj.d.ts.map +1 -1
  4. package/dist/generated/graphql-schemas/__mj.js +5178 -3
  5. package/dist/generated/graphql-schemas/__mj.js.map +1 -1
  6. package/dist/generic/FireAndForgetHeartbeat.d.ts +15 -1
  7. package/dist/generic/FireAndForgetHeartbeat.d.ts.map +1 -1
  8. package/dist/generic/FireAndForgetHeartbeat.js +16 -2
  9. package/dist/generic/FireAndForgetHeartbeat.js.map +1 -1
  10. package/dist/generic/OrphanedConversationDetailReconciler.d.ts +31 -0
  11. package/dist/generic/OrphanedConversationDetailReconciler.d.ts.map +1 -0
  12. package/dist/generic/OrphanedConversationDetailReconciler.js +215 -0
  13. package/dist/generic/OrphanedConversationDetailReconciler.js.map +1 -0
  14. package/dist/generic/PushStatusResolver.d.ts +38 -0
  15. package/dist/generic/PushStatusResolver.d.ts.map +1 -1
  16. package/dist/generic/PushStatusResolver.js +84 -1
  17. package/dist/generic/PushStatusResolver.js.map +1 -1
  18. package/dist/generic/ResolverBase.d.ts +13 -1
  19. package/dist/generic/ResolverBase.d.ts.map +1 -1
  20. package/dist/generic/ResolverBase.js +49 -47
  21. package/dist/generic/ResolverBase.js.map +1 -1
  22. package/dist/index.d.ts +3 -0
  23. package/dist/index.d.ts.map +1 -1
  24. package/dist/index.js +90 -2
  25. package/dist/index.js.map +1 -1
  26. package/dist/resolvers/AgentRunStatusPublisher.d.ts +55 -0
  27. package/dist/resolvers/AgentRunStatusPublisher.d.ts.map +1 -0
  28. package/dist/resolvers/AgentRunStatusPublisher.js +141 -0
  29. package/dist/resolvers/AgentRunStatusPublisher.js.map +1 -0
  30. package/dist/resolvers/ConversationRunEventsResolver.d.ts +90 -0
  31. package/dist/resolvers/ConversationRunEventsResolver.d.ts.map +1 -0
  32. package/dist/resolvers/ConversationRunEventsResolver.js +283 -0
  33. package/dist/resolvers/ConversationRunEventsResolver.js.map +1 -0
  34. package/dist/resolvers/DuplicateEntryCheckResolver.d.ts +98 -0
  35. package/dist/resolvers/DuplicateEntryCheckResolver.d.ts.map +1 -0
  36. package/dist/resolvers/DuplicateEntryCheckResolver.js +227 -0
  37. package/dist/resolvers/DuplicateEntryCheckResolver.js.map +1 -0
  38. package/dist/resolvers/FetchEntityVectorsResolver.d.ts.map +1 -1
  39. package/dist/resolvers/FetchEntityVectorsResolver.js +2 -1
  40. package/dist/resolvers/FetchEntityVectorsResolver.js.map +1 -1
  41. package/dist/resolvers/MergeRecordsResolver.d.ts +7 -0
  42. package/dist/resolvers/MergeRecordsResolver.d.ts.map +1 -1
  43. package/dist/resolvers/MergeRecordsResolver.js +16 -2
  44. package/dist/resolvers/MergeRecordsResolver.js.map +1 -1
  45. package/dist/resolvers/RealtimeClientSessionResolver.d.ts +44 -6
  46. package/dist/resolvers/RealtimeClientSessionResolver.d.ts.map +1 -1
  47. package/dist/resolvers/RealtimeClientSessionResolver.js +90 -34
  48. package/dist/resolvers/RealtimeClientSessionResolver.js.map +1 -1
  49. package/dist/resolvers/RunAIAgentResolver.d.ts +0 -16
  50. package/dist/resolvers/RunAIAgentResolver.d.ts.map +1 -1
  51. package/dist/resolvers/RunAIAgentResolver.js +23 -153
  52. package/dist/resolvers/RunAIAgentResolver.js.map +1 -1
  53. package/dist/resolvers/RunDecisionResolver.d.ts +149 -0
  54. package/dist/resolvers/RunDecisionResolver.d.ts.map +1 -0
  55. package/dist/resolvers/RunDecisionResolver.js +396 -0
  56. package/dist/resolvers/RunDecisionResolver.js.map +1 -0
  57. package/dist/resolvers/VectorizeEntityResolver.d.ts +1 -1
  58. package/dist/resolvers/VectorizeEntityResolver.d.ts.map +1 -1
  59. package/dist/resolvers/VectorizeEntityResolver.js +14 -10
  60. package/dist/resolvers/VectorizeEntityResolver.js.map +1 -1
  61. package/package.json +94 -92
  62. package/src/__tests__/AgentRunStatusPublisher.test.ts +175 -0
  63. package/src/__tests__/DuplicateEntryCheckResolver.test.ts +300 -0
  64. package/src/__tests__/RealtimeClientSessionResolver.test.ts +216 -2
  65. package/src/__tests__/RemoteBrowserActionRecovery.test.ts +7 -1
  66. package/src/__tests__/RemoteBrowserAudioStream.test.ts +7 -1
  67. package/src/__tests__/RemoteBrowserSnapshot.test.ts +7 -1
  68. package/src/__tests__/ResolverBase.clientOldValues.test.ts +52 -0
  69. package/src/__tests__/RunAIAgentResolver.historyFrom.test.ts +182 -12
  70. package/src/__tests__/RunAIAgentResolver.streaming.test.ts +18 -27
  71. package/src/__tests__/RunDecisionResolver.test.ts +699 -0
  72. package/src/__tests__/TaskGraphActionRunner.durablePayload.test.ts +2 -0
  73. package/src/__tests__/VectorizeEntityResolver.test.ts +106 -0
  74. package/src/__tests__/conversationRunEvents.test.ts +399 -0
  75. package/src/__tests__/orphanedConversationDetailReconciler.test.ts +283 -0
  76. package/src/__tests__/pushStatusFanOut.test.ts +206 -0
  77. package/src/auth/__tests__/backward-compatibility.test.ts +1 -0
  78. package/src/generated/graphql-schemas/__mj.ts +3637 -4
  79. package/src/generic/FireAndForgetHeartbeat.ts +16 -2
  80. package/src/generic/OrphanedConversationDetailReconciler.ts +267 -0
  81. package/src/generic/PushStatusResolver.ts +97 -1
  82. package/src/generic/ResolverBase.ts +49 -49
  83. package/src/index.ts +102 -2
  84. package/src/resolvers/AgentRunStatusPublisher.ts +175 -0
  85. package/src/resolvers/ConversationRunEventsResolver.ts +308 -0
  86. package/src/resolvers/DuplicateEntryCheckResolver.ts +232 -0
  87. package/src/resolvers/FetchEntityVectorsResolver.ts +2 -1
  88. package/src/resolvers/MergeRecordsResolver.ts +14 -2
  89. package/src/resolvers/RealtimeClientSessionResolver.ts +93 -35
  90. package/src/resolvers/RunAIAgentResolver.ts +28 -174
  91. package/src/resolvers/RunDecisionResolver.ts +435 -0
  92. package/src/resolvers/VectorizeEntityResolver.ts +16 -11
  93. package/src/resolvers/__tests__/MergeRecordsResolver.test.ts +93 -0
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@memberjunction/server",
3
- "version": "6.2.0-edge.1",
3
+ "version": "6.2.0-edge.2",
4
4
  "description": "MemberJunction: This project provides API access via GraphQL to the common data store.",
5
5
  "main": "./dist/index.js",
6
6
  "types": "./src/index.ts",
@@ -19,97 +19,98 @@
19
19
  "@as-integrations/express5": "^1.0.0",
20
20
  "@graphql-tools/schema": "^10.0.38",
21
21
  "@graphql-tools/utils": "^11.0.0",
22
- "@memberjunction/actions": "6.2.0-edge.1",
23
- "@memberjunction/actions-apollo": "6.2.0-edge.1",
24
- "@memberjunction/actions-base": "6.2.0-edge.1",
25
- "@memberjunction/actions-bizapps-accounting": "6.2.0-edge.1",
26
- "@memberjunction/actions-bizapps-crm": "6.2.0-edge.1",
27
- "@memberjunction/actions-bizapps-formbuilders": "6.2.0-edge.1",
28
- "@memberjunction/actions-bizapps-lms": "6.2.0-edge.1",
29
- "@memberjunction/actions-bizapps-social": "6.2.0-edge.1",
30
- "@memberjunction/ai": "6.2.0-edge.1",
31
- "@memberjunction/ai-agent-manager": "6.2.0-edge.1",
32
- "@memberjunction/ai-agent-manager-actions": "6.2.0-edge.1",
33
- "@memberjunction/ai-agents": "6.2.0-edge.1",
34
- "@memberjunction/ai-bridge-base": "6.2.0-edge.1",
35
- "@memberjunction/ai-bridge-ringcentral": "6.2.0-edge.1",
36
- "@memberjunction/ai-bridge-server": "6.2.0-edge.1",
37
- "@memberjunction/ai-bridge-teams": "6.2.0-edge.1",
38
- "@memberjunction/ai-bridge-twilio": "6.2.0-edge.1",
39
- "@memberjunction/ai-bridge-vonage": "6.2.0-edge.1",
40
- "@memberjunction/ai-core-plus": "6.2.0-edge.1",
41
- "@memberjunction/ai-engine-base": "6.2.0-edge.1",
42
- "@memberjunction/ai-mcp-client": "6.2.0-edge.1",
43
- "@memberjunction/ai-prompts": "6.2.0-edge.1",
44
- "@memberjunction/ai-provider-bundle": "6.2.0-edge.1",
45
- "@memberjunction/ai-vector-sync": "6.2.0-edge.1",
46
- "@memberjunction/ai-vectordb": "6.2.0-edge.1",
47
- "@memberjunction/ai-vectors-pinecone": "6.2.0-edge.1",
48
- "@memberjunction/aiengine": "6.2.0-edge.1",
49
- "@memberjunction/api-keys": "6.2.0-edge.1",
50
- "@memberjunction/auth-providers": "6.2.0-edge.1",
51
- "@memberjunction/clustering-engine": "6.2.0-edge.1",
52
- "@memberjunction/codegen-lib": "6.2.0-edge.1",
53
- "@memberjunction/communication-engine": "6.2.0-edge.1",
54
- "@memberjunction/communication-ms-graph": "6.2.0-edge.1",
55
- "@memberjunction/communication-sendgrid": "6.2.0-edge.1",
56
- "@memberjunction/communication-types": "6.2.0-edge.1",
57
- "@memberjunction/component-registry-client-sdk": "6.2.0-edge.1",
58
- "@memberjunction/computer-use": "6.2.0-edge.1",
59
- "@memberjunction/computer-use-engine": "6.2.0-edge.1",
60
- "@memberjunction/config": "6.2.0-edge.1",
61
- "@memberjunction/core": "6.2.0-edge.1",
62
- "@memberjunction/core-actions": "6.2.0-edge.1",
63
- "@memberjunction/core-entities": "6.2.0-edge.1",
64
- "@memberjunction/core-entities-server": "6.2.0-edge.1",
65
- "@memberjunction/credentials": "6.2.0-edge.1",
66
- "@memberjunction/data-context": "6.2.0-edge.1",
67
- "@memberjunction/data-context-server": "6.2.0-edge.1",
68
- "@memberjunction/doc-utils": "6.2.0-edge.1",
69
- "@memberjunction/encryption": "6.2.0-edge.1",
70
- "@memberjunction/entity-communications-base": "6.2.0-edge.1",
71
- "@memberjunction/entity-communications-server": "6.2.0-edge.1",
72
- "@memberjunction/esignature": "6.2.0-edge.1",
73
- "@memberjunction/external-change-detection": "6.2.0-edge.1",
74
- "@memberjunction/generic-database-provider": "6.2.0-edge.1",
75
- "@memberjunction/global": "6.2.0-edge.1",
76
- "@memberjunction/graphql-dataprovider": "6.2.0-edge.1",
77
- "@memberjunction/integration-engine": "6.2.0-edge.1",
78
- "@memberjunction/integration-engine-base": "6.2.0-edge.1",
79
- "@memberjunction/integration-progress-artifacts": "6.2.0-edge.1",
80
- "@memberjunction/integration-schema-builder": "6.2.0-edge.1",
81
- "@memberjunction/interactive-component-types": "6.2.0-edge.1",
82
- "@memberjunction/lists": "6.2.0-edge.1",
83
- "@memberjunction/lists-base": "6.2.0-edge.1",
84
- "@memberjunction/livekit-room-server": "6.2.0-edge.1",
85
- "@memberjunction/network-utils": "6.2.0-edge.1",
86
- "@memberjunction/notifications": "6.2.0-edge.1",
87
- "@memberjunction/postgresql-dataprovider": "6.2.0-edge.1",
88
- "@memberjunction/queue": "6.2.0-edge.1",
89
- "@memberjunction/record-comparison": "6.2.0-edge.1",
90
- "@memberjunction/redis-provider": "6.2.0-edge.1",
91
- "@memberjunction/remote-browser-base": "6.2.0-edge.1",
92
- "@memberjunction/remote-browser-cdp": "6.2.0-edge.1",
93
- "@memberjunction/remote-browser-selfhost": "6.2.0-edge.1",
94
- "@memberjunction/remote-browser-server": "6.2.0-edge.1",
95
- "@memberjunction/scheduling-actions": "6.2.0-edge.1",
96
- "@memberjunction/scheduling-base-types": "6.2.0-edge.1",
97
- "@memberjunction/scheduling-engine": "6.2.0-edge.1",
98
- "@memberjunction/scheduling-engine-base": "6.2.0-edge.1",
99
- "@memberjunction/schema-engine": "6.2.0-edge.1",
100
- "@memberjunction/search-engine": "6.2.0-edge.1",
101
- "@memberjunction/server-extensions-core": "6.2.0-edge.1",
102
- "@memberjunction/sql-dialect": "6.2.0-edge.1",
103
- "@memberjunction/sql-parser": "6.2.0-edge.1",
104
- "@memberjunction/sqlserver-dataprovider": "6.2.0-edge.1",
105
- "@memberjunction/storage": "6.2.0-edge.1",
106
- "@memberjunction/tag-engine": "6.2.0-edge.1",
107
- "@memberjunction/tag-engine-base": "6.2.0-edge.1",
108
- "@memberjunction/task-graph": "6.2.0-edge.1",
109
- "@memberjunction/templates": "6.2.0-edge.1",
110
- "@memberjunction/testing-engine": "6.2.0-edge.1",
111
- "@memberjunction/testing-engine-base": "6.2.0-edge.1",
112
- "@memberjunction/version-history": "6.2.0-edge.1",
22
+ "@memberjunction/actions": "6.2.0-edge.2",
23
+ "@memberjunction/actions-apollo": "6.2.0-edge.2",
24
+ "@memberjunction/actions-base": "6.2.0-edge.2",
25
+ "@memberjunction/actions-bizapps-accounting": "6.2.0-edge.2",
26
+ "@memberjunction/actions-bizapps-crm": "6.2.0-edge.2",
27
+ "@memberjunction/actions-bizapps-formbuilders": "6.2.0-edge.2",
28
+ "@memberjunction/actions-bizapps-lms": "6.2.0-edge.2",
29
+ "@memberjunction/actions-bizapps-social": "6.2.0-edge.2",
30
+ "@memberjunction/ai": "6.2.0-edge.2",
31
+ "@memberjunction/ai-agent-manager": "6.2.0-edge.2",
32
+ "@memberjunction/ai-agent-manager-actions": "6.2.0-edge.2",
33
+ "@memberjunction/ai-agents": "6.2.0-edge.2",
34
+ "@memberjunction/ai-bridge-base": "6.2.0-edge.2",
35
+ "@memberjunction/ai-bridge-ringcentral": "6.2.0-edge.2",
36
+ "@memberjunction/ai-bridge-server": "6.2.0-edge.2",
37
+ "@memberjunction/ai-bridge-teams": "6.2.0-edge.2",
38
+ "@memberjunction/ai-bridge-twilio": "6.2.0-edge.2",
39
+ "@memberjunction/ai-bridge-vonage": "6.2.0-edge.2",
40
+ "@memberjunction/ai-core-plus": "6.2.0-edge.2",
41
+ "@memberjunction/ai-engine-base": "6.2.0-edge.2",
42
+ "@memberjunction/ai-mcp-client": "6.2.0-edge.2",
43
+ "@memberjunction/ai-prompts": "6.2.0-edge.2",
44
+ "@memberjunction/ai-provider-bundle": "6.2.0-edge.2",
45
+ "@memberjunction/ai-vector-dupe": "6.2.0-edge.2",
46
+ "@memberjunction/ai-vector-sync": "6.2.0-edge.2",
47
+ "@memberjunction/ai-vectordb": "6.2.0-edge.2",
48
+ "@memberjunction/ai-vectors-pinecone": "6.2.0-edge.2",
49
+ "@memberjunction/aiengine": "6.2.0-edge.2",
50
+ "@memberjunction/api-keys": "6.2.0-edge.2",
51
+ "@memberjunction/auth-providers": "6.2.0-edge.2",
52
+ "@memberjunction/clustering-engine": "6.2.0-edge.2",
53
+ "@memberjunction/codegen-lib": "6.2.0-edge.2",
54
+ "@memberjunction/communication-engine": "6.2.0-edge.2",
55
+ "@memberjunction/communication-ms-graph": "6.2.0-edge.2",
56
+ "@memberjunction/communication-sendgrid": "6.2.0-edge.2",
57
+ "@memberjunction/communication-types": "6.2.0-edge.2",
58
+ "@memberjunction/component-registry-client-sdk": "6.2.0-edge.2",
59
+ "@memberjunction/computer-use": "6.2.0-edge.2",
60
+ "@memberjunction/computer-use-engine": "6.2.0-edge.2",
61
+ "@memberjunction/config": "6.2.0-edge.2",
62
+ "@memberjunction/core": "6.2.0-edge.2",
63
+ "@memberjunction/core-actions": "6.2.0-edge.2",
64
+ "@memberjunction/core-entities": "6.2.0-edge.2",
65
+ "@memberjunction/core-entities-server": "6.2.0-edge.2",
66
+ "@memberjunction/credentials": "6.2.0-edge.2",
67
+ "@memberjunction/data-context": "6.2.0-edge.2",
68
+ "@memberjunction/data-context-server": "6.2.0-edge.2",
69
+ "@memberjunction/doc-utils": "6.2.0-edge.2",
70
+ "@memberjunction/encryption": "6.2.0-edge.2",
71
+ "@memberjunction/entity-communications-base": "6.2.0-edge.2",
72
+ "@memberjunction/entity-communications-server": "6.2.0-edge.2",
73
+ "@memberjunction/esignature": "6.2.0-edge.2",
74
+ "@memberjunction/external-change-detection": "6.2.0-edge.2",
75
+ "@memberjunction/generic-database-provider": "6.2.0-edge.2",
76
+ "@memberjunction/global": "6.2.0-edge.2",
77
+ "@memberjunction/graphql-dataprovider": "6.2.0-edge.2",
78
+ "@memberjunction/integration-engine": "6.2.0-edge.2",
79
+ "@memberjunction/integration-engine-base": "6.2.0-edge.2",
80
+ "@memberjunction/integration-progress-artifacts": "6.2.0-edge.2",
81
+ "@memberjunction/integration-schema-builder": "6.2.0-edge.2",
82
+ "@memberjunction/interactive-component-types": "6.2.0-edge.2",
83
+ "@memberjunction/lists": "6.2.0-edge.2",
84
+ "@memberjunction/lists-base": "6.2.0-edge.2",
85
+ "@memberjunction/livekit-room-server": "6.2.0-edge.2",
86
+ "@memberjunction/network-utils": "6.2.0-edge.2",
87
+ "@memberjunction/notifications": "6.2.0-edge.2",
88
+ "@memberjunction/postgresql-dataprovider": "6.2.0-edge.2",
89
+ "@memberjunction/queue": "6.2.0-edge.2",
90
+ "@memberjunction/record-comparison": "6.2.0-edge.2",
91
+ "@memberjunction/redis-provider": "6.2.0-edge.2",
92
+ "@memberjunction/remote-browser-base": "6.2.0-edge.2",
93
+ "@memberjunction/remote-browser-cdp": "6.2.0-edge.2",
94
+ "@memberjunction/remote-browser-selfhost": "6.2.0-edge.2",
95
+ "@memberjunction/remote-browser-server": "6.2.0-edge.2",
96
+ "@memberjunction/scheduling-actions": "6.2.0-edge.2",
97
+ "@memberjunction/scheduling-base-types": "6.2.0-edge.2",
98
+ "@memberjunction/scheduling-engine": "6.2.0-edge.2",
99
+ "@memberjunction/scheduling-engine-base": "6.2.0-edge.2",
100
+ "@memberjunction/schema-engine": "6.2.0-edge.2",
101
+ "@memberjunction/search-engine": "6.2.0-edge.2",
102
+ "@memberjunction/server-extensions-core": "6.2.0-edge.2",
103
+ "@memberjunction/sql-dialect": "6.2.0-edge.2",
104
+ "@memberjunction/sql-parser": "6.2.0-edge.2",
105
+ "@memberjunction/sqlserver-dataprovider": "6.2.0-edge.2",
106
+ "@memberjunction/storage": "6.2.0-edge.2",
107
+ "@memberjunction/tag-engine": "6.2.0-edge.2",
108
+ "@memberjunction/tag-engine-base": "6.2.0-edge.2",
109
+ "@memberjunction/task-graph": "6.2.0-edge.2",
110
+ "@memberjunction/templates": "6.2.0-edge.2",
111
+ "@memberjunction/testing-engine": "6.2.0-edge.2",
112
+ "@memberjunction/testing-engine-base": "6.2.0-edge.2",
113
+ "@memberjunction/version-history": "6.2.0-edge.2",
113
114
  "@octokit/auth-app": "^7.1.5",
114
115
  "@octokit/rest": "^21.1.1",
115
116
  "@types/compression": "^1.8.1",
@@ -148,6 +149,7 @@
148
149
  "zod": "^3.25.0"
149
150
  },
150
151
  "devDependencies": {
152
+ "@memberjunction/conversations-runtime": "6.2.0-edge.2",
151
153
  "@types/express": "^5.0.6",
152
154
  "pkgroll": "2.23.0",
153
155
  "rimraf": "6.1.2",
@@ -0,0 +1,175 @@
1
+ /**
2
+ * Pins the status messages RunAIAgentResolver publishes for one run, and shows a host
3
+ * driving the same publisher into ConversationStreaming.
4
+ *
5
+ * The wire assertions are the contract the resolver already published before this class
6
+ * existed: a significant step and not a noise step, streamed text, the partial result,
7
+ * then the completion.
8
+ */
9
+ import 'reflect-metadata';
10
+ import { describe, it, expect, vi } from 'vitest';
11
+ import type { PubSubEngine } from 'type-graphql';
12
+
13
+ vi.mock('type-graphql', () => {
14
+ const decoratorFactory = (..._args: unknown[]) => (..._decorated: unknown[]) => undefined;
15
+ const exportNames = [
16
+ 'Resolver', 'Query', 'Mutation', 'Subscription', 'Arg', 'Args', 'ArgsType', 'Ctx', 'Root',
17
+ 'Info', 'Field', 'FieldResolver', 'ObjectType', 'InputType', 'InterfaceType', 'Authorized',
18
+ 'UseMiddleware', 'Extensions', 'Directive', 'ID', 'Int', 'Float', 'GraphQLISODateTime',
19
+ 'GraphQLTimestamp', 'registerEnumType', 'createMethodDecorator', 'createParamDecorator',
20
+ 'buildSchema', 'buildSchemaSync', 'PubSub',
21
+ ];
22
+ return Object.fromEntries(exportNames.map((name) => [name, decoratorFactory]));
23
+ });
24
+
25
+ vi.mock('@memberjunction/graphql-dataprovider', () => ({
26
+ GraphQLDataProvider: { Instance: { PushStatusUpdates: vi.fn() } },
27
+ }));
28
+
29
+ import type { ExecuteAgentResult } from '@memberjunction/ai-core-plus';
30
+ import { ConversationStreaming, type IConversationsRuntimeContext } from '@memberjunction/conversations-runtime';
31
+ import { AgentRunStatusPublisher } from '../resolvers/AgentRunStatusPublisher.js';
32
+ import type { UserPayload } from '../types.js';
33
+
34
+ const run = {
35
+ ID: 'run-1',
36
+ ConversationDetailID: 'detail-1',
37
+ ErrorMessage: undefined as string | undefined,
38
+ Steps: [{ StepName: 'Answer' }],
39
+ GetAll: () => ({ ID: 'run-1', ConversationDetailID: 'detail-1', Agent: 'Betty' }),
40
+ };
41
+
42
+ function publisher() {
43
+ const publish = vi.fn().mockResolvedValue(undefined);
44
+ const pubSub = { publish } as unknown as PubSubEngine;
45
+ const userPayload = { sessionId: 'session-1', userRecord: { ID: 'user-1' } } as UserPayload;
46
+ return { publish, made: new AgentRunStatusPublisher(pubSub, userPayload, 'session-1') };
47
+ }
48
+
49
+ function envelopes(publish: ReturnType<typeof vi.fn>): Array<Record<string, unknown>> {
50
+ return publish.mock.calls.map((call) => {
51
+ const payload = call[1] as { message: string; sessionId: string; ownerUserId: string };
52
+ expect(payload.sessionId).toBe('session-1');
53
+ expect(payload.ownerUserId).toBe('user-1');
54
+ return JSON.parse(payload.message) as Record<string, unknown>;
55
+ });
56
+ }
57
+
58
+ describe('AgentRunStatusPublisher', () => {
59
+ it('publishes the resolver sequence: a significant step, streamed text, the partial result, then the completion', () => {
60
+ const { publish, made } = publisher();
61
+
62
+ made.OnProgress({ step: 'initialization', message: 'noise', metadata: { agentRun: run } });
63
+ made.OnProgress({
64
+ step: 'prompt_execution',
65
+ message: 'Writing',
66
+ percentage: 40,
67
+ metadata: { agentRun: run, agentName: 'Betty' },
68
+ });
69
+ made.OnStreaming({ content: 'Hel', isComplete: false, stepType: 'prompt', kind: 'final-response' });
70
+ made.PublishFinal({ success: true, agentRun: run, payload: { text: 'Hello' } } as ExecuteAgentResult, 'detail-1', '{"text":"Hello"}');
71
+
72
+ const sent = envelopes(publish);
73
+ expect(sent.map((row) => (row.data as { type: string }).type)).toEqual([
74
+ 'progress',
75
+ 'streaming',
76
+ 'partial_result',
77
+ 'complete',
78
+ ]);
79
+ expect(sent[0].type).toBe('ExecutionProgress');
80
+ expect(sent[0].resolver).toBe('RunAIAgentResolver');
81
+ expect((sent[1].data as { streaming: { content: string } }).streaming.content).toBe('Hel');
82
+ expect((sent[2].data as { partialResult: { currentStep: string } }).partialResult.currentStep).toBe('Answer');
83
+ const completion = sent[3].data as { type: string; conversationDetailId: string; result: string; agentRunId: string };
84
+ expect(completion.conversationDetailId).toBe('detail-1');
85
+ expect(completion.agentRunId).toBe('run-1');
86
+ expect(completion.result).toBe('{"text":"Hello"}');
87
+ });
88
+
89
+ it('stamps the reply row on the completion when that id differs from the run', () => {
90
+ const { publish, made } = publisher();
91
+ made.PublishFinal(
92
+ { success: true, agentRun: run, payload: { text: 'Hello' } } as ExecuteAgentResult,
93
+ 'reply-9',
94
+ '{"text":"Hello"}',
95
+ );
96
+
97
+ const sent = envelopes(publish);
98
+ const completion = sent[sent.length - 1].data as { type: string; conversationDetailId: string; agentRunId: string };
99
+ expect(completion.type).toBe('complete');
100
+ expect(completion.conversationDetailId).toBe('reply-9');
101
+ expect(completion.agentRunId).toBe('run-1');
102
+ expect(run.ConversationDetailID).toBe('detail-1');
103
+ });
104
+
105
+ it('publishes a failure completion when there is no run', () => {
106
+ const { publish, made } = publisher();
107
+ made.PublishFailure('detail-1', 'background blew up');
108
+
109
+ const sent = envelopes(publish);
110
+ expect(sent).toHaveLength(1);
111
+ expect(sent[0].type).toBe('StreamingContent');
112
+ const completion = sent[0].data as {
113
+ type: string;
114
+ agentRunId: string;
115
+ conversationDetailId: string;
116
+ success: boolean;
117
+ errorMessage: string;
118
+ result: string;
119
+ };
120
+ expect(completion.type).toBe('complete');
121
+ expect(completion.agentRunId).toBe('unknown');
122
+ expect(completion.conversationDetailId).toBe('detail-1');
123
+ expect(completion.success).toBe(false);
124
+ expect(completion.errorMessage).toBe('background blew up');
125
+ expect(completion.result).toBe(JSON.stringify({ success: false, errorMessage: 'background blew up' }));
126
+ });
127
+
128
+ it('logs when a significant step or a streamed chunk arrives before any run', () => {
129
+ const error = vi.spyOn(console, 'error').mockImplementation(() => undefined);
130
+ const { publish, made } = publisher();
131
+
132
+ made.OnProgress({ step: 'initialization', message: 'noise' });
133
+ made.OnProgress({ step: 'prompt_execution', message: 'Writing' });
134
+ made.OnStreaming({ content: 'Hel', isComplete: false });
135
+
136
+ expect(publish).not.toHaveBeenCalled();
137
+ expect(error).toHaveBeenCalledWith('❌ No agent run available for progress callback');
138
+ expect(error).toHaveBeenCalledWith('❌ No agent run available for streaming callback');
139
+ expect(error).toHaveBeenCalledTimes(2);
140
+ error.mockRestore();
141
+ });
142
+
143
+ it('routes a host publisher progress and completion to the reply row', async () => {
144
+ const { publish, made } = publisher();
145
+ made.OnProgress({
146
+ step: 'action_execution',
147
+ message: 'Looking',
148
+ percentage: 10,
149
+ metadata: { agentRun: run },
150
+ });
151
+ made.PublishFinal({ success: true, agentRun: run } as ExecuteAgentResult, 'detail-1', '{"ok":true}');
152
+
153
+ const streaming = new ConversationStreaming({
154
+ Notification: { Notify: vi.fn() },
155
+ Tasks: { RemoveByAgentRunId: vi.fn().mockReturnValue(false) },
156
+ } as IConversationsRuntimeContext);
157
+ const onReply = vi.fn();
158
+ streaming.registerMessageCallback('detail-1', onReply);
159
+ const completions: Array<{ conversationDetailId: string; agentRunId: string }> = [];
160
+ streaming.CompletionEvents$.subscribe((event) => completions.push(event));
161
+
162
+ const deliver = streaming as unknown as {
163
+ handlePushStatusUpdate(status: unknown): Promise<void>;
164
+ };
165
+ for (const message of envelopes(publish)) {
166
+ await deliver.handlePushStatusUpdate(message);
167
+ }
168
+
169
+ expect(onReply).toHaveBeenCalledOnce();
170
+ expect(onReply.mock.calls[0][0].conversationDetailId).toBe('detail-1');
171
+ expect(onReply.mock.calls[0][0].message).toBe('Looking');
172
+ expect(completions).toEqual([{ conversationDetailId: 'detail-1', agentRunId: 'run-1', success: true, errorMessage: undefined }]);
173
+ expect(streaming.GetRecentCompletion('detail-1')?.agentRunId).toBe('run-1');
174
+ });
175
+ });