我当前的代码在使用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就解决了问题