Resovles breaking changes introduced by pydantic 2.0.3 (fixes #66)

hugie/config.py

@@ -100,13 +100,13 @@ def modify(
 
     if overwrite:
         try:
-            srsly.write_json(path, config.dict())
+            srsly.write_json(path, config.model_dump())
             typer.secho(f"Updated config at {path}", fg=typer.colors.GREEN)
         except Exception:
             typer.secho(f"Failed to update config at {path}", fg=typer.colors.RED)
 
     else:
-        typer.secho(srsly.json_dumps(config.dict()), fg=typer.colors.YELLOW)
+        typer.secho(srsly.json_dumps(config.model_dump()), fg=typer.colors.YELLOW)
 
 
 if __name__ == "__main__":

hugie/endpoint.py

@@ -88,7 +88,7 @@ def create(
         data (str): Path to JSON data to create the endpoint
     """
 
-    data = InferenceEndpointConfig.from_json(data).dict()
+    data = InferenceEndpointConfig.from_json(data).model_dump()
 
     try:
         response = requests.post(settings.endpoint_url, headers=headers, json=data)
@@ -126,7 +126,7 @@ def update(
     """
     Update an endpoint
     """
-    data = InferenceEndpointConfig.from_json(data).dict()
+    data = InferenceEndpointConfig.from_json(data).model_dump()
 
     try:
         response = requests.put(

hugie/models.py

@@ -1,4 +1,6 @@
-from pydantic import BaseModel, BaseSettings
+from pydantic import BaseModel
+from pydantic_settings import BaseSettings
+from typing import Optional
 
 from hugie.utils import load_json
 
@@ -35,7 +37,7 @@ class InferenceEndpointConfig(BaseSettings):
     Config for the inference endpoint
     """
 
-    accountId: str = None
+    accountId: Optional[str] = None
     type: str = None
     compute: ComputeModel = ComputeModel()
     model: ModelModel = ModelModel()

setup.cfg

@@ -20,6 +20,7 @@ install_requires =
     requests
     pydantic
     srsly
+    pydantic-settings
 python_requires = >= 3.6
 
 

tests/test_config.py

@@ -40,7 +40,7 @@ def data_path(tmp_path):
 
 def test_inference_endpoint_config_serialization(data_path):
     config = InferenceEndpointConfig.from_json(data_path)
-    config = config.dict()
+    config = config.model_dump()
     assert isinstance(config, dict)
     assert config["compute"]["instanceSize"] == "small"
     assert config["model"]["framework"] == "custom"