refactor: remove print.

Author: liqiang-fit2cloud

apps/application/flow/step_node/text_to_video_step_node/impl/base_text_to_video_node.py

@@ -37,7 +37,6 @@ def execute(self, model_id, prompt, negative_prompt, dialogue_number, dialogue_t
         self.context['dialogue_type'] = dialogue_type
         self.context['negative_prompt'] = self.generate_prompt_question(negative_prompt)
         video_urls = ttv_model.generate_video(question, negative_prompt)
-        print('video_urls', video_urls)
         # 保存图片
         if video_urls is None:
             return NodeResult({'answer': gettext('Failed to generate video')}, {})

apps/application/flow/workflow_manage.py

@@ -9,7 +9,6 @@
 import concurrent
 import json
 import threading
-import traceback
 from concurrent.futures import ThreadPoolExecutor
 from functools import reduce
 from typing import List, Dict
@@ -26,6 +25,7 @@
 from application.flow.step_node import get_node
 from common.handle.base_to_response import BaseToResponse
 from common.handle.impl.response.system_to_response import SystemToResponse
+from common.utils.logger import maxkb_logger
 
 executor = ThreadPoolExecutor(max_workers=200)
 
@@ -383,7 +383,7 @@ def run_chain(self, current_node, node_result_future=None):
                 current_node, node_result_future)
             return result
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
         return None
 
     def hand_node_result(self, current_node, node_result_future):
@@ -395,7 +395,7 @@ def hand_node_result(self, current_node, node_result_future):
                 list(result)
             return current_result
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             self.status = 500
             current_node.get_write_error_context(e)
             self.answer += str(e)
@@ -473,7 +473,7 @@ def hand_event_node_result(self, current_node, node_result_future):
             return current_result
         except Exception as e:
             # 添加节点
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             chunk = self.base_to_response.to_stream_chunk_response(self.params['chat_id'],
                                                                    self.params['chat_record_id'],
                                                                    current_node.id,

apps/common/auth/authenticate.py

@@ -6,7 +6,6 @@
     @date：2023/9/4 11:16
     @desc:  认证类
 """
-import traceback
 from importlib import import_module
 
 from django.conf import settings
@@ -18,6 +17,7 @@
 
 from common.exception.app_exception import AppAuthenticationFailed, AppEmbedIdentityFailed, AppChatNumOutOfBoundsFailed, \
     AppApiException
+from common.utils.logger import maxkb_logger
 
 token_cache = cache.caches['default']
 
@@ -88,7 +88,7 @@ def authenticate(self, request):
                     return handle.handle(request, token, token_details.get_token_details)
             raise AppAuthenticationFailed(1002, _('Authentication information is incorrect! illegal user'))
         except Exception as e:
-            traceback.print_stack()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             if isinstance(e, AppEmbedIdentityFailed) or isinstance(e, AppChatNumOutOfBoundsFailed) or isinstance(e,
                                                                                                                  AppApiException):
                 raise e

apps/common/handle/impl/text/html_split_handle.py

@@ -77,5 +77,5 @@ def get_content(self, file, save_image):
             content = buffer.decode(encoding)
             return html2text(content)
         except BaseException as e:
-            traceback.print_exception(e)
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             return f'{e}'

apps/knowledge/task/sync.py

@@ -7,7 +7,6 @@
     @desc:
 """
 
-import logging
 import traceback
 from typing import List
 

apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/asr_stt.py

@@ -1,13 +1,12 @@
 # coding=utf-8
-import traceback
 from typing import Dict, Any
 
 from common import forms
 from common.exception.app_exception import AppApiException
 from common.forms import BaseForm
 from models_provider.base_model_provider import BaseModelCredential, ValidCode
 from django.utils.translation import gettext as _
-
+from common.utils.logger import maxkb_logger
 
 class AliyunBaiLianAsrSTTModelCredential(BaseForm, BaseModelCredential):
     api_url = forms.TextInputField(_('API URL'), required=True)
@@ -41,7 +40,7 @@ def is_valid(self,
         try:
             model = provider.get_model(model_type, model_name, model_credential)
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             if isinstance(e, AppApiException):
                 raise e
             if raise_exception:

apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/embedding.py

@@ -6,7 +6,6 @@
     @date：2024/10/16 17:01
     @desc:
 """
-import traceback
 from typing import Dict, Any
 
 from django.utils.translation import gettext as _
@@ -16,6 +15,7 @@
 from common.forms import BaseForm, TooltipLabel
 from models_provider.base_model_provider import BaseModelCredential, ValidCode
 from models_provider.impl.aliyun_bai_lian_model_provider.model.embedding import AliyunBaiLianEmbedding
+from common.utils.logger import maxkb_logger
 
 class BaiLianEmbeddingModelParams(BaseForm):
     dimensions = forms.SingleSelect(
@@ -69,7 +69,7 @@ def is_valid(
             model: AliyunBaiLianEmbedding = provider.get_model(model_type, model_name, model_credential)
             model.embed_query(_("Hello"))
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             if isinstance(e, AppApiException):
                 raise e
             if raise_exception:

apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/image.py

@@ -6,11 +6,9 @@
     @date：2024/7/11 18:41
     @desc:
 """
-import traceback
 from typing import Dict
 
 from django.utils.translation import gettext_lazy as _, gettext
-from langchain_core.messages import HumanMessage
 
 from common import forms
 from common.exception.app_exception import AppApiException
@@ -69,7 +67,7 @@ def is_valid(
             model = provider.get_model(model_type, model_name, model_credential, **model_params)
             model.check_auth(model_credential.get('api_key'))
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             if isinstance(e, AppApiException):
                 raise e
             if raise_exception:

apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/itv.py

@@ -1,6 +1,5 @@
 # coding=utf-8
 
-import traceback
 from typing import Dict, Any
 
 from django.utils.translation import gettext_lazy as _, gettext
@@ -9,7 +8,7 @@
 from common.forms import BaseForm, PasswordInputField, SingleSelect, SliderField, TooltipLabel
 from common.forms.switch_field import SwitchField
 from models_provider.base_model_provider import BaseModelCredential, ValidCode
-
+from common.utils.logger import maxkb_logger
 
 class QwenModelParams(BaseForm):
     """
@@ -85,7 +84,7 @@ def is_valid(
             model = provider.get_model(model_type, model_name, model_credential, **model_params)
             res = model.check_auth()
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             if isinstance(e, AppApiException):
                 raise e
             if raise_exception:

apps/models_provider/impl/aliyun_bai_lian_model_provider/credential/llm.py

@@ -1,5 +1,4 @@
 # coding=utf-8
-import traceback
 from typing import Dict
 
 from langchain_core.messages import HumanMessage
@@ -9,7 +8,7 @@
 from common.exception.app_exception import AppApiException
 from common.forms import BaseForm, TooltipLabel
 from models_provider.base_model_provider import BaseModelCredential, ValidCode
-
+from common.utils.logger import maxkb_logger
 
 class BaiLianLLMModelParams(BaseForm):
     temperature = forms.SliderField(
@@ -76,7 +75,7 @@ def is_valid(
             else:
                 model.invoke([HumanMessage(content=gettext('Hello'))])
         except Exception as e:
-            traceback.print_exc()
+            maxkb_logger.error(f'Exception: {e}', exc_info=True)
             if isinstance(e, AppApiException):
                 raise e
             if raise_exception: