kobai-sdk 0.3.5rc3__tar.gz → 0.3.5rc6__tar.gz

{kobai_sdk-0.3.5rc3/kobai_sdk.egg-info → kobai_sdk-0.3.5rc6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kobai-sdk
-Version: 0.3.5rc3
+Version: 0.3.5rc6
 Summary: A package that enables interaction with a Kobai tenant.
 Author-email: Ryan Oattes <ryan@kobai.io>
 License:                                  Apache License

{kobai_sdk-0.3.5rc3 → kobai_sdk-0.3.5rc6}/kobai/tenant_client.py

@@ -219,13 +219,13 @@ class TenantClient:
         else:
             return True
 
-    def spark_generate_genie_views(self, domains = None, concepts = None, not_concepts=None, enforce_map=True):
+    def spark_generate_genie_views(self, domains = None, concepts = None, not_concepts=None, enforce_map=True, local=False):
 
         """
         Use the Spark Client to generate views for this tenant required to populate a Genie Data Room.
         """
 
-        tables = self.__get_view_sql(domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map)
+        tables = self.__get_view_sql(local, domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map)
         for t in tables:
             #print(t["sql"])
             try:
@@ -235,13 +235,13 @@ class TenantClient:
                 print(t["sql"])
         print("Updated " + str(len(tables)) + " views for Genie.")
 
-    def spark_remove_genie_views(self):
+    def spark_remove_genie_views(self, local=False):
 
         """
         Use the Spark Client to remove any views previously created for this tenant.
         """
 
-        tables = self.__get_view_sql()
+        tables = self.__get_view_sql(local)
         for t in tables:
             self.spark_client._SparkClient__run_sql("DROP VIEW " + t["table"])
         print("Removed " + str(len(tables)) + " views.")
@@ -263,7 +263,7 @@ class TenantClient:
 
         self.databricks_client = databricks_client.DatabricksClient(notebook_context, warehouse_id)
 
-    def databricks_build_genie(self, domains=None, concepts=None, not_concepts=None, enforce_map=True, add_questions=False):
+    def databricks_build_genie(self, domains=None, concepts=None, not_concepts=None, enforce_map=True, add_questions=False, local=False):
 
         """
         Use the Databricks Client to create a Genie Data Room for this tenant.
@@ -280,7 +280,7 @@ class TenantClient:
             response = self.databricks_client._DatabricksClient__api_post("/api/2.0/data-rooms", payload)
             room_id = response["id"]
 
-        for t in self.__get_view_sql(domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map):
+        for t in self.__get_view_sql(local, domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map):
             payload["table_identifiers"].append(t["table"])
             print(t["table"])
         response = self.databricks_client._DatabricksClient__api_patch("/api/2.0/data-rooms/" + room_id, payload)
@@ -326,15 +326,26 @@ class TenantClient:
 # Semantic Profile
 ########################################
 
-    def __get_descriptions(self):
+    #def get_descriptions(self):
+    #    response = self.api_client._TenantAPI__run_get("/episteme-svcs/api/descriptions", params={"schema": self.schema})
+    #    return response.json()
 
-        tenant_config = self.get_tenant_config()
-        descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)
-        return descriptions
+    def __get_descriptions(self, local):
 
-    def __get_view_sql(self, domains=None, concepts=None, not_concepts=None, enforce_map=True):
+        if local:
+            tenant_config = self.get_tenant_config()
+            descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)
+            return descriptions
+        else:
+            response = self.api_client._TenantAPI__run_get("/episteme-svcs/api/descriptions", params={"schema": self.schema})
+            return response.json()
+        #tenant_config = self.get_tenant_config()
+        #descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)
+        #return descriptions
+
+    def __get_view_sql(self, local, domains=None, concepts=None, not_concepts=None, enforce_map=True):
         sql_list = []
-        descriptions = self.__get_descriptions()
+        descriptions = self.__get_descriptions(local)
         
         for dom in descriptions["domains"]:
             for con in dom["concepts"]:

{kobai_sdk-0.3.5rc3 → kobai_sdk-0.3.5rc6/kobai_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kobai-sdk
-Version: 0.3.5rc3
+Version: 0.3.5rc6
 Summary: A package that enables interaction with a Kobai tenant.
 Author-email: Ryan Oattes <ryan@kobai.io>
 License:                                  Apache License

{kobai_sdk-0.3.5rc3 → kobai_sdk-0.3.5rc6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kobai-sdk"
-version = "0.3.5rc3"
+version = "0.3.5rc6"
 description = "A package that enables interaction with a Kobai tenant."
 readme = "README.md"
 authors = [{ name = "Ryan Oattes", email = "ryan@kobai.io" }]