@primo-ai/core 0.1.0

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 (194) hide show
  1. package/README.md +114 -0
  2. package/dist/agent.d.ts +66 -0
  3. package/dist/agent.d.ts.map +1 -0
  4. package/dist/agent.js +260 -0
  5. package/dist/agent.js.map +1 -0
  6. package/dist/checkpoint-store.d.ts +19 -0
  7. package/dist/checkpoint-store.d.ts.map +1 -0
  8. package/dist/checkpoint-store.js +79 -0
  9. package/dist/checkpoint-store.js.map +1 -0
  10. package/dist/concurrency-controller.d.ts +9 -0
  11. package/dist/concurrency-controller.d.ts.map +1 -0
  12. package/dist/concurrency-controller.js +71 -0
  13. package/dist/concurrency-controller.js.map +1 -0
  14. package/dist/config-merge.d.ts +12 -0
  15. package/dist/config-merge.d.ts.map +1 -0
  16. package/dist/config-merge.js +47 -0
  17. package/dist/config-merge.js.map +1 -0
  18. package/dist/config.d.ts +16 -0
  19. package/dist/config.d.ts.map +1 -0
  20. package/dist/config.js +144 -0
  21. package/dist/config.js.map +1 -0
  22. package/dist/context-builder.d.ts +29 -0
  23. package/dist/context-builder.d.ts.map +1 -0
  24. package/dist/context-builder.js +178 -0
  25. package/dist/context-builder.js.map +1 -0
  26. package/dist/dynamic-resolver.d.ts +7 -0
  27. package/dist/dynamic-resolver.d.ts.map +1 -0
  28. package/dist/dynamic-resolver.js +11 -0
  29. package/dist/dynamic-resolver.js.map +1 -0
  30. package/dist/errors.d.ts +30 -0
  31. package/dist/errors.d.ts.map +1 -0
  32. package/dist/errors.js +40 -0
  33. package/dist/errors.js.map +1 -0
  34. package/dist/event-bus.d.ts +8 -0
  35. package/dist/event-bus.d.ts.map +1 -0
  36. package/dist/event-bus.js +36 -0
  37. package/dist/event-bus.js.map +1 -0
  38. package/dist/event-system.d.ts +19 -0
  39. package/dist/event-system.d.ts.map +1 -0
  40. package/dist/event-system.js +58 -0
  41. package/dist/event-system.js.map +1 -0
  42. package/dist/fallback-runner.d.ts +30 -0
  43. package/dist/fallback-runner.d.ts.map +1 -0
  44. package/dist/fallback-runner.js +40 -0
  45. package/dist/fallback-runner.js.map +1 -0
  46. package/dist/gateways/builtin-gateway.d.ts +17 -0
  47. package/dist/gateways/builtin-gateway.d.ts.map +1 -0
  48. package/dist/gateways/builtin-gateway.js +58 -0
  49. package/dist/gateways/builtin-gateway.js.map +1 -0
  50. package/dist/gateways/gateway-chain.d.ts +9 -0
  51. package/dist/gateways/gateway-chain.d.ts.map +1 -0
  52. package/dist/gateways/gateway-chain.js +18 -0
  53. package/dist/gateways/gateway-chain.js.map +1 -0
  54. package/dist/gateways/openai-compatible-gateway.d.ts +10 -0
  55. package/dist/gateways/openai-compatible-gateway.d.ts.map +1 -0
  56. package/dist/gateways/openai-compatible-gateway.js +26 -0
  57. package/dist/gateways/openai-compatible-gateway.js.map +1 -0
  58. package/dist/harness.d.ts +44 -0
  59. package/dist/harness.d.ts.map +1 -0
  60. package/dist/harness.js +70 -0
  61. package/dist/harness.js.map +1 -0
  62. package/dist/hook-manager.d.ts +22 -0
  63. package/dist/hook-manager.d.ts.map +1 -0
  64. package/dist/hook-manager.js +92 -0
  65. package/dist/hook-manager.js.map +1 -0
  66. package/dist/index.d.ts +34 -0
  67. package/dist/index.d.ts.map +1 -0
  68. package/dist/index.js +34 -0
  69. package/dist/index.js.map +1 -0
  70. package/dist/llm-invoker.d.ts +34 -0
  71. package/dist/llm-invoker.d.ts.map +1 -0
  72. package/dist/llm-invoker.js +124 -0
  73. package/dist/llm-invoker.js.map +1 -0
  74. package/dist/loop-orchestrator.d.ts +48 -0
  75. package/dist/loop-orchestrator.d.ts.map +1 -0
  76. package/dist/loop-orchestrator.js +247 -0
  77. package/dist/loop-orchestrator.js.map +1 -0
  78. package/dist/model-factory.d.ts +13 -0
  79. package/dist/model-factory.d.ts.map +1 -0
  80. package/dist/model-factory.js +40 -0
  81. package/dist/model-factory.js.map +1 -0
  82. package/dist/model-profile.d.ts +17 -0
  83. package/dist/model-profile.d.ts.map +1 -0
  84. package/dist/model-profile.js +57 -0
  85. package/dist/model-profile.js.map +1 -0
  86. package/dist/model-resolver.d.ts +16 -0
  87. package/dist/model-resolver.d.ts.map +1 -0
  88. package/dist/model-resolver.js +28 -0
  89. package/dist/model-resolver.js.map +1 -0
  90. package/dist/parse-model.d.ts +6 -0
  91. package/dist/parse-model.d.ts.map +1 -0
  92. package/dist/parse-model.js +11 -0
  93. package/dist/parse-model.js.map +1 -0
  94. package/dist/pipeline.d.ts +28 -0
  95. package/dist/pipeline.d.ts.map +1 -0
  96. package/dist/pipeline.js +286 -0
  97. package/dist/pipeline.js.map +1 -0
  98. package/dist/plugin-manager.d.ts +42 -0
  99. package/dist/plugin-manager.d.ts.map +1 -0
  100. package/dist/plugin-manager.js +148 -0
  101. package/dist/plugin-manager.js.map +1 -0
  102. package/dist/processors/build-context.d.ts +3 -0
  103. package/dist/processors/build-context.d.ts.map +1 -0
  104. package/dist/processors/build-context.js +5 -0
  105. package/dist/processors/build-context.js.map +1 -0
  106. package/dist/processors/evaluate-iteration.d.ts +13 -0
  107. package/dist/processors/evaluate-iteration.d.ts.map +1 -0
  108. package/dist/processors/evaluate-iteration.js +196 -0
  109. package/dist/processors/evaluate-iteration.js.map +1 -0
  110. package/dist/processors/execute-tools.d.ts +4 -0
  111. package/dist/processors/execute-tools.d.ts.map +1 -0
  112. package/dist/processors/execute-tools.js +69 -0
  113. package/dist/processors/execute-tools.js.map +1 -0
  114. package/dist/processors/gate-tool.d.ts +3 -0
  115. package/dist/processors/gate-tool.d.ts.map +1 -0
  116. package/dist/processors/gate-tool.js +6 -0
  117. package/dist/processors/gate-tool.js.map +1 -0
  118. package/dist/processors/index.d.ts +10 -0
  119. package/dist/processors/index.d.ts.map +1 -0
  120. package/dist/processors/index.js +10 -0
  121. package/dist/processors/index.js.map +1 -0
  122. package/dist/processors/invoke-llm.d.ts +13 -0
  123. package/dist/processors/invoke-llm.d.ts.map +1 -0
  124. package/dist/processors/invoke-llm.js +138 -0
  125. package/dist/processors/invoke-llm.js.map +1 -0
  126. package/dist/processors/prepare-step.d.ts +3 -0
  127. package/dist/processors/prepare-step.d.ts.map +1 -0
  128. package/dist/processors/prepare-step.js +5 -0
  129. package/dist/processors/prepare-step.js.map +1 -0
  130. package/dist/processors/process-input.d.ts +3 -0
  131. package/dist/processors/process-input.d.ts.map +1 -0
  132. package/dist/processors/process-input.js +20 -0
  133. package/dist/processors/process-input.js.map +1 -0
  134. package/dist/processors/process-output.d.ts +3 -0
  135. package/dist/processors/process-output.d.ts.map +1 -0
  136. package/dist/processors/process-output.js +5 -0
  137. package/dist/processors/process-output.js.map +1 -0
  138. package/dist/processors/process-step-output.d.ts +3 -0
  139. package/dist/processors/process-step-output.d.ts.map +1 -0
  140. package/dist/processors/process-step-output.js +23 -0
  141. package/dist/processors/process-step-output.js.map +1 -0
  142. package/dist/processors/provider-history-compat.d.ts +7 -0
  143. package/dist/processors/provider-history-compat.d.ts.map +1 -0
  144. package/dist/processors/provider-history-compat.js +166 -0
  145. package/dist/processors/provider-history-compat.js.map +1 -0
  146. package/dist/provider-capabilities.d.ts +4 -0
  147. package/dist/provider-capabilities.d.ts.map +1 -0
  148. package/dist/provider-capabilities.js +47 -0
  149. package/dist/provider-capabilities.js.map +1 -0
  150. package/dist/retry.d.ts +6 -0
  151. package/dist/retry.d.ts.map +1 -0
  152. package/dist/retry.js +26 -0
  153. package/dist/retry.js.map +1 -0
  154. package/dist/serialize.d.ts +12 -0
  155. package/dist/serialize.d.ts.map +1 -0
  156. package/dist/serialize.js +22 -0
  157. package/dist/serialize.js.map +1 -0
  158. package/dist/session-manager.d.ts +17 -0
  159. package/dist/session-manager.d.ts.map +1 -0
  160. package/dist/session-manager.js +143 -0
  161. package/dist/session-manager.js.map +1 -0
  162. package/dist/session-persistence.d.ts +13 -0
  163. package/dist/session-persistence.d.ts.map +1 -0
  164. package/dist/session-persistence.js +63 -0
  165. package/dist/session-persistence.js.map +1 -0
  166. package/dist/session-storage.d.ts +19 -0
  167. package/dist/session-storage.d.ts.map +1 -0
  168. package/dist/session-storage.js +103 -0
  169. package/dist/session-storage.js.map +1 -0
  170. package/dist/state-machine.d.ts +13 -0
  171. package/dist/state-machine.d.ts.map +1 -0
  172. package/dist/state-machine.js +49 -0
  173. package/dist/state-machine.js.map +1 -0
  174. package/dist/storage-replay-backend.d.ts +7 -0
  175. package/dist/storage-replay-backend.d.ts.map +1 -0
  176. package/dist/storage-replay-backend.js +14 -0
  177. package/dist/storage-replay-backend.js.map +1 -0
  178. package/dist/sub-agent.d.ts +13 -0
  179. package/dist/sub-agent.d.ts.map +1 -0
  180. package/dist/sub-agent.js +87 -0
  181. package/dist/sub-agent.js.map +1 -0
  182. package/dist/task-manager.d.ts +22 -0
  183. package/dist/task-manager.d.ts.map +1 -0
  184. package/dist/task-manager.js +151 -0
  185. package/dist/task-manager.js.map +1 -0
  186. package/dist/token-counter.d.ts +9 -0
  187. package/dist/token-counter.d.ts.map +1 -0
  188. package/dist/token-counter.js +74 -0
  189. package/dist/token-counter.js.map +1 -0
  190. package/dist/tool-registry.d.ts +31 -0
  191. package/dist/tool-registry.d.ts.map +1 -0
  192. package/dist/tool-registry.js +147 -0
  193. package/dist/tool-registry.js.map +1 -0
  194. package/package.json +37 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,KAAK,qBAAqB,EAAE,MAAM,eAAe,CAAC;AAC3E,OAAO,EAAE,KAAK,EAAE,KAAK,iBAAiB,EAAE,KAAK,cAAc,EAAE,MAAM,YAAY,CAAC;AAChF,OAAO,EAAE,UAAU,EAAE,KAAK,iBAAiB,EAAE,KAAK,cAAc,EAAE,KAAK,eAAe,EAAE,KAAK,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACvI,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,UAAU,EAAE,KAAK,WAAW,EAAE,MAAM,qBAAqB,CAAC;AACnG,OAAO,EAAE,eAAe,EAAE,KAAK,YAAY,EAAE,MAAM,YAAY,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,KAAK,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACxE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,WAAW,EAAE,KAAK,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACzE,OAAO,EAAE,YAAY,EAAE,KAAK,eAAe,EAAE,KAAK,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAClG,OAAO,EAAE,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,KAAK,YAAY,EAAE,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,cAAc,EAAE,KAAK,eAAe,EAAE,KAAK,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AACxG,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,KAAK,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAE,KAAK,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AACtF,OAAO,EAAE,cAAc,EAAE,KAAK,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAClF,OAAO,EAAE,cAAc,EAAE,KAAK,WAAW,EAAE,MAAM,cAAc,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,YAAY,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,UAAU,EACV,SAAS,EACT,kBAAkB,EAClB,kBAAkB,EAClB,KAAK,iBAAiB,GACvB,MAAM,aAAa,CAAC"}
