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

kobai/ai_rag.py

@@ -143,7 +143,7 @@ def __generate_sentences_from_questions(tc: AIContext, debug):
     ss.sql(full_sql)
 
 
-def encode_to_delta_local(tc: AIContext, st_model: SentenceTransformer, replica_schema=None):
+def encode_to_delta_local(tc: AIContext, st_model: Union[SentenceTransformer, Embeddings], replica_schema=None):
     """
     Encode Semantic Data to Vectors in Delta Table
 
@@ -172,8 +172,13 @@ def encode_to_delta_local(tc: AIContext, st_model: SentenceTransformer, replica_
         content_list = [r["content"] for r in sentences_df.collect()]
         id_list = [r["id"] for r in sentences_df.collect()]
 
-        vector_list = st_model.encode(
-            content_list, normalize_embeddings=True, show_progress_bar=True)
+        if isinstance(st_model, SentenceTransformer):
+            vector_list = st_model.encode(
+                content_list, normalize_embeddings=True, show_progress_bar=True)
+        else:
+            vector_list = st_model.embed_documents(content_list)
+        #vector_list = st_model.encode(
+        #    content_list, normalize_embeddings=True, show_progress_bar=True)
 
         schema_v = StructType([
             StructField("id", IntegerType(), True),

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,9 +449,9 @@ 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):
+    def rag_encode_to_delta_local(self, st_model: Union[SentenceTransformer, Embeddings], replica_schema=None):
         """
         Encode Semantic Data to Vectors in Delta Table
 

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

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

kobai_sdk-0.2.8rc11.dist-info/RECORD

@@ -0,0 +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=Gqj_CF2lLjMc0CcxquOAgXhiuY6pKwyJumR2NK7QdOo,14559
+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=ktw75dmmX95gD6qIkG-mAmdo48RkF96SgmpRmfDWblo,39422
+kobai_sdk-0.2.8rc11.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+kobai_sdk-0.2.8rc11.dist-info/METADATA,sha256=jyrtCYbqTYKrHMHBOmWUrDfkK3NFqtqxmQyTW-vZ05A,19205
+kobai_sdk-0.2.8rc11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+kobai_sdk-0.2.8rc11.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
+kobai_sdk-0.2.8rc11.dist-info/RECORD,,

kobai_sdk-0.2.8rc9.dist-info/RECORD

@@ -1,13 +0,0 @@
-kobai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kobai/ai_query.py,sha256=d0WOPKtQ-bI_zW3-_6guEJX0t55OcxdXIdgXaD-zKK0,9413
-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=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,,