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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vikingdb-python-sdk
-Version: 0.1.13
+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

{vikingdb_python_sdk-0.1.13 → vikingdb_python_sdk-0.1.15}/vikingdb/_client.py

@@ -144,7 +144,7 @@ class Client(Service, ABC):
             response = self.session.post(
                 url,
                 headers=request.headers,
-                data=request.body,
+                data=request.body.encode("utf-8"),
                 timeout=request_timeout,
             )
         except Exception as exc:
@@ -154,9 +154,7 @@ class Client(Service, ABC):
                     message=f"failed to run session.post {api}: {exc}",
                 ) from exc
 
-        payload_text_attr = getattr(response, "text", "")
-        payload_text = payload_text_attr if isinstance(payload_text_attr, str) else ""
-        payload_text = payload_text or ""
+        payload_text = response.content.decode("utf-8", errors="replace") if response.content else ""
 
         if response.status_code != 200:
             error = VikingAPIException.from_response(
@@ -167,7 +165,7 @@ class Client(Service, ABC):
             raise error
 
         try:
-            return response.json()
+            return json.loads(response.content.decode("utf-8"))
         except (ValueError, JSONDecodeError) as exc:
             raise VikingAPIException(
                 DEFAULT_UNKNOWN_ERROR_CODE,
@@ -287,14 +285,12 @@ class Client(Service, ABC):
         response = self.session.post(
             url,
             headers=request.headers,
-            data=request.body,
+            data=request.body.encode("utf-8"),
             stream=True,
             timeout=request_timeout,
         )
         if response.status_code != 200:
-            payload_text_attr = getattr(response, "text", "")
-            payload_text = payload_text_attr if isinstance(payload_text_attr, str) else ""
-            payload_text = payload_text or ""
+            payload_text = response.content.decode("utf-8", errors="replace") if response.content else ""
             raise VikingAPIException.from_response(
                 payload_text,
                 request_id=request_id,

{vikingdb_python_sdk-0.1.13 → 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.13 → 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.13 → 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.13 → 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.13 → 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.13 → vikingdb_python_sdk-0.1.15}/vikingdb/vector/models/index.py

@@ -20,6 +20,7 @@ class FetchDataInIndexRequest(Model):
 class IndexDataItem(DataItem):
     dense_dim: Optional[int] = Field(default=None, alias="dense_dim")
     dense_vector: Optional[List[float]] = Field(default=None, alias="dense_vector")
+    sparse_vector: Optional[Dict[str, float]] = Field(default=None, alias="sparse_vector")
 
 
 class FetchDataInIndexResult(Model):

{vikingdb_python_sdk-0.1.13 → 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.13'
+__version__ = '0.1.15'

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vikingdb-python-sdk
-Version: 0.1.13
+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