Praxos-python 0.3.2__tar.gz → 0.3.3__tar.gz

{praxos_python-0.3.2 → praxos_python-0.3.3}/.gitignore

@@ -174,4 +174,6 @@ cython_debug/
 .pypirc
 
 .vscode/
-tests/*
+tests/*
+src/.DS_Store
+*/.DS_Store

{praxos_python-0.3.2 → praxos_python-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Praxos-python
-Version: 0.3.2
+Version: 0.3.3
 Summary: Python SDK for Praxos with Intelligent Search
 Author-email: Masoud Kermani Poor <masoud@praxos.ai>, Soheil Sadabadi <soheil@praxos.ai>
 License-File: LICENSE

{praxos_python-0.3.2 → praxos_python-0.3.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "Praxos-python"
-version = "0.3.2"
+version = "0.3.3"
 description = "Python SDK for Praxos with Intelligent Search"
 authors = [{ name = "Masoud Kermani Poor", email = "masoud@praxos.ai" }, {name = "Soheil Sadabadi", email = "soheil@praxos.ai"}]
 readme = "README.md"

{praxos_python-0.3.2 → praxos_python-0.3.3}/src/praxos_python/models/environment.py

@@ -676,6 +676,28 @@ class SyncEnvironment(BaseEnvironmentAttributes):
             json_data=payload
         )
         return SyncSource(client=self._client, **response_data)
+
+    def enrich(self, node_ids: Union[str, List[str]], k: int = 2) -> Dict[str, Any]:
+        """
+        Enriches a given node or list of nodes by finding the closest entity and retrieving all entities up to k hops away.
+        
+        Args:
+            node_ids: A single node ID or a list of node IDs to enrich.
+            k: The number of hops to traverse for enrichment.
+        
+        Returns:
+            A dictionary containing the enriched data for each node.
+        """
+        if not node_ids:
+            raise ValueError("node_ids cannot be empty")
+
+        payload = {
+            "node_ids": node_ids,
+            "k": k,
+        }
+
+        response_data = self._client._request("POST", "/enrich", json_data=payload)
+        return response_data
     
     def get_sources(self) -> List[SyncSource]:
         """Gets all sources for the environment."""