polly-python 4.6.0__tar.gz → 4.7.0__tar.gz

{polly_python-4.6.0/polly_python.egg-info → polly_python-4.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polly_python
-Version: 4.6.0
+Version: 4.7.0
 Summary: Polly SDK
 Home-page: https://github.com/ElucidataInc/polly-python
 Project-URL: Documentation, https://docs.elucidata.io

polly_python-4.7.0/polly/__init__.py

@@ -0,0 +1 @@
+__version__ = "4.7.0"

{polly_python-4.6.0 → polly_python-4.7.0}/polly/help.py

@@ -273,8 +273,11 @@ def get_txt(
         and (not function_name or function_name == fun.split(".")[-1])
         and should_show_fun
     ):
-        import rst2txt
-        from docutils.core import publish_string
+        try:
+            import rst2txt
+            from docutils.core import publish_string
+        except ImportError:
+            return None
 
         # converting rst to txt
         txt = publish_string(

{polly_python-4.6.0 → polly_python-4.7.0}/polly/polly_kg.py

@@ -221,21 +221,15 @@ class PollyKG:
         """
         if not query or query == "":
             raise InvalidParameterException("query")
-
-        # Validate query_type - only CYPHER is supported
+        query_url = f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/queries/"
         if query_type != "CYPHER":
             raise InvalidParameterException(
                 f"Invalid query_type: '{query_type}'. Only 'CYPHER' query type is supported."
             )
-
-        query_url = f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/queries/"
         payload = {
             "data": {
                 "type": "kg_query",
-                "attributes": {
-                    "query_string": query,
-                    "query_type": query_type,
-                },
+                "attributes": {"query_string": query, "query_type": query_type},
             }
         }
         try:
@@ -328,14 +322,11 @@ class PollyKG:
         """
         if not query or query == "":
             raise InvalidParameterException("query")
-
-        # Validate query_type - only CYPHER is supported
+        query_url = f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/queries/"
         if query_type != "CYPHER":
             raise InvalidParameterException(
                 f"Invalid query_type: '{query_type}'. Only 'CYPHER' query type is supported."
             )
-
-        query_url = f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/queries/"
         is_async = True
         payload = {
             "data": {
@@ -517,9 +508,10 @@ class PollyKG:
             print(e)
             raise
 
-    def get_summary(self) -> dict:
+    def get_summary(self, stats: bool = False) -> dict:
         """Retrieve a summary of the Polly Knowledge Graph.
-
+        Params:
+            stats: A boolean flag to retrieve the detailed counts edges.
         Returns:
             dict: A dictionary containing summary information about the graph,
                  such as node counts, edge counts, and other metadata.
@@ -530,9 +522,14 @@ class PollyKG:
             RequestFailureException: Raised when the request fails due to an unexpected error.
         """
         try:
-            response = self.session.get(
-                f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/summary"
-            )
+            if stats:
+                response = self.session.get(
+                    f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/summary?stats=true"
+                )
+            else:
+                response = self.session.get(
+                    f"{self.polly_kg_neo4j_endpoint}/kg/{self.kg_id}/versions/{self.version_id}/summary"
+                )
             return handle_success_and_error_response(response)
         except Exception as e:
             print(e)
@@ -683,6 +680,7 @@ class PollyKG:
         data = new_schema.get("data", {})
         attributes = data.get("attributes", {})
         schema = attributes.get("schema", {})
+        custom_attributes = attributes.get("custom_attributes", {})
 
         # Extract basic information
         node_types = schema.get("node_types", [])
@@ -761,19 +759,24 @@ class PollyKG:
         # Build legacy format response
         from datetime import datetime
 
+        # Build metadata with standard fields
+        metadata = {
+            "name": "schema",
+            "kg_id": str(kg_id),
+            "kg_version": str(version_id),
+            "last_updated": datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
+            "computed_at": datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
+        }
+
+        data_source = custom_attributes.get("data_source")
+        if data_source is not None:
+            metadata["data_source"] = data_source
+
         legacy_format = {
             "data": {
                 "type": "kg_schema",
                 "attributes": {
-                    "metadata": {
-                        "name": "schema",
-                        "kg_id": str(kg_id),
-                        "kg_version": str(version_id),
-                        "last_updated": datetime.utcnow().strftime(
-                            "%Y-%m-%dT%H:%M:%SZ"
-                        ),
-                        "computed_at": datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ"),
-                    },
+                    "metadata": metadata,
                     "node_types": node_types,
                     "edge_types": edge_types,
                     "nodes": transformed_nodes,
@@ -829,23 +832,6 @@ class PollyKG:
             ResourceNotFoundError: Raised when no knowledge graphs are found.
             AccessDeniedError: Raised when the user does not have permission to access the KGs.
             RequestFailureException: Raised when the request fails due to an unexpected error.
-
-        Example:
-            >>> kg = PollyKG()
-            >>> # Get only published KGs
-            >>> kgs = kg.get_all_kgs()
-            >>> for kg_info in kgs:
-            ...     print(f"{kg_info['kg_name']} (ID: {kg_info['kg_id']})")
-            base kg v3 (ID: 14_base_kg_v3)
-
-            >>> # Get all KGs including unpublished ones with instance details
-            >>> kgs_with_instances = kg.get_all_kgs(
-            ...     include_unpublished=True,
-            ...     include_instances=True,
-            ...     include_terminated=True
-            ... )
-            >>> for kg_info in kgs_with_instances:
-            ...     print(f"{kg_info['kg_name']} - {len(kg_info.get('instances', []))} instances")
         """
         try:
             kg_url = f"{self.polly_kg_neo4j_endpoint}/kg"

{polly_python-4.6.0 → polly_python-4.7.0/polly_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polly_python
-Version: 4.6.0
+Version: 4.7.0
 Summary: Polly SDK
 Home-page: https://github.com/ElucidataInc/polly-python
 Project-URL: Documentation, https://docs.elucidata.io

polly_python-4.6.0/polly/__init__.py

@@ -1 +0,0 @@
-__version__ = "4.6.0"