bug fix, dashscope import error

.gitignore

@@ -5,4 +5,5 @@ dist
 test
 build
 !_ai_summary_cache_chatgpt.json
-!_ai_summary_cache_tongyi.json
+!_ai_summary_cache_tongyi.json
+.DS_Store

mkdocs_ai_summary/chatgpt_api.py

@@ -1,4 +1,3 @@
-import json
 import logging
 from hashlib import md5
 from openai import OpenAI

mkdocs_ai_summary/plugin.py

@@ -9,7 +9,6 @@
 import re
 import logging
 
-from .tongyi_api import AiSummaryError
 
 logger = logging.getLogger("mkdocs.plugins.ai-summary")
 
@@ -59,7 +58,7 @@ def on_page_markdown(
         match api:
             case "tongyi":
                 try:
-                    from .tongyi_api import get_summary
+                    from .tongyi_api import get_summary, AiSummaryRequestError
                 except ImportError as e:
                     logger.warning("tongyi is not available", repr(e))
                     return markdown
@@ -75,8 +74,8 @@ def on_page_markdown(
                         model=model,
                         logger=logger,
                     )
-                except AiSummaryError as e:
-                    logger.warning("Ask AI Error", repr(e))
+                except AiSummaryRequestError as e:
+                    logger.warning("Request Tongyi AI Error", repr(e))
                     return markdown
                 except Exception as e:
                     logger.warning(repr(e))

mkdocs_ai_summary/tongyi_api.py

@@ -1,14 +1,14 @@
+import logging
 from hashlib import md5
 from http import HTTPStatus
 from dashscope import Generation
-import logging
 
 from .cache import with_cache, load_cache, save_cache
 
 MAX_LENGTH = 6000
 
 
-class AiSummaryError(Exception):
+class AiSummaryRequestError(Exception):
 
     def __init__(self, *args: object) -> None:
         super().__init__(*args)
@@ -30,7 +30,7 @@ def ask(prompt, model="qwen-turbo"):
     if response.status_code == HTTPStatus.OK:
         return response["output"]["text"]
     else:
-        raise AiSummaryError(
+        raise AiSummaryRequestError(
             "Request id: %s, Status code: %s, error code: %s, error message: %s"
             % (
                 response.request_id,

pyproject.toml

@@ -7,7 +7,7 @@ where = ["."]
 
 [project]
 name = "mkdocs-ai-summary"
-version = "0.2.5"
+version = "0.2.6"
 requires-python = ">=3.10"
 dependencies = [
     "mkdocs>=1.5.3",