@relayfx/sdk 0.3.4 → 0.3.5

package/dist/ai.js

@@ -32,7 +32,7 @@ import {
   exports_tool_executor,
   exports_tool_output,
   exports_turn_policy
-} from "./index-25gwk9tj.js";
+} from "./index-8fpd6kvj.js";
 import"./index-nb39b5ae.js";
 
 // src/ai.ts

package/dist/http-server.js

@@ -4,7 +4,7 @@ import {
   Service,
   Service1 as Service2,
   StateVersionConflict
-} from "./index-cphmds30.js";
+} from "./index-cyfx5y3r.js";
 import"./index-nb39b5ae.js";
 
 // src/http-server.ts

package/dist/{index-pr9133xv.js → index-5dqkw53p.js}

@@ -8,7 +8,7 @@ import {
   SqlFailure,
   make,
   normalizeAcquisitionCause
-} from "./index-01rpd00m.js";
+} from "./index-cq8t8ka1.js";
 
 // src/migration-errors.ts
 import { Cause, Effect, Function } from "effect";

package/dist/{index-25gwk9tj.js → index-8fpd6kvj.js}

@@ -3,7 +3,7 @@ import {
   __export
 } from "./index-nb39b5ae.js";
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/turn-policy.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/turn-policy.js
 var exports_turn_policy = {};
 __export(exports_turn_policy, {
   untilToolCall: () => untilToolCall,
@@ -74,7 +74,7 @@ var both = dual(2, (first, second) => ({
 }));
 var defaultPolicy = forever;
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/agent-event.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/agent-event.js
 var exports_agent_event = {};
 __export(exports_agent_event, {
   addUsage: () => addUsage,
@@ -184,7 +184,7 @@ class ResumeMismatch extends Schema2.TaggedErrorClass()("@batonfx/core/ResumeMis
 }) {
 }
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/tool-context.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/tool-context.js
 var exports_tool_context = {};
 __export(exports_tool_context, {
   testLayer: () => testLayer,
@@ -202,7 +202,7 @@ var layerDefault = Layer2.sync(ToolContext, () => ToolContext.of({
 }));
 var testLayer = (implementation) => Layer2.succeed(ToolContext, ToolContext.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/tool-executor.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/tool-executor.js
 var exports_tool_executor = {};
 __export(exports_tool_executor, {
   testLayer: () => testLayer2,
@@ -427,7 +427,7 @@ function router(routes) {
 }
 var testLayer2 = (implementation) => Layer3.succeed(ToolExecutor, ToolExecutor.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/approvals.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/approvals.js
 var exports_approvals = {};
 __export(exports_approvals, {
   testLayer: () => testLayer3,
@@ -442,7 +442,7 @@ var autoApprove = Layer4.succeed(Approvals, Approvals.of({ check: () => Effect4.
 var denyAll = Layer4.succeed(Approvals, Approvals.of({ check: () => Effect4.succeed({ _tag: "Denied" }) }));
 var testLayer3 = (implementation) => Layer4.succeed(Approvals, Approvals.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/memory.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/memory.js
 var exports_memory = {};
 __export(exports_memory, {
   testLayer: () => testLayer4,
@@ -510,7 +510,7 @@ var layerNoop = Layer5.succeed(Memory, Memory.of(noop));
 var noopLayer = layerNoop;
 var testLayer4 = (implementation) => Layer5.succeed(Memory, Memory.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/session.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/session.js
 var exports_session = {};
 __export(exports_session, {
   testLayer: () => testLayer5,
@@ -789,7 +789,7 @@ var layerMemory = Layer6.effect(SessionStore, Ref.make(initialState).pipe(Effect
 var memoryLayer = layerMemory;
 var testLayer5 = (implementation) => Layer6.succeed(SessionStore, SessionStore.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/tool-output.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/tool-output.js
 var exports_tool_output = {};
 __export(exports_tool_output, {
   testLayer: () => testLayer6,
@@ -865,7 +865,7 @@ var bound = Function2.dual(2, (result, options) => Effect7.gen(function* () {
   return boundedFromOriginal(encoded, bytes, options.maxBytes, [path.value]);
 }));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/compaction.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/compaction.js
 var exports_compaction = {};
 __export(exports_compaction, {
   truncate: () => truncate,
@@ -1134,7 +1134,7 @@ var truncate = (maxTokens) => ({
 });
 var testLayer7 = (implementation) => Layer8.succeed(Compaction, Compaction.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/instructions.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/instructions.js
 var exports_instructions = {};
 __export(exports_instructions, {
   testLayer: () => testLayer8,
@@ -1183,7 +1183,7 @@ var renderUpdate = dual3(2, (epoch, context) => Effect9.gen(function* () {
 var layer2 = (sources) => Layer9.succeed(Instructions, Instructions.of({ sources: [...sources] }));
 var testLayer8 = (implementation) => Layer9.succeed(Instructions, Instructions.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/model-middleware.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/model-middleware.js
 var exports_model_middleware = {};
 __export(exports_model_middleware, {
   layerIdentity: () => layerIdentity,
@@ -1199,7 +1199,7 @@ var layerIdentity = Layer10.succeed(ModelMiddleware, []);
 var identityLayer = layerIdentity;
 var layer3 = (middleware) => Layer10.succeed(ModelMiddleware, middleware);
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/model-registry.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/model-registry.js
 var exports_model_registry = {};
 __export(exports_model_registry, {
   testLayer: () => testLayer9,
@@ -1326,7 +1326,7 @@ var operate = Function4.dual(2, (selection, effect) => Effect11.gen(function* ()
 var stream = Function4.dual(2, (selection, operation) => Stream2.unwrap(Service.pipe(Effect11.map((service) => service.stream(selection, operation)))));
 var provide = operate;
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/model-resilience.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/model-resilience.js
 var exports_model_resilience = {};
 __export(exports_model_resilience, {
   testLayer: () => testLayer10,
@@ -1379,7 +1379,7 @@ var apply = Function5.dual(2, (model, resilience) => ({
   streamText: (options) => retryStream(() => model.streamText(options), (error) => Response5.makePart("error", { error }), resilience)
 }));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/permissions.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/permissions.js
 var exports_permissions = {};
 __export(exports_permissions, {
   testLayer: () => testLayer11,
@@ -1513,7 +1513,7 @@ var ruleStoreMemory = (initialRules = []) => Layer13.effect(RuleStore, Ref4.make
 var ruleStoreTestLayer = (implementation) => Layer13.succeed(RuleStore, RuleStore.of(implementation));
 var testLayer11 = (implementation) => Layer13.succeed(Permissions, Permissions.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/skill-source.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/skill-source.js
 var exports_skill_source = {};
 __export(exports_skill_source, {
   testLayer: () => testLayer12,
@@ -1593,7 +1593,7 @@ var selectListings = Function6.dual(3, (skills, budgetTokens, recentlyUsed) => {
   return selected;
 });
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/steering.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/steering.js
 var exports_steering = {};
 __export(exports_steering, {
   testLayer: () => testLayer13,
@@ -1651,7 +1651,7 @@ var layer7 = (options = {}) => Layer15.effect(Steering, Effect15.gen(function* (
 }));
 var testLayer13 = (implementation) => Layer15.succeed(Steering, Steering.of(implementation));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/agent.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/agent.js
 var exports_agent = {};
 __export(exports_agent, {
   streamObject: () => streamObject,
@@ -1670,7 +1670,7 @@ import { Cause as Cause4, Channel, Effect as Effect18, Equal, Exit as Exit2, Fib
 import { dual as dual7 } from "effect/Function";
 import { AiError as AiError3, Chat, LanguageModel as LanguageModel5, Prompt as Prompt9, Response as Response7, Telemetry, Tokenizer as Tokenizer2, Tool as Tool7, Toolkit as Toolkit4 } from "effect/unstable/ai";
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/tool-authorization.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/tool-authorization.js
 import { Cause as Cause3, Context as Context15, Effect as Effect16, Layer as Layer16, Option as Option10, Schema as Schema13 } from "effect";
 import { dual as dual5 } from "effect/Function";
 import { Prompt as Prompt8, Response as Response6, Tool as Tool5 } from "effect/unstable/ai";
@@ -1841,7 +1841,7 @@ var make4 = (options = {}) => ({
 });
 var fromPermissions = dual5((args) => args.length === 2 || args.length === 1 && ("evaluate" in args[0]), (permissions, options = {}) => make4({ ...options, permissions }));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/tool-registry.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/tool-registry.js
 import { Effect as Effect17, HashMap as HashMap3, Option as Option11 } from "effect";
 import { dual as dual6 } from "effect/Function";
 import { Tool as Tool6, Toolkit as Toolkit3 } from "effect/unstable/ai";
@@ -1894,7 +1894,7 @@ var select = dual6(2, (registry, names) => {
   };
 });
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/agent.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/agent.js
 var AgentTypeId = Symbol.for("@batonfx/core/Agent");
 var ModelLayerTypeId = Symbol.for("@batonfx/core/Agent/ModelLayer");
 var classifyOtherFailure = () => "other";
@@ -1911,6 +1911,7 @@ function make5(nameOrOptions, options = {}) {
         ...curriedOptions.model === undefined ? {} : { model: curriedOptions.model },
         ...curriedOptions.memory === undefined ? {} : { memory: curriedOptions.memory },
         ...curriedOptions.authorization === undefined ? {} : { authorization: curriedOptions.authorization },
+        ...curriedOptions.toolExecution === undefined ? {} : { toolExecution: curriedOptions.toolExecution },
         ...curriedOptions.metadata === undefined ? {} : { metadata: curriedOptions.metadata }
       });
     }
@@ -1922,6 +1923,7 @@ function make5(nameOrOptions, options = {}) {
       ...curriedOptions.model === undefined ? {} : { model: curriedOptions.model },
       ...curriedOptions.memory === undefined ? {} : { memory: curriedOptions.memory },
       ...curriedOptions.authorization === undefined ? {} : { authorization: curriedOptions.authorization },
+      ...curriedOptions.toolExecution === undefined ? {} : { toolExecution: curriedOptions.toolExecution },
       ...curriedOptions.metadata === undefined ? {} : { metadata: curriedOptions.metadata }
     });
   }
@@ -1952,6 +1954,7 @@ function make5(nameOrOptions, options = {}) {
     ...resolved.model === undefined ? {} : { model: resolved.model },
     ...resolved.memory === undefined ? {} : { memory: resolved.memory },
     ...resolved.authorization === undefined ? {} : { authorization: resolved.authorization },
+    ...resolved.toolExecution === undefined ? {} : { toolExecution: resolved.toolExecution },
     ...resolved.metadata === undefined ? {} : { metadata: resolved.metadata },
     toolDeclarations: (declaredTools ?? Object.values(toolkit.tools)).map((tool) => ({
       tool,
@@ -2269,6 +2272,12 @@ var streamInternal = (agent, options, structured) => Stream4.unwrap(Effect18.gen
     });
   }
   const progressPolicy = decodedProgressPolicy.value;
+  if (agent.toolExecution !== undefined && (!Number.isSafeInteger(agent.toolExecution.concurrency) || agent.toolExecution.concurrency <= 0)) {
+    return yield* AgentError.make({
+      message: "Agent.toolExecution.concurrency must be a positive safe integer",
+      turn: 0
+    });
+  }
   if (options.compaction?.contextWindow !== undefined && (!Number.isFinite(options.compaction.contextWindow) || options.compaction.contextWindow <= 0)) {
     return yield* AgentError.make({
       message: "RunOptions.compaction.contextWindow must be a positive finite number",
@@ -2598,13 +2607,9 @@ var streamInternal = (agent, options, structured) => Stream4.unwrap(Effect18.gen
     const metadata = droppedProgress === 0 ? {} : { metadata: { toolProgress: { dropped: droppedProgress } } };
     switch (outcome._tag) {
       case "Success":
-        return (isSkillActivationCall(call, registry) ? Effect18.succeed(successResult(call, outcome)) : boundedSuccessResult(call, outcome)).pipe(Effect18.map((result) => {
-          state.pending.push(result);
-          return Stream4.fromIterable([{ _tag: "ToolExecutionCompleted", turn, call, result, ...metadata }]);
-        }));
+        return (isSkillActivationCall(call, registry) ? Effect18.succeed(successResult(call, outcome)) : boundedSuccessResult(call, outcome)).pipe(Effect18.map((result) => Stream4.fromIterable([{ _tag: "ToolExecutionCompleted", turn, call, result, ...metadata }])));
       case "DomainFailure": {
         const result = domainFailureResult(call, outcome);
-        state.pending.push(result);
         return Effect18.succeed(Stream4.fromIterable([{ _tag: "ToolExecutionCompleted", turn, call, result, ...metadata }]));
       }
       case "Suspend":
@@ -2841,6 +2846,10 @@ var streamInternal = (agent, options, structured) => Stream4.unwrap(Effect18.gen
     }
     return modelPart;
   };
+  const recordPending = (event) => Effect18.sync(() => {
+    if (event._tag === "ToolExecutionCompleted")
+      state.pending.push(event.result);
+  });
   const transformPart = (turn, part) => applyPartChain(chain, part, { agentName: agent.name, turn }).pipe(Effect18.flatMap(Option12.match({
     onSome: (transformed) => Effect18.succeed(Option12.some(transformed)),
     onNone: () => part.type === "tool-call" ? Effect18.fail(MiddlewareViolation.make({
@@ -2931,7 +2940,13 @@ var streamInternal = (agent, options, structured) => Stream4.unwrap(Effect18.gen
         return Option12.isSome(failure2) && AiError3.isAiError(failure2.value) ? Stream4.fail(AgentError.make({ message: errorMessage(failure2.value), turn, cause: failure2.value })) : Stream4.failCause(cause);
       }));
     };
-    const parts = Stream4.unwrap(applyPromptChain(chain, Prompt9.make(prompt), { agentName: agent.name, turn }).pipe(Effect18.map((transformedPrompt) => attempt(transformedPrompt, true).pipe(Stream4.flatMap(({ accept, part, messages }) => Stream4.fromEffect(accept).pipe(Stream4.flatMap(() => partEvents(turn, part, messages, activeRegistry))))))));
+    const parts = Stream4.unwrap(applyPromptChain(chain, Prompt9.make(prompt), { agentName: agent.name, turn }).pipe(Effect18.map((transformedPrompt) => attempt(transformedPrompt, true).pipe(Stream4.mapEffect(({ accept, part, messages }) => accept.pipe(Effect18.as({ part, messages }))), (accepted) => {
+      const concurrency = agent.toolExecution?.concurrency ?? 1;
+      if (concurrency === 1) {
+        return accepted.pipe(Stream4.flatMap(({ part, messages }) => partEvents(turn, part, messages, activeRegistry)), Stream4.tap(recordPending));
+      }
+      return accepted.pipe(Stream4.mapEffect(({ part, messages }) => Stream4.runCollect(partEvents(turn, part, messages, activeRegistry)), { concurrency }), Stream4.flatMap(Stream4.fromIterable), Stream4.tap(recordPending));
+    }))));
     const resilientParts = Option12.match(resilienceService, {
       onNone: () => parts,
       onSome: (resilience) => Stream4.unwrap(LanguageModel5.LanguageModel.pipe(Effect18.map((model) => parts.pipe(Stream4.provideService(LanguageModel5.LanguageModel, apply(model, {
@@ -3088,7 +3103,7 @@ var streamInternal = (agent, options, structured) => Stream4.unwrap(Effect18.gen
         name: checkpoint.call.name,
         params: checkpoint.call.params,
         providerExecuted: false
-      }), checkpoint.messages, registry, suspension2.authorization_stage, suspension2.token);
+      }), checkpoint.messages, registry, suspension2.authorization_stage, suspension2.token).pipe(Stream4.tap(recordPending));
     })))), Stream4.concat(Stream4.unwrap(afterTurn(0).pipe(Effect18.map((result) => {
       next = result.next;
       return result.events;
@@ -3185,7 +3200,7 @@ var generatePersistedObject = dual7(2, (agent, options) => Stream4.runFold(persi
   })
 }))));
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/agent-tool.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/agent-tool.js
 var exports_agent_tool = {};
 __export(exports_agent_tool, {
   asTool: () => asTool
@@ -3256,7 +3271,7 @@ var asTool = Function7.dual((args) => args.length !== 1 || ("name" in args[0]),
   return lazyHandled(toolkit, name, handler);
 });
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/guardrail.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/guardrail.js
 var exports_guardrail = {};
 __export(exports_guardrail, {
   validateInput: () => validateInput,
@@ -3337,7 +3352,7 @@ var filterOutput = (keep) => ({
   transformPart: (part, context) => Effect20.succeed(part.type === "tool-call" || keep(part, context) ? Option13.some(part) : Option13.none())
 });
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/handoff.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/handoff.js
 var exports_handoff = {};
 __export(exports_handoff, {
   transferTool: () => transferTool,
@@ -3417,7 +3432,7 @@ var supervisor = (options) => {
   };
 };
 
-// ../../node_modules/.bun/@batonfx+core@0.6.1/node_modules/@batonfx/core/dist/index.js
+// ../../node_modules/.bun/@batonfx+core@0.6.2/node_modules/@batonfx/core/dist/index.js
 import { AiError as AiError5, Chat as Chat2, EmbeddingModel, IdGenerator, LanguageModel as LanguageModel7, Model as Model2, Prompt as Prompt13, Response as Response9, ResponseIdTracker, Telemetry as Telemetry2, Tokenizer as Tokenizer3, Tool as Tool10, Toolkit as Toolkit7 } from "effect/unstable/ai";
 
 export { exports_turn_policy, exports_agent_event, exports_tool_context, exports_tool_executor, exports_approvals, exports_memory, exports_session, exports_tool_output, exports_compaction, exports_instructions, exports_model_middleware, exports_model_registry, exports_model_resilience, exports_permissions, exports_skill_source, exports_steering, exports_agent, exports_agent_tool, exports_guardrail, exports_handoff, AiError5 as AiError, Chat2 as Chat, EmbeddingModel, IdGenerator, Model2 as Model, Prompt13 as Prompt, Response9 as Response, ResponseIdTracker, Telemetry2 as Telemetry, Tokenizer3 as Tokenizer, Tool10 as Tool, Toolkit7 as Toolkit };