Merge pull request #164 from CambioML/dev

Fix check key in Pydantic BaseModel class bug

Author: Cambio ML

uniflow/op/model/model_server.py

@@ -98,7 +98,10 @@ def __init__(
         self._model_config = model_config
         self._example_keys = None
 
-        if "few_shot_prompt" in prompt_template and prompt_template.few_shot_prompt:
+        if (
+            "few_shot_prompt" in prompt_template.model_fields
+            and prompt_template.few_shot_prompt  # noqa: W503
+        ):
             self._example_keys = list(
                 prompt_template.few_shot_prompt[0].model_dump().keys()
             )