vikingdb-python-sdk 0.1.27__tar.gz → 0.1.29__tar.gz

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.27
+Version: 0.1.29
 Summary: vikingdb Python SDK
 License: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/vikingdb-python-sdk/blob/main/README.md
@@ -24,6 +24,7 @@ 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.27 → vikingdb_python_sdk-0.1.29}/vikingdb/knowledge/__init__.py

@@ -38,7 +38,7 @@ from .exceptions import (
     PointNotExistException,
 )
 from .models.base import CollectionMeta
-from .models.doc import DocInfo, DocInfo, ListDocsResponse, ListDocsV2Response, SearchDocsByFilterResponse, AddDocRequest, AddDocV2Request, ListDocsRequest, ListDocsV2Request, SearchDocsByFilterRequest, ListDocsFilter, DedupOptions, MetaItem, AddDocResponse, AddDocResponseData
+from .models.doc import DocInfo, DocInfo, ListDocsResponse, ListDocsV2Response, SearchDocsByFilterResponse, AddDocRequest, AddDocV2Request, ListDocsRequest, ListDocsV2Request, SearchDocsByFilterRequest, ListDocsFilter, DedupOptions, MetaItem, AddDocResponse, AddDocResponseData, AddDocV2Response, AddDocV2ResponseData
 from .models.point import (
     PointInfo,
     ListPointsResponse,
@@ -104,6 +104,8 @@ __all__ = [
     "AddDocV2Request",
     "AddDocResponse",
     "AddDocResponseData",
+    "AddDocV2Response",
+    "AddDocV2ResponseData",
     "ListDocsRequest",
     "ListDocsV2Request",
     "SearchDocsByFilterRequest",

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/vikingdb/knowledge/collection.py

@@ -9,7 +9,7 @@ from typing import Mapping, Optional, Union, List
 import warnings
 
 from .models.base import CollectionMeta, CommonResponse, Model
-from .models.doc import DocInfo, ListDocsResponse, ListDocsV2Response, SearchDocsByFilterResponse, AddDocRequest, ListDocsRequest, ListDocsV2Request, SearchDocsByFilterRequest, MetaItem, AddDocV2Request, AddDocResponse
+from .models.doc import DocInfo, ListDocsResponse, ListDocsV2Response, SearchDocsByFilterResponse, AddDocRequest, ListDocsRequest, ListDocsV2Request, SearchDocsByFilterRequest, MetaItem, AddDocV2Request, AddDocResponse, AddDocV2Response
 from .models.point import (
     ListPointsResponse,
     PointInfo,
@@ -57,7 +57,7 @@ class KnowledgeCollection:
         *,
         headers: Optional[Mapping[str, str]] = None,
         timeout: Optional[int] = None,
-    ) -> AddDocResponse:
+    ) -> AddDocV2Response:
         req_payload = (
             request.model_dump(by_alias=True, exclude_none=True)  # type: ignore[attr-defined]
             if isinstance(request, Model)
@@ -65,7 +65,7 @@ class KnowledgeCollection:
         )
         payload = {**self._meta_payload, **req_payload}
         res = self.client.json_exception("AddDocV2", {}, json.dumps(payload), headers=headers, timeout=timeout)
-        response = AddDocResponse.parse_with(res)
+        response = AddDocV2Response.parse_with(res)
         return response
 
     def delete_doc(

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/vikingdb/knowledge/models/doc.py

@@ -33,6 +33,8 @@ __all__ = [
     "SearchDocsByFilterResponse",
     "AddDocResponseData",
     "AddDocResponse",
+    "AddDocV2ResponseData",
+    "AddDocV2Response",
 ]
 
 class ListDocsFilter(Model):
@@ -174,9 +176,16 @@ class AddDocResponseData(Model):
     resource_id: Optional[str] = Field(default=None, alias="resource_id")
     project: Optional[str] = Field(default=None, alias="project")
     doc_id: Optional[str] = Field(default=None, alias="doc_id")
-    task_id: Optional[int] = Field(default=None, alias="task_id")
     dedup_info: Optional[DedupInfo] = Field(default=None, alias="dedup_info")
-    more_info: Optional[str] = Field(default=None, alias="more_info")
 
 class AddDocResponse(DataApiResponse):
     result: Optional[AddDocResponseData] = None
+
+class AddDocV2ResponseData(Model):
+    collection_name: Optional[str] = Field(default=None, alias="collection_name")
+    resource_id: Optional[str] = Field(default=None, alias="resource_id")
+    project: Optional[str] = Field(default=None, alias="project")
+    doc_id: Optional[str] = Field(default=None, alias="doc_id")
+
+class AddDocV2Response(DataApiResponse):
+    result: Optional[AddDocV2ResponseData] = None

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/vikingdb/vector/models/__init__.py

@@ -46,6 +46,7 @@ from .index import (
     FetchDataInIndexRequest,
     FetchDataInIndexResponse,
     FetchDataInIndexResult,
+    Highlight,
     IndexDataItem,
     Instruction,
     ModelRerank,
@@ -60,6 +61,7 @@ from .index import (
     SearchResponse,
     SearchResult,
     TensorRerank,
+    VectorWeightedSumResult,
 )
 
 __all__ = [
@@ -89,6 +91,7 @@ __all__ = [
     "FetchDataInIndexResponse",
     "FetchDataInIndexResult",
     "FullModalData",
+    "Highlight",
     "IndexDataItem",
     "IndexMeta",
     "Instruction",
@@ -111,4 +114,5 @@ __all__ = [
     "UpdateDataRequest",
     "UpdateDataResponse",
     "UpdateDataResult",
+    "VectorWeightedSumResult",
 ]

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/vikingdb/vector/models/index.py

@@ -57,6 +57,11 @@ class SearchBase(RecallBase):
     return_download_url: Optional[bool] = Field(default=None, alias="return_download_url")
     return_analyzed_result: Optional[bool] = Field(default=None, alias="return_analyzed_result")
     return_detail_info: Optional[bool] = Field(default=None, alias="return_detail_info")
+    ignore_unknown_fields: Optional[bool] = Field(default=None, alias="ignore_unknown_fields")
+
+
+class Highlight(Model):
+    enable: Optional[bool] = Field(default=None, alias="enable")
 
 
 class Instruction(Model):
@@ -86,6 +91,7 @@ class SearchItemResult(Model):
     score: Optional[float] = Field(default=None, alias="score")
     origin_score: Optional[float] = Field(default=None, alias="origin_score")
     addition_score: Optional[float] = Field(default=None, alias="addition_score")
+    highlight: Optional[Mapping[str, Mapping[str, List[int]]]] = Field(default=None, alias="highlight")
 
 
 class SearchResult(Model):
@@ -137,6 +143,7 @@ class SearchByKeywordsRequest(SearchBase):
     mode: Optional[str] = Field(default=None, alias="mode")
     keywords: Optional[Sequence[str]] = Field(default=None, alias="keywords")
     query: Optional[str] = Field(default=None, alias="query")
+    highlight: Optional[Highlight] = Field(default=None, alias="highlight")
     case_sensitive: Optional[bool] = Field(default=None, alias="case_sensitive")
     fields: Optional[Sequence[str]] = Field(default=None, alias="fields")
     bm25_k1: Optional[float] = Field(default=None, alias="bm25_k1")
@@ -149,15 +156,31 @@ class SearchByRandomRequest(SearchBase):
 
 class AggRequest(RecallBase):
     op: str = Field(alias="op")
+    group_by: Optional[str] = Field(default=None, alias="group_by")
+    # Deprecated: use group_by instead.
     field: Optional[str] = Field(default=None, alias="field")
+    calc_value_on: Optional[str] = Field(default=None, alias="calc_value_on")
+    ids: Optional[Sequence[Any]] = Field(default=None, alias="ids")
+    weights: Optional[Sequence[float]] = Field(default=None, alias="weights")
     cond: Optional[Mapping[str, Any]] = Field(default=None, alias="cond")
     order: Optional[str] = Field(default=None, alias="order")
 
 
+class VectorWeightedSumResult(Model):
+    vector: List[float] = Field(default_factory=list, alias="vector")
+    found_count: Optional[int] = Field(default=None, alias="found_count")
+    found_ids: List[Any] = Field(default_factory=list, alias="found_ids")
+    weight_sum: Optional[float] = Field(default=None, alias="weight_sum")
+
+
 class AggResult(Model):
     agg: Dict[str, Any] = Field(default_factory=dict, alias="agg")
     op: Optional[str] = Field(default=None, alias="op")
+    group_by: Optional[str] = Field(default=None, alias="group_by")
+    # Deprecated: use group_by instead.
     field: Optional[str] = Field(default=None, alias="field")
+    calc_value_on: Optional[str] = Field(default=None, alias="calc_value_on")
+    vector_weighted_sum: Optional[VectorWeightedSumResult] = Field(default=None, alias="vector_weighted_sum")
 
 
 class AggResponse(DataApiResponse):

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/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.27'
+__version__ = '0.1.29'

{vikingdb_python_sdk-0.1.27 → vikingdb_python_sdk-0.1.29}/vikingdb_python_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.27
+Version: 0.1.29
 Summary: vikingdb Python SDK
 License: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/vikingdb-python-sdk/blob/main/README.md
@@ -24,6 +24,7 @@ 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)