Add enums for structured data mode

Author: BenConstable9

text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/__init__.py

@@ -0,0 +1,12 @@
+from pydantic import BaseModel, Field
+
+
+class Source(BaseModel):
+    sql_query: str
+    sql_rows: list[dict]
+    markdown_table: str
+
+
+class AnswerWithSources(BaseModel):
+    answer: str
+    sources: list[str] = Field(default_factory=list)

text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/answer_with_sources.py

@@ -0,0 +1,26 @@
+from pydantic import BaseModel, RootModel, Field
+from enum import StrEnum
+from typing import Literal
+
+
+class RequestType(StrEnum):
+    DISAMBIGUATION = "disambiguation"
+    CLARIFICATION = "clarification"
+
+
+class ClarificationRequest(BaseModel):
+    type: Literal[RequestType.CLARIFICATION]
+    question: str
+    other_user_choices: list[str]
+
+
+class DismabiguationRequest(BaseModel):
+    type: Literal[RequestType.DISAMBIGUATION]
+    question: str
+    matching_columns: list[str]
+    matching_filter_values: list[str]
+    other_user_choices: list[str]
+
+
+class UserInformationRequest(RootModel):
+    root: DismabiguationRequest = Field(..., discriminator="type")