Fix importing issues (#9)

Author: nithinjamespallicka

CHANGELOG.md

@@ -1,3 +1,10 @@
+## [0.0.2](https://github.com/Datuanalytics/datu-core/tree/0.0.2) - 2025-08-18
+
+### Changed
+
+- Packaging changes
+
+
 ## [0.0.1](https://github.com/Datuanalytics/datu-core/tree/0.0.1) - 2025-08-18
 
 ### Added

README.md

@@ -54,7 +54,7 @@ python -m venv .venv
 source .venv/bin/activate  # On Windows use: .venv\Scripts\activate
 
 #Install datu core
-pip install datu-core
+pip install 'datu-core[all]'
 
 ```
 

pyproject.toml

@@ -23,7 +23,6 @@ dependencies = [
     "langchain-openai>=0.3,<0.4",
     "pydantic-settings>=2.8.1",
     "types-psycopg2>=2.9.21.20250318",
-    "dbt-core>=1.9.3",
     "types-pyyaml>=6.0.12.20250402",
     "sql-metadata>=2.17.0",
     "sentence-transformers>=2.5.1",
@@ -69,7 +68,6 @@ dev = [
 ]
 postgres = [
     "psycopg2-binary>=2.9.6",
-    "dbt-postgres>=1.9.0",
 ]
 sqldb = [
     "pyodbc>=5.2.0",
@@ -84,6 +82,13 @@ docs = [
     "mkdocs-macros-plugin~=1.3.7",
     "mkdocstrings-python~=1.16.10"
 ]
+all = [
+    # postgres
+    "psycopg2-binary>=2.9.6",
+
+    #sqldb
+    "pyodbc>=5.2.0",
+]
 
 [[project.maintainers]]
 name = "Datu"

src/datu/factory/db_connector.py

@@ -1,8 +1,6 @@
 """Factory class to create database connectors based on dbt profiles"""
 
 from datu.integrations.dbt.config import get_dbt_profiles_settings
-from datu.integrations.postgre_sql.postgre_connector import PostgreSQLConnector
-from datu.integrations.sql_server.sqldb_connector import SQLServerConnector
 
 
 class DBConnectorFactory:
@@ -39,8 +37,12 @@ def get_connector(
         db_type = config.type
 
         if db_type == "postgres":
+            from datu.integrations.postgre_sql.postgre_connector import PostgreSQLConnector
+
             return PostgreSQLConnector(config)
         elif db_type == "sqlserver":
+            from datu.integrations.sql_server.sqldb_connector import SQLServerConnector
+
             return SQLServerConnector(config)
         else:
             raise ValueError(f"Unsupported database type: {db_type}")

tests/services/test_graph_rag.py

@@ -21,7 +21,8 @@ def clean_test_graph_cache():
         shutil.rmtree(TEST_GRAPH_DIR)
     os.makedirs(TEST_GRAPH_DIR)
     yield
-    shutil.rmtree(TEST_GRAPH_DIR)
+    if os.path.exists(TEST_GRAPH_DIR):
+        shutil.rmtree(TEST_GRAPH_DIR)
 
 
 def test_init_with_dict_schema():
@@ -82,21 +83,6 @@ def test_initialize_graph_rebuild_and_cache():
     assert set(builder2.graph.edges) == set(builder.graph.edges)
 
 
-def test_graph_contains_expected_edges():
-    """Verify specific graph edges exist after initialization and triple extraction."""
-    schema = SchemaTestFixtures.sample_schema()
-    extractor = SchemaTripleExtractor(schema)
-    extractor.create_schema_triples()
-    builder = SchemaGraphBuilder(triples=extractor.triples, is_rag_outdated=True)
-    builder.initialize_graph()
-    test_graph = builder.graph
-
-    assert test_graph.has_edge("orders", "order_id")
-    assert test_graph.has_edge("order_id", "int") or any(
-        d["label"] == "has_data_type" for _, _, d in test_graph.edges(data=True)
-    )
-
-
 @pytest.mark.requires_service
 def test_schema_rag_run_query_returns_filtered_schema_dict():
     """Test SchemaRAG end-to-end run_query method returns filtered schema."""