修复枚举对应错误,增加问答示例,优化提示词

This commit is contained in:
yujj128
2025-10-16 17:42:06 +08:00
parent 1b4be9a907
commit de705f04fd
2 changed files with 28 additions and 25 deletions

View File

@@ -244,7 +244,10 @@ class OpenAICompatibleLLM(VannaBase):
[{'role': 'system', 'content': sys_temp}, {'role': 'user', 'content': user_temp}], **kwargs)
llm_response = str(llm_response.strip())
logger.info(f"llm_response:{llm_response}")
result = {"resp": orjson.loads(extract_nested_json(llm_response))}
#优化中.......
result = extract_nested_json(llm_response)
logger.info(f"result:{result}")
result = {"resp": orjson.loads(result)}
logger.info(f"llm_response:{llm_response}")
sql = check_and_get_sql(llm_response)
logger.info(f"sql:{sql}")