vikingdb-python-sdk 0.1.4__tar.gz → 0.1.7__tar.gz

{vikingdb_python_sdk-0.1.4/vikingdb_python_sdk.egg-info → vikingdb_python_sdk-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.4
+Version: 0.1.7
 Summary: vikingdb Python SDK
 License-Expression: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/volc-vikingdb-python-sdk/README.md
@@ -21,7 +21,7 @@ Requires-Dist: numpy>=1.21.6
 Requires-Dist: httpx>=0.24.1
 Requires-Dist: requests>=2.28.1
 Requires-Dist: pydantic>=2.5.3
-Requires-Dist: urllib3>=2.0.7
+Requires-Dist: urllib3>=1.21.1
 Requires-Dist: volcengine>=1.0.0
 Requires-Dist: aiohttp>=3.10.0
 Dynamic: license-file

{vikingdb_python_sdk-0.1.4 → vikingdb_python_sdk-0.1.7}/pyproject.toml

@@ -25,7 +25,7 @@ dependencies = [
     "httpx>=0.24.1",
     "requests>=2.28.1",
     "pydantic>=2.5.3",
-    "urllib3>=2.0.7",
+    "urllib3>=1.21.1",
     "volcengine>=1.0.0",
     "aiohttp>=3.10.0",
 ]

{vikingdb_python_sdk-0.1.4 → vikingdb_python_sdk-0.1.7}/vikingdb/memory/client.py

@@ -123,6 +123,13 @@ class VikingMem(Client):
                 {},
                 _get_common_viking_request_header(),
             ),
+            "TriggerUpdateProfile": ApiInfo(
+                "POST",
+                "/api/memory/profile/trigger_update",
+                {},
+                {},
+                _get_common_viking_request_header(),
+            ),
             # Event APIs
             "AddEvent": ApiInfo(
                 "POST",

{vikingdb_python_sdk-0.1.4 → vikingdb_python_sdk-0.1.7}/vikingdb/memory/collection.py

@@ -52,7 +52,8 @@ class Collection:
         assistant_id=None, 
         group_id=None, 
         headers=None,
-        timeout=None
+        timeout=None,
+        update_profiles=None,
     ):
         """
         Add an event to the collection
@@ -65,6 +66,7 @@ class Collection:
             group_id (str, optional): Group ID
             headers (dict, optional): Custom request headers
             timeout (int, optional): Timeout in seconds
+            update_profiles (optional): Profiles to update along with the event
             
         Returns:
             dict: Event information
@@ -83,6 +85,8 @@ class Collection:
             params["assistant_id"] = assistant_id
         if group_id is not None:
             params["group_id"] = group_id
+        if update_profiles is not None:
+            params["update_profiles"] = update_profiles
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
@@ -97,7 +101,8 @@ class Collection:
         assistant_id=None, 
         group_id=None, 
         headers=None,
-        timeout=None
+        timeout=None,
+        update_profiles=None,
     ):
         """
         Add an event to the collection asynchronously
@@ -110,6 +115,7 @@ class Collection:
             group_id (str, optional): Group ID
             headers (dict, optional): Custom request headers
             timeout (int, optional): Timeout in seconds
+            update_profiles (optional): Profiles to update along with the event
             
         Returns:
             dict: Event information
@@ -128,6 +134,8 @@ class Collection:
             params["assistant_id"] = assistant_id
         if group_id is not None:
             params["group_id"] = group_id
+        if update_profiles is not None:
+            params["update_profiles"] = update_profiles
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
 
