DH-5698/fix the uri validator

Dataherald · Apr 4, 2024 · 37a7c38 · 37a7c38
1 parent c8c6a28
commit 37a7c38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataherald/sql_database/models/types.py b/dataherald/sql_database/models/types.py
@@ -104,7 +104,7 @@ class DatabaseConnection(BaseModel):
 
     @classmethod
     def get_dialect(cls, input_string):
-        pattern = r"([^:/]+):/+([^/]+)/([^/]+)"
+        pattern = r"([^:/]+):/+([^/]+)/?([^/]+)"
         match = re.match(pattern, input_string)
         if not match:
             raise InvalidURIFormatError(f"Invalid URI format: {input_string}")