fix: SQL connection parsing error (#312)

Author: zakahan

veadk/memory/short_term_memory.py

@@ -137,6 +137,12 @@ class ShortTermMemory(BaseModel):
     def model_post_init(self, __context: Any) -> None:
         if self.db_url:
             logger.info("The `db_url` is set, ignore `backend` option.")
+            if self.db_url.count("@") > 1 or self.db_url.count(":") > 2:
+                logger.warning(
+                    "Multiple `@` or `:` symbols detected in the database URL. "
+                    "Please encode `username` or `password` with `urllib.parse.quote_plus`. "
+                    "Examples: p@ssword→p%40ssword."
+                )
             self._session_service = DatabaseSessionService(db_url=self.db_url)
         else:
             if self.backend == "database":

veadk/memory/short_term_memory_backends/mysql_backend.py

@@ -21,6 +21,7 @@
 )
 from pydantic import Field
 from typing_extensions import override
+from urllib.parse import quote_plus
 
 import veadk.config  # noqa E401
 from veadk.configs.database_configs import MysqlConfig
@@ -33,7 +34,9 @@ class MysqlSTMBackend(BaseShortTermMemoryBackend):
     mysql_config: MysqlConfig = Field(default_factory=MysqlConfig)
 
     def model_post_init(self, context: Any) -> None:
-        self._db_url = f"mysql+pymysql://{self.mysql_config.user}:{self.mysql_config.password}@{self.mysql_config.host}/{self.mysql_config.database}"
+        encoded_username = quote_plus(self.mysql_config.user)
+        encoded_password = quote_plus(self.mysql_config.password)
+        self._db_url = f"mysql+pymysql://{encoded_username}:{encoded_password}@{self.mysql_config.host}/{self.mysql_config.database}"
 
     @cached_property
     @override

veadk/memory/short_term_memory_backends/postgresql_backend.py

@@ -22,6 +22,7 @@
 )
 from pydantic import Field
 from typing_extensions import override
+from urllib.parse import quote_plus
 
 import veadk.config  # noqa E401
 from veadk.configs.database_configs import PostgreSqlConfig
@@ -34,7 +35,9 @@ class PostgreSqlSTMBackend(BaseShortTermMemoryBackend):
     postgresql_config: PostgreSqlConfig = Field(default_factory=PostgreSqlConfig)
 
     def model_post_init(self, context: Any) -> None:
-        self._db_url = f"postgresql://{self.postgresql_config.user}:{self.postgresql_config.password}@{self.postgresql_config.host}:{self.postgresql_config.port}/{self.postgresql_config.database}"
+        encoded_username = quote_plus(self.postgresql_config.user)
+        encoded_password = quote_plus(self.postgresql_config.password)
+        self._db_url = f"postgresql://{encoded_username}:{encoded_password}@{self.postgresql_config.host}:{self.postgresql_config.port}/{self.postgresql_config.database}"
         logger.debug(self._db_url)
 
     @cached_property