kobai-sdk 0.3.3rc1__tar.gz → 0.3.4rc1__tar.gz

{kobai_sdk-0.3.3rc1/kobai_sdk.egg-info → kobai_sdk-0.3.4rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kobai-sdk
-Version: 0.3.3rc1
+Version: 0.3.4rc1
 Summary: A package that enables interaction with a Kobai tenant.
 Author-email: Ryan Oattes <ryan@kobai.io>
 License:                                  Apache License
@@ -221,8 +221,6 @@ Requires-Dist: azure-identity
 Requires-Dist: azure-storage-blob
 Requires-Dist: langchain-core
 Requires-Dist: langchain-community
-Requires-Dist: langchain_openai
-Requires-Dist: databricks_langchain
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: bumpver; extra == "dev"

{kobai_sdk-0.3.3rc1 → kobai_sdk-0.3.4rc1}/kobai/genie.py

@@ -1,4 +1,3 @@
-import json
 
 def get_genie_descriptions(solution_id, structure, schema):
     for di, dom in enumerate(structure["domains"]):
@@ -82,7 +81,6 @@ def recurse_parent_props(uri, structure, props, rels, visited=None):
                         recurse_parent_props(icon, structure, props, rels, visited)
 
 def add_map_count(structure):
-    #mapping_defs = get_tenant_mapping_config(structure["solution_id"])
 
     for dom in structure["domains"]:
         for con in dom["concepts"]:
@@ -93,41 +91,11 @@ def add_map_count(structure):
                     map_count = map_count + 1
             con["map_count"] = map_count
 
-#def get_tenant_mapping_config(solution_id): 
-#    mapping_sql = f"""
-#        select s.id, mapd.concept_type_uri
-#        from studio.solutions s
-#        inner join studio.mapping_defs mapd 
-#        on s.id = mapd.solution_id
-#        where s.id = {solution_id}
-#    """
-
-#    mapping_rows = app_db.run_query(mapping_sql)
-
-#    mapping_defs = []
-#    for row in mapping_rows:
-#        mapping_def = {"solution_id": row[0], "concept_type_uri": row[1]}
-#        mapping_defs.append(mapping_def)
-#    return mapping_defs
-
 ############################
 # Question Config
 ############################
 
 def get_tenant_question_config(solution_id, structure):
-    #question_sql = f"""
-    #    select s.id solution_id, q.id, q.description, q.definition, a.id api_id
-    #    from studio.models m
-    #    inner join studio.queries q 
-    #    on m.id = q.model_id
-    #    inner join studio.solutions s 
-    #    on m.id = s.model_id
-    #    left join studio.api a 
-    #    on q.id = a.query_id
-    #    where s.id = {solution_id}
-    #"""
-
-    #question_rows = app_db.run_query(question_sql)
 
     question_defs = []
     for row in structure["queries"]:

{kobai_sdk-0.3.3rc1 → kobai_sdk-0.3.4rc1}/kobai/tenant_client.py

@@ -5,8 +5,6 @@ import urllib.parse
 
 from pyspark.sql import SparkSession
 
-from langchain_community.chat_models import ChatDatabricks
-from databricks_langchain import DatabricksEmbeddings
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.embeddings import Embeddings
 
@@ -281,9 +279,10 @@ class TenantClient:
 
         for t in self.__get_view_sql(domains=domains, concepts=concepts, not_concepts=not_concepts, enforce_map=enforce_map):
             payload["table_identifiers"].append(t["table"])
+            print(t["table"])
         response = self.databricks_client._DatabricksClient__api_patch("/api/2.0/data-rooms/" + room_id, payload)
 
-        payload = {"title":"Notes","content":"When filtering for a named entity, use a like comparison instead of equality. All tables are denormalized, so columns may have repeated rows for the same primary identifier. You should handle this by putting each table in a subquery and using the DISTINCT keyword.","instruction_type":"TEXT_INSTRUCTION"}
+        payload = {"title":"Notes","content":"When filtering for a named entity, use a like comparison instead of equality. All tables are denormalized, so columns may have repeated rows for the same primary identifier. You should handle this by putting each table in a subquery and using the DISTINCT keyword. The first column in each view is a unique identifier that should only be used for joins, and never shown to a user. Find another column to identify the subject of the table.","instruction_type":"TEXT_INSTRUCTION"}
         instructions = self.databricks_client._DatabricksClient__api_get("/api/2.0/data-rooms/" + room_id + "/instructions")
         inst_id = "-1"
 
@@ -325,10 +324,6 @@ class TenantClient:
 ########################################
 
     def __get_descriptions(self):
-        
-        #params={"schema": self.schema, "tenant_id": self.model_id}
-        #response = self.api_client._TenantAPI__run_get("/episteme-svcs/api/descriptions", params=params)
-        #return response.json()
 
         tenant_config = self.get_tenant_config()
         descriptions = get_genie_descriptions(self.model_id, tenant_config, self.schema)

{kobai_sdk-0.3.3rc1 → kobai_sdk-0.3.4rc1/kobai_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kobai-sdk
-Version: 0.3.3rc1
+Version: 0.3.4rc1
 Summary: A package that enables interaction with a Kobai tenant.
 Author-email: Ryan Oattes <ryan@kobai.io>
 License:                                  Apache License
@@ -221,8 +221,6 @@ Requires-Dist: azure-identity
 Requires-Dist: azure-storage-blob
 Requires-Dist: langchain-core
 Requires-Dist: langchain-community
-Requires-Dist: langchain_openai
-Requires-Dist: databricks_langchain
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: bumpver; extra == "dev"

{kobai_sdk-0.3.3rc1 → kobai_sdk-0.3.4rc1}/kobai_sdk.egg-info/requires.txt

@@ -5,8 +5,6 @@ azure-identity
 azure-storage-blob
 langchain-core
 langchain-community
-langchain_openai
-databricks_langchain
 
 [dev]
 black

{kobai_sdk-0.3.3rc1 → kobai_sdk-0.3.4rc1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kobai-sdk"
-version = "0.3.3rc1"
+version = "0.3.4rc1"
 description = "A package that enables interaction with a Kobai tenant."
 readme = "README.md"
 authors = [{ name = "Ryan Oattes", email = "ryan@kobai.io" }]
@@ -24,9 +24,7 @@ dependencies = [
     "azure-identity",
     "azure-storage-blob",
     "langchain-core",
-    "langchain-community",
-    "langchain_openai",
-    "databricks_langchain"
+    "langchain-community"
     ]
 requires-python = ">=3.11"