dbgraph 0.2.2__tar.gz → 0.2.4__tar.gz

{dbgraph-0.2.2 → dbgraph-0.2.4}/.idea/workspace.xml

@@ -5,7 +5,11 @@
   </component>
   <component name="ChangeListManager">
     <list default="true" id="263a96da-9a82-40fd-89b2-ba71dd2c194d" name="Changes" comment="update renderer">
-      <change beforePath="$PROJECT_DIR$/pyproject.toml" beforeDir="false" afterPath="$PROJECT_DIR$/pyproject.toml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/data/northwind-index/corpus.mmindex.json" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/corpus.mmindex.json" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/data/northwind-index/data.csc.index.npy" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/data.csc.index.npy" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/data/northwind-index/indices.csc.index.npy" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/indices.csc.index.npy" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/data/northwind-index/indptr.csc.index.npy" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/indptr.csc.index.npy" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/data/northwind-index/vocab.index.json" beforeDir="false" afterPath="$PROJECT_DIR$/data/northwind-index/vocab.index.json" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -67,6 +71,7 @@
     &quot;ASKED_SHARE_PROJECT_CONFIGURATION_FILES&quot;: &quot;true&quot;,
     &quot;ModuleVcsDetector.initialDetectionPerformed&quot;: &quot;true&quot;,
     &quot;Python tests.Python tests for tests.test_dbgraph.TestDBGraph.executor&quot;: &quot;Run&quot;,
+    &quot;Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph.executor&quot;: &quot;Run&quot;,
     &quot;Python tests.Python tests for tests.test_load_graph_sql.TestLoadGraphSQL.test_load.executor&quot;: &quot;Run&quot;,
     &quot;Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.executor&quot;: &quot;Run&quot;,
     &quot;Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test__build_links.executor&quot;: &quot;Run&quot;,
@@ -86,8 +91,8 @@
     &quot;settings.editor.selected.configurable&quot;: &quot;editor.preferences.folding&quot;
   }
 }</component>
-  <component name="RunManager" selected="Python tests.Python tests in tests">
-    <configuration name="Python tests for tests.test_load_graph_sql.TestLoadGraphSQL.test_load" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+  <component name="RunManager" selected="Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph">
+    <configuration name="Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -100,7 +105,7 @@
       <option name="DEBUG_JUST_MY_CODE" value="false" />
       <option name="RUN_TOOL" value="" />
       <option name="_new_additionalArguments" value="&quot;&quot;" />
-      <option name="_new_target" value="&quot;tests.test_load_graph_sql.TestLoadGraphSQL.test_load&quot;" />
+      <option name="_new_target" value="&quot;tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
@@ -174,11 +179,11 @@
     </configuration>
     <recent_temporary>
       <list>
+        <item itemvalue="Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph" />
         <item itemvalue="Python tests.Python tests in tests" />
         <item itemvalue="Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder" />
         <item itemvalue="Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test__build_links" />
         <item itemvalue="Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" />
-        <item itemvalue="Python tests.Python tests for tests.test_load_graph_sql.TestLoadGraphSQL.test_load" />
       </list>
     </recent_temporary>
   </component>
@@ -245,7 +250,39 @@
       <option name="project" value="LOCAL" />
       <updated>1788922273132</updated>
     </task>
-    <option name="localTasksCounter" value="7" />
+    <task id="LOCAL-00007" summary="update renderer">
+      <option name="closed" value="true" />
+      <created>1788922706985</created>
+      <option name="number" value="00007" />
+      <option name="presentableId" value="LOCAL-00007" />
+      <option name="project" value="LOCAL" />
+      <updated>1788922706985</updated>
+    </task>
+    <task id="LOCAL-00008" summary="update renderer">
+      <option name="closed" value="true" />
+      <created>1788928883769</created>
+      <option name="number" value="00008" />
+      <option name="presentableId" value="LOCAL-00008" />
+      <option name="project" value="LOCAL" />
+      <updated>1788928883769</updated>
+    </task>
+    <task id="LOCAL-00009" summary="update renderer">
+      <option name="closed" value="true" />
+      <created>1788937576614</created>
+      <option name="number" value="00009" />
+      <option name="presentableId" value="LOCAL-00009" />
+      <option name="project" value="LOCAL" />
+      <updated>1788937576614</updated>
+    </task>
+    <task id="LOCAL-00010" summary="update renderer">
+      <option name="closed" value="true" />
+      <created>1788937586437</created>
+      <option name="number" value="00010" />
+      <option name="presentableId" value="LOCAL-00010" />
+      <option name="project" value="LOCAL" />
+      <updated>1788937586437</updated>
+    </task>
+    <option name="localTasksCounter" value="11" />
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">

