kobai-sdk 0.2.8rc13__tar.gz → 0.2.8rc15__tar.gz

{kobai_sdk-0.2.8rc13 → kobai_sdk-0.2.8rc15}/PKG-INFO

@@ -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 → kobai_sdk-0.2.8rc15}/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_sdk-0.2.8rc13 → kobai_sdk-0.2.8rc15}/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 → kobai_sdk-0.2.8rc15}/kobai_sdk.egg-info/PKG-INFO

@@ -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 → kobai_sdk-0.2.8rc15}/pyproject.toml

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