股市数据库问答

目标

借助大语言模型,使用自然语言查询数据库中的内容。

效果示例

构建思路

  1. 我们需要把自然语言转化为SQL语言,选择语言模型ChatOpenAI组件(或其他语言模型组件);

  2. 实现在数据库中对数据的检索,可以选择SQLAgent 🚧组件。

  • 连接示例

  • 工作原理

用户输入的自然语言由大语言模型转化为SQL语言,SQLAgent在用户指定的数据库中执行SQL语言。

  • 组件参数设置

    • ChatOpenAI

      • Model Name:选择gpt-3.5-turbo,可以换为任意模型

      • OpenAI API Key:API接口密钥

      • OpenAI API Base/OpenAI Proxy:服务的地址,填写其中之一即可(如果是自己代理的国内的地址则填写OpenAI Proxy参数)

    • SQLAgent

      • Database URI:数据库对应的字符串,Agent可以使用这个字符串连接到对应的数据库并执行各种操作

      • 其他参数(若使用OpenAI模型,以下参数可以使用默认值,不必修改。若使用其他模型,可能需要调整最终效果表现才好)

        • Format Instructions:指导Agent运行的方式

        • input_variables:最终的提示词所需要的输入变量列表,对应Suffix中的变量,即{query}和{agent_scratchpad}

        • prefix:提示词模板的开头

        • suffix:提示词模板的结尾

Last updated