更新model_name抽取方式 (#657)

* 更新model_name抽取方式 * update * update * update * update --------- Co-authored-by: yinjiaqi <[email protected]>
baidubce · Dec 12, 2024 · 214c5eb · 214c5eb
1 parent 5138b79
commit 214c5eb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/utils/json_schema_to_model.py b/python/utils/json_schema_to_model.py
@@ -13,6 +13,7 @@
 
 
 NON_ALPHANUMERIC = re.compile(r"[^a-zA-Z0-9]+")
+STARTS_WITH_NUMBER = re.compile(r'[0-9]+')
 UPPER_CAMEL_CASE = re.compile(r"[A-Z][a-zA-Z0-9]+")
 LOWER_CAMEL_CASE = re.compile(r"[a-z][a-zA-Z0-9]+")
 
@@ -22,7 +23,7 @@ class BadJsonSchema(Exception):
 
 def _to_camel_case(name: str) -> str:
     if any(NON_ALPHANUMERIC.finditer(name)):
-        return "".join(term.lower().title() for term in NON_ALPHANUMERIC.split(name))
+        return  "".join(term.lower().title() if not STARTS_WITH_NUMBER.match(term) else term.lower() for term in NON_ALPHANUMERIC.split(name))
     if UPPER_CAMEL_CASE.match(name):
         return name
     if LOWER_CAMEL_CASE.match(name):