Skip to content

Commit 2550f9c

Browse files
committed
Fix bad agent update
1 parent c416cc7 commit 2550f9c

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

text_2_sql/text_2_sql_core/src/text_2_sql_core/payloads/interaction_payloads.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ class PayloadBase(InteractionPayloadBase):
4848
class DismabiguationRequestsPayload(InteractionPayloadBase):
4949
class Body(InteractionPayloadBase):
5050
class DismabiguationRequest(InteractionPayloadBase):
51-
ASSISTANT_question: str | None = Field(..., alias="ASSISTANTQuestion")
51+
assistant_question: str | None = Field(..., alias="AssistantQuestion")
5252
user_choices: list[str] | None = Field(default=None, alias="userChoices")
5353

5454
disambiguation_requests: list[DismabiguationRequest] | None = Field(

text_2_sql/text_2_sql_core/src/text_2_sql_core/prompts/disambiguation_and_sql_query_generation_agent.yaml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ system_message:
164164
{
165165
\"disambiguation_requests\": [
166166
{
167-
\"agent_question\": \"I'm sorry, I couldn't find any relevant database schemas for your request about [REQUEST TYPE]. I focus on providing answers in the context of the use case. Could you please provide more context or rephrase your question?\",
167+
\"assistant_question\": \"I'm sorry, I couldn't find any relevant database schemas for your request about [REQUEST TYPE]. I focus on providing answers in the context of the use case. Could you please provide more context or rephrase your question?\",
168168
\"user_choices\": []
169169
}
170170
]
@@ -183,14 +183,14 @@ system_message:
183183
{
184184
\"disambiguation_requests\": [
185185
{
186-
\"agent_question\": \"Did you mean the 'Customer Name' column or the 'Client Name' column?\",
186+
\"assistant_question\": \"Did you mean the 'Customer Name' column or the 'Client Name' column?\",
187187
\"user_choices\": [
188188
\"Customer Name\",
189189
\"Client Name\"
190190
]
191191
},
192192
{
193-
\"agent_question\": \"Which sort of bike do you mean?\",
193+
\"assistant_question\": \"Which sort of bike do you mean?\",
194194
\"user_choices\": [
195195
\"Mountain Bike\",
196196
\"Road Bike\"
@@ -208,7 +208,7 @@ system_message:
208208
{
209209
\"disambiguation_requests\": [
210210
{
211-
\"agent_question\": \"Could you please rephrase your question or provide more context? I'm having trouble understanding the specifics of your request.\",
211+
\"assistant_question\": \"Could you please rephrase your question or provide more context? I'm having trouble understanding the specifics of your request.\",
212212
\"user_choices\": []
213213
}
214214
]
@@ -223,7 +223,7 @@ system_message:
223223
{
224224
\"disambiguation_requests\": [
225225
{
226-
\"agent_question\": \"Could you clarify the details of your request so I can assist you better?\",
226+
\"assistant_question\": \"Could you clarify the details of your request so I can assist you better?\",
227227
\"user_choices\": []
228228
}
229229
]
@@ -262,11 +262,11 @@ system_message:
262262
{
263263
\"disambiguation_requests\": [
264264
{
265-
\"agent_question\": \"<specific_question>\",
265+
\"assistant_question\": \"<specific_question>\",
266266
\"user_choices\": [\"<choice1>\", \"<choice2>\"]
267267
},
268268
{
269-
\"agent_question\": \"<specific_question>\",
269+
\"assistant_question\": \"<specific_question>\",
270270
\"user_choices\": [\"<choice1>\", \"<choice2>\"]
271271
}
272272
]

0 commit comments

Comments
 (0)