package/dist/index.js ADDED
@@ -0,0 +1,34 @@
1
+ // @primo-ai/core — Agent Loop, Processor Pipeline, Context, Tool Registry
2
+ export { PipelineRunner } from './pipeline.js';
3
+ export { Agent } from './agent.js';
4
+ export { LLMInvoker } from './llm-invoker.js';
5
+ export { resolveModel, registerProvider, parseModel } from './model-resolver.js';
6
+ export { streamWithRetry } from './retry.js';
7
+ export { PluginManager } from './plugin-manager.js';
8
+ export { EventBus } from './event-bus.js';
9
+ export { EventSystem, REPLAY_SENTINEL } from './event-system.js';
10
+ export { StorageReplayBackend } from './storage-replay-backend.js';
11
+ export { HookManager } from './hook-manager.js';
12
+ export { ToolRegistry } from './tool-registry.js';
13
+ export { FilesystemSessionStorage } from './session-storage.js';
14
+ export { SessionPersistence } from './session-persistence.js';
15
+ export { SessionManagerImpl } from './session-manager.js';
16
+ export { createSubAgentTool } from './sub-agent.js';
17
+ export { deepMerge } from './config-merge.js';
18
+ export { resolveDynamic } from './dynamic-resolver.js';
19
+ export { matchProfile, applyProfile } from './model-profile.js';
20
+ export { ConfigLoader } from './config.js';
21
+ export { ConcurrencyController } from './concurrency-controller.js';
22
+ export { FallbackRunner } from './fallback-runner.js';
23
+ export { TaskManagerImpl } from './task-manager.js';
24
+ export { StateMachine } from './state-machine.js';
25
+ export { ModelFactory } from './model-factory.js';
26
+ export { LoopOrchestrator } from './loop-orchestrator.js';
27
+ export { serialize, deserialize } from './serialize.js';
28
+ export { InMemoryCheckpointStore, JsonlCheckpointStore } from './checkpoint-store.js';
29
+ export { ContextBuilder } from './context-builder.js';
30
+ export { HarnessAPIImpl } from './harness.js';
31
+ export { TiktokenCounter } from './token-counter.js';
32
+ export { OpenAICompatibleGateway } from './gateways/openai-compatible-gateway.js';
33
+ export { AgentForgeError, RecoverableError, FatalError, AuthError, ModelNotFoundError, ToolExecutionError, } from './errors.js';
34
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,0EAA0E;AAE1E,OAAO,EAAE,cAAc,EAA8B,MAAM,eAAe,CAAC;AAC3E,OAAO,EAAE,KAAK,EAA+C,MAAM,YAAY,CAAC;AAChF,OAAO,EAAE,UAAU,EAA2F,MAAM,kBAAkB,CAAC;AACvI,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,UAAU,EAAoB,MAAM,qBAAqB,CAAC;AACnG,OAAO,EAAE,eAAe,EAAqB,MAAM,YAAY,CAAC;AAChE,OAAO,EAAE,aAAa,EAAsB,MAAM,qBAAqB,CAAC;AACxE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,EAAE,WAAW,EAA2B,MAAM,mBAAmB,CAAC;AACzE,OAAO,EAAE,YAAY,EAAkD,MAAM,oBAAoB,CAAC;AAClG,OAAO,EAAE,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,YAAY,EAAqB,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,cAAc,EAAoD,MAAM,sBAAsB,CAAC;AACxG,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAmB,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,gBAAgB,EAAoB,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AACtF,OAAO,EAAE,cAAc,EAA8B,MAAM,sBAAsB,CAAC;AAClF,OAAO,EAAE,cAAc,EAAoB,MAAM,cAAc,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAClF,OAAO,EACL,eAAe,EACf,gBAAgB,EAChB,UAAU,EACV,SAAS,EACT,kBAAkB,EAClB,kBAAkB,GAEnB,MAAM,aAAa,CAAC"}
@@ -0,0 +1,34 @@
1
+ import type { LanguageModel } from 'ai';
2
+ import type { TokenUsage, Tracer } from '@primo-ai/sdk';
3
+ import { type RetryOptions } from './retry.js';
4
+ export interface LLMInvokerOptions {
5
+ model: LanguageModel;
6
+ system?: string;
7
+ retryOptions?: RetryOptions;
8
+ tracer?: Tracer;
9
+ eventBus?: {
10
+ emit: (type: string, data: unknown) => void;
11
+ };
12
+ }
13
+ export interface LLMInvokeInput {
14
+ messages: unknown[];
15
+ tools?: Record<string, unknown>;
16
+ providerOptions?: Record<string, Record<string, unknown>>;
17
+ }
18
+ export interface LLMInvokeResult {
19
+ response: string;
20
+ tokenUsage: TokenUsage;
21
+ }
22
+ export interface LLMStreamHandle {
23
+ fullStream: AsyncIterable<unknown>;
24
+ usage: Promise<TokenUsage>;
25
+ reasoning: Promise<string | undefined>;
26
+ }
27
+ export declare function extractTokenUsage(usage: unknown): TokenUsage;
28
+ export declare class LLMInvoker {
29
+ private options;
30
+ constructor(options: LLMInvokerOptions);
31
+ invoke(input: LLMInvokeInput): Promise<LLMInvokeResult>;
32
+ stream(input: LLMInvokeInput): LLMStreamHandle;
33
+ }
34
+ //# sourceMappingURL=llm-invoker.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llm-invoker.d.ts","sourceRoot":"","sources":["../src/llm-invoker.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAExD,OAAO,EAAmB,KAAK,YAAY,EAAE,MAAM,YAAY,CAAC;AAEhE,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,aAAa,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE;QAAE,IAAI,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,KAAK,IAAI,CAAA;KAAE,CAAC;CAC5D;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,UAAU,CAAC;CACxB;AAED,MAAM,WAAW,eAAe;IAC9B,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;IACnC,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3B,SAAS,EAAE,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;CACxC;AAID,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,OAAO,GAAG,UAAU,CAU5D;AAED,qBAAa,UAAU;IACrB,OAAO,CAAC,OAAO,CAAoB;gBAEvB,OAAO,EAAE,iBAAiB;IAIhC,MAAM,CAAC,KAAK,EAAE,cAAc,GAAG,OAAO,CAAC,eAAe,CAAC;IAsD7D,MAAM,CAAC,KAAK,EAAE,cAAc,GAAG,eAAe;CA4D/C"}
@@ -0,0 +1,124 @@
1
+ import { streamText } from 'ai';
2
+ import { SpanType } from '@primo-ai/sdk';
3
+ import { streamWithRetry } from './retry.js';
4
+ export function extractTokenUsage(usage) {
5
+ const u = usage;
6
+ return {
7
+ input: typeof u?.inputTokens === 'number'
8
+ ? u.inputTokens
9
+ : u?.inputTokens?.total ?? 0,
10
+ output: typeof u?.outputTokens === 'number'
11
+ ? u.outputTokens
12
+ : u?.outputTokens?.total ?? 0,
13
+ };
14
+ }
15
+ export class LLMInvoker {
16
+ options;
17
+ constructor(options) {
18
+ this.options = options;
19
+ }
20
+ async invoke(input) {
21
+ const span = this.options.tracer?.startSpan(SpanType.MODEL_STEP);
22
+ try {
23
+ return await streamWithRetry(async () => {
24
+ const streamOpts = {
25
+ model: this.options.model,
26
+ system: this.options.system,
27
+ messages: input.messages,
28
+ maxRetries: 0,
29
+ };
30
+ if (input.tools && Object.keys(input.tools).length > 0) {
31
+ streamOpts.tools = input.tools;
32
+ }
33
+ if (input.providerOptions) {
34
+ streamOpts.providerOptions = input.providerOptions;
35
+ }
36
+ span?.setAttribute('llm.model', this.options.model.modelId ?? 'unknown');
37
+ const result = streamText(streamOpts);
38
+ const chunks = [];
39
+ let usage = null;
40
+ for await (const event of result.fullStream) {
41
+ switch (event.type) {
42
+ case 'text-delta':
43
+ chunks.push(event.text);
44
+ break;
45
+ case 'error':
46
+ throw event.error;
47
+ case 'finish-step':
48
+ usage = event.usage;
49
+ break;
50
+ }
51
+ }
52
+ if (!usage) {
53
+ try {
54
+ usage = await result.usage;
55
+ }
56
+ catch (err) {
57
+ this.options.eventBus?.emit('llm:usage_unavailable', { error: err instanceof Error ? err.message : String(err) });
58
+ }
59
+ }
60
+ return {
61
+ response: chunks.join(''),
62
+ tokenUsage: extractTokenUsage(usage),
63
+ };
64
+ }, this.options.retryOptions ?? { maxRetries: 3, baseDelay: 1000 });
65
+ }
66
+ finally {
67
+ span?.end();
68
+ }
69
+ }
70
+ stream(input) {
71
+ const span = this.options.tracer?.startSpan(SpanType.LLM_STREAM);
72
+ const streamOpts = {
73
+ model: this.options.model,
74
+ system: this.options.system,
75
+ messages: input.messages,
76
+ maxRetries: 0,
77
+ };
78
+ if (input.tools && Object.keys(input.tools).length > 0) {
79
+ streamOpts.tools = input.tools;
80
+ }
81
+ if (input.providerOptions) {
82
+ streamOpts.providerOptions = input.providerOptions;
83
+ }
84
+ span?.setAttribute('llm.model', this.options.model.modelId ?? 'unknown');
85
+ // Retry only the initial streamText() call (connection phase).
86
+ // Subsequent stream iteration errors are not retried here.
87
+ const maxRetries = this.options.retryOptions?.maxRetries ?? 3;
88
+ let result;
89
+ for (let attempt = 0; attempt <= maxRetries; attempt++) {
90
+ try {
91
+ result = streamText(streamOpts);
92
+ break;
93
+ }
94
+ catch (error) {
95
+ if (attempt >= maxRetries) {
96
+ span?.end();
97
+ throw error;
98
+ }
99
+ }
100
+ }
101
+ // Suppress AI_NoOutputGeneratedError: when the model returns only tool-calls
102
+ // (no text), the AI SDK's internal flush rejects. Since we consume fullStream
103
+ // directly to extract tool-call events, this error is expected and harmless.
104
+ Promise.resolve(result.text).catch(() => { });
105
+ // End the span when usage resolves (i.e. after the stream finishes),
106
+ // matching the lifecycle span pattern used in invoke().
107
+ const endSpan = () => { span?.end(); };
108
+ return {
109
+ fullStream: result.fullStream,
110
+ usage: Promise.resolve(result.usage)
111
+ .then((u) => { endSpan(); return extractTokenUsage(u); })
112
+ .catch((err) => {
113
+ endSpan();
114
+ this.options.eventBus?.emit('llm:usage_unavailable', { error: err instanceof Error ? err.message : String(err) });
115
+ return { input: 0, output: 0 };
116
+ }),
117
+ reasoning: Promise.resolve(result.reasoningText).catch((err) => {
118
+ this.options.eventBus?.emit('llm:reasoning_error', { error: err instanceof Error ? err.message : String(err) });
119
+ return undefined;
120
+ }),
121
+ };
122
+ }
123
+ }
124
+ //# sourceMappingURL=llm-invoker.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"llm-invoker.js","sourceRoot":"","sources":["../src/llm-invoker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAGhC,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,eAAe,EAAqB,MAAM,YAAY,CAAC;AA6BhE,MAAM,UAAU,iBAAiB,CAAC,KAAc;IAC9C,MAAM,CAAC,GAAG,KAAmB,CAAC;IAC9B,OAAO;QACL,KAAK,EAAE,OAAO,CAAC,EAAE,WAAW,KAAK,QAAQ;YACvC,CAAC,CAAC,CAAC,CAAC,WAAW;YACf,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,IAAI,CAAC;QAC9B,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,KAAK,QAAQ;YACzC,CAAC,CAAC,CAAC,CAAC,YAAY;YAChB,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,KAAK,IAAI,CAAC;KAChC,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,UAAU;IACb,OAAO,CAAoB;IAEnC,YAAY,OAA0B;QACpC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAqB;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QACjE,IAAI,CAAC;YACH,OAAO,MAAM,eAAe,CAAC,KAAK,IAAI,EAAE;gBACtC,MAAM,UAAU,GAA4B;oBAC1C,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;oBACzB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;oBAC3B,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,UAAU,EAAE,CAAC;iBACd,CAAC;gBAEF,IAAI,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvD,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;gBACjC,CAAC;gBACD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;oBAC1B,UAAU,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;gBACrD,CAAC;gBAED,IAAI,EAAE,YAAY,CAAC,WAAW,EAAG,IAAI,CAAC,OAAO,CAAC,KAAwC,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;gBAE7G,MAAM,MAAM,GAAG,UAAU,CAAC,UAAyD,CAAC,CAAC;gBAErF,MAAM,MAAM,GAAa,EAAE,CAAC;gBAC5B,IAAI,KAAK,GAAY,IAAI,CAAC;gBAE1B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;oBAC5C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;wBACnB,KAAK,YAAY;4BACf,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;4BACxB,MAAM;wBACR,KAAK,OAAO;4BACV,MAAM,KAAK,CAAC,KAAK,CAAC;wBACpB,KAAK,aAAa;4BAChB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;4BACpB,MAAM;oBACV,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,IAAI,CAAC;wBAAC,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;oBAAC,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBAC/C,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;oBACpH,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;oBACzB,UAAU,EAAE,iBAAiB,CAAC,KAAK,CAAC;iBACrC,CAAC;YACJ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC;gBAAS,CAAC;YACT,IAAI,EAAE,GAAG,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAED,MAAM,CAAC,KAAqB;QAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAEjE,MAAM,UAAU,GAA4B;YAC1C,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;YACzB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;YAC3B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,UAAU,EAAE,CAAC;SACd,CAAC;QAEF,IAAI,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvD,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QACjC,CAAC;QACD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC1B,UAAU,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC;QACrD,CAAC;QAED,IAAI,EAAE,YAAY,CAAC,WAAW,EAAG,IAAI,CAAC,OAAO,CAAC,KAAwC,CAAC,OAAO,IAAI,SAAS,CAAC,CAAC;QAE7G,+DAA+D;QAC/D,2DAA2D;QAC3D,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,UAAU,IAAI,CAAC,CAAC;QAC9D,IAAI,MAAqC,CAAC;QAE1C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,UAAU,EAAE,OAAO,EAAE,EAAE,CAAC;YACvD,IAAI,CAAC;gBACH,MAAM,GAAG,UAAU,CAAC,UAAyD,CAAC,CAAC;gBAC/E,MAAM;YACR,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;oBAC1B,IAAI,EAAE,GAAG,EAAE,CAAC;oBACZ,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;QACH,CAAC;QAED,6EAA6E;QAC7E,8EAA8E;QAC9E,6EAA6E;QAC7E,OAAO,CAAC,OAAO,CAAC,MAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAE9C,qEAAqE;QACrE,wDAAwD;QACxD,MAAM,OAAO,GAAG,GAAS,EAAE,GAAG,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC;QAE7C,OAAO;YACL,UAAU,EAAE,MAAO,CAAC,UAAU;YAC9B,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,MAAO,CAAC,KAAK,CAAC;iBAClC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,OAAO,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;iBACxD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACb,OAAO,EAAE,CAAC;gBACV,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAClH,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YACjC,CAAC,CAAC;YACJ,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,MAAO,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC9D,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,qBAAqB,EAAE,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAChH,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC;SACH,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,48 @@
1
+ import type { CheckpointStore, PipelineContext, PipelineStageConfig, StageMutation, StreamEvent } from '@primo-ai/sdk';
2
+ import type { PipelineRunner } from './pipeline.js';
3
+ import type { HookManager } from './hook-manager.js';
4
+ import type { EventBus } from './event-bus.js';
5
+ import { StateMachine, type AgentState } from './state-machine.js';
6
+ import { serialize } from './serialize.js';
7
+ /** Result of runLoop / resumeLoop, carrying both the final context and retry statistics. */
8
+ export interface LoopResult {
9
+ context: PipelineContext;
10
+ compatRetries: number;
11
+ }
12
+ export interface LoopOptions {
13
+ maxIterations: number;
14
+ signal?: globalThis.AbortSignal;
15
+ modelString: string;
16
+ sessionId: string;
17
+ maxCompatRetries?: number;
18
+ /** When true, saves a checkpoint after each completed iteration */
19
+ autoCheckpoint?: boolean;
20
+ /** Max times a processor can retryFrom the same stage per loop (default: 3) */
21
+ maxProcessorRetries?: number;
22
+ }
23
+ export declare class LoopOrchestrator {
24
+ private runner;
25
+ private hookManager;
26
+ readonly stateMachine: StateMachine;
27
+ private checkpointStore;
28
+ private eventBus?;
29
+ private preLoopStages;
30
+ private loopStages;
31
+ private postLoopStages;
32
+ constructor(runner: PipelineRunner, hookManager: HookManager, checkpointStore?: CheckpointStore<ReturnType<typeof serialize>>, eventBus?: EventBus, stageConfig?: PipelineStageConfig);
33
+ get state(): AgentState;
34
+ applyMutation(mutation: StageMutation): void;
35
+ runLoop(ctx: PipelineContext, options: LoopOptions): Promise<LoopResult>;
36
+ resumeLoop(sessionId: string, options: LoopOptions): Promise<LoopResult>;
37
+ streamLoop(ctx: PipelineContext, options: LoopOptions): AsyncGenerator<string>;
38
+ streamEvents(ctx: PipelineContext, options: LoopOptions): AsyncGenerator<StreamEvent>;
39
+ private streamCore;
40
+ private resetToRunning;
41
+ private saveCheckpoint;
42
+ private computeLoopStages;
43
+ /** Handle compat retry logic. Returns updated loopCtx on retry, throws on unrecoverable error. */
44
+ private handleLoopError;
45
+ private afterIteration;
46
+ private finalizeState;
47
+ }
48
+ //# sourceMappingURL=loop-orchestrator.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"loop-orchestrator.d.ts","sourceRoot":"","sources":["../src/loop-orchestrator.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAEV,eAAe,EACf,eAAe,EAEf,mBAAmB,EACnB,aAAa,EAEb,WAAW,EAEZ,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,OAAO,EAAE,YAAY,EAAE,KAAK,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,SAAS,EAAe,MAAM,gBAAgB,CAAC;AAUxD,4FAA4F;AAC5F,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,eAAe,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,WAAW;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,CAAC;IAChC,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mEAAmE;IACnE,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,+EAA+E;IAC/E,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC9B;AAOD,qBAAa,gBAAgB;IASzB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,WAAW;IATrB,QAAQ,CAAC,YAAY,eAAsB;IAC3C,OAAO,CAAC,eAAe,CAAgD;IACvE,OAAO,CAAC,QAAQ,CAAC,CAAW;IAC5B,OAAO,CAAC,aAAa,CAAc;IACnC,OAAO,CAAC,UAAU,CAAc;IAChC,OAAO,CAAC,cAAc,CAAc;gBAG1B,MAAM,EAAE,cAAc,EACtB,WAAW,EAAE,WAAW,EAChC,eAAe,CAAC,EAAE,eAAe,CAAC,UAAU,CAAC,OAAO,SAAS,CAAC,CAAC,EAC/D,QAAQ,CAAC,EAAE,QAAQ,EACnB,WAAW,CAAC,EAAE,mBAAmB;IAWnC,IAAI,KAAK,IAAI,UAAU,CAEtB;IAED,aAAa,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI;IAoCtC,OAAO,CAAC,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;IAaxE,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC;IAavE,UAAU,CAAC,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC;IAW9E,YAAY,CAAC,GAAG,EAAE,eAAe,EAAE,OAAO,EAAE,WAAW,GAAG,cAAc,CAAC,WAAW,CAAC;YAQ7E,UAAU;IA+FzB,OAAO,CAAC,cAAc;YAOR,cAAc;IAI5B,OAAO,CAAC,iBAAiB;IAWzB,kGAAkG;YACpF,eAAe;YA2Bf,cAAc;IAW5B,OAAO,CAAC,aAAa;CAQtB"}
@@ -0,0 +1,247 @@
1
+ import { applyReactiveRules } from './processors/provider-history-compat.js';
2
+ import { StateMachine } from './state-machine.js';
3
+ import { serialize, deserialize } from './serialize.js';
4
+ import { JsonlCheckpointStore } from './checkpoint-store.js';
5
+ import { join } from 'node:path';
6
+ const PRE_LOOP_STAGES = ['processInput', 'buildContext'];
7
+ const LOOP_STAGES = [
8
+ 'prepareStep', 'gateLLM', 'invokeLLM', 'processStepOutput', 'gateTool', 'executeTools', 'evaluateIteration',
9
+ ];
10
+ const POST_LOOP_STAGES = ['processOutput'];
11
+ export class LoopOrchestrator {
12
+ runner;
13
+ hookManager;
14
+ stateMachine = new StateMachine();
15
+ checkpointStore;
16
+ eventBus;
17
+ preLoopStages;
18
+ loopStages;
19
+ postLoopStages;
20
+ constructor(runner, hookManager, checkpointStore, eventBus, stageConfig) {
21
+ this.runner = runner;
22
+ this.hookManager = hookManager;
23
+ this.checkpointStore = checkpointStore ?? new JsonlCheckpointStore(join(process.cwd(), '.agentforge', 'checkpoints'));
24
+ this.eventBus = eventBus;
25
+ this.preLoopStages = stageConfig?.preLoop ?? PRE_LOOP_STAGES;
26
+ this.loopStages = stageConfig?.loop ?? LOOP_STAGES;
27
+ this.postLoopStages = stageConfig?.postLoop ?? POST_LOOP_STAGES;
28
+ }
29
+ get state() {
30
+ return this.stateMachine.current;
31
+ }
32
+ applyMutation(mutation) {
33
+ if (this.stateMachine.current !== 'pending') {
34
+ throw new Error(`Stage mutations only allowed before first run (current: ${this.stateMachine.current})`);
35
+ }
36
+ const arr = mutation.phase === 'preLoop' ? this.preLoopStages
37
+ : mutation.phase === 'loop' ? this.loopStages
38
+ : this.postLoopStages;
39
+ switch (mutation.type) {
40
+ case 'insert': {
41
+ const idx = arr.indexOf(mutation.after);
42
+ if (idx === -1)
43
+ throw new Error(`Stage "${mutation.after}" not found in ${mutation.phase}`);
44
+ arr.splice(idx + 1, 0, mutation.stage);
45
+ break;
46
+ }
47
+ case 'remove': {
48
+ const idx = arr.indexOf(mutation.stage);
49
+ if (idx === -1)
50
+ throw new Error(`Stage "${mutation.stage}" not found in ${mutation.phase}`);
51
+ arr.splice(idx, 1);
52
+ break;
53
+ }
54
+ case 'replace': {
55
+ if (mutation.phase === 'loop' && !mutation.stages.includes('invokeLLM')) {
56
+ throw new Error('Cannot replace loop stages: required stage "invokeLLM" is missing');
57
+ }
58
+ arr.length = 0;
59
+ arr.push(...mutation.stages);
60
+ break;
61
+ }
62
+ }
63
+ }
64
+ // ---------------------------------------------------------------------------
65
+ // Public: run (non-streaming) — delegates to streamCore
66
+ // ---------------------------------------------------------------------------
67
+ async runLoop(ctx, options) {
68
+ const runState = { compatRetries: 0 };
69
+ let finalCtx = ctx;
70
+ for await (const event of this.streamCore(ctx, options, runState)) {
71
+ if (event.type === 'complete') {
72
+ finalCtx = event.context;
73
+ }
74
+ }
75
+ return { context: finalCtx, compatRetries: runState.compatRetries };
76
+ }
77
+ async resumeLoop(sessionId, options) {
78
+ const checkpoint = await this.checkpointStore.load(sessionId);
79
+ if (!checkpoint)
80
+ throw new Error(`No checkpoint found for session: ${sessionId}`);
81
+ const ctx = deserialize(checkpoint);
82
+ const result = await this.runLoop(ctx, options);
83
+ await this.checkpointStore.delete(sessionId);
84
+ return result;
85
+ }
86
+ // ---------------------------------------------------------------------------
87
+ // Public: streamLoop (yields text deltas)
88
+ // ---------------------------------------------------------------------------
89
+ async *streamLoop(ctx, options) {
90
+ for await (const event of this.streamCore(ctx, options)) {
91
+ if (event.type === 'text_delta')
92
+ yield event.text;
93
+ if (event.type === 'suspended')
94
+ yield ` [suspended: ${event.reason}]`;
95
+ }
96
+ }
97
+ // ---------------------------------------------------------------------------
98
+ // Public: streamEvents (yields StreamEvents)
99
+ // ---------------------------------------------------------------------------
100
+ async *streamEvents(ctx, options) {
101
+ yield* this.streamCore(ctx, options);
102
+ }
103
+ // ---------------------------------------------------------------------------
104
+ // Core streaming loop — single source of truth for streamLoop + streamEvents
105
+ // ---------------------------------------------------------------------------
106
+ async *streamCore(ctx, options, sharedRunState) {
107
+ const { signal, maxIterations, modelString, sessionId } = options;
108
+ const maxCompatRetries = options.maxCompatRetries ?? 3;
109
+ const maxProcessorRetries = options.maxProcessorRetries ?? 3;
110
+ const runState = sharedRunState ?? { compatRetries: 0 };
111
+ const processorRetryCounts = new Map();
112
+ this.resetToRunning();
113
+ let loopCtx = ctx;
114
+ try {
115
+ // Pre-loop stages
116
+ for await (const event of this.runner.stream(loopCtx, this.preLoopStages, { signal })) {
117
+ if (signal?.aborted)
118
+ throw new DOMException('Agent stream aborted', 'AbortError');
119
+ if (event.type === 'abort')
120
+ throw new Error(`Agent aborted: ${event.reason}`);
121
+ if (event.type === 'suspended') {
122
+ await this.saveCheckpoint(sessionId, loopCtx);
123
+ this.stateMachine.transition('paused');
124
+ yield event;
125
+ return;
126
+ }
127
+ if (event.type === 'complete')
128
+ loopCtx = event.context;
129
+ yield event;
130
+ }
131
+ // Agentic loop
132
+ for (let i = 0; i < maxIterations; i++) {
133
+ if (signal?.aborted)
134
+ throw new DOMException('Agent stream aborted', 'AbortError');
135
+ const { ctx: stepCtx, stages } = this.computeLoopStages(loopCtx, i);
136
+ loopCtx = stepCtx;
137
+ let loopBreak = false;
138
+ let compatRetry = false;
139
+ try {
140
+ for await (const event of this.runner.stream(loopCtx, stages, { signal })) {
141
+ if (signal?.aborted)
142
+ throw new DOMException('Agent stream aborted', 'AbortError');
143
+ if (event.type === 'abort') {
144
+ const abortEvent = event;
145
+ if (abortEvent.retryFrom) {
146
+ const count = (processorRetryCounts.get(abortEvent.retryFrom) ?? 0) + 1;
147
+ if (count > maxProcessorRetries) {
148
+ this.eventBus?.emit('processor:retry_exhausted', { stage: abortEvent.retryFrom, retries: count - 1, max: maxProcessorRetries });
149
+ throw new Error(`Processor retry budget exhausted for ${abortEvent.retryFrom}: ${abortEvent.reason}`);
150
+ }
151
+ processorRetryCounts.set(abortEvent.retryFrom, count);
152
+ loopCtx = { ...loopCtx, iteration: { ...loopCtx.iteration, loopDirective: { action: 'retry', retryFrom: abortEvent.retryFrom } } };
153
+ loopBreak = true;
154
+ break;
155
+ }
156
+ throw new Error(`Agent aborted: ${abortEvent.reason}`);
157
+ }
158
+ if (event.type === 'suspended') {
159
+ await this.saveCheckpoint(sessionId, loopCtx);
160
+ this.stateMachine.transition('paused');
161
+ yield event;
162
+ return;
163
+ }
164
+ if (event.type === 'complete')
165
+ loopCtx = event.context;
166
+ yield event;
167
+ }
168
+ }
169
+ catch (error) {
170
+ loopCtx = await this.handleLoopError(error, loopCtx, i, sessionId, modelString, maxCompatRetries, runState);
171
+ compatRetry = true;
172
+ }
173
+ if (compatRetry)
174
+ continue;
175
+ if (loopBreak)
176
+ continue;
177
+ await this.afterIteration(loopCtx, sessionId, options.autoCheckpoint);
178
+ if (loopCtx.iteration.loopDirective?.action === 'stop')
179
+ break;
180
+ }
181
+ if (signal?.aborted)
182
+ throw new DOMException('Agent stream aborted', 'AbortError');
183
+ // Post-loop stages
184
+ for await (const event of this.runner.stream(loopCtx, this.postLoopStages, { signal })) {
185
+ if (event.type === 'abort')
186
+ throw new Error(`Agent aborted: ${event.reason}`);
187
+ yield event;
188
+ }
189
+ this.stateMachine.transition('completed');
190
+ }
191
+ catch (e) {
192
+ this.finalizeState(e);
193
+ throw e;
194
+ }
195
+ }
196
+ // ---------------------------------------------------------------------------
197
+ // Shared helpers
198
+ // ---------------------------------------------------------------------------
199
+ resetToRunning() {
200
+ if (this.stateMachine.current !== 'pending') {
201
+ this.stateMachine.transition('pending');
202
+ }
203
+ this.stateMachine.transition('running');
204
+ }
205
+ async saveCheckpoint(sessionId, ctx) {
206
+ await this.checkpointStore.save(sessionId, serialize(ctx));
207
+ }
208
+ computeLoopStages(ctx, step) {
209
+ const prevDirective = ctx.iteration.loopDirective;
210
+ const newCtx = { ...ctx, iteration: { ...ctx.iteration, step, loopDirective: undefined } };
211
+ const retryFrom = prevDirective?.action === 'retry' ? prevDirective.retryFrom : undefined;
212
+ const stages = retryFrom ? this.loopStages.slice(this.loopStages.indexOf(retryFrom)) : this.loopStages;
213
+ return { ctx: newCtx, stages };
214
+ }
215
+ /** Handle compat retry logic. Returns updated loopCtx on retry, throws on unrecoverable error. */
216
+ async handleLoopError(error, loopCtx, step, sessionId, modelString, maxCompatRetries, runState) {
217
+ await this.hookManager.invoke('error', { error, stage: 'invokeLLM', sessionId }, {});
218
+ const compatResult = applyReactiveRules(loopCtx.session.messageHistory ?? [], modelString, error);
219
+ if (!compatResult)
220
+ throw error;
221
+ runState.compatRetries++;
222
+ this.eventBus?.emit('compat:retry', { step, sessionId, retryCount: runState.compatRetries, maxRetries: maxCompatRetries });
223
+ this.eventBus?.emit('compat:diff', { step, sessionId, diff: compatResult.diff });
224
+ if (runState.compatRetries > maxCompatRetries) {
225
+ this.eventBus?.emit('compat:retry_exhausted', { step, sessionId, retryCount: runState.compatRetries, maxRetries: maxCompatRetries });
226
+ throw error;
227
+ }
228
+ return { ...loopCtx, session: { ...loopCtx.session, messageHistory: compatResult.history } };
229
+ }
230
+ async afterIteration(loopCtx, sessionId, autoCheckpoint) {
231
+ await this.hookManager.invoke('iteration.end', { step: loopCtx.iteration.step, sessionId }, {});
232
+ if (autoCheckpoint) {
233
+ await this.saveCheckpoint(sessionId, loopCtx);
234
+ }
235
+ }
236
+ finalizeState(e) {
237
+ if (this.stateMachine.current !== 'running')
238
+ return;
239
+ if (e instanceof DOMException && e.name === 'AbortError') {
240
+ this.stateMachine.transition('cancelled');
241
+ }
242
+ else {
243
+ this.stateMachine.transition('error');
244
+ }
245
+ }
246
+ }
247
+ //# sourceMappingURL=loop-orchestrator.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"loop-orchestrator.js","sourceRoot":"","sources":["../src/loop-orchestrator.ts"],"names":[],"mappings":"AAcA,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,OAAO,EAAE,YAAY,EAAmB,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAA2B,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AACtF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,MAAM,eAAe,GAAoB,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;AAC1E,MAAM,WAAW,GAAoB;IACnC,aAAa,EAAE,SAAS,EAAE,WAAW,EAAE,mBAAmB,EAAE,UAAU,EAAE,cAAc,EAAE,mBAAmB;CAC5G,CAAC;AACF,MAAM,gBAAgB,GAAoB,CAAC,eAAe,CAAC,CAAC;AAyB5D,MAAM,OAAO,gBAAgB;IASjB;IACA;IATD,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;IACnC,eAAe,CAAgD;IAC/D,QAAQ,CAAY;IACpB,aAAa,CAAc;IAC3B,UAAU,CAAc;IACxB,cAAc,CAAc;IAEpC,YACU,MAAsB,EACtB,WAAwB,EAChC,eAA+D,EAC/D,QAAmB,EACnB,WAAiC;QAJzB,WAAM,GAAN,MAAM,CAAgB;QACtB,gBAAW,GAAX,WAAW,CAAa;QAKhC,IAAI,CAAC,eAAe,GAAG,eAAe,IAAI,IAAI,oBAAoB,CAChE,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,aAAa,CAAC,CAClD,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,aAAa,GAAG,WAAW,EAAE,OAAO,IAAI,eAAe,CAAC;QAC7D,IAAI,CAAC,UAAU,GAAG,WAAW,EAAE,IAAI,IAAI,WAAW,CAAC;QACnD,IAAI,CAAC,cAAc,GAAG,WAAW,EAAE,QAAQ,IAAI,gBAAgB,CAAC;IAClE,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;IACnC,CAAC;IAED,aAAa,CAAC,QAAuB;QACnC,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,2DAA2D,IAAI,CAAC,YAAY,CAAC,OAAO,GAAG,CAAC,CAAC;QAC3G,CAAC;QACD,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa;YAC3D,CAAC,CAAC,QAAQ,CAAC,KAAK,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU;gBAC7C,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC;QAExB,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtB,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACxC,IAAI,GAAG,KAAK,CAAC,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,UAAU,QAAQ,CAAC,KAAK,kBAAkB,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC5F,GAAG,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACvC,MAAM;YACR,CAAC;YACD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACxC,IAAI,GAAG,KAAK,CAAC,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,UAAU,QAAQ,CAAC,KAAK,kBAAkB,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC5F,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBACnB,MAAM;YACR,CAAC;YACD,KAAK,SAAS,CAAC,CAAC,CAAC;gBACf,IAAI,QAAQ,CAAC,KAAK,KAAK,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;oBACxE,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;gBACvF,CAAC;gBACD,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC;gBACf,GAAG,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAC7B,MAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,wDAAwD;IACxD,8EAA8E;IAE9E,KAAK,CAAC,OAAO,CAAC,GAAoB,EAAE,OAAoB;QACtD,MAAM,QAAQ,GAAiB,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;QACpD,IAAI,QAAQ,GAAG,GAAG,CAAC;QAEnB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,CAAC;YAClE,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC9B,QAAQ,GAAI,KAAsC,CAAC,OAAO,CAAC;YAC7D,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,QAAQ,CAAC,aAAa,EAAE,CAAC;IACtE,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,OAAoB;QACtD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9D,IAAI,CAAC,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,oCAAoC,SAAS,EAAE,CAAC,CAAC;QAClF,MAAM,GAAG,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAChD,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC7C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,8EAA8E;IAC9E,0CAA0C;IAC1C,8EAA8E;IAE9E,KAAK,CAAC,CAAC,UAAU,CAAC,GAAoB,EAAE,OAAoB;QAC1D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,EAAE,CAAC;YACxD,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;gBAAE,MAAM,KAAK,CAAC,IAAI,CAAC;YAClD,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW;gBAAE,MAAM,gBAAiB,KAA4B,CAAC,MAAM,GAAG,CAAC;QAChG,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,6CAA6C;IAC7C,8EAA8E;IAE9E,KAAK,CAAC,CAAC,YAAY,CAAC,GAAoB,EAAE,OAAoB;QAC5D,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED,8EAA8E;IAC9E,6EAA6E;IAC7E,8EAA8E;IAEtE,KAAK,CAAC,CAAC,UAAU,CACvB,GAAoB,EACpB,OAAoB,EACpB,cAA6B;QAE7B,MAAM,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC;QAClE,MAAM,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,CAAC,CAAC;QACvD,MAAM,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC7D,MAAM,QAAQ,GAAG,cAAc,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;QACxD,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAyB,CAAC;QAE9D,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,IAAI,OAAO,GAAG,GAAG,CAAC;QAClB,IAAI,CAAC;YACH,kBAAkB;YAClB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;gBACtF,IAAI,MAAM,EAAE,OAAO;oBAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;gBAClF,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;oBAAE,MAAM,IAAI,KAAK,CAAC,kBAAmB,KAAyB,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnG,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC/B,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;oBAC9C,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;oBACvC,MAAM,KAAK,CAAC;oBACZ,OAAO;gBACT,CAAC;gBACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;oBAAE,OAAO,GAAI,KAAsC,CAAC,OAAO,CAAC;gBACzF,MAAM,KAAK,CAAC;YACd,CAAC;YAED,eAAe;YACf,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,IAAI,MAAM,EAAE,OAAO;oBAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;gBAElF,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBACpE,OAAO,GAAG,OAAO,CAAC;gBAElB,IAAI,SAAS,GAAG,KAAK,CAAC;gBACtB,IAAI,WAAW,GAAG,KAAK,CAAC;gBACxB,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;wBAC1E,IAAI,MAAM,EAAE,OAAO;4BAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;wBAClF,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;4BAC3B,MAAM,UAAU,GAAG,KAAqE,CAAC;4BACzF,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;gCACzB,MAAM,KAAK,GAAG,CAAC,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;gCACxE,IAAI,KAAK,GAAG,mBAAmB,EAAE,CAAC;oCAChC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,2BAA2B,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,SAAS,EAAE,OAAO,EAAE,KAAK,GAAG,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;oCAChI,MAAM,IAAI,KAAK,CAAC,wCAAwC,UAAU,CAAC,SAAS,KAAK,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;gCACxG,CAAC;gCACD,oBAAoB,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;gCACtD,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,EAAE,GAAG,OAAO,CAAC,SAAS,EAAE,aAAa,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC;gCACnI,SAAS,GAAG,IAAI,CAAC;gCACjB,MAAM;4BACR,CAAC;4BACD,MAAM,IAAI,KAAK,CAAC,kBAAkB,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;wBACzD,CAAC;wBACD,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;4BAC/B,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;4BAC9C,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;4BACvC,MAAM,KAAK,CAAC;4BACZ,OAAO;wBACT,CAAC;wBACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;4BAAE,OAAO,GAAI,KAAsC,CAAC,OAAO,CAAC;wBACzF,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,gBAAgB,EAAE,QAAQ,CAAC,CAAC;oBAC5G,WAAW,GAAG,IAAI,CAAC;gBACrB,CAAC;gBACD,IAAI,WAAW;oBAAE,SAAS;gBAC1B,IAAI,SAAS;oBAAE,SAAS;gBAExB,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,cAAc,CAAC,CAAC;gBACtE,IAAI,OAAO,CAAC,SAAS,CAAC,aAAa,EAAE,MAAM,KAAK,MAAM;oBAAE,MAAM;YAChE,CAAC;YAED,IAAI,MAAM,EAAE,OAAO;gBAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;YAElF,mBAAmB;YACnB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;gBACvF,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;oBAAE,MAAM,IAAI,KAAK,CAAC,kBAAmB,KAAyB,CAAC,MAAM,EAAE,CAAC,CAAC;gBACnG,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACtB,MAAM,CAAC,CAAC;QACV,CAAC;IACH,CAAC;IAED,8EAA8E;IAC9E,iBAAiB;IACjB,8EAA8E;IAEtE,cAAc;QACpB,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC5C,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC1C,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,SAAiB,EAAE,GAAoB;QAClE,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,iBAAiB,CACvB,GAAoB,EACpB,IAAY;QAEZ,MAAM,aAAa,GAAG,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC;QAClD,MAAM,MAAM,GAAG,EAAE,GAAG,GAAG,EAAE,SAAS,EAAE,EAAE,GAAG,GAAG,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,CAAC;QAC3F,MAAM,SAAS,GAAG,aAAa,EAAE,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1F,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC;QACvG,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;IACjC,CAAC;IAED,kGAAkG;IAC1F,KAAK,CAAC,eAAe,CAC3B,KAAc,EACd,OAAwB,EACxB,IAAY,EACZ,SAAiB,EACjB,WAAmB,EACnB,gBAAwB,EACxB,QAAsB;QAEtB,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,WAA4B,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;QACtG,MAAM,YAAY,GAAG,kBAAkB,CACrC,OAAO,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,EACpC,WAAW,EACX,KAAK,CACN,CAAC;QACF,IAAI,CAAC,YAAY;YAAE,MAAM,KAAK,CAAC;QAE/B,QAAQ,CAAC,aAAa,EAAE,CAAC;QACzB,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,aAAa,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC3H,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;QACjF,IAAI,QAAQ,CAAC,aAAa,GAAG,gBAAgB,EAAE,CAAC;YAC9C,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,wBAAwB,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,aAAa,EAAE,UAAU,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACrI,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,cAAc,EAAE,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;IAC/F,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,OAAwB,EACxB,SAAiB,EACjB,cAAwB;QAExB,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;QAChG,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,CAAU;QAC9B,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,KAAK,SAAS;YAAE,OAAO;QACpD,IAAI,CAAC,YAAY,YAAY,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACzD,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,13 @@
1
+ import type { LanguageModel } from 'ai';
2
+ import type { ModelGateway } from '@primo-ai/sdk';
3
+ type ProviderFactory = (modelId: string) => LanguageModel;
4
+ export declare class ModelFactory {
5
+ private chain;
6
+ private providerGateway;
7
+ constructor();
8
+ resolve(modelString: string): Promise<LanguageModel>;
9
+ registerGateway(gateway: ModelGateway): void;
10
+ registerProvider(name: string, factory: ProviderFactory): void;
11
+ }
12
+ export {};
13
+ //# sourceMappingURL=model-factory.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"model-factory.d.ts","sourceRoot":"","sources":["../src/model-factory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAGlD,KAAK,eAAe,GAAG,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;AA0B1D,qBAAa,YAAY;IACvB,OAAO,CAAC,KAAK,CAAsB;IACnC,OAAO,CAAC,eAAe,CAAyB;;IAM1C,OAAO,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAI1D,eAAe,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAI5C,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,GAAG,IAAI;CAG/D"}
@@ -0,0 +1,40 @@
1
+ import { GatewayChain } from './gateways/gateway-chain.js';
2
+ class ProviderGateway {
3
+ name = 'custom-providers';
4
+ providers = new Map();
5
+ addProvider(name, factory) {
6
+ this.providers.set(name, factory);
7
+ }
8
+ canResolve(modelString) {
9
+ const idx = modelString.indexOf('/');
10
+ if (idx < 1)
11
+ return false;
12
+ return this.providers.has(modelString.slice(0, idx));
13
+ }
14
+ async resolve(modelString) {
15
+ const idx = modelString.indexOf('/');
16
+ const provider = modelString.slice(0, idx);
17
+ const modelId = modelString.slice(idx + 1);
18
+ const factory = this.providers.get(provider);
19
+ if (!factory)
20
+ throw new Error(`Unknown provider: "${provider}"`);
21
+ return factory(modelId);
22
+ }
23
+ }
24
+ export class ModelFactory {
25
+ chain = new GatewayChain();
26
+ providerGateway = new ProviderGateway();
27
+ constructor() {
28
+ this.chain.register(this.providerGateway);
29
+ }
30
+ async resolve(modelString) {
31
+ return this.chain.resolve(modelString);
32
+ }
33
+ registerGateway(gateway) {
34
+ this.chain.register(gateway);
35
+ }
36
+ registerProvider(name, factory) {
37
+ this.providerGateway.addProvider(name, factory);
38
+ }
39
+ }
40
+ //# sourceMappingURL=model-factory.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"model-factory.js","sourceRoot":"","sources":["../src/model-factory.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAI3D,MAAM,eAAe;IACnB,IAAI,GAAG,kBAAkB,CAAC;IAClB,SAAS,GAAiC,IAAI,GAAG,EAAE,CAAC;IAE5D,WAAW,CAAC,IAAY,EAAE,OAAwB;QAChD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACpC,CAAC;IAED,UAAU,CAAC,WAAmB;QAC5B,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,GAAG,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QAC1B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,WAAmB;QAC/B,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,QAAQ,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,QAAQ,GAAG,CAAC,CAAC;QACjE,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,OAAO,YAAY;IACf,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC;IAC3B,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAEhD;QACE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,WAAmB;QAC/B,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACzC,CAAC;IAED,eAAe,CAAC,OAAqB;QACnC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED,gBAAgB,CAAC,IAAY,EAAE,OAAwB;QACrD,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;CACF"}
@@ -0,0 +1,17 @@
1
+ import type { PipelineContext, ModelProfile } from '@primo-ai/sdk';
2
+ /**
3
+ * Find the first ModelProfile whose modelPattern matches the given model string.
4
+ * String patterns use simple `includes`; RegExp patterns use `test()`.
5
+ */
6
+ export declare function matchProfile(model: string, profiles: ModelProfile[]): ModelProfile | undefined;
7
+ /**
8
+ * Apply a ModelProfile to a PipelineContext, returning a new context
9
+ * (original is not mutated).
10
+ *
11
+ * - extraPromptFragments are converted from PromptFragment[] to strings
12
+ * by extracting `.content`, then appended to promptFragments.
13
+ * - systemPromptSuffix is appended as a plain string after fragments.
14
+ * - toolOverrides.exclude removes matching tools; description overrides update them.
15
+ */
16
+ export declare function applyProfile(ctx: PipelineContext, profile: ModelProfile): PipelineContext;
17
+ //# sourceMappingURL=model-profile.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"model-profile.d.ts","sourceRoot":"","sources":["../src/model-profile.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEnE;;;GAGG;AACH,wBAAgB,YAAY,CAC1B,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,YAAY,EAAE,GACvB,YAAY,GAAG,SAAS,CAS1B;AAED;;;;;;;;GAQG;AACH,wBAAgB,YAAY,CAC1B,GAAG,EAAE,eAAe,EACpB,OAAO,EAAE,YAAY,GACpB,eAAe,CAgCjB"}