Merge pull request #47 from kontext-tech/32-add-data-sources-postgresql

feat: added support for PostgreSQL using psycopg 3
kontext-tech · Sep 9, 2024 · 4f78764 · 4f78764
2 parents f25483a + 59bd7a1
commit 4f78764
Show file tree

Hide file tree

Showing 6 changed files with 105 additions and 5 deletions.
diff --git a/kontext_copilot/data/models/_db_models.py b/kontext_copilot/data/models/_db_models.py
@@ -21,7 +21,7 @@ class Setting(Base):
 class DataSourceType(str, Enum):
     SQLite = "SQLite"
     DuckDB = "DuckDB"
-    # PostgreSQL = "PostgreSQL"
+    PostgreSQL = "PostgreSQL"
     # MySQL = "MySQL"
     SQLServer = "SQLServer"
     # Oracle = "Oracle"

diff --git a/kontext_copilot/services/_data_provider_service/_service.py b/kontext_copilot/services/_data_provider_service/_service.py
@@ -20,6 +20,7 @@ def get_data_provider(source: DataSourceModel):
             DataSourceType.SQLite,
             DataSourceType.DuckDB,
             DataSourceType.SQLServer,
+            DataSourceType.PostgreSQL,
         ]
         if source.type in supported_types:
             return BaseProvider(source)

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,6 +25,7 @@ duckdb = "^1.0.0"
 pyarrow = "^17.0.0"
 duckdb-engine = "^0.13.1"
 pyodbc = "^5.1.0"
+psycopg = {extras = ["binary"], version = "^3.2.1"}
 [tool.commitizen]
 name = "cz_conventional_commits"
 tag_format = "v$version"

diff --git a/ui/components/data-source/create-form.vue b/ui/components/data-source/create-form.vue
@@ -123,6 +123,9 @@ watch(
          } else if (newVal == DataSourceType.SQLServer) {
             dataSourceCreateModel.value.connStr =
                "mssql+pyodbc://user:password@host:port/databasename?driver=ODBC+Driver+18+for+SQL+Server&TrustServerCertificate=yes"
+         } else if (newVal == DataSourceType.PostgreSQL) {
+            dataSourceCreateModel.value.connStr =
+               "postgresql+psycopg://user:password@host:port/databasename?key=value&key=value"
          }
       }
    }

diff --git a/ui/types/Schemas.ts b/ui/types/Schemas.ts
@@ -144,7 +144,7 @@ export interface PromptListModel {
 export enum DataSourceType {
    SQLite = "SQLite",
    DuckDB = "DuckDB",
-   // PostgreSQL = "PostgreSQL",
+   PostgreSQL = "PostgreSQL",
    // MySQL = "MySQL",
    SQLServer = "SQLServer"
    // Oracle = "Oracle",