DH-5698/fix the uri validator (#455)

Dataherald · May 7, 2024 · 8aecf63 · 8aecf63
1 parent bb4a5e1
commit 8aecf63
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}")