dbgraph 0.2.3__tar.gz → 0.2.5__tar.gz

{dbgraph-0.2.3 → dbgraph-0.2.5}/.idea/workspace.xml

@@ -10,7 +10,11 @@
       <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" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/persistent/models.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/persistent/models.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/pyproject.toml" beforeDir="false" afterPath="$PROJECT_DIR$/pyproject.toml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/tests/test_sql_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/test_sql_graph_persistent.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/uv.lock" beforeDir="false" afterPath="$PROJECT_DIR$/uv.lock" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -78,6 +82,7 @@
     "Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test__build_links.executor": "Run",
     "Python tests.Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph.executor": "Run",
     "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.executor": "Run",
+    "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph.executor": "Run",
     "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_load_graph.executor": "Run",
     "Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_save_graph.executor": "Run",
     "Python tests.Python tests in test_load_graph_sql.py.executor": "Run",
@@ -92,7 +97,7 @@
     "settings.editor.selected.configurable": "editor.preferences.folding"
   }
 }]]></component>
-  <component name="RunManager" selected="Python tests.Python tests for tests.test_json_graph_loader.TestJSONGraphLoader.test_get_neighbors_sub_graph">
+  <component name="RunManager" selected="Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_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="" />
@@ -144,7 +149,7 @@
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -157,7 +162,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_sql_graph_builder.TestSQLGraphBuilder.test_build_graph&quot;" />
+      <option name="_new_target" value="&quot;tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
@@ -180,11 +185,11 @@
     </configuration>
     <recent_temporary>
       <list>
+        <item itemvalue="Python tests.Python tests for tests.test_sql_graph_persistent.TestSQLGraphPersistent.test_delete_graph" />
         <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" />
       </list>
     </recent_temporary>
   </component>
@@ -267,7 +272,23 @@
       <option name="project" value="LOCAL" />
       <updated>1788928883769</updated>
     </task>
-    <option name="localTasksCounter" value="9" />
+    <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">
@@ -297,12 +318,7 @@
       <breakpoints>
         <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
           <url>file://$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py</url>
-          <line>330</line>
-          <option name="timeStamp" value="4" />
-        </line-breakpoint>
-        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
-          <url>file://$PROJECT_DIR$/dbgraph/persistent/sql_graph_persistent.py</url>
-          <line>351</line>
+          <line>353</line>
           <option name="timeStamp" value="5" />
         </line-breakpoint>
       </breakpoints>

{dbgraph-0.2.3 → dbgraph-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: dbgraph
-Version: 0.2.3
+Version: 0.2.5
 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.3 → dbgraph-0.2.5}/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)

{dbgraph-0.2.3 → dbgraph-0.2.5}/dbgraph/persistent/models.py

@@ -66,6 +66,7 @@ class AssetModel(Base):
         ForeignKey(
             "graphs.graph_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -116,6 +117,7 @@ class LinkModel(Base):
         ForeignKey(
             "assets.asset_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -125,6 +127,7 @@ class LinkModel(Base):
         ForeignKey(
             "assets.asset_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -134,6 +137,7 @@ class LinkModel(Base):
         ForeignKey(
             "graphs.graph_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         nullable=False,
         index=True,
@@ -167,6 +171,7 @@ class AssetAspectModel(Base):
         ForeignKey(
             "assets.asset_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         primary_key=True,
     )
@@ -198,6 +203,7 @@ class LinkAspectModel(Base):
         ForeignKey(
             "links.link_id",
             ondelete="CASCADE",
+            onupdate="CASCADE"
         ),
         primary_key=True,
     )

{dbgraph-0.2.3 → dbgraph-0.2.5}/dbgraph/persistent/sql_graph_persistent.py

@@ -52,6 +52,8 @@ class SQLGraphPersistent(GraphPersistent):
     def delete_graph(self, graph_id: str):
         with Session(self.engine) as session:
             session.execute(delete(GraphModel).where(GraphModel.graph_id == graph_id))
+            session.execute(delete(AssetModel).where(AssetModel.graph_id == graph_id))
+            session.execute(delete(LinkModel).where(LinkModel.graph_id == graph_id))
             session.commit()
 
     def insert_assets(self, assets: list[Asset], graph_id: str):

{dbgraph-0.2.3 → dbgraph-0.2.5}/pyproject.toml

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

{dbgraph-0.2.3 → dbgraph-0.2.5}/tests/test_sql_graph_persistent.py

@@ -441,6 +441,15 @@ class TestSQLGraphPersistent(unittest.TestCase):
         graph = self.persistent.load_graph(self.graph_id)
         self.assertEqual(self.graph, graph)
 
+    def test_delete_graph(self):
+        self.persistent.save_graph(self.graph, self.graph_id)
+        graph = self.persistent.load_graph(self.graph_id)
+        self.assertEqual(self.graph, graph)
+        self.persistent.delete_graph(self.graph_id)
+        graph = self.persistent.load_graph(self.graph_id)
+        self.assertEqual(0, len(graph.assets))
+        self.assertEqual(0, len(graph.links))
+
     def test_get_graph_info(self):
         graph_info = self.persistent.get_graph_info("1")
         self.assertEqual(graph_info[1], "graph-1")

{dbgraph-0.2.3 → dbgraph-0.2.5}/uv.lock

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