chore(internal): minor restructuring (#84)

Finch-API · Aug 31, 2023 · fafab61 · fafab61
1 parent fff39ad
commit fafab61
Showing 1 changed file with 7 additions and 27 deletions.
diff --git a/src/finch/types/hris/benefit_create_params.py b/src/finch/types/hris/benefit_create_params.py
@@ -2,38 +2,18 @@
 
 from __future__ import annotations
 
-from typing import Optional
-from typing_extensions import Literal, TypedDict
+from typing_extensions import TypedDict
+
+from .benefit_type import BenefitType
+from .benefit_frequency import BenefitFrequency
 
 __all__ = ["BenefitCreateParams"]
 
 
 class BenefitCreateParams(TypedDict, total=False):
     description: str
 
-    frequency: Optional[Literal["one_time", "every_paycheck"]]
-
-    type: Optional[
-        Literal[
-            "401k",
-            "401k_roth",
-            "401k_loan",
-            "403b",
-            "403b_roth",
-            "457",
-            "457_roth",
-            "s125_medical",
-            "s125_dental",
-            "s125_vision",
-            "hsa_pre",
-            "hsa_post",
-            "fsa_medical",
-            "fsa_dependent_care",
-            "simple_ira",
-            "simple",
-            "commuter",
-            "custom_post_tax",
-            "custom_pre_tax",
-        ]
-    ]
+    frequency: BenefitFrequency
+
+    type: BenefitType
     """Type of benefit."""