kobai-sdk 0.2.8rc8__tar.gz → 0.2.8rc10__tar.gz

{kobai_sdk-0.2.8rc8 → kobai_sdk-0.2.8rc10}/PKG-INFO

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

{kobai_sdk-0.2.8rc8 → kobai_sdk-0.2.8rc10}/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_sdk-0.2.8rc8 → kobai_sdk-0.2.8rc10}/kobai/tenant_client.py

@@ -440,7 +440,7 @@ class TenantClient:
         context.api_client = self.api_client
         return context
 
-    def rag_generate_sentences(self, replica_schema=None, concept_white_list=None, use_questions=False):
+    def rag_generate_sentences(self, replica_schema=None, concept_white_list=None, use_questions=False, debug=False):
         """
         Extract Semantic Data from Graph to Delta Table
 
@@ -449,7 +449,7 @@ class TenantClient:
         concept_white_list ([str]) OPTIONAL: A list of Domain and Concept names for extraction.
         use_questions (bool) OPTIONAL: Extract facts from published Kobai questions. 
         """
-        ai_rag.generate_sentences(self.get_ai_context(), replica_schema=replica_schema, concept_white_list=concept_white_list, use_questions=use_questions)
+        ai_rag.generate_sentences(self.get_ai_context(), replica_schema=replica_schema, concept_white_list=concept_white_list, use_questions=use_questions, debug=debug)
 
     def rag_encode_to_delta_local(self, st_model: SentenceTransformer, replica_schema=None):
         """

{kobai_sdk-0.2.8rc8 → kobai_sdk-0.2.8rc10}/kobai_sdk.egg-info/PKG-INFO

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

{kobai_sdk-0.2.8rc8 → kobai_sdk-0.2.8rc10}/pyproject.toml

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