Update docs and postgresql connection

Author: BenConstable9

text_2_sql/__init__.py

@@ -224,7 +224,8 @@ If there is no pre-built script for your database engine, take one of the above
 
 ## Running
 
-1. Create your `.env` file based on the provided sample `.env.example`. Place this file in the same place as the `.env.example`.
+**Execute all these commands in the `text_2_sql` directory.**
+1. Create your `.env` file based on the provided sample `text_2_sql/.env.example`. Place this file in the same place in `text_2_sql/.env`.
 2. Package and install the `text_2_sql_core` library. See [build](https://docs.astral.sh/uv/concepts/projects/build/) if you want to build as a wheel and install on an agent. Or you can run from within a `uv` environment and skip packaging.
     - Install the optional dependencies if you need a database connector other than TSQL. `uv sync --extra <DATABASE ENGINE>`
 3. Run `data_dictionary <DATABASE ENGINE>`

text_2_sql/data_dictionary/README.md

@@ -16,6 +16,11 @@ def __init__(self):
 
         self.database_engine = DatabaseEngine.POSTGRESQL
 
+    @property
+    def engine_specific_rules(self) -> str:
+        """Get the engine specific rules."""
+        return ""
+
     @property
     def engine_specific_fields(self) -> list[str]:
         """Get the engine specific fields."""
@@ -64,7 +69,7 @@ async def query_execution(
         connection_string = os.environ["Text2Sql__DatabaseConnectionString"]
 
         # Establish an asynchronous connection to the PostgreSQL database
-        async with psycopg.AsyncConnection.connect(connection_string) as conn:
+        async with await psycopg.AsyncConnection.connect(connection_string) as conn:
             # Create an asynchronous cursor
             async with conn.cursor() as cursor:
                 await cursor.execute(sql_query)

text_2_sql/text_2_sql_core/src/text_2_sql_core/connectors/postgresql_sql.py

@@ -5,6 +5,7 @@
 import logging
 import typer
 from rich import print as rich_print
+from tenacity import RetryError
 
 logging.basicConfig(level=logging.INFO)
 
@@ -112,8 +113,18 @@ def create(
 
     try:
         asyncio.run(data_dictionary_creator.create_data_dictionary())
+    except RetryError as e:
+        # Fetch the actual exception
+        e = e.last_attempt.exception()
+        logging.error(e)
+        rich_print("Text2SQL Data Dictionary Creator Failed ❌")
+
+        rich_print(f"Error Messages: {e}")
+
+        raise typer.Exit(code=1)
     except Exception as e:
         logging.error(e)
+
         rich_print("Text2SQL Data Dictionary Creator Failed ❌")
 
         rich_print(f"Error Messages: {e}")

text_2_sql/text_2_sql_core/src/text_2_sql_core/data_dictionary/cli.py

@@ -21,7 +21,7 @@ class ForeignKeyRelationship(BaseModel):
     column: str = Field(..., alias="Column")
     foreign_column: str = Field(..., alias="ForeignColumn")
 
-    model_config = ConfigDict(populate_by_name=True, arbitrary_types_allowed=True)
+    model_config = ConfigDict(populate_by_name=True)
 
 
 class EntityRelationship(BaseModel):
@@ -39,7 +39,7 @@ class EntityRelationship(BaseModel):
     foreign_database: Optional[str] = Field(default=None, alias="ForeignDatabase")
     foreign_catalog: Optional[str] = Field(default=None, alias="ForeignCatalog")
 
-    model_config = ConfigDict(populate_by_name=True, arbitrary_types_allowed=True)
+    model_config = ConfigDict(populate_by_name=True)
 
     def pivot(self):
         """A method to pivot the entity relationship."""
@@ -128,12 +128,10 @@ class ColumnItem(BaseModel):
     name: str = Field(..., alias="Name")
     data_type: str = Field(..., alias="DataType")
     definition: Optional[str] = Field(..., alias="Definition")
-    distinct_values: Optional[list[any]] = Field(
-        None, alias="DistinctValues", exclude=True
-    )
-    sample_values: Optional[list[any]] = Field(None, alias="SampleValues")
+    distinct_values: Optional[list] = Field(None, alias="DistinctValues", exclude=True)
+    sample_values: Optional[list] = Field(None, alias="SampleValues")
 
-    model_config = ConfigDict(populate_by_name=True, arbitrary_types_allowed=True)
+    model_config = ConfigDict(populate_by_name=True)
 
     def value_store_entry(
         self, entity, distinct_value, excluded_fields_for_database_engine