当前位置:   article > 正文

探索.NET平台下自然语言转换为SQL的Nl2Sql项目

nl2sql 自然语言生成sql

        随着技术的发展,人工智能在各个领域的应用已经不再是新鲜事。在数据库查询领域能够自然地将人类语言转换为SQL语句将为不懂技术的人士提供极大的便捷,同时也能大幅提高专业开发者的工作效率。今天,我带大家深入了解一个非常有趣的项目——Nl2Sql,这个项目是基于.NET平台和Semantic Kernel的工具,它可以将自然语言转换为SQL查询语句

https://github.com/microsoft/kernel-memory/tree/NL2SQL/examples/200-dotnet-nl2sql

GPT-4带来的新突破

        GPT-4的出现,使得基于自然语言处理的技术跨越了一个新的门槛,尤其是在自然语言转换成SQL语句的能力上有了显著提升。Nl2Sql工具就是利用GPT-4和Semantic Kernel的强大功能,为我们提供了一个实验和测试平台,能够基于自然语言表达生成SQL查询语句。

项目结构与样例信息

在开源的Nl2Sql项目中,我们可以看到以下几个组成部分:

  1. nl2sql.config - 包含了设置说明、数据模式和语义提示。

  2. nl2sql.console - 控制台应用,用于将自然语言目标转换成SQL查询。

  3. nl2sql.library - 支持库,同样用于自然语言到SQL的转换。

  4. nl2sql.harness - 开发调试工具,用于实时逆向工程化数据库模式。

  5. nl2sql.sln - Visual Studio解决方案文件。

  6. 929ca463f2ca49baba9e578d0df47bbc.png

运行样例的第一步是进行初始设置和配置。

81f66e3aadae34afed04e47963e31d2b.png

核心方法解析

        我们来看一看Nl2Sql功能的实现关键。该程序采用.NET的开发框架,通过一系列方法来完成从输入的自然语言到输出的SQL查询的转换。这个过程涉及到自然语言理解、数据库模式解析和查询生成等环节。

  1. private async Task ExecuteConsoleAsync(CancellationToken stoppingToken)
  2. {
  3. var schemaNames = SchemaDefinitions.GetNames().ToArray();
  4. await SchemaProvider.InitializeAsync(
  5. this._memory,
  6. schemaNames.Select(s => Path.Combine(Repo.RootConfigFolder, "schema", $"{s}.json"))).ConfigureAwait(false);
  7. this.WriteIntroduction(schemaNames);
  8. while (!stoppingToken.IsCancellationRequested)
  9. {
  10. var objective = await ReadInputAsync().ConfigureAwait(false);
  11. if (string.IsNullOrWhiteSpace(objective))
  12. {
  13. continue;
  14. }
  15. var result =
  16. await this._queryGenerator.SolveObjectiveAsync(objective).ConfigureAwait(false);
  17. await ProcessQueryAsync(result).ConfigureAwait(false);
  18. }
  19. this.WriteLine();
  20. // Capture console input with cancellation detection
  21. async Task<string?> ReadInputAsync()
  22. {
  23. this.Write(SystemColor, "# ");
  24. var inputTask = Console.In.ReadLineAsync(stoppingToken).AsTask();
  25. var objective = await inputTask.ConfigureAwait(false);
  26. // Null response occurs when blocking input is cancelled (CTRL+C)
  27. if (null == objective)
  28. {
  29. this.WriteLine();
  30. this.WriteLine(FocusColor, "Cancellation detected...");
  31. // Yield to sync stoppingToken state
  32. await Task.Delay(TimeSpan.FromMilliseconds(300), stoppingToken).ConfigureAwait(false);
  33. }
  34. else if (string.IsNullOrWhiteSpace(objective))
  35. {
  36. this.WriteLine(FocusColor, $"Please provide a query related to the defined schemas.{Environment.NewLine}");
  37. }
  38. else
  39. {
  40. this.ClearLine(previous: true);
  41. this.WriteLine(QueryColor, $"# {objective}");
  42. }
  43. return objective;
  44. }
  45. // Display query result and (optionally) execute.
  46. async Task ProcessQueryAsync(SqlQueryResult? result)
  47. {
  48. if (result == null)
  49. {
  50. this.WriteLine(FocusColor, $"Unable to translate request into a query.{Environment.NewLine}");
  51. return;
  52. }
  53. this.WriteLine(SystemColor, $"{Environment.NewLine}SCHEMA:");
  54. this.WriteLine(QueryColor, result.Schema);
  55. this.WriteLine(SystemColor, $"{Environment.NewLine}QUERY:");
  56. this.WriteLine(QueryColor, result.Query);
  57. if (!this.Confirm($"{Environment.NewLine}Execute?"))
  58. {
  59. this.WriteLine();
  60. this.WriteLine();
  61. return;
  62. }
  63. await Task.Delay(300, stoppingToken).ConfigureAwait(false); // Human feedback window
  64. this.ClearLine();
  65. this.Write(SystemColor, "Executing...");
  66. await ProcessDataAsync(
  67. result.Schema,
  68. result.Query,
  69. reader =>
  70. {
  71. this.ClearLine();
  72. this.WriteData(reader);
  73. }).ConfigureAwait(false);
  74. }
  75. // Execute query and display the resulting data-set.
  76. async Task ProcessDataAsync(string schema, string query, Action<IDataReader> callback)
  77. {
  78. try
  79. {
  80. using var connection = await this._sqlProvider.ConnectAsync(schema).ConfigureAwait(false);
  81. using var command = connection.CreateCommand();
  82. #pragma warning disable CA2100 // Review SQL queries for security vulnerabilities
  83. command.CommandText = query;
  84. #pragma warning restore CA2100 // Review SQL queries for security vulnerabilities
  85. using var reader = await command.ExecuteReaderAsync(stoppingToken).ConfigureAwait(false);
  86. callback.Invoke(reader);
  87. }
  88. #pragma warning disable CA1031 // Do not catch general exception types
  89. catch (Exception exception)
  90. #pragma warning restore CA1031 // Do not catch general exception types
  91. {
  92. this.ClearLine();
  93. this.WriteLine(FocusColor, exception.Message);
  94. }
  95. }
  96. }

        代码结构非常清晰,主要包含数据读取、SQL命令执行、命令行输入处理和数据展示等功能。整个转换过程使用了异步编程模式,保证了良好的用户交互体验和程序性能。

