add types

fluent-qa · Dec 17, 2024 · 1248d64 · 1248d64
1 parent 6fcd836
commit 1248d64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/qpydao/database_client.py b/src/qpydao/database_client.py
@@ -36,7 +36,7 @@ class DatabaseClient:
     """Database client, both synchronous and asynchronous."""
 
     def __init__(self, config: DatabaseConfig = None):
-        self.config = config
+        self.config:DatabaseConfig = config
         self._engine = None
         self._async_engine = None
 
@@ -249,7 +249,7 @@ def get_db(self, db_name: str = "default") -> DatabaseClient:
 
 
 databases: Databases = Databases()
-db = databases.default_client()
+db:DatabaseConfig = databases.default_client()
 
 def init_database(database: DatabaseClient, schema_name: str = ""):
     """Init postgresql database

diff --git a/src/qpydao/models.py b/src/qpydao/models.py
@@ -41,7 +41,6 @@ def check_db_url(self):
             _validate_param("port", self.host)
             _validate_param("database", self.host)
 
-
 class SqlRequestModel(BaseModel):
     """SqlRequestModel: Model for SQLRequest."""