diff --git a/src/ai-bundle/src/Profiler/DataCollector.php b/src/ai-bundle/src/Profiler/DataCollector.php
index e3041a2d6..546ecdabd 100644
--- a/src/ai-bundle/src/Profiler/DataCollector.php
+++ b/src/ai-bundle/src/Profiler/DataCollector.php
@@ -95,7 +95,7 @@ public function getToolCalls(): array
 
     /**
      * @return array{
-     *     model: Model,
+     *     model: string,
      *     input: array<mixed>|string|object,
      *     options: array<string, mixed>,
      *     result: string|iterable<mixed>|object|null