vikingdb-python-sdk 0.1.24__tar.gz → 0.1.26__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (47) hide show
  1. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/PKG-INFO +1 -1
  2. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/__init__.py +6 -1
  3. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/client.py +2 -0
  4. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/collection.py +35 -1
  5. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/doc.py +41 -1
  6. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/version.py +1 -1
  7. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb_python_sdk.egg-info/PKG-INFO +1 -1
  8. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/LICENSE.txt +0 -0
  9. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/README.md +0 -0
  10. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/pyproject.toml +0 -0
  11. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/setup.cfg +0 -0
  12. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/__init__.py +0 -0
  13. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/_client.py +0 -0
  14. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/auth.py +0 -0
  15. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/exceptions.py +0 -0
  16. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/exceptions.py +0 -0
  17. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/__init__.py +0 -0
  18. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/base.py +0 -0
  19. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/chat.py +0 -0
  20. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/point.py +0 -0
  21. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/rerank.py +0 -0
  22. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/search.py +0 -0
  23. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/knowledge/models/service_chat.py +0 -0
  24. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/memory/__init__.py +0 -0
  25. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/memory/client.py +0 -0
  26. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/memory/collection.py +0 -0
  27. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/memory/exceptions.py +0 -0
  28. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/memory/types.py +0 -0
  29. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/request_options.py +0 -0
  30. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/__init__.py +0 -0
  31. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/base.py +0 -0
  32. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/client.py +0 -0
  33. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/collection.py +0 -0
  34. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/embedding.py +0 -0
  35. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/exceptions.py +0 -0
  36. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/index.py +0 -0
  37. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/models/__init__.py +0 -0
  38. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/models/base.py +0 -0
  39. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/models/collection.py +0 -0
  40. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/models/embedding.py +0 -0
  41. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/models/index.py +0 -0
  42. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/models/rerank.py +0 -0
  43. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb/vector/rerank.py +0 -0
  44. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb_python_sdk.egg-info/SOURCES.txt +0 -0
  45. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb_python_sdk.egg-info/dependency_links.txt +0 -0
  46. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb_python_sdk.egg-info/requires.txt +0 -0
  47. {vikingdb_python_sdk-0.1.24 → vikingdb_python_sdk-0.1.26}/vikingdb_python_sdk.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vikingdb-python-sdk
3
- Version: 0.1.24
3
+ Version: 0.1.26
4
4
  Summary: vikingdb Python SDK
5
5
  License: Apache-2.0
6
6
  Project-URL: Documentation, https://github.com/volcengine/vikingdb-python-sdk/blob/main/README.md
@@ -38,7 +38,7 @@ from .exceptions import (
38
38
  PointNotExistException,
39
39
  )
40
40
  from .models.base import CollectionMeta
