commit
cc034e02e4
|
|
@ -96,9 +96,6 @@ public class PrimaryStagePlanFn : IFunctionCallback
|
||||||
var conv = _services.GetRequiredService<IConversationService>();
|
var conv = _services.GetRequiredService<IConversationService>();
|
||||||
var wholeDialogs = conv.GetDialogHistory();
|
var wholeDialogs = conv.GetDialogHistory();
|
||||||
|
|
||||||
// Append text
|
|
||||||
wholeDialogs.Last().Content += "\n\nYou must analyze the table description to infer the table relations. Only output the JSON result.";
|
|
||||||
|
|
||||||
var completion = CompletionProvider.GetChatCompletion(_services,
|
var completion = CompletionProvider.GetChatCompletion(_services,
|
||||||
provider: plannerAgent.LlmConfig.Provider,
|
provider: plannerAgent.LlmConfig.Provider,
|
||||||
model: plannerAgent.LlmConfig.Model);
|
model: plannerAgent.LlmConfig.Model);
|
||||||
|
|
|
||||||
|
|
@ -74,7 +74,7 @@ public class SqlValidateFn : IFunctionCallback
|
||||||
Message = "Correct SQL Statement",
|
Message = "Correct SQL Statement",
|
||||||
Data = new Dictionary<string, object>
|
Data = new Dictionary<string, object>
|
||||||
{
|
{
|
||||||
{ "original_sql", validateSql },
|
{ "original_sql", sql },
|
||||||
{ "error_message", ex.Message },
|
{ "error_message", ex.Message },
|
||||||
{ "table_structure", ddl }
|
{ "table_structure", ddl }
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue