stackgen-sdk 0.5.0rc4__tar.gz → 0.5.0rc5__tar.gz

{stackgen_sdk-0.5.0rc4 → stackgen_sdk-0.5.0rc5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: stackgen-sdk
-Version: 0.5.0rc4
+Version: 0.5.0rc5
 Summary: StackGen SDK — Aiden, SRE, and Vault from one client
 Project-URL: Homepage, https://appcd-dev.github.io/stackgen-sdk/
 Project-URL: Documentation, https://appcd-dev.github.io/stackgen-sdk/

{stackgen_sdk-0.5.0rc4 → stackgen_sdk-0.5.0rc5}/src/stackgen/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.5.0rc4'
-__version_tuple__ = version_tuple = (0, 5, 0, 'rc4')
+__version__ = version = '0.5.0rc5'
+__version_tuple__ = version_tuple = (0, 5, 0, 'rc5')
 
 __commit_id__ = commit_id = None

{stackgen_sdk-0.5.0rc4 → stackgen_sdk-0.5.0rc5}/src/stackgen/context_graph/namespace.py

@@ -68,9 +68,11 @@ class ContextGraphNamespace:
         """GET /records/types"""
         return self._req(self._url("/records/types"))
 
-    def get_record(self, type: str, id: str) -> Any:
-        """GET /records/{type}/{id}"""
-        return self._req(self._url(f"/records/{_escape(type)}/{_escape(id)}"))
+    def get_record(self, type: str, id: str, **params: Any) -> Any:
+        """GET /records/{type}/{id} (optional ``include_facets``)."""
+        return self._req(
+            self._url(f"/records/{_escape(type)}/{_escape(id)}", params)
+        )
 
     def query_records(self, body: dict[str, Any]) -> Any:
         """POST /records/query"""
@@ -178,8 +180,16 @@ class ContextGraphNamespace:
         """GET /entity-link-rules"""
         return self._req(self._url("/entity-link-rules", params))
 
-    def search_graph_tools(self, body: dict[str, Any]) -> Any:
-        """POST /graph/tools/search"""
+    def search_graph_tools(self, query_or_body: str | dict[str, Any]) -> Any:
+        """POST /graph/tools/search.
+
+        Pass a query string or a full GraphSearchRequest body.
+        """
+        body = (
+            {"query": query_or_body}
+            if isinstance(query_or_body, str)
+            else query_or_body
+        )
         payload = json.dumps(body).encode("utf-8")
         return self._req(
             self._url("/graph/tools/search"),