41
- from .models.doc import DocInfo, DocInfo, ListDocsResponse, AddDocRequest, AddDocV2Request, ListDocsRequest, DedupOptions, MetaItem, AddDocResponse, AddDocResponseData
41
+ from .models.doc import DocInfo, DocInfo, ListDocsResponse, ListDocsV2Response, SearchDocsByFilterResponse, AddDocRequest, AddDocV2Request, ListDocsRequest, ListDocsV2Request, SearchDocsByFilterRequest, ListDocsFilter, DedupOptions, MetaItem, AddDocResponse, AddDocResponseData
42
42
  from .models.point import (
43
43
  PointInfo,
44
44
  ListPointsResponse,
@@ -104,6 +104,11 @@ __all__ = [
104
104
  "AddDocResponse",
105
105
  "AddDocResponseData",
106
106
  "ListDocsRequest",
107
+ "ListDocsV2Request",
108
+ "SearchDocsByFilterRequest",
109
+ "ListDocsFilter",
110
+ "ListDocsV2Response",
111
+ "SearchDocsByFilterResponse",
107
112
  "PointInfo",
108
113
  "DedupOptions",
109
114
  "MetaItem",
@@ -66,6 +66,8 @@ class VikingKnowledge(Client):
66
66
  "DeleteDoc": ApiInfo("POST", "/api/knowledge/doc/delete", {}, {}, header),
67
67
  "GetDocInfo": ApiInfo("POST", "/api/knowledge/doc/info", {}, {}, header),
68
68
  "ListDocs": ApiInfo("POST", "/api/knowledge/doc/list", {}, {}, header),
69
+ "ListDocsV2": ApiInfo("POST", "/api/knowledge/doc/v2/list", {}, {}, header),
70
+ "SearchDocsByFilter": ApiInfo("POST", "/api/knowledge/doc/v2/search_by_filter", {}, {}, header),
69
71
  "UpdateDocMeta": ApiInfo("POST", "/api/knowledge/doc/update_meta", {}, {}, header),
70
72
  "UpdateDoc": ApiInfo("POST", "/api/knowledge/doc/update", {}, {}, header),
71
73
  # Point
@@ -9,7 +9,7 @@ from typing import Mapping, Optional, Union, List
9
9
  import warnings
10
10
 
11
11
  from .models.base import CollectionMeta, CommonResponse, Model
12
- from .models.doc import DocInfo, ListDocsResponse, AddDocRequest, ListDocsRequest, MetaItem, AddDocV2Request, AddDocResponse
12
+ from .models.doc import DocInfo, ListDocsResponse, ListDocsV2Response, SearchDocsByFilterResponse, AddDocRequest, ListDocsRequest, ListDocsV2Request, SearchDocsByFilterRequest, MetaItem, AddDocV2Request, AddDocResponse
13
13
  from .models.point import (
14
14
  ListPointsResponse,
15
15
  PointInfo,
@@ -117,6 +117,40 @@ class KnowledgeCollection:
117
117
  response = ListDocsResponse.parse_with(res)
118
118
  return response
119
119
 
120
+ def list_docs_v2(
121
+ self,
122
+ request: Union[ListDocsV2Request, Mapping[str, object]],
123
+ *,
124
+ headers: Optional[Mapping[str, str]] = None,
125
+ timeout: Optional[int] = None,
126
+ ) -> ListDocsV2Response:
127
+ req_payload = (
128
+ request.model_dump(by_alias=True, exclude_none=True) # type: ignore[attr-defined]
129
+ if isinstance(request, Model)
130
+ else dict(request)
131
+ )
132
+ payload: dict = {**self._meta_payload, **req_payload}
133
+ res = self.client.json_exception("ListDocsV2", {}, json.dumps(payload), headers=headers, timeout=timeout)
134
+ response = ListDocsV2Response.parse_with(res)
135
+ return response
136
+
137
+ def search_docs_by_filter(
138
+ self,
139
+ request: Union[SearchDocsByFilterRequest, Mapping[str, object]],
140
+ *,
141
+ headers: Optional[Mapping[str, str]] = None,
142
+ timeout: Optional[int] = None,
143
+ ) -> SearchDocsByFilterResponse:
144
+ req_payload = (
145
+ request.model_dump(by_alias=True, exclude_none=True) # type: ignore[attr-defined]
146
+ if isinstance(request, Model)
147
+ else dict(request)
148
+ )
149
+ payload: dict = {**self._meta_payload, **req_payload}
150
+ res = self.client.json_exception("SearchDocsByFilter", {}, json.dumps(payload), headers=headers, timeout=timeout)
151
+ response = SearchDocsByFilterResponse.parse_with(res)
152
+ return response
153
+
120
154
  def update_doc_meta(
121
155
  self,
122
156
  doc_id: str,
@@ -13,6 +13,9 @@ from .base import DataApiResponse, Model
13
13
  __all__ = [
14
14
  "AddDocRequest",
15
15
  "ListDocsRequest",
16
+ "ListDocsFilter",
17
+ "ListDocsV2Request",
18
+ "SearchDocsByFilterRequest",
16
19
  "LarkFile",
17
20
  "MetaItem",
18
21
  "DedupOptions",
@@ -24,17 +27,32 @@ __all__ = [
24
27
  "DocInfo",
25
28
  "ListDocsResult",
26
29
  "ListDocsResponse",
30
+ "ListDocsV2Result",
31
+ "ListDocsV2Response",
32
+ "SearchDocsByFilterResult",
33
+ "SearchDocsByFilterResponse",
27
34
  "AddDocResponseData",
28
35
  "AddDocResponse",
29
36
  ]
30
37
 
38
+ class ListDocsFilter(Model):
39
+ doc_id_list: Optional[List[str]] = Field(default=None, alias="doc_id_list")
40
+
31
41
  class ListDocsRequest(Model):
32
42
  offset: int = Field(default=0, alias="offset")
33
43
  limit: int = Field(default=-1, alias="limit")
34
44
  doc_type: Optional[str] = Field(default=None, alias="doc_type")
35
- filter: Optional[Dict[str, Any]] = Field(default=None, alias="filter")
45
+ filter: Optional["ListDocsFilter"] = Field(default=None, alias="filter")
36
46
  return_token_usage: Optional[bool] = Field(default=False, alias="return_token_usage")
37
47
 
48
+ class ListDocsV2Request(Model):
49
+ limit: Optional[int] = Field(default=None, alias="limit")
50
+ next_token: Optional[str] = Field(default=None, alias="next_token")
51
+
52
+ class SearchDocsByFilterRequest(Model):
53
+ filter: Dict[str, Any] = Field(alias="filter")
54
+ limit: Optional[int] = Field(default=None, alias="limit")
55
+
38
56
  class LarkFile(Model):
39
57
  url: Optional[str] = Field(default=None, alias="url")
40
58
  obj_type: Optional[str] = Field(default=None, alias="obj_type")
@@ -127,6 +145,28 @@ class ListDocsResult(Model):
127
145
  class ListDocsResponse(DataApiResponse):
128
146
  result: Optional[ListDocsResult] = None
129
147
 
148
+ class ListDocsV2Result(Model):
149
+ collection_name: Optional[str] = Field(default=None, alias="collection_name")
150
+ doc_list: Sequence[DocInfo] = Field(default_factory=list, alias="doc_list")
151
+ count: Optional[int] = Field(default=None, alias="count")
152
+ total_num: Optional[int] = Field(default=None, alias="total_num")
153
+ has_more: Optional[bool] = Field(default=None, alias="has_more")
154
+ next_token: Optional[str] = Field(default=None, alias="next_token")
155
+
156
+
157
+ class ListDocsV2Response(DataApiResponse):
158
+ result: Optional[ListDocsV2Result] = None
159
+
160
+ class SearchDocsByFilterResult(Model):
161
+ collection_name: Optional[str] = Field(default=None, alias="collection_name")
162
+ doc_list: Sequence[DocInfo] = Field(default_factory=list, alias="doc_list")
163
+ count: Optional[int] = Field(default=None, alias="count")
164
+ total_num: Optional[int] = Field(default=None, alias="total_num")
165
+
166
+
167
+ class SearchDocsByFilterResponse(DataApiResponse):
168
+ result: Optional[SearchDocsByFilterResult] = None
169
+
130
170
  class AddDocResponseData(Model):
131
171
  collection_name: Optional[str] = Field(default=None, alias="collection_name")
132
172
  resource_id: Optional[str] = Field(default=None, alias="resource_id")
@@ -1,4 +1,4 @@
1
1
  # Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd.
2
2
  # SPDX-License-Identifier: Apache-2.0
3
3
 
4
- __version__ = '0.1.24'
4
+ __version__ = '0.1.26'
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vikingdb-python-sdk
3
- Version: 0.1.24
3
+ Version: 0.1.26
4
4
  Summary: vikingdb Python SDK
5
5
  License: Apache-2.0
6
6
  Project-URL: Documentation, https://github.com/volcengine/vikingdb-python-sdk/blob/main/README.md