Add buffered history

Author: BenConstable9

text_2_sql/autogen/src/autogen_text_2_sql/creators/llm_agent_creator.py

@@ -11,6 +11,7 @@
     AnswerAgentWithFollowUpQuestionsAgentOutput,
     UserMessageRewriteAgentOutput,
 )
+from autogen_core.model_context import BufferedChatCompletionContext
 
 
 class LLMAgentCreator:
@@ -132,4 +133,9 @@ def create(cls, name: str, **kwargs) -> AssistantAgent:
             ),
         )
 
+        if "context_size" in agent_file:
+            agent.model_context = BufferedChatCompletionContext(
+                buffer_size=agent_file["context_size"]
+            )
+
         return agent

text_2_sql/text_2_sql_core/src/text_2_sql_core/prompts/answer_agent.yaml

@@ -27,3 +27,4 @@ system_message: |
     Make sure your response directly addresses every part of the user's question.
 
   </instructions>
+context_size: 8

text_2_sql/text_2_sql_core/src/text_2_sql_core/prompts/user_message_rewrite_agent.yaml

@@ -166,3 +166,4 @@ system_message: |
       ```
   </examples>
 structured_output: true
+context_size: 5