kobai-sdk 0.2.8rc14__py3-none-any.whl → 0.2.9__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/test.py

@@ -0,0 +1,5 @@
+import llm_config, ai_query
+
+llm_config = llm_config.LLMConfig(api_key="sV9LuoA5n0PwqggMXOYMhhZlt56FpgnMXFohimPhD7Ug3CnBLbO8JQQJ99ALACYeBjFXJ3w3AAABACOGZm8X", llm_provider="azure_openai")
+llm_config.get_azure_ad_token()
+ai_query.followup_question_1(question="abc", data={}, question_name="sample", llm_config=llm_config)

{kobai_sdk-0.2.8rc14.dist-info → kobai_sdk-0.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: kobai-sdk
-Version: 0.2.8rc14
+Version: 0.2.9
 Summary: A package that enables interaction with a Kobai tenant.
 Author-email: Ryan Oattes <ryan@kobai.io>
 License:                                  Apache License
@@ -223,12 +223,14 @@ Requires-Dist: langchain-core
 Requires-Dist: langchain-community
 Requires-Dist: langchain_openai
 Requires-Dist: databricks_langchain
+Requires-Dist: sentence-transformers
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: bumpver; extra == "dev"
 Requires-Dist: isort; extra == "dev"
 Requires-Dist: pip-tools; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
+Dynamic: license-file
 
 # Kobai SDK for Python (Alpha)
 

kobai_sdk-0.2.9.dist-info/RECORD

@@ -0,0 +1,14 @@
+kobai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kobai/ai_query.py,sha256=xZh_OyakU01gIrnzaW4v_TdfzG51nPu0ntXJw1WEnvw,9424
+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=ocAjiX7ewb7JjCy7PiISX2in55dLVpkf8HGQv2fE6Fc,39565
+kobai/test.py,sha256=LHCKClyFNPgLA1hF62RlDzilaFw2Nd3BWcaLtOc5xdc,320
+kobai_sdk-0.2.9.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+kobai_sdk-0.2.9.dist-info/METADATA,sha256=HmKJKi_XVB2MECaQXziP8Cvfe594I09ztK_2Ne7KmHI,19260
+kobai_sdk-0.2.9.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+kobai_sdk-0.2.9.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
+kobai_sdk-0.2.9.dist-info/RECORD,,

{kobai_sdk-0.2.8rc14.dist-info → kobai_sdk-0.2.9.dist-info}/WHEEL

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

kobai_sdk-0.2.8rc14.dist-info/RECORD

@@ -1,13 +0,0 @@
-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/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=ocAjiX7ewb7JjCy7PiISX2in55dLVpkf8HGQv2fE6Fc,39565
-kobai_sdk-0.2.8rc14.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-kobai_sdk-0.2.8rc14.dist-info/METADATA,sha256=8R-oSGwdmDhVmZh2MH25hBFoNXSjS_RyAtSpEwA2thg,19205
-kobai_sdk-0.2.8rc14.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-kobai_sdk-0.2.8rc14.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
-kobai_sdk-0.2.8rc14.dist-info/RECORD,,