Fix bad chat history instantiation

Author: BenConstable9

text_2_sql/autogen/src/autogen_text_2_sql/autogen_text_2_sql.py

@@ -12,7 +12,6 @@
 from autogen_text_2_sql.custom_agents.parallel_query_solving_agent import (
     ParallelQuerySolvingAgent,
 )
-from autogen_agentchat.agents import UserProxyAgent
 from autogen_agentchat.messages import TextMessage
 import json
 import os
@@ -30,26 +29,10 @@
     ProcessingUpdateBody,
     ProcessingUpdate,
 )
-from autogen_agentchat.base import Response, TaskResult
+from autogen_agentchat.base import TaskResult
 from typing import AsyncGenerator
 
 
-class EmptyResponseUserProxyAgent(UserProxyAgent):
-    """UserProxyAgent that automatically responds with empty messages."""
-
-    def __init__(self, name):
-        super().__init__(name=name)
-        self._has_responded = False
-
-    async def on_messages_stream(self, messages, sender=None, config=None):
-        """Auto-respond with empty message and return Response object."""
-        message = TextMessage(content="", source=self.name)
-        if not self._has_responded:
-            self._has_responded = True
-            yield message
-        yield Response(chat_message=message)
-
-
 class AutoGenText2Sql:
     def __init__(self, engine_specific_rules: str, **kwargs: dict):
         self.target_engine = os.environ["Text2Sql__DatabaseEngine"].upper()
@@ -71,11 +54,7 @@ def get_all_agents(self):
 
         self.answer_agent = LLMAgentCreator.create("answer_agent")
 
-        # Auto-responding UserProxyAgent
-        self.user_proxy = EmptyResponseUserProxyAgent(name="user_proxy")
-
         agents = [
-            self.user_proxy,
             self.query_rewrite_agent,
             self.parallel_query_solving_agent,
             self.answer_agent,

text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/chat_history.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 from pydantic import BaseModel, Field
-from text_2_sql_core.payloads.agent_response import AgentResponse
+from text_2_sql_core.payloads.agent_response import AgentRequestBody
 from datetime import datetime, timezone
 
 
@@ -13,4 +13,4 @@ class ChatHistoryItem(BaseModel):
         description="Timestamp in UTC",
         default_factory=lambda: datetime.now(timezone.utc),
     )
-    agent_response: AgentResponse
+    agent_response: AgentRequestBody