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

2025-10-16 17:42:06 +08:00
parent 1b4be9a907
commit de705f04fd
2 changed files with 28 additions and 25 deletions
--- a/service/cus_vanna_srevice.py
+++ b/service/cus_vanna_srevice.py
@@ -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}")