agentworkshop 0.7.49 → 0.7.50

package/server/services/workshop/aml/twin/provider-health.ts

@@ -1,8 +1,69 @@
-import { sha256 } from './contracts'
-import type { TwinPhysicsProvider, PhysicsProviderManifest, ProviderHealth, ProviderValidationResult, TwinProviderHealth } from './provider-contracts'
-export function normalizeProviderHealth(value:ProviderHealth|undefined):ProviderHealth{return value??{status:'healthy',checkedAt:new Date().toISOString(),detail:'not checked'}}
-export function providerHealthAllowsUse(value:ProviderHealth|undefined){return Boolean(value)&&value.status!=='unhealthy'}
-export async function checkProviderHealth(provider:TwinPhysicsProvider):Promise<ProviderHealth>{try{if(typeof provider.health==='function')return normalizeProviderHealth(await provider.health());return{status:'healthy',checkedAt:new Date().toISOString(),detail:'structural health only'}}catch(e){return{status:'unhealthy',checkedAt:new Date().toISOString(),detail:e instanceof Error?e.message:String(e),errors:['PROVIDER_HEALTH_FAILED']}}}
-export function providerHashOf(v:TwinPhysicsProvider|PhysicsProviderManifest){return sha256('manifest'in v?v.manifest:v)}
-export function validateProviderDefinition(provider:TwinPhysicsProvider):ProviderValidationResult{const errors:string[]=[];const warnings:string[]=[];const raw=(provider as any)?.manifest??{};const providerId=String(raw.providerId??raw.physicsModelId??'');const version=String(raw.version??raw.providerVersion??'');const sceneKinds=Array.isArray(raw.sceneKinds)?raw.sceneKinds.map(String):raw.sceneId?[String(raw.sceneId)]:[];if(!providerId)errors.push('PROVIDER_ID_MISSING');if(!version)errors.push('PROVIDER_VERSION_MISSING');if(raw.apiVersion&&raw.apiVersion!=='twin-provider.v1')errors.push('PROVIDER_API_VERSION_UNSUPPORTED');if(!sceneKinds.length)errors.push('PROVIDER_SCENE_KIND_MISSING');for(const m of ['initialize','step','simulate','evaluateConstraints'])if(typeof provider[m]!=='function')errors.push(`PROVIDER_METHOD_MISSING:${m}`);const manifest=errors.length?undefined:{providerId,version,apiVersion:'twin-provider.v1' as const,sceneKinds,backend:raw.backend==='pytorch'?'pytorch':'typescript',stateVariables:Array.isArray(raw.stateVariables)?raw.stateVariables.map((id:string)=>({id})):[],controlVariables:Array.isArray(raw.controlVariables)?raw.controlVariables.map((id:string)=>({id})):[],disturbanceVariables:Array.isArray(raw.disturbanceVariables)?raw.disturbanceVariables.map((id:string)=>({id})):[],parameterPriors:raw.parameterPriors??{},capabilities:raw.capabilities??{},artifactContract:raw.artifactContract};return{valid:errors.length===0,errors,warnings,manifest,providerHash:manifest?providerHashOf(manifest):undefined}}
-export async function healthOf(provider:TwinPhysicsProvider,hash?:string):Promise<TwinProviderHealth>{const m:any=provider.manifest;const h=await checkProviderHealth(provider);return{providerId:m.providerId??m.physicsModelId,providerVersion:m.providerVersion??m.version,status:h.status==='unhealthy'?'failed':h.status,checkedAt:h.checkedAt??new Date().toISOString(),detail:h.detail??`hash=${hash??providerHashOf(provider)}`,errors:h.errors}}
+import type { ProviderHealth, ProviderHealthStatus, TwinPhysicsProvider } from './provider-contracts'
+
+export interface ProviderHealthOptions { timeoutMs?: number, now?: () => number }
+
+function isRecord(value: unknown): value is Record<string, unknown> {
+  return Boolean(value) && typeof value === 'object'
+}
+function status(value: unknown): ProviderHealthStatus {
+  const raw = String(value ?? 'unknown').toLowerCase()
+  if (raw === 'healthy' || raw === 'degraded' || raw === 'unhealthy' || raw === 'unknown') return raw
+  if (raw === 'ok' || raw === 'ready' || raw === 'pass' || raw === 'passed') return 'healthy'
+  if (raw === 'warn' || raw === 'warning') return 'degraded'
+  if (raw === 'failed' || raw === 'fail' || raw === 'error') return 'unhealthy'
+  return 'unknown'
+}
+
+export function normalizeProviderHealth(value: unknown, checkedAt = new Date().toISOString()): ProviderHealth {
+  if (value === true) return { status: 'healthy', checkedAt, detail: 'Provider health check passed' }
+  if (value === false) return { status: 'unhealthy', checkedAt, detail: 'Provider health check failed' }
+  if (!isRecord(value)) return { status: 'unknown', checkedAt, detail: 'Provider does not expose a health result' }
+  return {
+    ...value,
+    status: status(value.status ?? value.state),
+    checkedAt: typeof value.checkedAt === 'string' && value.checkedAt ? value.checkedAt : checkedAt,
+    detail: value.detail == null ? (value.message == null ? undefined : String(value.message)) : String(value.detail),
+    latencyMs: Number.isFinite(Number(value.latencyMs)) ? Number(value.latencyMs) : undefined,
+  }
+}
+
+function withTimeout<T>(promise: Promise<T>, timeoutMs: number): Promise<T> {
+  if (!Number.isFinite(timeoutMs) || timeoutMs <= 0) return promise
+  return new Promise<T>((resolve, reject) => {
+    const timer = setTimeout(() => reject(new Error('Provider health check timed out')), timeoutMs)
+    promise.then(
+      (value) => {
+        clearTimeout(timer)
+        resolve(value)
+      },
+      (error) => {
+        clearTimeout(timer)
+        reject(error)
+      },
+    )
+  })
+}
+
+export async function checkProviderHealth(provider: TwinPhysicsProvider, options: ProviderHealthOptions = {}): Promise<ProviderHealth> {
+  const started = options.now?.() ?? Date.now()
+  const checkedAt = new Date(started).toISOString()
+  const hook = (provider as unknown as { health?: unknown }).health
+  if (hook == null) return { status: 'unknown', checkedAt, detail: 'Provider health hook is not declared' }
+  try {
+    const raw = typeof hook === 'function' ? (hook as () => unknown)() : hook
+    const result = normalizeProviderHealth(await withTimeout(Promise.resolve(raw), options.timeoutMs ?? 0), checkedAt)
+    if (result.latencyMs == null) result.latencyMs = Math.max(0, (options.now?.() ?? Date.now()) - started)
+    return result
+  }
+  catch (error) {
+    const detail = error instanceof Error ? error.message : String(error)
+    return { status: 'unhealthy', checkedAt, detail, errors: [detail], latencyMs: Math.max(0, (options.now?.() ?? Date.now()) - started) }
+  }
+}
+
+export function providerHealthAllowsUse(health: ProviderHealth): boolean {
+  return health.status !== 'unhealthy'
+}
+export function providerHealthIsHealthy(health: ProviderHealth): boolean {
+  return health.status === 'healthy'
+}

