Optimize GetConnectionString

Author: yileicn

src/Infrastructure/BotSharp.Abstraction/MLTasks/IText2SqlHook.cs

@@ -6,7 +6,7 @@ public interface IText2SqlHook : IHookBase
 {
     // Get database type
     string GetDatabaseType(RoleDialogModel message);
-    string? GetConnectionString(RoleDialogModel message);
+    string? GetConnectionString(RoleDialogModel message, string dataSource = null);
     Task SqlGenerated(RoleDialogModel message);
     Task SqlExecuting(RoleDialogModel message);
     Task SqlExecuted(RoleDialogModel message);

src/Plugins/BotSharp.Plugin.SqlDriver/Functions/ExecuteQueryFn.cs

@@ -4,18 +4,15 @@ public class ExecuteQueryFn : IFunctionCallback
 {
     public string Name => "execute_sql";
     public string Indication => "Performing data retrieval operation.";
-    private readonly SqlDriverSetting _setting;
     private readonly SqlExecuteService _sqlExecuteService;
     private readonly IServiceProvider _services;
     private readonly ILogger _logger;
 
     public ExecuteQueryFn(IServiceProvider services, 
-        SqlDriverSetting setting,
         SqlExecuteService sqlExecuteService,
         ILogger<ExecuteQueryFn> logger)
     {
         _services = services;
-        _setting = setting;
         _sqlExecuteService = sqlExecuteService;
         _logger = logger;
     }
@@ -26,8 +23,8 @@ public async Task<bool> Execute(RoleDialogModel message)
         //var refinedArgs = await RefineSqlStatement(message, args);
         var dbHook = _services.GetRequiredService<IText2SqlHook>();
         var dbType = dbHook.GetDatabaseType(message);
-        var connectionString = _setting.Connections.FirstOrDefault(x => x.Name.Equals(args.DataSource, StringComparison.OrdinalIgnoreCase))?.ConnectionString;
-        var dbConnectionString = dbHook.GetConnectionString(message) ?? connectionString ?? throw new Exception("database connection is not found");
+        var dbConnectionString = dbHook.GetConnectionString(message, args.DataSource) ?? 
+            throw new Exception("database connection is not found");
 
         // Print all the SQL statements for debugging
         _logger.LogInformation("Executing SQL Statements: {SqlStatements}", string.Join("\r\n", args.SqlStatements));

src/Plugins/BotSharp.Plugin.SqlDriver/Functions/SqlSelectFn.cs

@@ -5,13 +5,10 @@ public class SqlSelectFn : IFunctionCallback
     public string Name => "sql_select";
     private readonly IServiceProvider _services;
     private readonly SqlExecuteService _sqlExecuteService;
-    private readonly SqlDriverSetting _settings;
 
     public SqlSelectFn(IServiceProvider services, 
-        SqlExecuteService sqlExecuteService,
-        SqlDriverSetting settings)
+        SqlExecuteService sqlExecuteService)
     {
-        _settings = settings;
         _services = services;
         _sqlExecuteService = sqlExecuteService;
     }
@@ -30,7 +27,6 @@ public async Task<bool> Execute(RoleDialogModel message)
         var dbHook = _services.GetRequiredService<IText2SqlHook>();
         var dbType = dbHook.GetDatabaseType(message);
         var dbConnectionString = dbHook.GetConnectionString(message) ?? 
-            _settings.Connections.FirstOrDefault(c => c.DbType == dbType)?.ConnectionString ??
             throw new Exception("database connectdion is not found");
 
         var result = await (dbType switch