1 files changed, 11 insertions, 6 deletions
diff --git a/src/Agent.Core/Runtime/IAgentRuntimeObserver.cs b/src/Agent.Core/Runtime/IAgentRuntimeObserver.cs
index e4e26e3..5098a92 100644
--- a/src/Agent.Core/Runtime/IAgentRuntimeObserver.cs
+++ b/src/Agent.Core/Runtime/IAgentRuntimeObserver.cs
@@ -5,15 +5,20 @@ namespace Agent.Core.Runtime;
 
 public interface IAgentRuntimeObserver
 {
-    void LlmRequest(Guid taskId, int iteration, IReadOnlyList<OllamaMessage> messages);
-    void LlmResponse(Guid taskId, int iteration, OllamaResult result);
-    void ToolResult(Guid taskId, string toolName, ToolExecutionResult result);
+    void LlmRequest(AgentLlmCallContext context, IReadOnlyList<OllamaMessage> messages);
+    void LlmResponse(AgentLlmCallContext context, OllamaResult result);
+    void LlmFailure(AgentLlmCallContext context, string errorCode, string message);
+    void ToolResult(AgentLlmCallContext context, string toolName, ToolExecutionResult result);
 }
 
+public sealed record AgentLlmCallContext(Guid CallId, Guid SessionId, Guid TaskId, Guid? PlannerTaskId,
+    string Phase, int Iteration, int Replan, string Model);
+
 public sealed class NullAgentRuntimeObserver : IAgentRuntimeObserver
 {
     public static NullAgentRuntimeObserver Instance { get; } = new();
-    public void LlmRequest(Guid taskId, int iteration, IReadOnlyList<OllamaMessage> messages) { }
-    public void LlmResponse(Guid taskId, int iteration, OllamaResult result) { }
-    public void ToolResult(Guid taskId, string toolName, ToolExecutionResult result) { }
+    public void LlmRequest(AgentLlmCallContext context, IReadOnlyList<OllamaMessage> messages) { }
+    public void LlmResponse(AgentLlmCallContext context, OllamaResult result) { }
+    public void LlmFailure(AgentLlmCallContext context, string errorCode, string message) { }
+    public void ToolResult(AgentLlmCallContext context, string toolName, ToolExecutionResult result) { }
 }