We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 66688d1 commit fbd96eaCopy full SHA for fbd96ea
dataherald/api/types/requests.py
@@ -1,4 +1,5 @@
1
-from pydantic import BaseModel
+from pydantic import BaseModel, validator
2
+from sql_metadata import Parser
3
4
from dataherald.types import LLMConfig
5
@@ -17,6 +18,14 @@ class SQLGenerationRequest(BaseModel):
17
18
sql: str | None
19
metadata: dict | None
20
21
+ @validator("sql")
22
+ def validate_model_name(cls, v: str | None):
23
+ try:
24
+ Parser(v).tables # noqa: B018
25
+ except Exception as e:
26
+ raise ValueError(f"SQL {v} is malformed. Please check the syntax.") from e
27
+ return v
28
+
29
30
class StreamSQLGenerationRequest(BaseModel):
31
finetuning_id: str | None
0 commit comments