datasetseek 0.1.0__py3-none-any.whl

datasetseek/__init__.py

@@ -0,0 +1,5 @@
+from .client import Client
+from .exceptions import DatasetSeekError, AuthenticationError, APIError
+
+__version__ = "0.1.0"
+__all__ = ["Client", "DatasetSeekError", "AuthenticationError", "APIError"]

datasetseek/client.py

@@ -0,0 +1,39 @@
+import requests
+from .exceptions import AuthenticationError, APIError
+
+
+class Client:
+    def __init__(self, api_key: str, base_url: str = "http://127.0.0.1:8001/sdk"):
+        """
+        初始化客户端
+        :param api_key: 你的 API Key (dsk_开头)
+        :param base_url: 老系统的服务端地址 (上线后改成你的公网域名)
+        """
+        if not api_key:
+            raise ValueError("api_key 不能为空")
+
+        self.api_key = api_key
+        self.base_url = base_url.rstrip("/")
+        self.session = requests.Session()
+        # 自动加上 Bearer 前缀
+        self.session.headers.update({
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        })
+
+    def search(self, q: str, limit: int = 20):
+        """搜索数据集"""
+        url = f"{self.base_url}/v1/search"
+        params = {"q": q, "limit": limit}
+
+        try:
+            resp = self.session.get(url, params=params, timeout=10)
+        except requests.RequestException as e:
+            raise APIError(f"网络请求失败: {e}")
+
+        if resp.status_code == 401:
+            raise AuthenticationError("API Key 无效或已过期")
+        if resp.status_code != 200:
+            raise APIError(f"服务端错误: HTTP {resp.status_code} - {resp.text}")
+
+        return resp.json().get("data", [])

datasetseek/exceptions.py

@@ -0,0 +1,11 @@
+class DatasetSeekError(Exception):
+    """所有异常的基类"""
+    pass
+
+class AuthenticationError(DatasetSeekError):
+    """API Key 无效或未提供"""
+    pass
+
+class APIError(DatasetSeekError):
+    """服务端返回错误"""
+    pass

datasetseek-0.1.0.dist-info/METADATA

@@ -0,0 +1,22 @@
+Metadata-Version: 2.5
+Name: datasetseek
+Version: 0.1.0
+Summary: DatasetSeek 官方 Python SDK
+Requires-Python: >=3.8
+Requires-Dist: requests>=2.28
+Description-Content-Type: text/markdown
+
+# datasetseek
+
+DatasetSeek 官方 Python SDK。
+
+## 安装
+pip install datasetseek-test
+
+## 快速开始
+```python
+from datasetseek import Client
+
+client = Client(api_key="你的dsk_key")
+results = client.search("测试")
+print(results)

datasetseek-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+datasetseek/__init__.py,sha256=3_gGr4WaoIS6GnnWAIPNo93FZDJ1T8TCrWmaECkb6Ic,201
+datasetseek/client.py,sha256=oyyNU7eXLKd63P1cVJ5UDAVZE849IesdI0RTIG8YX0c,1400
+datasetseek/exceptions.py,sha256=TM6t6FeI4opyvzY3yRHmv9l6ZaUmQHZtOYVMgHnNWoU,253
+datasetseek-0.1.0.dist-info/METADATA,sha256=NyuYcQjkC0EiuOpv5SBCsnPm8qicUqyq4Ywl-pYjBP4,424
+datasetseek-0.1.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+datasetseek-0.1.0.dist-info/RECORD,,

datasetseek-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.4
+Root-Is-Purelib: true
+Tag: py3-none-any