refactor: remove print

Author: wxg0103

apps/common/handle/impl/common_handle.py

@@ -7,6 +7,8 @@
     @desc:
 """
 import io
+import logging
+
 import uuid_utils.compat as uuid
 from functools import reduce
 from io import BytesIO
@@ -73,7 +75,7 @@ def handle_images(deps, archive: ZipFile) -> []:
             image_io = archive.read(dep.target)
             image = openpyxl_Image(BytesIO(image_io))
         except Exception as e:
-            print(e)
+            logging.getLogger("max_kb_error").error(f"Error reading image {dep.target}: {e}")
             continue
         image.embed = dep.id  # 文件rId
         image.target = dep.target  # 文件地址

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

@@ -7,6 +7,7 @@
     @desc:
 """
 import io
+import logging
 import os
 import re
 import traceback
@@ -78,7 +79,7 @@ def get_paragraph_element_txt(paragraph_element, doc: Document, images_list, get
             return paragraph_element.text
         return ""
     except Exception as e:
-        print(e)
+        logging.getLogger("max_kb_error").error(f'Error getting paragraph element text: {e}')
     return ""
 
 

apps/common/utils/fork.py

@@ -173,6 +173,6 @@ def fork(self):
 
 
 def handler(base_url, response: Fork.Response):
-    print(base_url.url, base_url.tag.text if base_url.tag else None, response.content)
+    logging.getLogger("max_kb").info(base_url.url, base_url.tag.text if base_url.tag else None, response.content)
 
 # ForkManage('https://bbs.fit2cloud.com/c/de/6', ['.md-content']).fork(3, set(), handler)

apps/knowledge/serializers/knowledge.py

@@ -617,7 +617,7 @@ def handler(child_link: ChildLink, response: Fork.Response):
                         document_name = child_link.tag.text if child_link.tag is not None and len(
                             child_link.tag.text.strip()) > 0 else child_link.url
                         paragraphs = get_split_model('web.md').parse(response.content)
-                        print(child_link.url.strip())
+                        logging.getLogger("max_kb").info(child_link.url.strip())
                         first = QuerySet(Document).filter(
                             meta__source_url=child_link.url.strip(),
                             knowledge=knowledge

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

@@ -48,8 +48,6 @@ def is_valid(
         try:
             model = provider.get_model(model_type, model_name, model_credential, **model_params)
             res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext('Hello')}])])
-            for chunk in res:
-                print(chunk)
         except Exception as e:
             traceback.print_exc()
             if isinstance(e, AppApiException):

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

@@ -109,7 +109,6 @@ def is_valid(
         try:
             model = provider.get_model(model_type, model_name, model_credential, **model_params)
             res = model.check_auth()
-            print(res)
         except Exception as e:
             traceback.print_exc()
             if isinstance(e, AppApiException):

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/tti.py

@@ -6,10 +6,12 @@
 from django.utils.translation import gettext
 from langchain_community.chat_models import ChatTongyi
 from langchain_core.messages import HumanMessage
-
+import logging
 from models_provider.base_model_provider import MaxKBBaseModel
 from models_provider.impl.base_tti import BaseTextToImage
 
+max_kb_error = logging.getLogger("max_kb_error")
+
 
 class QwenTextToImageModel(MaxKBBaseModel, BaseTextToImage):
     api_key: str
@@ -54,6 +56,6 @@ def generate_image(self, prompt: str, negative_prompt: str = None):
             for result in rsp.output.results:
                 file_urls.append(result.url)
         else:
-            print('sync_call Failed, status_code: %s, code: %s, message: %s' %
-                  (rsp.status_code, rsp.code, rsp.message))
+            max_kb_error.error('sync_call Failed, status_code: %s, code: %s, message: %s' %
+                               (rsp.status_code, rsp.code, rsp.message))
         return file_urls

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/tts.py

@@ -49,7 +49,6 @@ def text_to_speech(self, text):
         if audio is None:
             raise Exception('Failed to generate audio')
         if type(audio) == str:
-            print(audio)
             raise Exception(audio)
         return audio
 

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

@@ -31,8 +31,6 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
         try:
             model = provider.get_model(model_type, model_name, model_credential)
             res = model.stream([HumanMessage(content=[{"type": "text", "text": gettext("Hello")}])])
-            for chunk in res:
-                print(chunk)
         except Exception as e:
             traceback.print_exc()
             if isinstance(e, AppApiException):

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

@@ -32,7 +32,6 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje
         try:
             model: BedrockEmbeddingModel = provider.get_model(model_type, model_name, model_credential)
             aa = model.embed_query(_('Hello'))
-            print(aa)
         except AppApiException:
             raise
         except Exception as e: