sdpy-kit 0.1.3__tar.gz → 0.1.4__tar.gz

{sdpy_kit-0.1.3 → sdpy_kit-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: sdpy-kit
-Version: 0.1.3
+Version: 0.1.4
 Summary: SDPY Kit——统一数据库访问层（PG/MySQL/SQLite）与 AI/MCP/Redis/文件/文档/向量工具集
 Author-email: Clark <changhongyuan@126.com>
 License-Expression: MIT
@@ -17,23 +17,34 @@ Classifier: Topic :: Database
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.13
-Requires-Dist: fastmcp>=4.0.3
 Requires-Dist: firecrawl-anydoc>=0.2.4
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: markdown-it-py>=3.0
 Requires-Dist: numpy>=2.0
 Requires-Dist: pimd[pdfa]>=2.2.5
-Requires-Dist: psycopg2-binary>=2.9.12
 Requires-Dist: pydantic-settings>=2.0
 Requires-Dist: pydantic>=2.0
-Requires-Dist: pymysql>=1.2.0
 Requires-Dist: python-dotenv>=1.2.3
-Requires-Dist: redis>=5.0.0
 Requires-Dist: requests>=2.34.0
 Requires-Dist: sqlalchemy~=2.0.51
 Requires-Dist: sqliteai-vector>=1.0.0
 Requires-Dist: sqlmodel>=0.0.39
-Requires-Dist: zvec>=0.6.0
+Provides-Extra: all
+Requires-Dist: fastmcp>=4.0.3; extra == 'all'
+Requires-Dist: psycopg2-binary>=2.9.12; extra == 'all'
+Requires-Dist: pymysql>=1.2.0; extra == 'all'
+Requires-Dist: redis>=5.0.0; extra == 'all'
+Requires-Dist: zvec>=0.6.0; extra == 'all'
+Provides-Extra: mcp
+Requires-Dist: fastmcp>=4.0.3; extra == 'mcp'
+Provides-Extra: mysql
+Requires-Dist: pymysql>=1.2.0; extra == 'mysql'
+Provides-Extra: pg
+Requires-Dist: psycopg2-binary>=2.9.12; extra == 'pg'
+Provides-Extra: redis
+Requires-Dist: redis>=5.0.0; extra == 'redis'
+Provides-Extra: zvec
+Requires-Dist: zvec>=0.6.0; extra == 'zvec'
 Description-Content-Type: text/markdown
 
 # sdpy-kit

{sdpy_kit-0.1.3 → sdpy_kit-0.1.4}/kit/__init__.py

@@ -23,7 +23,7 @@ Author: Clark Chang
 数据库访问通过 kit.db 子包提供（基于 SQLModel）。
 """
 
-__version__ = "0.1.3"
+__version__ = "0.1.4"
 
 from kit.db import (
     MYSQL,

{sdpy_kit-0.1.3 → sdpy_kit-0.1.4}/kit/ai/base_service.py

@@ -127,7 +127,7 @@ class BaseAIService[RequestT, ResponseT](ABC):
             data = await self._client.post_json(payload)
             result = self._parse_response(data)
         except AIError as exc:
-            logger.info(
+            logger.error(
                 f"********** {desc} ********** "
                 f"耗时 {time.perf_counter() - start:.1f}秒（失败：{exc}）"
             )

{sdpy_kit-0.1.3 → sdpy_kit-0.1.4}/kit/ai/chat_service.py

@@ -116,6 +116,7 @@ class ChatService(BaseAIService[ChatRequest, str]):
         logger.info(f"请求参数: {params}")
         start = time.perf_counter()
         note = ""
+        failed = False
         try:
             async for parsed in self._client.stream_json(payload):
                 if parsed is None:
@@ -130,12 +131,14 @@ class ChatService(BaseAIService[ChatRequest, str]):
                     yield delta
         except AIError as exc:
             note = f"（失败：{exc}）"
+            failed = True
             raise
         except BaseException:  # noqa: BLE001 - 中断路径（提前 break/取消）也需收尾日志
             note = "（中断）"
             raise
         finally:
-            logger.info(
+            log = logger.error if failed else logger.info
+            log(
                 f"********** {desc} ********** "
                 f"耗时 {time.perf_counter() - start:.1f}秒{note}"
             )

{sdpy_kit-0.1.3 → sdpy_kit-0.1.4}/kit/ai/embedding_service.py

@@ -146,7 +146,8 @@ class EmbeddingService(BaseAIService[EmbeddingRequest, list[list[float]]]):
 
             return [vec for vec in results if vec is not None]
         finally:
-            logger.info(
+            log = logger.error if failures else logger.info
+            log(
                 f"********** {desc} ********** "
                 f"耗时 {time.perf_counter() - start:.1f}秒{note}"
             )

{sdpy_kit-0.1.3 → sdpy_kit-0.1.4}/pyproject.toml

@@ -34,21 +34,34 @@ dependencies = [
     "numpy>=2.0",
     "sqlalchemy~=2.0.51",
     "sqlmodel>=0.0.39",
-    "pymysql>=1.2.0",
-    "psycopg2-binary>=2.9.12",
-    "redis>=5.0.0",
-    "zvec>=0.6.0",
     "sqliteai-vector>=1.0.0",
     "firecrawl-anydoc>=0.2.4",
     "pimd[pdfa]>=2.2.5",
     "markdown-it-py>=3.0",
-    "fastmcp>=4.0.3",
     "httpx>=0.28.1",
     "requests>=2.34.0",
     "python-dotenv>=1.2.3",
 ]
 dynamic = ["version"]
 
+# 关键节点：可选后端按 extras 拆分（0.1.4）。fastmcp/zvec/redis/psycopg2/pymysql
+# 仅在对应 Kit（mcp_kit/zvec_kit/redis_kit/pg_kit/mysql_kit）被实际使用时才需
+# 安装，且均为延迟导入——不装 extras 时 import kit 与基础 Kit 全部可用。
+# 单机 SQLite 场景安装基础版即可，避免拖入重依赖（原全量安装多占约 60MB）。
+[project.optional-dependencies]
+pg = ["psycopg2-binary>=2.9.12"]
+mysql = ["pymysql>=1.2.0"]
+redis = ["redis>=5.0.0"]
+zvec = ["zvec>=0.6.0"]
+mcp = ["fastmcp>=4.0.3"]
+all = [
+    "psycopg2-binary>=2.9.12",
+    "pymysql>=1.2.0",
+    "redis>=5.0.0",
+    "zvec>=0.6.0",
+    "fastmcp>=4.0.3",
+]
+
 [tool.hatch.version]
 path = "kit/__init__.py"