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

kobai/ai_query.py

@@ -73,17 +73,17 @@ def format_docs(docs):
 def input_only(inpt):
     return inpt["question"]
 
-def followup_question(user_question, question_results, question_name, question_def, embedding_model: Union[SentenceTransformer, Embeddings], chat_model: BaseChatModel, use_inmem_vectors=False):
+def followup_question(user_question, question_results, question_name, question_def, embedding_model: Union[SentenceTransformer, Embeddings], chat_model: BaseChatModel, use_inmem_vectors=False, k=50):
     
     row_texts = process_question_results(question_def, question_results)
     question_documents = [Document(page_content=r, metadata={"source": "kobai"}) for r in row_texts]
 
     if use_inmem_vectors:
         question_retriever = InMemoryVectorStore.from_documents(question_documents, embedding=embedding_model).as_retriever(
-    search_kwargs={"k": 5}
+    search_kwargs={"k": k}
 )
     else:
-        question_retriever = QuestionRetriever(documents=question_documents)
+        question_retriever = QuestionRetriever(documents=question_documents, k=k)
 
     output_parser = StrOutputParser()
 

kobai/ai_rag.py

@@ -201,15 +201,15 @@ def encode_to_delta_local(tc: AIContext, st_model: Union[SentenceTransformer, Em
             .whenMatchedUpdate(set={"vector": "s.vector"}) \
             .execute()
 
-    ss.sql(f"""
-          CREATE FUNCTION IF NOT EXISTS {schema}.cos_sim(a ARRAY<FLOAT>, b ARRAY<FLOAT>)
-            RETURNS FLOAT
-            LANGUAGE PYTHON
-            AS $$
-                import numpy as np
-                return float(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b)))
-            $$
-          """)
+    #ss.sql(f"""
+    #      CREATE FUNCTION IF NOT EXISTS {schema}.cos_sim(a ARRAY<FLOAT>, b ARRAY<FLOAT>)
+    #        RETURNS FLOAT
+    #       LANGUAGE PYTHON
+    #        AS $$
+    #           import numpy as np
+    #            return float(np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b)))
+    #        $$
+    #      """)
 
 
 def rag_delta(tc: AIContext, emb_model: Union[SentenceTransformer, Embeddings], chat_model: BaseChatModel, question, k=5, replica_schema=None):

kobai/tenant_client.py

@@ -82,6 +82,7 @@ class TenantClient:
         access = credential.authenticate()
 
         oauth_token = access.serialize()
+        print(oauth_token)
         user_name = json.loads(access.serialize())["username"]
 
         if override_username is not None:
@@ -478,7 +479,7 @@ class TenantClient:
 # AI Functions
 ########################################
 
-    def followup_question(self, user_question, question_id=None, use_inmem_vectors=False):
+    def followup_question(self, user_question, question_id=None, use_inmem_vectors=False, k=50):
         """
         Use LLM to answer question in the context of a Kobai Studio question.
 
@@ -499,7 +500,7 @@ class TenantClient:
         question_def = self.get_question(question_id)
         question_name = question_def["description"]
 
-        return ai_query.followup_question(user_question, question_results, question_name, question_def, self.embedding_model, self.chat_model, use_inmem_vectors=use_inmem_vectors)
+        return ai_query.followup_question(user_question, question_results, question_name, question_def, self.embedding_model, self.chat_model, use_inmem_vectors=use_inmem_vectors, k=k)
 
     def init_ai_components(self, embedding_model: Union[SentenceTransformer, Embeddings] = None, chat_model: BaseChatModel = None):
         """

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: kobai-sdk
-Version: 0.2.8rc12
+Version: 0.2.8rc13
 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/RECORD

@@ -0,0 +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/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_sdk-0.2.8rc12.dist-info → kobai_sdk-0.2.8rc13.dist-info}/WHEEL

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

kobai_sdk-0.2.8rc12.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=hc8M7M9azY-tFovsXdrjHJTmemFr8L74X2dy0j-XUoY,14685
-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=OUHQMmqWlezenkmMJH9p4wOz5CsAntmO25v5vWdepVE,39464
-kobai_sdk-0.2.8rc12.dist-info/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-kobai_sdk-0.2.8rc12.dist-info/METADATA,sha256=vrkBxo0XcZ1uG4lPHs-XfL38rxu-az55yw4NuJ_v5rk,19205
-kobai_sdk-0.2.8rc12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-kobai_sdk-0.2.8rc12.dist-info/top_level.txt,sha256=ns1El3BrTTHKvoAgU1XtiSaVIudYeCXbEEUVY8HFDZ4,6
-kobai_sdk-0.2.8rc12.dist-info/RECORD,,