dbgraph 0.2.8__tar.gz → 0.2.9__tar.gz

{dbgraph-0.2.8 → dbgraph-0.2.9}/.idea/workspace.xml

@@ -6,6 +6,12 @@
   <component name="ChangeListManager">
     <list default="true" id="263a96da-9a82-40fd-89b2-ba71dd2c194d" name="Changes" comment="update sql builder and tests">
       <change beforePath="$PROJECT_DIR$/.idea/dbgraph.iml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/dbgraph.iml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/__init__.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/__init__.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/entity/aspect.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/entity/aspect.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/dbgraph/persistent/orm_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/dbgraph/persistent/orm_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/units/test_orm_graph_persistent.py" beforeDir="false" afterPath="$PROJECT_DIR$/tests/units/test_orm_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" />
@@ -50,6 +56,9 @@
     &quot;repository&quot;: &quot;dbgraph&quot;
   }
 }</component>
+  <component name="ProblemsViewState">
+    <option name="selectedTabId" value="ProjectErrors" />
+  </component>
   <component name="ProjectColorInfo">{
   &quot;associatedIndex&quot;: 0,
   &quot;fromUser&quot;: false
@@ -137,7 +146,7 @@
       <recent name="C:\Users\minhden\Projects\dbgraph\tests\units" />
     </key>
   </component>
-  <component name="RunManager" selected="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm">
+  <component name="RunManager" selected="Python tests.Python tests in units">
     <configuration name="Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
@@ -155,7 +164,7 @@
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_cat_stats_aspect" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -168,11 +177,11 @@
       <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.integrates.test_trino_builder.TestTrinoBuilder.test__get_cat_stats_aspect&quot;" />
+      <option name="_new_target" value="&quot;tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -185,11 +194,11 @@
       <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.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm&quot;" />
+      <option name="_new_target" value="&quot;tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
@@ -200,36 +209,36 @@
       <option name="ADD_CONTENT_ROOTS" value="true" />
       <option name="ADD_SOURCE_ROOTS" value="true" />
       <option name="DEBUG_JUST_MY_CODE" value="false" />
-      <option name="RUN_TOOL" value="" />
+      <option name="RUN_TOOL" value="true" />
       <option name="_new_additionalArguments" value="&quot;&quot;" />
-      <option name="_new_target" value="&quot;tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count&quot;" />
+      <option name="_new_target" value="&quot;tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph&quot;" />
       <option name="_new_targetType" value="&quot;PYTHON&quot;" />
       <method v="2" />
     </configuration>
-    <configuration name="Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
+    <configuration name="Python tests in units" type="tests" factoryName="Autodetect" temporary="true" nameIsGenerated="true">
       <module name="dbgraph" />
       <option name="ENV_FILES" value="" />
       <option name="INTERPRETER_OPTIONS" value="" />
       <option name="PARENT_ENVS" value="true" />
       <option name="SDK_HOME" value="" />
-      <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
+      <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/tests/units" />
       <option name="IS_MODULE_SDK" value="true" />
       <option name="ADD_CONTENT_ROOTS" value="true" />
       <option name="ADD_SOURCE_ROOTS" value="true" />
       <option name="DEBUG_JUST_MY_CODE" value="false" />
-      <option name="RUN_TOOL" value="true" />
+      <option name="RUN_TOOL" value="" />
       <option name="_new_additionalArguments" value="&quot;&quot;" />
-      <option name="_new_target" value="&quot;tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph&quot;" />
-      <option name="_new_targetType" value="&quot;PYTHON&quot;" />
+      <option name="_new_target" value="&quot;$PROJECT_DIR$/tests/units&quot;" />
+      <option name="_new_targetType" value="&quot;PATH&quot;" />
       <method v="2" />
     </configuration>
     <recent_temporary>
       <list>
+        <item itemvalue="Python tests.Python tests in units" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_orm" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_sql_graph_builder.TestSQLGraphBuilder.test_build_graph" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_table_row_count" />
         <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test_build_graph" />
-        <item itemvalue="Python tests.Python tests for tests.integrates.test_trino_builder.TestTrinoBuilder.test__get_cat_stats_aspect" />
       </list>
     </recent_temporary>
   </component>

{dbgraph-0.2.8 → dbgraph-0.2.9}/PKG-INFO

@@ -1,30 +1,30 @@
 Metadata-Version: 2.5
 Name: dbgraph
-Version: 0.2.8
+Version: 0.2.9
 Summary: dbgraph is a database profiling tool which model the database as a graph
 Requires-Python: >=3.11
-Requires-Dist: bm25s>=0.3.9
-Requires-Dist: build>=1.6.0
+Requires-Dist: bm25s==0.3.11
+Requires-Dist: build==1.6.1
 Requires-Dist: grip>=4.6.2
-Requires-Dist: langchain-openai>=1.6.0
-Requires-Dist: langchain>=1.3.16
-Requires-Dist: matplotlib>=3.11.0
+Requires-Dist: langchain-openai==1.6.5
+Requires-Dist: langchain==1.4.2
+Requires-Dist: matplotlib==3.11.2
 Requires-Dist: mkdocs-gen-files>=0.6.1
 Requires-Dist: mkdocs-material>=9.7.7
 Requires-Dist: mkdocs>=1.6.1
 Requires-Dist: mkdocstrings[python]>=1.0.6
-Requires-Dist: openai>=2.45.0
+Requires-Dist: openai==3.19.2
 Requires-Dist: psutil>=7.2.2
-Requires-Dist: pydantic>=2.13.4
+Requires-Dist: pydantic==2.13.5
 Requires-Dist: pydot>=4.0.1
 Requires-Dist: pympler>=1.1
 Requires-Dist: pyrefly>=0.60.0
 Requires-Dist: pystemmer>=3.1.0
-Requires-Dist: python-dotenv>=1.2.2
-Requires-Dist: rustworkx>=0.18.0
-Requires-Dist: sqlalchemy>=2.0.52
-Requires-Dist: tqdm>=4.68.4
-Requires-Dist: trino>=0.339.0
+Requires-Dist: python-dotenv==1.2.3
+Requires-Dist: rustworkx==0.18.1
+Requires-Dist: sqlalchemy==2.0.54
+Requires-Dist: tqdm==4.70.1
+Requires-Dist: trino==0.340.0
 Description-Content-Type: text/markdown
 
 # DBGraph - Building schema graph with LLM Assistance

{dbgraph-0.2.8 → dbgraph-0.2.9}/dbgraph/__init__.py

@@ -6,6 +6,7 @@ from dbgraph.entity.link_type import LinkType
 from dbgraph.builder.sql_graph_builder import SQLGraphBuilder
 from dbgraph.descriptor.langchain_graph_descriptor import LangchainGraphDescriptor
 from dbgraph.render.markdown_renderer import MarkdownRenderer
+from dbgraph.persistent.orm_graph_persistent import ORMGraphPersistent
 
 __all__ = [
     "Asset",
@@ -16,4 +17,5 @@ __all__ = [
     "SQLGraphBuilder",
     "LangchainGraphDescriptor",
     "MarkdownRenderer",
+    "ORMGraphPersistent"
 ]

{dbgraph-0.2.8 → dbgraph-0.2.9}/dbgraph/entity/aspect.py

@@ -134,7 +134,7 @@ class RColumnStatisticsAspect(Aspect):
 
     non_null_count: int
     null_count: int
-    is_textual: bool
+    is_textual: bool = False
     numerical_stats: RNumericalStatistics | None = None
     categorical_stats: RCategoricalStatistics | None = None
     temporal_stats: RTemporalStatistics | None = None

{dbgraph-0.2.8 → dbgraph-0.2.9}/dbgraph/persistent/orm_graph_persistent.py

@@ -212,6 +212,7 @@ class ORMGraphPersistent(GraphPersistent):
                                 categorical_stats=categorical_aspect,
                                 numerical_stats=numerical_aspect,
                                 temporal_stats=temp_aspect,
+                                is_textual=aspect_data["is_textual"]
                             )
                 case "semantic_properties":
                     aspect = SemanticAspect(**aspect_data)

{dbgraph-0.2.8 → dbgraph-0.2.9}/pyproject.toml

@@ -4,33 +4,33 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dbgraph"
-version = "0.2.8"
+version = "0.2.9"
 description = "dbgraph is a database profiling tool which model the database as a graph"
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "bm25s>=0.3.9",
-    "build>=1.6.0",
+    "bm25s==0.3.11",
+    "build==1.6.1",
     "grip>=4.6.2",
-    "langchain>=1.3.16",
-    "langchain-openai>=1.6.0",
-    "matplotlib>=3.11.0",
+    "langchain==1.4.2",
+    "langchain-openai==1.6.5",
+    "matplotlib==3.11.2",
     "mkdocs>=1.6.1",
     "mkdocs-gen-files>=0.6.1",
     "mkdocs-material>=9.7.7",
     "mkdocstrings[python]>=1.0.6",
-    "openai>=2.45.0",
+    "openai==3.19.2",
     "psutil>=7.2.2",
-    "pydantic>=2.13.4",
+    "pydantic==2.13.5",
     "pydot>=4.0.1",
     "pympler>=1.1",
     "pyrefly>=0.60.0",
     "pystemmer>=3.1.0",
-    "python-dotenv>=1.2.2",
-    "rustworkx>=0.18.0",
-    "sqlalchemy>=2.0.52",
-    "tqdm>=4.68.4",
-    "trino>=0.339.0",
+    "python-dotenv==1.2.3",
+    "rustworkx==0.18.1",
+    "sqlalchemy==2.0.54",
+    "tqdm==4.70.1",
+    "trino==0.340.0",
 ]
 
 [tool.setuptools.packages.find]

{dbgraph-0.2.8 → dbgraph-0.2.9}/tests/units/test_orm_graph_persistent.py

@@ -84,6 +84,7 @@ class TestORMGraphPersistent(unittest.TestCase):
                         name="column-1-1-statistic",
                         non_null_count=0,
                         null_count=0,
+                        is_textual=False,
                         numerical_stats=RNumericalStatistics(min=0, max=0, mean=0),
                         categorical_stats=None,
                     ),
@@ -108,6 +109,7 @@ class TestORMGraphPersistent(unittest.TestCase):
                         non_null_count=0,
                         null_count=0,
                         numerical_stats=None,
+                        is_textual=False,
                         categorical_stats=RCategoricalStatistics(
                             value_counts={"1": 1, "2": 2}
                         ),
@@ -132,6 +134,7 @@ class TestORMGraphPersistent(unittest.TestCase):
                         name="column-2-1-statistic",
                         non_null_count=0,
                         null_count=0,
+                        is_textual=False,
                         numerical_stats=None,
                         categorical_stats=None,
                         temporal_stats=RTemporalStatistics(