package/server/services/workshop/aml/twin/provider-loader.ts

@@ -1,46 +1,80 @@
+/* eslint-disable @typescript-eslint/no-explicit-any, @stylistic/max-statements-per-line, @typescript-eslint/no-unused-expressions */
 import { getTwinProviderRegistry, type TwinProviderRegistry } from './provider-registry'
-import type { TwinPluginContribution, TwinPluginExtBridge, TwinPluginBridgeOperation, TwinProviderFilter, TwinProviderSummary, TwinRegistrationSource, TwinPhysicsProvider, ScenePack, SolverAdapter } from './provider-contracts'
+import type { TwinPluginContribution, TwinPluginExtBridge, TwinPluginBridgeOperation, TwinProviderFilter, TwinProviderSummary, TwinRegistrationSource, TwinPhysicsProvider, ScenePack, SolverAdapter } from './provider-contracts'
+
 export interface TwinPluginBridgeGlobal { __twinPluginExt?: any }
-function sourceName(source: TwinRegistrationSource|string|undefined): { pluginName:string, pluginVersion?:string, source?:string } { if(typeof source==='string') return {pluginName:source}; return {pluginName:String(source?.pluginName??source?.plugin??source?.name??'external-twin-plugin'),pluginVersion:source?.pluginVersion??source?.version,source:source?.source??source?.scope} }
-function normalizePack(payload:any): ScenePack { return { ...payload, sceneKind: payload.sceneKind ?? payload.scenePackId ?? 'unknown-scene', sceneSchemaVersion: payload.sceneSchemaVersion ?? payload.version ?? '1.0.0' } }
+function sourceName(source: TwinRegistrationSource | string | undefined): { pluginName: string, pluginVersion?: string, source?: string } { if (typeof source === 'string') return { pluginName: source }; return { pluginName: String(source?.pluginName ?? source?.plugin ?? source?.name ?? 'external-twin-plugin'), pluginVersion: source?.pluginVersion ?? source?.version, source: source?.source ?? source?.scope } }
+function normalizePack(payload: any): ScenePack { return { ...payload, sceneKind: payload.sceneKind ?? payload.scenePackId ?? 'unknown-scene', sceneSchemaVersion: payload.sceneSchemaVersion ?? payload.version ?? '1.0.0' } }
 function registryApi(registry: TwinProviderRegistry) {
   return {
-    register(kind:string, payload:unknown, source?:TwinRegistrationSource) {
-      const s=sourceName(source)
-      if(kind==='registerPhysicsProvider') return registry.registerPhysicsProvider(payload as TwinPhysicsProvider, s)
-      if(kind==='registerScenePack') return registry.registerScenePack(normalizePack(payload as Record<string, unknown>), s)
-      if(kind==='registerSolverAdapter') return registry.registerSolverAdapter(payload as SolverAdapter, s)
-      if(kind==='registerObjectiveProfile' || kind==='registerTrainingAdapter') return { ok:true, accepted:true, capability:kind }
+    register(kind: string, payload: unknown, source?: TwinRegistrationSource) {
+      const s = sourceName(source)
+      if (kind === 'registerPhysicsProvider') return registry.registerPhysicsProvider(payload as TwinPhysicsProvider, s)
+      if (kind === 'registerScenePack') return registry.registerScenePack(normalizePack(payload as Record<string, unknown>), s)
+      if (kind === 'registerSolverAdapter') return registry.registerSolverAdapter(payload as SolverAdapter, s)
+      if (kind === 'registerObjectiveProfile' || kind === 'registerTrainingAdapter') return { ok: true, accepted: true, capability: kind }
       throw new Error(`Unknown Twin registration kind: ${kind}`)
     },
-    registerEntry(entry:{ kind:string, payload:unknown, source?:TwinRegistrationSource }) { return this.register(entry.kind, entry.payload, entry.source) },
-    registerTwinPluginContribution(pluginName:string, contribution:TwinPluginContribution, generation?:number) { return registry.registerTwinPluginContribution(pluginName, contribution, generation) },
-    unregisterTwinPluginContribution(pluginName:string, generation?:number) { return registry.unregisterTwinPluginContribution(pluginName, generation) },
-    registerPhysicsProvider(payload: TwinPhysicsProvider, source?: TwinRegistrationSource) { const s=sourceName(source); return registry.registerTwinPluginContribution(s.pluginName,{providers:[payload]},undefined) },
-    registerScenePack(payload: any, source?: TwinRegistrationSource) { const s=sourceName(source); return registry.registerTwinPluginContribution(s.pluginName,{scenePacks:[normalizePack(payload)]},undefined) },
-    registerObjectiveProfile(_payload: unknown, _source?: TwinRegistrationSource) { return { ok:true, accepted:true, capability:'objective-profile' } },
-    registerTrainingAdapter(_payload: unknown, _source?: TwinRegistrationSource) { return { ok:true, accepted:true, capability:'training-adapter' } },
-    registerSolverAdapter(payload: SolverAdapter, source?: TwinRegistrationSource) { const s=sourceName(source); return registry.registerTwinPluginContribution(s.pluginName,{solverAdapters:[payload]},undefined) },
+    registerEntry(entry: { kind: string, payload: unknown, source?: TwinRegistrationSource }) { return this.register(entry.kind, entry.payload, entry.source) },
+    registerTwinPluginContribution(pluginName: string, contribution: TwinPluginContribution, generation?: number) { return registry.registerTwinPluginContribution(pluginName, contribution, generation) },
+    unregisterTwinPluginContribution(pluginName: string, generation?: number) { return registry.unregisterTwinPluginContribution(pluginName, generation) },
+    registerPhysicsProvider(payload: TwinPhysicsProvider, source?: TwinRegistrationSource) { const s = sourceName(source); return registry.registerTwinPluginContribution(s.pluginName, { providers: [payload] }, undefined) },
+    registerScenePack(payload: any, source?: TwinRegistrationSource) { const s = sourceName(source); return registry.registerTwinPluginContribution(s.pluginName, { scenePacks: [normalizePack(payload)] }, undefined) },
+    registerObjectiveProfile(_payload: unknown, _source?: TwinRegistrationSource) { return { ok: true, accepted: true, capability: 'objective-profile' } },
+    registerTrainingAdapter(_payload: unknown, _source?: TwinRegistrationSource) { return { ok: true, accepted: true, capability: 'training-adapter' } },
+    registerSolverAdapter(payload: SolverAdapter, source?: TwinRegistrationSource) { const s = sourceName(source); return registry.registerTwinPluginContribution(s.pluginName, { solverAdapters: [payload] }, undefined) },
     listProviders(filter?: TwinProviderFilter): TwinProviderSummary[] { return registry.listProviders(filter) },
     listTwinProviders(filter?: TwinProviderFilter): TwinProviderSummary[] { return registry.listProviders(filter) },
-    getProviderHealth(providerId:string, version?:string) { return registry.getProviderHealth(providerId,version) },
-    resolveProvider(providerId:string, version?:string) { return registry.resolveProvider(providerId,version) },
-    validateProvider(providerId:string, version?:string) { return registry.validateProvider(providerId,version) },
-    retireProvider(providerId:string, version?:string) { return registry.retireProvider(providerId,version??'',undefined) },
-    unregisterPlugin(source?: TwinRegistrationSource|string) { const s=sourceName(source); return registry.unregisterTwinPluginContribution(s.pluginName) },
-    removePlugin(source?: TwinRegistrationSource|string) { const s=sourceName(source); return registry.unregisterTwinPluginContribution(s.pluginName) },
+    getProviderHealth(providerId: string, version?: string) { return registry.getProviderHealth(providerId, version) },
+    resolveProvider(providerId: string, version?: string) { return registry.resolveProvider(providerId, version) },
+    validateProvider(providerId: string, version?: string) { return registry.validateProvider(providerId, version) },
+    retireProvider(providerId: string, version?: string) { return registry.retireProvider(providerId, version ?? '', undefined) },
+    unregisterPlugin(source?: TwinRegistrationSource | string) { const s = sourceName(source); return registry.unregisterTwinPluginContribution(s.pluginName) },
+    removePlugin(source?: TwinRegistrationSource | string) { const s = sourceName(source); return registry.unregisterTwinPluginContribution(s.pluginName) },
+  }
+}
+function isNativeBridge(value: unknown): value is TwinPluginExtBridge { return Boolean(value && typeof value === 'object' && Array.isArray((value as any).pending) && typeof (value as any).registerTwinPluginContribution === 'function' && typeof (value as any).drain === 'function') }
+export function createTwinPluginBridge(): TwinPluginExtBridge {
+  const pending: TwinPluginBridgeOperation[] = []
+  let drainPending: (() => void) | undefined
+  const bridge: any = {
+    pending,
+    get _drain() { return drainPending },
+    set _drain(value: (() => void) | undefined) { drainPending = value },
+    registerTwinPluginContribution(pluginName: string, contribution: TwinPluginContribution, generation?: number) {
+      pending.push({ op: 'register', pluginName, contribution, generation }); drainPending?.()
+    },
+    unregisterTwinPluginContribution(pluginName: string, generation?: number) {
+      pending.push({ op: 'unregister', pluginName, generation }); drainPending?.()
+    },
+    register(kind: string, payload: unknown, source?: TwinRegistrationSource) {
+      const pluginName = sourceName(source).pluginName
+      if (kind === 'registerPhysicsProvider') return this.registerTwinPluginContribution(pluginName, { providers: [payload] })
+      if (kind === 'registerScenePack') return this.registerTwinPluginContribution(pluginName, { scenePacks: [normalizePack(payload)] })
+      if (kind === 'registerSolverAdapter') return this.registerTwinPluginContribution(pluginName, { solverAdapters: [payload] })
+      return { ok: true, accepted: true, capability: kind }
+    },
+    registerEntry(entry: any) { return this.register(entry.kind, entry.payload, entry.source) },
+    unregisterPlugin(source?: TwinRegistrationSource | string) { return this.unregisterTwinPluginContribution(sourceName(source).pluginName) },
+    removePlugin(source?: TwinRegistrationSource | string) { return this.unregisterPlugin(source) },
+    drain(onRegister: (pluginName: string, c: TwinPluginContribution, generation?: number) => unknown, onUnregister: (pluginName: string, generation?: number) => unknown = () => {}) {
+      drainPending = () => {
+        for (const op of pending.splice(0)) op.op === 'register' ? onRegister(op.pluginName, op.contribution!, op.generation) : onUnregister(op.pluginName, op.generation)
+      }
+      drainPending()
+    },
   }
+  return bridge
 }
-function isNativeBridge(value: unknown): value is TwinPluginExtBridge { return Boolean(value && typeof value==='object' && Array.isArray((value as any).pending) && typeof (value as any).registerTwinPluginContribution==='function' && typeof (value as any).drain==='function') }
-export function createTwinPluginBridge(): TwinPluginExtBridge { const pending:TwinPluginBridgeOperation[]=[]; let drainPending:(()=>void)|undefined; const bridge:TwinPluginExtBridge={pending,get _drain(){return drainPending},set _drain(v){drainPending=v},registerTwinPluginContribution(pluginName,contribution,generation){pending.push({op:'register',pluginName,contribution,generation});drainPending?.()},unregisterTwinPluginContribution(pluginName,generation){pending.push({op:'unregister',pluginName,generation});drainPending?.()},drain(onRegister,onUnregister=()=>{}){drainPending=()=>{for(const op of pending.splice(0))op.op==='register'?onRegister(op.pluginName,op.contribution,op.generation):onUnregister(op.pluginName,op.generation)};drainPending()}};return bridge }
-export function loadTwinPluginContribution(pluginName:string, contribution:TwinPluginContribution,generation?:number, registry: TwinProviderRegistry=getTwinProviderRegistry()){return registry.registerTwinPluginContribution(pluginName,contribution,generation)}
-export function unloadTwinPluginContribution(pluginName:string,generation?:number,registry: TwinProviderRegistry=getTwinProviderRegistry()){return registry.unregisterTwinPluginContribution(pluginName,generation)}
-export function attachTwinPluginBridge(registry: TwinProviderRegistry=getTwinProviderRegistry()): any {
-  const global=globalThis as unknown as TwinPluginBridgeGlobal; const existing=global.__twinPluginExt; const api=registryApi(registry)
+
+export function loadTwinPluginContribution(pluginName: string, contribution: TwinPluginContribution, generation?: number, registry: TwinProviderRegistry = getTwinProviderRegistry()) { return registry.registerTwinPluginContribution(pluginName, contribution, generation) }
+export function unloadTwinPluginContribution(pluginName: string, generation?: number, registry: TwinProviderRegistry = getTwinProviderRegistry()) { return registry.unregisterTwinPluginContribution(pluginName, generation) }
+export function attachTwinPluginBridge(registry: TwinProviderRegistry = getTwinProviderRegistry()): any {
+  const global = globalThis as unknown as TwinPluginBridgeGlobal; const existing = global.__twinPluginExt; const api = registryApi(registry)
   // SDK-created bridge exposes attach(registry): preserve its pending queue and let
   // it dispatch future kind/payload registrations into the typed Registry API.
-  if(existing && typeof existing.attach==='function') { existing.attach(api); return existing }
-  const bridge=isNativeBridge(existing)?existing:createTwinPluginBridge(); global.__twinPluginExt=bridge
+  if (existing && typeof existing.attach === 'function') { existing.attach(api); return existing }
+  const bridge = isNativeBridge(existing) ? existing : createTwinPluginBridge(); global.__twinPluginExt = bridge
   const registerApi = registryApi(registry) as any
   // Make the native bridge compatible with SDK's kind/payload/source dispatch.
   ;(bridge as any).register = (kind: string, payload: unknown, source?: TwinRegistrationSource) => {
@@ -49,19 +83,16 @@ export function attachTwinPluginBridge(registry: TwinProviderRegistry=getTwinPro
     return fn(payload, source)
   }
   ;(bridge as any).registerEntry = (entry: any) => (bridge as any).register(entry?.kind, entry?.payload, entry?.source)
-  ;(bridge as any).unregisterPlugin = (source?: TwinRegistrationSource|string) => registerApi.unregisterPlugin(source)
-  ;(bridge as any).removePlugin = (source?: TwinRegistrationSource|string) => registerApi.unregisterPlugin(source)
-  ;(bridge as any).listProviders = (filter?:TwinProviderFilter)=>registry.listProviders(filter)
-  ;(bridge as any).listTwinProviders = (filter?:TwinProviderFilter)=>registry.listProviders(filter)
-  ;(bridge as any).getProviderHealth = (providerId:string, version?:string)=>registry.getProviderHealth(providerId,version)
-  ;(bridge as any).resolveProvider = (providerId:string, version?:string)=>registry.resolveProvider(providerId,version)
-  ;(bridge as any).validateProvider = (providerId:string, version?:string)=>registry.validateProvider(providerId,version)
-  ;(bridge as any).retireProvider = (providerId:string, version?:string)=>registry.retireProvider(providerId,version??'',undefined)
-  bridge.drain((pluginName,contribution,generation)=>registry.registerTwinPluginContribution(pluginName,contribution,generation),(pluginName,generation)=>registry.unregisterTwinPluginContribution(pluginName,generation))
+  ;(bridge as any).unregisterPlugin = (source?: TwinRegistrationSource | string) => registerApi.unregisterPlugin(source)
+  ;(bridge as any).removePlugin = (source?: TwinRegistrationSource | string) => registerApi.unregisterPlugin(source)
+  ;(bridge as any).listProviders = (filter?: TwinProviderFilter) => registry.listProviders(filter)
+  ;(bridge as any).listTwinProviders = (filter?: TwinProviderFilter) => registry.listProviders(filter)
+  ;(bridge as any).getProviderHealth = (providerId: string, version?: string) => registry.getProviderHealth(providerId, version)
+  ;(bridge as any).resolveProvider = (providerId: string, version?: string) => registry.resolveProvider(providerId, version)
+  ;(bridge as any).validateProvider = (providerId: string, version?: string) => registry.validateProvider(providerId, version)
+  ;(bridge as any).retireProvider = (providerId: string, version?: string) => registry.retireProvider(providerId, version ?? '', undefined)
+  bridge.drain((pluginName, contribution, generation) => registry.registerTwinPluginContribution(pluginName, contribution, generation), (pluginName, generation) => registry.unregisterTwinPluginContribution(pluginName, generation))
   return bridge
 }
-export const attachTwinProviderRegistry=attachTwinPluginBridge
-export const createTwinPluginExt=createTwinPluginBridge
-
-
-
+export const attachTwinProviderRegistry = attachTwinPluginBridge
+export const createTwinPluginExt = createTwinPluginBridge