monitoring

Author: ilvalerione

src/Observability/AgentMonitoring.php

@@ -235,15 +235,15 @@ protected function getContext(Agent $agent): array
         ];
     }
 
-    public function getMessageId(Message $message): string
+    public function getMessageId(Message $message, string $prefix = ''): string
     {
         $content = $message->getContent();
 
-        if (!\is_string($content)) {
-            $content = \json_encode($content, \JSON_UNESCAPED_UNICODE);
+        if (\is_array($content)) {
+            $content = \json_encode($content);
         }
 
-        return \md5($content.$message->getRole());
+        return $prefix.\md5($content.$message->getRole());
     }
 
     protected function getBaseClassName(string $class): string

src/Observability/HandleInferenceEvents.php

@@ -21,14 +21,14 @@ public function messageSaving(Agent $agent, string $event, MessageSaving $data):
 
         $label = $this->getBaseClassName($data->message::class);
 
-        $this->segments[$this->getMessageId($data->message).'-save'] = $this->inspector
+        $this->segments[$this->getMessageId($data->message, 'save')] = $this->inspector
             ->startSegment(self::SEGMENT_TYPE.'.chathistory', "save_message( {$label} )")
             ->setColor(self::STANDARD_COLOR);
     }
 
     public function messageSaved(Agent $agent, string $event, MessageSaved $data): void
     {
-        $id = $this->getMessageId($data->message).'-save';
+        $id = $this->getMessageId($data->message, 'save');
 
         if (!\array_key_exists($id, $this->segments)) {
             return;
@@ -45,6 +45,7 @@ public function messageSaved(Agent $agent, string $event, MessageSaved $data): v
             ] : []
         ));
         $segment->end();
+        unset($this->segments[$id]);
     }
 
     public function inferenceStart(Agent $agent, string $event, InferenceStart $data): void
@@ -55,19 +56,22 @@ public function inferenceStart(Agent $agent, string $event, InferenceStart $data
 
         $label = $this->getBaseClassName($data->message::class);
 
-        $this->segments[$this->getMessageId($data->message).'-inference'] = $this->inspector
+        $this->segments[$this->getMessageId($data->message, 'inference')] = $this->inspector
             ->startSegment(self::SEGMENT_TYPE.'.inference', "inference( {$label} )")
             ->setColor(self::STANDARD_COLOR);
     }
 
     public function inferenceStop(Agent $agent, string $event, InferenceStop $data): void
     {
-        $id = $this->getMessageId($data->message).'-inference';
+        $id = $this->getMessageId($data->message, 'inference');
 
-        if (\array_key_exists($id, $this->segments)) {
-            $segment = $this->segments[$id]->end();
-            $segment->addContext('Message', $data->message)
-                ->addContext('Response', $data->response);
+        if (!\array_key_exists($id, $this->segments)) {
+            return;
         }
+
+        $segment = $this->segments[$id]->end();
+        $segment->addContext('Message', $data->message)
+            ->addContext('Response', $data->response);
+        unset($this->segments[$id]);
     }
 }

src/Observability/HandleStructuredEvents.php

@@ -40,30 +40,32 @@ protected function extracting(AgentInterface $agent, string $event, Extracting $
             return;
         }
 
-        $id = $this->getMessageId($data->message).'-extract';
+        $id = $this->getMessageId($data->message, 'extract');
 
         $this->segments[$id] = $this->inspector->startSegment(self::SEGMENT_TYPE.'.structured-output', 'extract_output')
             ->setColor(self::STANDARD_COLOR);
     }
 
     protected function extracted(AgentInterface $agent, string $event, Extracted $data): void
     {
-        $id = $this->getMessageId($data->message).'-extract';
+        $id = $this->getMessageId($data->message, 'extract');
 
-        if (\array_key_exists($id, $this->segments)) {
-            $segment = $this->segments[$id]->end();
-            $segment->addContext(
-                'Data',
-                [
-                    'response' => $data->message->jsonSerialize(),
-                    'json' => $data->json,
-                ]
-            )->addContext(
-                'Schema',
-                $data->schema
-            );
-            unset($this->segments[$id]);
+        if (!\array_key_exists($id, $this->segments)) {
+            return;
         }
+
+        $segment = $this->segments[$id]->end();
+        $segment->addContext(
+            'Data',
+            [
+                'response' => $data->message->jsonSerialize(),
+                'json' => $data->json,
+            ]
+        )->addContext(
+            'Schema',
+            $data->schema
+        );
+        unset($this->segments[$id]);
     }
 
     protected function deserializing(AgentInterface $agent, string $event, Deserializing $data): void