@smythos/sre 1.7.9 → 1.7.15

package/src/subsystems/ObservabilityManager/Telemetry.service/connectors/OTel/OTelContextRegistry.ts

@@ -0,0 +1,35 @@
+// OTelContextRegistry.ts
+import { trace, Span } from '@opentelemetry/api';
+
+interface ProcessContext {
+    processId: string;
+    agentId: string;
+    rootSpan: Span;
+    currentSpan?: Span;
+}
+
+export class OTelContextRegistry {
+    private static registry = new Map<string, ProcessContext>();
+
+    static key(agentId: string, processId: string) {
+        return `${agentId}:${processId}`;
+    }
+
+    static startProcess(agentId: string, processId: string, span: Span) {
+        this.registry.set(this.key(agentId, processId), { agentId, processId, rootSpan: span });
+    }
+
+    static get(agentId: string, processId: string) {
+        return this.registry.get(this.key(agentId, processId));
+    }
+
+    static update(agentId: string, processId: string, ctx: Partial<ProcessContext>) {
+        const key = this.key(agentId, processId);
+        const current = this.registry.get(key);
+        if (current) this.registry.set(key, { ...current, ...ctx });
+    }
+
+    static endProcess(agentId: string, processId: string) {
+        this.registry.delete(this.key(agentId, processId));
+    }
+}

package/src/subsystems/ObservabilityManager/Telemetry.service/index.ts

@@ -0,0 +1,12 @@
+import { ConnectorService, ConnectorServiceProvider } from '@sre/Core/ConnectorsService';
+import { TConnectorService } from '@sre/types/SRE.types';
+
+import { OTel } from './connectors/OTel/OTel.class';
+
+//const console = Logger('LogService');
+
+export class TelemetryService extends ConnectorServiceProvider {
+    public register() {
+        ConnectorService.register(TConnectorService.Telemetry, 'OTel', OTel);
+    }
+}

package/src/types/LLM.types.ts

@@ -499,6 +499,8 @@ export enum TLLMEvent {
     Interrupted = 'interrupted',
     /** Fallback : emitted when the response is using a fallback model */
     Fallback = 'fallback',
+    /** Requested : emitted when a request is sent to the LLM */
+    Requested = 'requested',
 }
 
 export interface ILLMRequestContext {

package/src/types/SRE.types.ts

@@ -9,11 +9,12 @@ import { AccountService } from '@sre/Security/Account.service';
 import { VaultService } from '@sre/Security/Vault.service';
 import { RouterService } from '@sre/IO/Router.service';
 import { ManagedVaultService } from '@sre/Security/ManagedVault.service';
-import { LogService } from '@sre/IO/Log.service';
+import { LogService } from '@sre/ObservabilityManager/Log.service';
 import { ComponentService } from '@sre/AgentManager/Component.service';
 import { ModelsProviderService } from '@sre/LLMManager/ModelsProvider.service';
 import { CodeService } from '@sre/ComputeManager/Code.service';
 import { SchedulerService } from '@sre/AgentManager/Scheduler.service';
+import { TelemetryService } from '@sre/ObservabilityManager/Telemetry.service';
 
 export type TServiceRegistry = {
     Storage?: StorageService;
@@ -32,6 +33,7 @@ export type TServiceRegistry = {
     ModelsProvider?: ModelsProviderService;
     Code?: CodeService;
     Scheduler?: SchedulerService;
+    Telemetry?: TelemetryService;
 };
 
 export enum TConnectorService {
@@ -51,6 +53,7 @@ export enum TConnectorService {
     ModelsProvider = 'ModelsProvider',
     Code = 'Code',
     Scheduler = 'Scheduler',
+    Telemetry = 'Telemetry',
 }
 
 export type SREConnectorConfig = {