@relayfx/sdk 0.0.40 → 0.0.42

package/dist/types/ai/index.d.ts

@@ -1,3 +1,4 @@
 export * as EmbeddingModelRegistration from "./embedding-model/embedding-model-registration";
 export * as LanguageModelRegistration from "./language-model/language-model-registration";
+export { Agent, AgentEvent, AgentTool, AiError, Approvals, Chat, Compaction, EmbeddingModel, Guardrail, Handoff, IdGenerator, Instructions, LanguageModel, Memory, Model, ModelMiddleware, ModelRegistry, ModelResilience, Permissions, Prompt, Response, ResponseIdTracker, Session, SkillSource, Steering, Telemetry, Tokenizer, Tool, ToolContext, ToolExecutor, Toolkit, ToolOutput, TurnPolicy, } from "@batonfx/core";
 export declare const aiPackage = "./index";

package/dist/types/relay/ai.d.ts

@@ -0,0 +1,2 @@
+export { Agent, AgentEvent, AgentTool, AiError, Approvals, Chat, Compaction, EmbeddingModel, Guardrail, Handoff, IdGenerator, Instructions, LanguageModel, Memory, Model, ModelMiddleware, ModelRegistry, ModelResilience, Permissions, Prompt, Response, ResponseIdTracker, Session, SkillSource, Steering, Telemetry, Tokenizer, Tool, ToolContext, ToolExecutor, Toolkit, ToolOutput, TurnPolicy, } from "../ai/index";
+export declare const relayAiPackage = "@relayfx/sdk/ai";

package/dist/types/relay/client.d.ts

@@ -1,5 +1,6 @@
 import { AddressBook, AgentRegistry, EventLog, ExecutionService, SkillRegistry, WaitService } from "../runtime/index";
-import { Address, Agent, Content, Envelope, Execution, Ids, Shared, Skill } from "../schema/index";
+import type { Agent as BatonAgent } from "@batonfx/core";
+import { Address, Agent, Content, Envelope, Execution, Ids, Shared, Skill, Tool } from "../schema/index";
 import { ClusterRegistryRepository, EnvelopeRepository, ExecutionRepository, ScheduleRepository, SessionRepository, SteeringRepository } from "../store-sql/index";
 import { Context, Effect, Layer, Schema, Stream } from "effect";
 import { Sharding, ShardingConfig } from "effect/unstable/cluster";
@@ -34,9 +35,26 @@ export interface StartExecutionByAgentDefinitionInput {
     readonly wait_id?: Ids.WaitId;
     readonly metadata?: Shared.Metadata;
 }