@@ -566,6 +574,65 @@ class Collection:
 
         res = await self.client.async_json_exception("DeleteProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
         return res
+
+    def trigger_update_profile(
+        self,
+        update_profile_type=None,
+        filters=None,
+        headers=None,
+        timeout=None,
+    ):
+        """
+        Trigger profile update tasks for the collection
+        
+        Args:
+            update_profile_type (list|str, optional): Profile types to update
+            filters (dict, optional): Filter conditions, e.g., user_id, end_time
+            headers (dict, optional): Custom request headers
+            timeout (int, optional): Timeout in seconds
+        
+        Returns:
+            dict: Trigger result
+        """
+        params = {}
+        if self.collection_name is not None:
+            params["collection_name"] = self.collection_name
+        if self.project_name is not None:
+            params["project_name"] = self.project_name
+        if update_profile_type is not None:
+            params["update_profile_type"] = update_profile_type
+        if filters is not None:
+            params["filters"] = filters
+        if self.resource_id is not None:
+            params["resource_id"] = self.resource_id
+
+        res = self.client.json_exception("TriggerUpdateProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        return res
+
+    async def async_trigger_update_profile(
+        self,
+        update_profile_type=None,
+        filters=None,
+        headers=None,
+        timeout=None,
+    ):
+        """
+        Trigger profile update tasks for the collection asynchronously
+        """
+        params = {}
+        if self.collection_name is not None:
+            params["collection_name"] = self.collection_name
+        if self.project_name is not None:
+            params["project_name"] = self.project_name
+        if update_profile_type is not None:
+            params["update_profile_type"] = update_profile_type
+        if filters is not None:
+            params["filters"] = filters
+        if self.resource_id is not None:
+            params["resource_id"] = self.resource_id
+
+        res = await self.client.async_json_exception("TriggerUpdateProfile", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
+        return res
     
     def batch_delete_profile(
         self, 
@@ -635,7 +702,8 @@ class Collection:
         metadata=None, 
         profiles=None,
         headers=None,
-        timeout=None
+        timeout=None,
+        store_file=None
     ):
         """
         Add session messages to the collection
@@ -653,7 +721,7 @@ class Collection:
         """
         params = {
             "session_id": session_id,
-            "messages": messages
+            "messages": messages,
         }
         if self.collection_name is not None:
             params["collection_name"] = self.collection_name
@@ -665,6 +733,8 @@ class Collection:
             params["profiles"] = profiles
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
+        if store_file is not None:
+            params["store_file"] = store_file
 
         res = self.client.json_exception("AddSession", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
         return res
@@ -676,7 +746,8 @@ class Collection:
         metadata=None, 
         profiles=None,
         headers=None,
-        timeout=None
+        timeout=None,
+        store_file=None
     ):
         """
         Add session messages to the collection asynchronously
@@ -694,7 +765,7 @@ class Collection:
         """
         params = {
             "session_id": session_id,
-            "messages": messages
+            "messages": messages,
         }
         if self.collection_name is not None:
             params["collection_name"] = self.collection_name
@@ -706,6 +777,8 @@ class Collection:
             params["profiles"] = profiles
         if self.resource_id is not None:
             params["resource_id"] = self.resource_id
+        if store_file is not None:
+            params["store_file"] = store_file
 
         res = await self.client.async_json_exception("AddSession", {}, json.dumps(params, cls=EnumEncoder), headers=headers, timeout=timeout)
         return res

{vikingdb_python_sdk-0.1.4 → vikingdb_python_sdk-0.1.7}/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.4'
+__version__ = '0.1.7'

{vikingdb_python_sdk-0.1.4 → vikingdb_python_sdk-0.1.7/vikingdb_python_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vikingdb-python-sdk
-Version: 0.1.4
+Version: 0.1.7
 Summary: vikingdb Python SDK
 License-Expression: Apache-2.0
 Project-URL: Documentation, https://github.com/volcengine/volc-vikingdb-python-sdk/README.md
@@ -21,7 +21,7 @@ Requires-Dist: numpy>=1.21.6
 Requires-Dist: httpx>=0.24.1
 Requires-Dist: requests>=2.28.1
 Requires-Dist: pydantic>=2.5.3
-Requires-Dist: urllib3>=2.0.7
+Requires-Dist: urllib3>=1.21.1
 Requires-Dist: volcengine>=1.0.0
 Requires-Dist: aiohttp>=3.10.0
 Dynamic: license-file

{vikingdb_python_sdk-0.1.4 → vikingdb_python_sdk-0.1.7}/vikingdb_python_sdk.egg-info/requires.txt

@@ -2,6 +2,6 @@ numpy>=1.21.6
 httpx>=0.24.1
 requests>=2.28.1
 pydantic>=2.5.3
-urllib3>=2.0.7
+urllib3>=1.21.1
 volcengine>=1.0.0
 aiohttp>=3.10.0