@ai-agent-forge/plugin-sdk 0.86.4 → 0.86.5

package/dist/public-api.d.ts

@@ -1755,6 +1755,40 @@ export interface ProviderHeaderTransformRequest {
  * result.
  */
 export type ProviderHeaderTransform = (request: ProviderHeaderTransformRequest) => Record<string, string | null> | Promise<Record<string, string | null>>;
+/**
+ * Host closure capabilities handed to every {@link StreamFnWrapper} call.
+ * `model`/`context`/`options` stay untyped on this face because the SDK keeps
+ * no dependency on the provider message model; the host and embedded plugins
+ * cast at their boundary.
+ */
+export interface StreamFnWrapperHostV1 {
+    /**
+     * Resolve one complete model object (provider catalog entry: compat,
+     * sampling params, baseUrl) from the host's merged model catalog.
+     * `undefined` when the host does not know the provider/model pair —
+     * callers must treat that as "no failover candidate".
+     */
+    resolveModel(provider: string, modelId: string): unknown | undefined;
+}
+/**
+ * The remainder of the stream-function chain ending at the host's default
+ * stream call. A wrapper calls it with the request it wants issued (same
+ * context, possibly a substituted model and sanitized options) and consumes
+ * the returned event stream.
+ */
+export type StreamFnNext = (model: unknown, context: unknown, options: unknown) => unknown;
+/**
+ * Experimental request-time stream-function middleware (provider-failover
+ * family): observe the provider stream's terminal state, substitute the
+ * request model, and replay. Wrappers run in registration order — the first
+ * registered wrapper is the outermost, its `next` covering every later
+ * wrapper plus the host default. Must return (a promise of) the event stream
+ * the caller consumes; failures belong inside the stream protocol, matching
+ * the host stream-function contract. A session whose caller supplied an
+ * explicit stream function bypasses the host default assembly entirely and
+ * never sees these wrappers.
+ */
+export type StreamFnWrapper = (model: unknown, context: unknown, options: unknown, next: StreamFnNext, host: StreamFnWrapperHostV1) => unknown;
 /**
  * Hook positions on the default agent loop. Links run in registration order
  * after the host's legacy chain head.
@@ -1937,6 +1971,7 @@ export interface CapabilityAPI {
      */
     registerEntryProjection(projection: EntryProjection): DisposableRegistration;
     registerProviderHeaderTransform(name: string, transform: ProviderHeaderTransform): DisposableRegistration;
+    registerStreamFnWrapper(name: string, wrapper: StreamFnWrapper): DisposableRegistration;
     registerSkill(definition: SkillDefinition): DisposableRegistration;
     registerRetryPolicy(name: string, policy: RetryPolicy): DisposableRegistration;
     registerCompactionPolicy(name: string, policy: CompactionPolicy): DisposableRegistration;