Update namign

Author: BenConstable9

text_2_sql/autogen/src/autogen_text_2_sql/__init__.py

@@ -1,6 +1,6 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 from autogen_text_2_sql.autogen_text_2_sql import AutoGenText2Sql
-from text_2_sql_core.payloads.agent_response import AgentRequestBody
+from text_2_sql_core.payloads.agent_request_response_pair import AgentRequestBody
 
 __all__ = ["AutoGenText2Sql", "AgentRequestBody"]

text_2_sql/autogen/src/autogen_text_2_sql/autogen_text_2_sql.py

@@ -17,8 +17,8 @@
 import os
 from datetime import datetime
 
-from text_2_sql_core.payloads.agent_response import (
-    AgentResponse,
+from text_2_sql_core.payloads.agent_request_response_pair import (
+    AgentRequestResponsePair,
     AgentRequestBody,
     AnswerWithSources,
     Source,
@@ -161,7 +161,7 @@ async def process_question(
         self,
         request: AgentRequestBody,
         chat_history: list[ChatHistoryItem] = None,
-    ) -> AsyncGenerator[AgentResponse | ProcessingUpdate, None]:
+    ) -> AsyncGenerator[AgentRequestResponsePair | ProcessingUpdate, None]:
         """Process the complete question through the unified system.
 
         Args:
@@ -229,7 +229,7 @@ async def process_question(
                     logging.error("Unexpected TaskResult: %s", message)
                     raise ValueError("Unexpected TaskResult")
 
-                payload = AgentResponse(request=request, response=response)
+                payload = AgentRequestResponsePair(request=request, response=response)
 
             if payload is not None:
                 logging.debug("Payload: %s", payload)

…xt_2_sql_core/payloads/agent_response.py …/payloads/agent_request_response_pair.pytext_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/agent_response.py renamed to text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/agent_request_response_pair.py text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/agent_response.py renamed to text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/agent_request_response_pair.py

@@ -83,7 +83,7 @@ def add_defaults_to_injected_parameters(cls, values):
         return values
 
 
-class AgentResponse(BaseModel):
+class AgentRequestResponsePair(BaseModel):
     header: AgentResponseHeader | None = Field(default=None)
     request: AgentRequestBody
     response: AgentResponseBody

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

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