DEV Community
·
文本到SQL：使用Nebius AI Studio生成SQL（第二部分）
💡
原文英文，约600词，阅读约需3分钟。
📝
内容提要
本文介绍了如何使用Nebius AI Studio和Postgres数据库进行文档查询。通过创建向量和余弦相似度检索数据库，并结合OpenAI模型生成SQL查询。示例展示了如何按国家字母顺序列出供应商数量及相应SQL语句。
🎯
关键要点
- 本文介绍了如何使用Nebius AI Studio和Postgres数据库进行文档查询。
- 通过创建向量和余弦相似度检索数据库，并结合OpenAI模型生成SQL查询。
- 示例展示了如何按国家字母顺序列出供应商数量及相应SQL语句。
- 使用Nebius客户端和数据库连接参数开始脚本。
- 查询函数从数据库中检索嵌入，使用余弦相似度形成查询。
- 数据库结果增强了OpenAI风格的提示，提示发送给Nebius Qwen2.5-Coder-7B-Instruct模型。
- 编写查询以按国家字母顺序列出供应商数量，并从数据库检索相关嵌入。
- 生成的SQL查询选择国家列并计算每个国家的供应商数量。
- 使用pgAdmin或类似工具测试结果。
- 文档需要进行多次修改以生成有效的SQL查询。
- LLM可以生成标准SQL函数，结果令人鼓舞，下一篇文章将展示如何构建运行SQL的代理。
❓
延伸问答
如何使用Nebius AI Studio进行文档查询？
使用Nebius AI Studio可以通过创建向量和余弦相似度检索Postgres数据库中的文档，并结合OpenAI模型生成SQL查询。
生成SQL查询的过程是怎样的？
生成SQL查询的过程包括从数据库检索嵌入、使用余弦相似度形成查询，并将结果增强后发送给OpenAI模型生成SQL语句。
如何按国家字母顺序列出供应商数量？
可以使用SQL语句：SELECT country, COUNT(*) AS number_of_suppliers FROM suppliers GROUP BY country ORDER BY country; 来按国家字母顺序列出供应商数量。
在使用Nebius AI Studio时需要注意哪些事项？
需要确保文档中只包含必要的表名，避免包含数据库名，并且要注意表之间的关系，以便生成有效的SQL查询。
如何测试生成的SQL查询结果？
可以使用pgAdmin或类似工具来测试生成的SQL查询结果，确保查询的正确性。
Nebius AI Studio生成的SQL查询有什么优点？
Nebius AI Studio生成的SQL查询能够使用标准SQL函数，结果令人鼓舞，并且可以处理复杂的查询需求。
➡️
