1 files changed, 11 insertions, 3 deletions
diff --git a/src/Agent.Core/Runtime/PlannerService.cs b/src/Agent.Core/Runtime/PlannerService.cs
index 110e476..bf896f0 100644
--- a/src/Agent.Core/Runtime/PlannerService.cs
+++ b/src/Agent.Core/Runtime/PlannerService.cs
@@ -48,12 +48,18 @@ public sealed class PlannerService(IOllamaClient ollama, AgentRuntimeSettings se
         catch
         {
             // Fallback: single-task plan with conservative estimates.
-            return CreateFallbackGraph(request);
+            TaskGraph fallback = CreateFallbackGraph(request);
+            session.TaskGraphs[taskId] = fallback;
+            return fallback;
         }
 
         PlanResult? plan = ParsePlan(result.Content);
         if (plan is null || plan.Tasks.Count == 0)
-            return CreateFallbackGraph(request);
+        {
+            TaskGraph fallback = CreateFallbackGraph(request);
+            session.TaskGraphs[taskId] = fallback;
+            return fallback;
+        }
 
         TaskGraph graph;
         try
@@ -62,7 +68,9 @@ public sealed class PlannerService(IOllamaClient ollama, AgentRuntimeSettings se
         }
         catch
         {
-            return CreateFallbackGraph(request);
+            TaskGraph fallback = CreateFallbackGraph(request);
+            session.TaskGraphs[taskId] = fallback;
+            return fallback;
         }
         session.TaskGraphs[taskId] = graph;
         return graph;