From 7132d07921f52f74616f80e59d824c52d446ce86 Mon Sep 17 00:00:00 2001 From: yujj128 Date: Thu, 25 Sep 2025 08:35:49 +0800 Subject: [PATCH] =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=B7=BB=E5=8A=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- service/cus_vanna_srevice.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/service/cus_vanna_srevice.py b/service/cus_vanna_srevice.py index 577673d..9585194 100644 --- a/service/cus_vanna_srevice.py +++ b/service/cus_vanna_srevice.py @@ -14,7 +14,9 @@ from util.utils import extract_nested_json, check_and_get_sql, get_chart_type_fr import json from template.template import get_base_template from datetime import datetime +import logging +logger = logging.getLogger(__name__) class OpenAICompatibleLLM(VannaBase): def __init__(self, client=None, config_file=None): @@ -192,15 +194,14 @@ class OpenAICompatibleLLM(VannaBase): sys_temp = sql_temp['system'].format(engine=config("DB_ENGINE", default='mysql'), lang='中文', schema=ddl_list, documentation=doc_list, data_training=question_sql_list) - print("sys_temp", sys_temp) + logger.info(f"sys_temp:{sys_temp}") user_temp = sql_temp['user'].format(question=question, current_time=datetime.now().strftime('%Y-%m-%d %H:%M:%S')) - print("user_temp", user_temp) + logger.info(f"user_temp:{user_temp}") llm_response = self.submit_prompt( [{'role': 'system', 'content': sys_temp}, {'role': 'user', 'content': user_temp}], **kwargs) - print(llm_response) + logger.info(f"llm_response:{llm_response}") result = {"resp": orjson.loads(extract_nested_json(llm_response))} - print("result", result) sql = check_and_get_sql(llm_response) # ---------------生成图表 char_type = get_chart_type_from_sql_answer(llm_response) @@ -213,12 +214,13 @@ class OpenAICompatibleLLM(VannaBase): **kwargs) print(llm_response2) result['chart'] = orjson.loads(extract_nested_json(llm_response2)) + logger.info(f"chart_response:{result}") return result except Exception as e: raise e def generate_rewritten_question(self, last_question: str, new_question: str, **kwargs) -> str: - print("new_question---------------", new_question) + logger.info(f"generate_rewritten_question---------------{new_question}") if last_question is None: return new_question