vikingdb-python-sdk 0.1.14__tar.gz → 0.1.15__tar.gz

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: vikingdb-python-sdk
-Version: 0.1.14
+Version: 0.1.15
 Summary: vikingdb Python SDK
 License: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/volc-vikingdb-python-sdk/README.md
@@ -24,7 +24,6 @@ Requires-Dist: typing_extensions>=4.9.0
 Requires-Dist: urllib3>=1.21.1
 Requires-Dist: volcengine>=1.0.0
 Requires-Dist: aiohttp>=3.10.0
-Dynamic: license-file
 
 ## VikingDB Python SDK (v2)
 

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb/knowledge/collection.py

@@ -8,7 +8,7 @@ import json
 from typing import Mapping, Optional, Union, List
 import warnings
 
-from .models.base import CollectionMeta, Model
+from .models.base import CollectionMeta, CommonResponse, Model
 from .models.doc import DocInfo, ListDocsResponse, AddDocRequest, ListDocsRequest, MetaItem, AddDocV2Request, AddDocResponse
 from .models.point import (
     ListPointsResponse,
@@ -74,10 +74,11 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ):
+    ) -> CommonResponse:
         payload = {**self._meta_payload, "doc_id": doc_id}
         res = self.client.json_exception("DeleteDoc", {}, json.dumps(payload), headers=headers, timeout=timeout)
-        return res
+        response = CommonResponse.model_validate(res)
+        return response
 
     def get_doc(
         self,
@@ -123,10 +124,11 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ):
+    ) -> CommonResponse:
         payload = {**self._meta_payload, "doc_id": doc_id, "meta": [item.model_dump(by_alias=True) for item in meta]}
         res = self.client.json_exception("UpdateDocMeta", {}, json.dumps(payload), headers=headers, timeout=timeout)
-        return res
+        response = CommonResponse.model_validate(res)
+        return response
 
     def update_doc(
         self,
@@ -135,10 +137,11 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ):
+    ) -> CommonResponse:
         payload = {**self._meta_payload, "doc_id": doc_id, "doc_name": doc_name}
         res = self.client.json_exception("UpdateDoc", {}, json.dumps(payload), headers=headers, timeout=timeout)
-        return res
+        response = CommonResponse.model_validate(res)
+        return response
 
     def get_point(
         self,
@@ -183,7 +186,7 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ):
+    ) -> PointAddResponse:
         req_payload = (
             request.model_dump(by_alias=True, exclude_none=True)  # type: ignore[attr-defined]
             if isinstance(request, Model)
@@ -201,7 +204,7 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ):
+    ) -> CommonResponse:
         upd_payload = (
             update.model_dump(by_alias=True, exclude_none=True)  # type: ignore[attr-defined]
             if isinstance(update, Model)
@@ -209,7 +212,8 @@ class KnowledgeCollection:
         )
         payload = {**self._meta_payload, "point_id": point_id, **upd_payload}
         res = self.client.json_exception("UpdatePoint", {}, json.dumps(payload), headers=headers, timeout=timeout)
-        return res
+        response = CommonResponse.model_validate(res)
+        return response
 
     def delete_point(
         self,
@@ -217,7 +221,7 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ):
+    ) -> CommonResponse:
         req_payload = (
             request.model_dump(by_alias=True, exclude_none=True)  # type: ignore[attr-defined]
             if isinstance(request, Model)
@@ -225,7 +229,8 @@ class KnowledgeCollection:
         )
         payload = {**self._meta_payload, **req_payload}
         res = self.client.json_exception("DeletePoint", {}, json.dumps(payload), headers=headers, timeout=timeout)
-        return res
+        response = CommonResponse.model_validate(res)
+        return response
 
     def search_collection(
         self,

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb/knowledge/models/base.py

@@ -29,10 +29,10 @@ class CollectionMeta(Model):
 
 
 class CommonResponse(Model):
-    api: Optional[str] = Field(default=None, alias="api")
+    code: Optional[int] = Field(default=None, alias="code")
     message: Optional[str] = Field(default=None, alias="message")
-    code: Optional[str] = Field(default=None, alias="code")
     request_id: Optional[str] = Field(default=None, alias="request_id")
+    data: Optional[Any] = Field(default=None, alias="data")
 
 
 class DataApiResponse(Model):

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb/knowledge/models/point.py

@@ -54,8 +54,8 @@ class UpdatePointRequest(Model):
 class ListPointsRequest(Model):
     offset: int = Field(default=0, alias="offset")
     limit: int = Field(default=-1, alias="limit")
-    doc_ids: Optional[Dict[str, Any]] = Field(default=None, alias="doc_ids")
-    point_ids: Optional[Dict[str, Any]] = Field(default=None, alias="point_ids")
+    doc_ids: Optional[List[str]] = Field(default=None, alias="doc_ids")
+    point_ids: Optional[List[str]] = Field(default=None, alias="point_ids")
     get_attachment_link: Optional[bool] = Field(default=False, alias="get_attachment_link")
 
 class DeletePointRequest(Model):

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb/knowledge/models/rerank.py

@@ -18,7 +18,7 @@ __all__ = [
 
 class RerankDataItem(Model):
     query: object = Field(alias="query")
-    content: str = Field(alias="content")
+    content: str = Field(default=None, alias="content")
     title: Optional[str] = Field(default=None, alias="title")
     image: Union[str, List[str]] = Field(default=None, alias="image")
 

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb/knowledge/models/search.py

@@ -21,7 +21,7 @@ __all__ = [
 ]
 
 class SearchCollectionRequest(Model):
-    query: Any = Field(alias="query")
+    query: str = Field(alias="query")
     limit: int = Field(default=10, alias="limit")
     dense_weight: float = Field(default=0.5, alias="dense_weight")
     rerank_switch: bool = Field(default=False, alias="rerank_switch")
@@ -40,8 +40,8 @@ class SearchResponse(DataApiResponse):
 
 
 class SearchKnowledgeRequest(Model):
-    query: Any = Field(alias="query")
-    image_query: Optional[Any] = Field(default=None, alias="image_query")
+    query: str = Field(alias="query")
+    image_query: Optional[str] = Field(default=None, alias="image_query")
     pre_processing: Optional[Dict[str, Any]] = Field(default=None, alias="pre_processing")
     post_processing: Optional[Dict[str, Any]] = Field(default=None, alias="post_processing")
     query_param: Optional[Dict[str, Any]] = Field(default=None, alias="query_param")

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb/version.py

@@ -1,4 +1,4 @@
 # Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd.
 # SPDX-License-Identifier: Apache-2.0
 
-__version__ = '0.1.14'
+__version__ = '0.1.15'

{vikingdb_python_sdk-0.1.14 → vikingdb_python_sdk-0.1.15}/vikingdb_python_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: vikingdb-python-sdk
-Version: 0.1.14
+Version: 0.1.15
 Summary: vikingdb Python SDK
 License: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/volc-vikingdb-python-sdk/README.md
@@ -24,7 +24,6 @@ Requires-Dist: typing_extensions>=4.9.0
 Requires-Dist: urllib3>=1.21.1
 Requires-Dist: volcengine>=1.0.0
 Requires-Dist: aiohttp>=3.10.0
-Dynamic: license-file
 
 ## VikingDB Python SDK (v2)