kobai-sdk 0.2.8rc7__py3-none-any.whl → 0.2.8rc9__py3-none-any.whl

kobai/ai_rag.py

@@ -48,7 +48,7 @@ def ai_run_question_remote(tc: AIContext, question_id, dynamic_filters: dict = N
 
     return response.json()
 
-def generate_sentences(tc: AIContext, replica_schema=None, concept_white_list=None, use_questions=False):
+def generate_sentences(tc: AIContext, replica_schema=None, concept_white_list=None, use_questions=False, debug=False):
     """
     Extract Semantic Data from Graph to Delta Table
 
@@ -79,10 +79,12 @@ def generate_sentences(tc: AIContext, replica_schema=None, concept_white_list=No
 
     print("Running the Extraction")
     for sql_statement in sql_statements:
+        if debug:
+            print(sql_statement)
         ss.sql(sql_statement)
 
     if use_questions:
-        __generate_sentences_from_questions(tc)
+        __generate_sentences_from_questions(tc, debug)
 
     if replica_schema is not None:
         print("Replicating Schema")
@@ -91,7 +93,7 @@ def generate_sentences(tc: AIContext, replica_schema=None, concept_white_list=No
             tc.schema, replica_schema, tc.model_id))
 
 
-def __generate_sentences_from_questions(tc: AIContext):
+def __generate_sentences_from_questions(tc: AIContext, debug):
     ss = tc.spark_session
 
     print("Getting Question Data")
@@ -114,6 +116,8 @@ def __generate_sentences_from_questions(tc: AIContext):
 
     sentences = []
     for p in published_queries:
+        if debug:
+            print("Running Question:", p)
         output = ai_run_question_remote(tc, p)
         for r in output:
             sentence = f"For {question_names[p]}: "

kobai/tenant_client.py

@@ -64,7 +64,7 @@ class TenantClient:
 # MS Entra Auth
 ########################################
 
-    def authenticate(self, client_id: str, tenant_id: str, run_ai_init: bool = True):
+    def authenticate(self, client_id: str, tenant_id: str, run_ai_init: bool = True, override_username: str = None):
 
         """
         Authenticate the TenantClient with the Kobai instance. Returns nothing, but stores bearer token in client.
@@ -84,6 +84,9 @@ class TenantClient:
         oauth_token = access.serialize()
         user_name = json.loads(access.serialize())["username"]
 
+        if override_username is not None:
+            user_name = override_username
+
         user_name_query_params={ 'userName' : user_name}
         tenants_response = self.api_client._TenantAPI__run_get('/user-mgmt-svcs/auth/tenants?'+urllib.parse.urlencode(user_name_query_params))
 

{kobai_sdk-0.2.8rc7.dist-info → kobai_sdk-0.2.8rc9.dist-info}/METADATA

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

{kobai_sdk-0.2.8rc7.dist-info → kobai_sdk-0.2.8rc9.dist-info}/RECORD

@@ -1,13 +1,13 @@
 kobai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kobai/ai_query.py,sha256=d0WOPKtQ-bI_zW3-_6guEJX0t55OcxdXIdgXaD-zKK0,9413
-kobai/ai_rag.py,sha256=CNP4Rl6LXnXYR88JT35A3NfznoNpEQUldA0jozChKzc,14143
+kobai/ai_rag.py,sha256=xAifSGLsW2Akx0pEfWi1nBKtZe9tqbItiEmyeK53W_o,14281
 kobai/databricks_client.py,sha256=fyqqMly2Qm0r1AHWsQjkYeNsDdH0G1JSgTkF9KJ55qA,2118
 kobai/demo_tenant_client.py,sha256=wlNc-bdI2wotRXo8ppUOalv4hYdBlek_WzJNARZV-AE,9293
 kobai/spark_client.py,sha256=opM_F-4Ut5Hq5zZjWMuLvUps9sDULvyPNZHXGL8dW1k,776
 kobai/tenant_api.py,sha256=9U6UbxpaAb-kpbuADXx3kbkNKaOzYy0I-GGwbpiCCOk,4212
-kobai/tenant_client.py,sha256=Uf1qNj8ASLyWf4ymE2pw1s4fxNQL-QW4GXVMfmuAuIU,39261
-kobai_sdk-0.2.8rc7.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-kobai_sdk-0.2.8rc7.dist-info/METADATA,sha256=e64A0BMY9_sLkuOBz8o0zRqzIVn1W-Pu0mqdTPgwN8s,19204
-kobai_sdk-0.2.8rc7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-kobai_sdk-0.2.8rc7.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
-kobai_sdk-0.2.8rc7.dist-info/RECORD,,
+kobai/tenant_client.py,sha256=EYen_QEAKs04zLF2oIKAVC84vrPp9vpRjS7NOITH1Ck,39377
+kobai_sdk-0.2.8rc9.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+kobai_sdk-0.2.8rc9.dist-info/METADATA,sha256=RXgBY0-y-dZJY7q1RcRKWi-EiL6PEp4hZkO6KmQ645k,19204
+kobai_sdk-0.2.8rc9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+kobai_sdk-0.2.8rc9.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
+kobai_sdk-0.2.8rc9.dist-info/RECORD,,