refacto: List editable claim fields

microsoft · Jan 16, 2024 · 28d2d79 · 28d2d79
1 parent 6834c50
commit 28d2d79
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -695,9 +695,7 @@ async def gpt_chat(call: CallModel) -> ActionModel:
                     "properties": {
                         "field": {
                             "description": "The claim field to update.",
-                            "enum": list(
-                                ClaimModel.model_json_schema()["properties"].keys()
-                            ),
+                            "enum": list(ClaimModel.editable_fields()),
                             "type": "string",
                         },
                         "value": {

diff --git a/models/claim.py b/models/claim.py
@@ -1,5 +1,6 @@
 from pydantic import BaseModel
 from typing import Optional
+from typing import Optional, Set
 
 
 class ClaimModel(BaseModel):
@@ -23,3 +24,7 @@ class ClaimModel(BaseModel):
     stolen_lost_items: Optional[str] = None
     vehicle_info: Optional[str] = None
     witnesses: Optional[str] = None
+
+    @staticmethod
+    def editable_fields() -> Set[str]:
+        return ClaimModel.model_json_schema()["properties"].keys()