用户交互设计

        在用户交互方面,该程序展现了良好的设计。用户可以在控制台中输入自然语言描述的目标,程序会解析这一描述并生成对应的SQL查询。如果成功生成,还会提供是否执行查询的选项,在用户确认后,能够将查询结果以格式化的方式显示在控制台上。

数据展示和交互

        在数据展示方面,开发人员精心设计了一整套显示系统,能够将查询结果分页显示,同时考虑到了窗口宽度和数据字段宽度的问题,确保了数据显示的友好性和阅读性。

  1. private void WriteData(IDataReader reader)
  2. {
  3. int maxPage = Console.WindowHeight - 10;
  4. var widths = GetWidths().ToArray();
  5. var isColumnTruncation = widths.Length < reader.FieldCount;
  6. var rowFormatter = string.Join('│', widths.Select((width, index) => width == -1 ? $"{{{index}}}" : $"{{{index},-{width}}}"));
  7. if (isColumnTruncation)
  8. {
  9. rowFormatter = string.Concat(rowFormatter, isColumnTruncation ? $"│{{{widths.Length}}}" : string.Empty);
  10. }
  11. WriteRow(GetColumns());
  12. WriteSeparator(widths);
  13. bool showData;
  14. do
  15. {
  16. int count = 0;
  17. while (reader.Read() && count < maxPage)
  18. {
  19. WriteRow(GetValues());
  20. count++;
  21. }
  22. if (count >= maxPage)
  23. {
  24. showData = this.Confirm($"...More?");
  25. this.ClearLine();
  26. if (!showData)
  27. {
  28. this.WriteLine();
  29. }
  30. }
  31. else
  32. {
  33. showData = false;
  34. this.WriteLine();
  35. }
  36. } while (showData);
  37. void WriteRow(IEnumerable<string> fields)
  38. {
  39. fields = TrimValues(fields).Concat(isColumnTruncation ? new[] { "..." } : Array.Empty<string>());
  40. this.WriteLine(SystemColor, rowFormatter, fields.ToArray());
  41. }
  42. IEnumerable<string> TrimValues(IEnumerable<string> fields)
  43. {
  44. int index = 0;
  45. int totalWidth = 0;
  46. foreach (var field in fields)
  47. {
  48. if (index >= widths.Length)
  49. {
  50. yield break;
  51. }
  52. var width = widths[index];
  53. ++index;
  54. if (width == -1)
  55. {
  56. var remainingWidth = Console.WindowWidth - totalWidth;
  57. yield return TrimValue(field, remainingWidth);
  58. yield break;
  59. }
  60. totalWidth += width + 1;
  61. yield return TrimValue(field, width);
  62. }
  63. }
  64. string TrimValue(string? value, int width)
  65. {
  66. value ??= string.Empty;
  67. if (value.Length <= width)
  68. {
  69. return value;
  70. }
  71. return string.Concat(value.AsSpan(0, width - 4), "...");
  72. }
  73. void WriteSeparator(int[] widths)
  74. {
  75. int totalWidth = 0;
  76. for (int index = 0; index < widths.Length; index++)
  77. {
  78. if (index > 0)
  79. {
  80. this.Write(SystemColor, "┼");
  81. }
  82. var width = widths[index];
  83. this.Write(SystemColor, new string('─', width == -1 ? Console.WindowWidth - totalWidth : width));
  84. totalWidth += width + 1;
  85. }
  86. if (isColumnTruncation)
  87. {
  88. this.Write(SystemColor, "┼───");
  89. }
  90. this.WriteLine();
  91. }
  92. IEnumerable<int> GetWidths()
  93. {
  94. if (reader.FieldCount == 1)
  95. {
  96. yield return -1;
  97. yield break;
  98. }
  99. int totalWidth = 0;
  100. for (int index = 0; index < reader.FieldCount; ++index)
  101. {
  102. if (index == reader.FieldCount - 1)
  103. {
  104. // Last field gets remaining width
  105. yield return -1;
  106. yield break;
  107. }
  108. var width = GetWidth(reader.GetFieldType(index));
  109. if (totalWidth + width > Console.WindowWidth - 11)
  110. {
  111. yield break;
  112. }
  113. totalWidth += width;
  114. yield return width;
  115. }
  116. }
  117. static int GetWidth(Type type)
  118. {
  119. if (!s_typeWidths.TryGetValue(type, out var width))
  120. {
  121. return 16; // Default width
  122. }
  123. return width;
  124. }
  125. IEnumerable<string> GetColumns()
  126. {
  127. for (int index = 0; index < reader.FieldCount; ++index)
  128. {
  129. var label = reader.GetName(index);
  130. yield return string.IsNullOrWhiteSpace(label) ? $"#{index + 1}" : label;
  131. }
  132. }
  133. IEnumerable<string> GetValues()
  134. {
  135. for (int index = 0; index < reader.FieldCount; ++index)
  136. {
  137. yield return reader.GetValue(index)?.ToString() ?? string.Empty;
  138. }
  139. }
  140. }

        整个程序对于输入的处理非常人性化,如果用户输入了无关的或格式不正确的指令,程序会给出相应的提示,引导用户重新输入。

结语

        作为一名.NET方向的技术博主,我认为Nl2Sql项目不仅展示了现代AI技术的强大能力,也体现了.NET生态在人工智能应用中的活跃度,并且推动了开发者之间的技术互助和分享。Nl2Sql证明了在未来,我们可以期待更多的自然语言处理工具来帮助我们更好地与数据交互。

        此篇博文的目的在于向您介绍Nl2Sql项目的设计理念、项目结构、核心功能以及用户交互体验。但要注意的是,该项目是否适用于特定的用例,仍然需要结合具体的业务场景和预期来评估。在整个数据检索和处理生态系统中,Nl2Sql展示的仅仅是其中的一环,但绝对是非常值得我们关注的那部分。

        希望今天的分享能够激发您的兴趣,同时也希望能够在这个领域看到更多.NET开发者的参与和贡献。让我们一同期待在人工智能和自然语言处理领域能够取得更多技术突破,为我们的开发和生活带来更多便捷!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/682766
推荐阅读
相关标签
  

闽ICP备14008679号