获取create_sql_chain适用于gpt-4

我当前的代码在使用gpt-3.5-turbo-instruct时可以正常工作,但在使用gpt-4时却无法工作。我喜欢使用这个框架,因为我只想得到SQL代码,而其他代理无法做到这一点。我该如何修改代码,以便可以使用其他GPT模型?

from langchain.chains import create_sql_query_chainconnection_string = ""db = SQLDatabase.from_uri(connection_string)llm = OpenAI(temperature=0, verbose=True, model='gpt-4')seed_prompt = """Given an input question, create a syntactically correct MySQL SQL query to run.Question: "Question here"SQLQuery: "SQL Query to run""""restrictions = """Never use LIMIT statement, use TOP statement instead.Format all numeric response ###,###,###,###.Only return relevant columns to the question.If a table or column does not exist, return table or column could not be found.Question: {input}"""prompt = seed_prompt + restrictionsPROMPT = PromptTemplate(    input_variables=["input"], template=prompt)database_chain = create_sql_query_chain(llm,db, prompt=PROMPT)sql_query = database_chain.invoke({"question": x})print(sql_query)

回答:

JK找到了解决方案!将OpenAI改为ChatOpenAI就解决了问题

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注