-export interface RegisterAgentInput extends Agent.DefineInput {
+export interface RegisterDefinedAgentInput extends Agent.DefineInput {
     readonly address?: Ids.AddressId;
 }
+export interface RegisterBatonAgentInput {
+    readonly id: Ids.AgentId;
+    readonly address?: Ids.AddressId;
+    readonly agent: BatonAgent.Agent<any>;
+    readonly model: Agent.ModelSelection;
+    readonly permissions?: ReadonlyArray<Tool.Permission>;
+    readonly skill_definition_ids?: ReadonlyArray<Ids.SkillDefinitionId>;
+    readonly permission_rules?: Agent.PermissionRuleset;
+    readonly max_tool_turns?: number;
+    readonly max_wait_turns?: number;
+    readonly token_budget?: number;
+    readonly child_run_presets?: Readonly<Record<string, Agent.ChildRunPreset>>;
+    readonly handoff_targets?: ReadonlyArray<Agent.HandoffTarget>;
+    readonly output_schema_ref?: string;
+    readonly metadata?: Shared.Metadata;
+}
+export type RegisterAgentInput = RegisterDefinedAgentInput | RegisterBatonAgentInput;
 export interface Interface {
     readonly registerAgent: (input: RegisterAgentInput) => Effect.Effect<Agent.DefinitionRegistered, ClientError>;
     readonly registerAgentDefinition: (input: Agent.RegisterDefinitionPayload) => Effect.Effect<Agent.DefinitionRegistered, ClientError>;

package/dist/types/runtime/tool/tool-runtime-service.d.ts

@@ -1,6 +1,6 @@
 import { Execution, Ids, Shared, Tool } from "../../schema/index";
-import { ToolCallRepository } from "../../store-sql/index";
-import { Tool as AiTool } from "effect/unstable/ai";
+import { EnvelopeRepository, ScheduleRepository, ToolCallRepository } from "../../store-sql/index";
+import { Tool as AiTool, Toolkit } from "effect/unstable/ai";
 import { Context, Effect, Layer, Schema } from "effect";
 import { Service as EventLogService } from "../execution/event-log-service";
 import { Service as WaitServiceService } from "../wait/wait-service";
@@ -53,7 +53,6 @@ declare const ToolCallInfo_base: Context.ServiceClass<ToolCallInfo, "@relayfx/ru
 export declare class ToolCallInfo extends ToolCallInfo_base {
 }
 export interface RegisteredTool {
-    readonly name: Shared.NonEmptyString;
     readonly definition: Tool.Definition;
     readonly tool: AiTool.Any;
     readonly requiredPermissions?: ReadonlyArray<string>;
@@ -82,6 +81,14 @@ export interface DynamicToolOptions {
     readonly metadata?: Shared.Metadata | undefined;
     readonly run: (input: Shared.JsonValue, context: ToolExecutionContext) => Effect.Effect<Shared.JsonValue, unknown, any>;
 }
+export interface ToolkitToolOptions {
+    readonly permissions?: ReadonlyArray<Tool.Permission> | undefined;
+    readonly requiredPermissions?: ReadonlyArray<string> | undefined;
+    readonly requirements?: ReadonlyArray<Tool.RuntimeRequirement> | undefined;
+    readonly needsApproval?: boolean | undefined;
+    readonly metadata?: Shared.Metadata | undefined;
+}
+export type ToolkitRegistrationOptions = ToolkitToolOptions | ((tool: AiTool.Any) => ToolkitToolOptions);
 export interface RunInput {
     readonly executionId: Ids.ExecutionId;
     readonly call: Tool.Call;
@@ -101,14 +108,13 @@ declare const Service_base: Context.ServiceClass<Service, "@relayfx/runtime/Tool
 export declare class Service extends Service_base {
 }
 export declare const toolFromRegistered: (tool: RegisteredTool) => AiTool.Any;
-export declare const tool: <const Name extends string, Input extends Schema.Constraint, Output extends Schema.Constraint>(name: Name, options: ToolOptions<Input, Output>) => RegisteredTool & {
-    readonly name: Name;
-};
-export declare const dynamicTool: <const Name extends string>(name: Name, options: DynamicToolOptions) => RegisteredTool & {
-    readonly name: Name;
-};
+export declare const toolsFromToolkit: <Tools extends Record<string, AiTool.Any>>(toolkit: Toolkit.WithHandler<Tools>, options?: ToolkitRegistrationOptions) => ReadonlyArray<RegisteredTool>;
+export declare const tool: <const Name extends string, Input extends Schema.Constraint, Output extends Schema.Constraint>(name: Name, options: ToolOptions<Input, Output>) => RegisteredTool;
+export declare const dynamicTool: <const Name extends string>(name: Name, options: DynamicToolOptions) => RegisteredTool;
 export declare const layer: (initialTools?: ReadonlyArray<RegisteredTool>) => Layer.Layer<Service, never, ToolCallRepository.Service | EventLogService | WaitServiceService>;
+export declare const layerFromToolkit: <Tools extends Record<string, AiTool.Any>>(toolkit: Toolkit.Toolkit<Tools>, options?: ToolkitRegistrationOptions) => Layer.Layer<Service, never, ToolCallRepository.Service | EventLogService | WaitServiceService | Exclude<AiTool.HandlersFor<Tools>, import("effect/Scope").Scope>>;
 export declare const memoryLayer: (tools?: ReadonlyArray<RegisteredTool>) => Layer.Layer<Service, never, never>;
+export declare const memoryLayerFromToolkit: <Tools extends Record<string, AiTool.Any>>(toolkit: Toolkit.Toolkit<Tools>, options?: ToolkitRegistrationOptions) => Layer.Layer<Service, never, Exclude<Exclude<Exclude<Exclude<Exclude<Exclude<AiTool.HandlersFor<Tools>, import("effect/Scope").Scope>, WaitServiceService>, ToolCallRepository.Service>, EnvelopeRepository.Service>, ScheduleRepository.Service>, EventLogService>>;
 export declare const testLayer: (implementation: Interface) => Layer.Layer<Service, never, never>;
 export declare const register: (registeredTool: RegisteredTool) => Effect.Effect<void, never, Service>;
 export declare const definitions: () => Effect.Effect<readonly Tool.Definition[], never, Service>;

package/package.json

@@ -1,13 +1,17 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "@relayfx/sdk",
-  "version": "0.0.40",
+  "version": "0.0.42",
   "type": "module",
   "exports": {
     ".": {
       "types": "./dist/types/relay/index.d.ts",
       "import": "./dist/index.js"
     },
+    "./ai": {
+      "types": "./dist/types/ai/index.d.ts",
+      "import": "./dist/ai.js"
+    },
     "./migrations/*": "./dist/migrations/*"
   },
   "types": "./dist/types/relay/index.d.ts",
@@ -15,14 +19,14 @@
     "dist"
   ],
   "scripts": {
-    "build": "bun build ./src/index.ts --outdir dist --target bun --format esm --external effect --external 'effect/*' --external '@effect/*' --external drizzle-orm --external 'drizzle-orm/*' --external pg && bun scripts/build-types.ts",
+    "build": "bun build ./src/index.ts ./src/ai.ts --outdir dist --target bun --format esm --external effect --external 'effect/*' --external '@effect/*' --external drizzle-orm --external 'drizzle-orm/*' --external pg && bun scripts/build-types.ts",
     "lint": "bun --cwd ../.. oxlint packages/relay/src packages/relay/scripts packages/relay/test",
     "test": "bun --cwd ../.. vitest run packages/relay/test --passWithNoTests",
     "typecheck": "bun tsc --noEmit"
   },
   "dependencies": {
-    "@batonfx/core": "0.2.0",
-    "@batonfx/providers": "0.2.0",
+    "@batonfx/core": "0.3.1",
+    "@batonfx/providers": "0.3.1",
     "@effect/ai-anthropic": "4.0.0-beta.93",
     "@effect/ai-openai": "4.0.0-beta.93",
     "@effect/ai-openai-compat": "4.0.0-beta.93",