diff --git a/src/LLM/AgentPromptGeneratorInterface.php b/src/LLM/AgentPromptGeneratorInterface.php
index 8081ac6..59a3fbb 100644
--- a/src/LLM/AgentPromptGeneratorInterface.php
+++ b/src/LLM/AgentPromptGeneratorInterface.php
@@ -5,13 +5,15 @@
 namespace LLM\Agents\LLM;
 
 use LLM\Agents\Agent\AgentInterface;
+use LLM\Agents\LLM\Prompt\Chat\Prompt;
 use LLM\Agents\LLM\Prompt\Chat\PromptInterface;
 
 interface AgentPromptGeneratorInterface
 {
     public function generate(
         AgentInterface $agent,
-        string|\Stringable $prompt,
-        ?array $sessionContext = null,
+        string|\Stringable $userPrompt,
+        PromptContextInterface $context,
+        PromptInterface $prompt = new Prompt(),
     ): PromptInterface;
 }
diff --git a/src/LLM/PromptContextInterface.php b/src/LLM/PromptContextInterface.php
new file mode 100644
index 0000000..f0d0c1d
--- /dev/null
+++ b/src/LLM/PromptContextInterface.php
@@ -0,0 +1,7 @@
+<?php
+
+declare(strict_types=1);
+
+namespace LLM\Agents\LLM;
+
+interface PromptContextInterface { }