kobai-sdk 0.2.8rc13__py3-none-any.whl → 0.2.8rc15__py3-none-any.whl

kobai/ai_rag.py

@@ -69,6 +69,7 @@ def generate_sentences(tc: AIContext, replica_schema=None, concept_white_list=No
 
     print("Dropping and Recreating the RAG Table")
     ss.sql(__create_rag_table_sql(tc.schema, tc.model_id))
+    ss.sql(__clear_rag_table_sql(tc.schema, tc.model_id))
 
     print("Generating Extraction SQL")
     sql_statements = []
@@ -89,6 +90,7 @@ def generate_sentences(tc: AIContext, replica_schema=None, concept_white_list=No
     if replica_schema is not None:
         print("Replicating Schema")
         ss.sql(__create_rag_table_sql(replica_schema, tc.model_id))
+        ss.sql(__clear_rag_table_sql(tc.schema, tc.model_id))
         ss.sql(__replicate_to_catalog_sql(
             tc.schema, replica_schema, tc.model_id))
 
@@ -274,8 +276,14 @@ def rag_delta(tc: AIContext, emb_model: Union[SentenceTransformer, Embeddings],
     return response
 
 
+#def __create_rag_table_sql(schema, model_id):
+#    return f"CREATE OR REPLACE TABLE {schema}.rag_{model_id} (id BIGINT GENERATED BY DEFAULT AS IDENTITY, content STRING, type string, concept_id string, vector ARRAY<FLOAT>) TBLPROPERTIES (delta.enableChangeDataFeed = true)"
+
 def __create_rag_table_sql(schema, model_id):
-    return f"CREATE OR REPLACE TABLE {schema}.rag_{model_id} (id BIGINT GENERATED BY DEFAULT AS IDENTITY, content STRING, type string, concept_id string, vector ARRAY<FLOAT>) TBLPROPERTIES (delta.enableChangeDataFeed = true)"
+    return f"CREATE TABLE IF NOT EXISTS {schema}.rag_{model_id} (id BIGINT GENERATED BY DEFAULT AS IDENTITY, content STRING, type string, concept_id string, vector ARRAY<FLOAT>) TBLPROPERTIES (delta.enableChangeDataFeed = true)"
+
+def __clear_rag_table_sql(schema, model_id):
+    return f"DELETE FROM {schema}.rag_{model_id}"
 
 
 def __replicate_to_catalog_sql(base_schema, target_schema, model_id):

kobai/tenant_client.py

@@ -479,7 +479,7 @@ class TenantClient:
 # AI Functions
 ########################################
 
-    def followup_question(self, user_question, question_id=None, use_inmem_vectors=False, k=50):
+    def followup_question(self, user_question, question_id=None, use_inmem_vectors=False, k=50, dynamic_filters: dict = None):
         """
         Use LLM to answer question in the context of a Kobai Studio question.
 
@@ -495,7 +495,7 @@ class TenantClient:
 
             question_id = suggestions[0]["id"]
         
-        question_results = self.run_question_remote(question_id)
+        question_results = self.run_question_remote(question_id, dynamic_filters=dynamic_filters)
 
         question_def = self.get_question(question_id)
         question_name = question_def["description"]

{kobai_sdk-0.2.8rc13.dist-info → kobai_sdk-0.2.8rc15.dist-info}/METADATA

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

{kobai_sdk-0.2.8rc13.dist-info → kobai_sdk-0.2.8rc15.dist-info}/RECORD

@@ -1,13 +1,13 @@
 kobai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kobai/ai_query.py,sha256=xZh_OyakU01gIrnzaW4v_TdfzG51nPu0ntXJw1WEnvw,9424
-kobai/ai_rag.py,sha256=Y6R0THtatIixAJvfN1V9NPFTzh8nPdtwBrlLpIFTWJk,14692
+kobai/ai_rag.py,sha256=8B3HM4GoGVrgxJG678NN4vGaDwZRYnQiK5SCGiMIYkM,15186
 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=7ckHbn5cciLdIkUTBocZ2O5WRyBNExEuJnIDOjoXOX0,39502
-kobai_sdk-0.2.8rc13.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-kobai_sdk-0.2.8rc13.dist-info/METADATA,sha256=AUtGYR-gSfiDeBMv0s0Rcd3k1v8FV3kHNjEBL9CmnQM,19205
-kobai_sdk-0.2.8rc13.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-kobai_sdk-0.2.8rc13.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
-kobai_sdk-0.2.8rc13.dist-info/RECORD,,
+kobai/tenant_client.py,sha256=ocAjiX7ewb7JjCy7PiISX2in55dLVpkf8HGQv2fE6Fc,39565
+kobai_sdk-0.2.8rc15.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+kobai_sdk-0.2.8rc15.dist-info/METADATA,sha256=TuX7YcuECiNlBxnpKJaMA4rttPhYX8so6i3IP-dDzhk,19205
+kobai_sdk-0.2.8rc15.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+kobai_sdk-0.2.8rc15.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
+kobai_sdk-0.2.8rc15.dist-info/RECORD,,

{kobai_sdk-0.2.8rc13.dist-info → kobai_sdk-0.2.8rc15.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.2)
+Generator: setuptools (76.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any