{dbgraph-0.2.2 → dbgraph-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: dbgraph
-Version: 0.2.2
+Version: 0.2.4
 Summary: dbgraph is a database profiling tool which model the database as a graph
 Requires-Python: >=3.11
 Requires-Dist: bm25s>=0.3.9

dbgraph-0.2.4/data/northwind-index/corpus.mmindex.json

@@ -0,0 +1 @@
+[0,137,356,500,662,862,1059,1262,1444,1689,1866,2017,2200,2342,2527,2655,2805,2959,3168,3372,3527,3689,3789,3894,4003,4127,4226,4318,4441,4544,4639,4738,4829,4936,5034,5136,5264,5398,5489,5598,5688,5781,5888,6009,6108,6207,6317,6439,6564,6681,6791,6920,7042,7173,7292,7424,7545,7685,7785,7905,8035,8136,8258,8383,8518,8645,8773,8889,8990,9106,9211,9315,9433,9523,9667,9812,9972,10071,10199,10358,10508,10670,10860,11021,11119,11220,11323,11430,11531,11658,11789,11901,11997,12115,12223,12331,12441,12561,12678,12784,12919]

dbgraph-0.2.4/data/northwind-index/vocab.index.json

@@ -0,0 +1 @@
+{"type":0,"discontinu":1,"store":2,"base":3,"system":4,"postal":5,"which":6,"ms":7,"first":8,"report":9,"object":10,"whom":11,"map":12,"file":13,"process":14,"metadata":15,"detail":16,"regard":17,"visual":18,"photo":19,"sold":20,"profil":21,"group":22,"tabl":23,"status":24,"given":25,"parent":26,"inform":27,"each":28,"scope":29,"discount":30,"destin":31,"content":32,"has":33,"dispatch":34,"relationship":35,"primari":36,"entiti":37,"purchas":38,"courtesi":39,"key":40,"customerdemograph":41,"yet":42,"telephon":43,"geograph":44,"imag":45,"segment":46,"note":47,"southern":48,"minimum":49,"provinc":50,"it":51,"ani":52,"between":53,"job":54,"referenc":55,"facsimil":56,"data":57,"full":58,"have":59,"junction":60,"quantiti":61,"threshold":62,"confect":63,"rang":64,"25":65,"titl":66,"profession":67,"phone":68,"zip":69,"correspond":70,"date":71,"foreign":72,"organiz":73,"countri":74,"logist":75,"citi":76,"address":77,"line":78,"record":79,"resid":80,"singl":81,"shipper":82,"salut":83,"employeeid":84,"weight":85,"busi":86,"when":87,"price":88,"descript":89,"serv":90,"definit":91,"handl":92,"binari":93,"url":94,"link":95,"assign":96,"identifi":97,"appli":98,"path":99,"time":100,"all":101,"area":102,"fax":103,"posit":104,"physic":105,"respons":106,"establish":107,"facilit":108,"product":109,"actual":110,"extens":111,"master":112,"street":113,"act":114,"integ":115,"provid":116,"associ":117,"mr":118,"where":119,"id":120,"per":121,"label":122,"last":123,"explain":124,"order":125,"been":126,"place":127,"pack":128,"supplier":129,"northern":130,"locat":131,"should":132,"lookup":133,"item":134,"current":135,"hire":136,"larg":137,"photograph":138,"charg":139,"role":140,"sale":141,"manag":142,"categor":143,"deliv":144,"often":145,"avail":146,"characterist":147,"dr":148,"entri":149,"number":150,"whether":151,"mani":152,"transact":153,"custom":154,"cost":155,"held":156,"contact":157,"refer":158,"personnel":159,"compani":160,"demograph":161,"packag":162,"size":163,"employe":164,"use":165,"name":166,"textual":167,"text":168,"case":169,"legal":170,"freight":171,"describ":172,"from":173,"code":174,"local":175,"level":176,"databas":177,"includ":178,"uniqu":179,"within":180,"identif":181,"biograph":182,"homepag":183,"specif":184,"state":185,"market":186,"beverag":187,"arriv":188,"internal":189,"new":190,"unit":191,"person":192,"percentag":193,"who":194,"categori":195,"be":196,"territori":197,"surnam":198,"expect":199,"flag":200,"addit":201,"repres":202,"logo":203,"carrier":204,"region":205,"hierarchi":206,"activ":207,"contain":208,"classifi":209,"home":210,"ship":211,"deliveri":212,"eastern":213,"websit":214,"stock":215,"indic":216,"birth":217,"about":218,"inventori":219,"western":220,"":221}

{dbgraph-0.2.2 → dbgraph-0.2.4}/dbgraph/descriptor/langchain_graph_descriptor.py

@@ -34,9 +34,9 @@ class LangchainGraphDescriptor(GraphDescriptor):
     def _get_semantic_aspect_table(
         self, asset: Asset, context: RDatabaseGraph
     ) -> SemanticAspect:
-        self.markdown_renderer.render(context)
+        context_str = self.markdown_renderer.render(context)
         system_prompt = (
-            self.table_system_prompt + "\n" + self.markdown_renderer.get_content()
+            self.table_system_prompt + "\n" + context_str
         )
         question = self.table_question_prompt + " " + asset.name
         description = self._generate(system_prompt, question)
@@ -49,9 +49,9 @@ class LangchainGraphDescriptor(GraphDescriptor):
     def _get_semantic_aspect_column(
         self, asset: Asset, context: RDatabaseGraph
     ) -> SemanticAspect:
-        self.markdown_renderer.render(context)
+        context_str = self.markdown_renderer.render(context)
         system_prompt = (
-            self.column_system_prompt + "\n" + self.markdown_renderer.get_content()
+            self.column_system_prompt + "\n" + context_str
         )
         question = self.column_question_prompt + " " + asset.name
         description = self._generate(system_prompt, question)

{dbgraph-0.2.2 → dbgraph-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dbgraph"
-version = "0.2.2"
+version = "0.2.4"
 description = "dbgraph is a database profiling tool which model the database as a graph"
 readme = "README.md"
 requires-python = ">=3.11"

{dbgraph-0.2.2 → dbgraph-0.2.4}/tests/test_json_graph_loader.py

@@ -19,7 +19,7 @@ class TestJSONGraphLoader(unittest.TestCase):
             "601fca0f-9c27-4833-86d0-48f8383437a0",
             "29fb86b7-13f6-4a23-ab24-26d68881624f",
         )
-        self.assertIsInstance(subgraphs, DatabaseGraph)
+        self.assertIsInstance(subgraphs, list)
 
     def test_get_neighbors_sub_graph(self):
         graph = self.graph_loader.load()

{dbgraph-0.2.2 → dbgraph-0.2.4}/uv.lock

@@ -284,7 +284,7 @@ wheels = [
 
 [[package]]
 name = "dbgraph"
-version = "0.2.1"
+version = "0.2.2"
 source = { editable = "." }
 dependencies = [
     { name = "bm25s" },

dbgraph-0.2.2/data/northwind-index/corpus.mmindex.json

@@ -1 +0,0 @@
-[0,136,354,497,658,857,1053,1255,1436,1680,1856,2006,2188,2329,2513,2640,2789,2942,3150,3353,3507,3668,3767,3871,3979,4102,4200,4291,4413,4515,4609,4707,4797,4903,5000,5101,5228,5361,5451,5559,5648,5740,5846,5966,6064,6162,6271,6392,6516,6632,6741,6869,6990,7120,7238,7369,7489,7628,7727,7846,7975,8075,8196,8320,8454,8580,8707,8822,8922,9037,9141,9244,9361,9450,9593,9737,9896,9994,10121,10279,10428,10589,10778,10938,11035,11135,11237,11343,11443,11569,11699,11810,11905,12022,12129,12236,12345,12464,12580,12685,12819]

dbgraph-0.2.2/data/northwind-index/vocab.index.json

@@ -1 +0,0 @@
-{"type":0,"identifi":1,"shipper":2,"hierarchi":3,"categor":4,"status":5,"path":6,"physic":7,"inventori":8,"current":9,"primari":10,"photograph":11,"carrier":12,"discontinu":13,"demograph":14,"held":15,"25":16,"name":17,"ship":18,"job":19,"profil":20,"often":21,"weight":22,"base":23,"local":24,"stock":25,"first":26,"group":27,"manag":28,"destin":29,"entiti":30,"referenc":31,"about":32,"id":33,"categori":34,"deliv":35,"territori":36,"actual":37,"facilit":38,"cost":39,"price":40,"citi":41,"activ":42,"tabl":43,"serv":44,"whom":45,"provid":46,"logist":47,"parent":48,"date":49,"relationship":50,"level":51,"courtesi":52,"legal":53,"area":54,"surnam":55,"extens":56,"yet":57,"master":58,"geograph":59,"size":60,"use":61,"contact":62,"databas":63,"appli":64,"packag":65,"internal":66,"each":67,"file":68,"transact":69,"integ":70,"ani":71,"address":72,"websit":73,"imag":74,"url":75,"ms":76,"characterist":77,"specif":78,"binari":79,"locat":80,"it":81,"number":82,"last":83,"person":84,"homepag":85,"rang":86,"who":87,"explain":88,"singl":89,"associ":90,"note":91,"segment":92,"assign":93,"system":94,"label":95,"textual":96,"scope":97,"quantiti":98,"lookup":99,"metadata":100,"home":101,"provinc":102,"be":103,"definit":104,"custom":105,"between":106,"eastern":107,"order":108,"text":109,"object":110,"expect":111,"mani":112,"beverag":113,"logo":114,"dr":115,"store":116,"when":117,"all":118,"code":119,"process":120,"biograph":121,"record":122,"sale":123,"data":124,"charg":125,"per":126,"item":127,"personnel":128,"organiz":129,"given":130,"countri":131,"descript":132,"from":133,"postal":134,"birth":135,"zip":136,"regard":137,"foreign":138,"southern":139,"hire":140,"telephon":141,"percentag":142,"junction":143,"employe":144,"classifi":145,"which":146,"where":147,"state":148,"uniqu":149,"visual":150,"has":151,"product":152,"addit":153,"entri":154,"arriv":155,"phone":156,"place":157,"deliveri":158,"indic":159,"contain":160,"salut":161,"fax":162,"correspond":163,"threshold":164,"facsimil":165,"detail":166,"confect":167,"profession":168,"unit":169,"map":170,"case":171,"handl":172,"region":173,"pack":174,"sold":175,"customerdemograph":176,"refer":177,"freight":178,"whether":179,"role":180,"dispatch":181,"discount":182,"time":183,"line":184,"content":185,"titl":186,"posit":187,"busi":188,"link":189,"inform":190,"minimum":191,"avail":192,"should":193,"resid":194,"respons":195,"identif":196,"photo":197,"market":198,"includ":199,"mr":200,"have":201,"larg":202,"employeeid":203,"northern":204,"within":205,"western":206,"repres":207,"describ":208,"full":209,"been":210,"supplier":211,"key":212,"act":213,"street":214,"establish":215,"purchas":216,"flag":217,"compani":218,"new":219,"report":220,"":221}