optionals

BoundaryML · Jun 19, 2024 · 930e908 · 930e908
1 parent 98c0035
commit 930e908
Show file tree

Hide file tree

Showing 2 changed files with 185 additions and 179 deletions.
diff --git a/integ-tests/python/baml_client/partial_types.py b/integ-tests/python/baml_client/partial_types.py
@@ -44,14 +44,14 @@ class ClassOptionalOutput2(BaseModel):
 
     prop1: Optional[str] = None
     prop2: Optional[str] = None
-    prop3: "Blah"
+    prop3: Optional["Blah"] = None
 
 class ClassWithImage(BaseModel):
 
 
     myImage: Optional[baml_py.Image] = None
     param2: Optional[str] = None
-    fake_image: "FakeImage"
+    fake_image: Optional["FakeImage"] = None
 
 class DynamicClassOne(BaseModel):
 
@@ -63,7 +63,7 @@ class DynamicClassTwo(BaseModel):
     model_config = ConfigDict(extra='allow')
 
     hi: Optional[str] = None
-    some_class: "SomeClassNestedDynamic"
+    some_class: Optional["SomeClassNestedDynamic"] = None
     status: Optional[Union[types.DynEnumOne, str]] = None
 
 class DynamicOutput(BaseModel):
@@ -105,7 +105,7 @@ class InnerClass(BaseModel):
 
     prop1: Optional[str] = None
     prop2: Optional[str] = None
-    inner: "InnerClass2"
+    inner: Optional["InnerClass2"] = None
 
 class InnerClass2(BaseModel):
 
@@ -129,7 +129,7 @@ class OptionalTest_Prop1(BaseModel):
 class OptionalTest_ReturnType(BaseModel):
 
 
-    omega_1: "OptionalTest_Prop1"
+    omega_1: Optional["OptionalTest_Prop1"] = None
     omega_2: Optional[str] = None
     omega_3: List[Optional[types.OptionalTest_CategoryType]]
 
@@ -168,8 +168,8 @@ class SearchParams(BaseModel):
 
     dateRange: Optional[int] = None
     location: List[Optional[str]]
-    jobTitle: "WithReasoning"
-    company: "WithReasoning"
+    jobTitle: Optional["WithReasoning"] = None
+    company: Optional["WithReasoning"] = None
     description: List["WithReasoning"]
     tags: List[Optional[Union[Optional[types.Tag], Optional[str]]]]
 
@@ -192,7 +192,7 @@ class TestClassNested(BaseModel):
 
 
     prop1: Optional[str] = None
-    prop2: "InnerClass"
+    prop2: Optional["InnerClass"] = None
 
 class TestClassWithEnum(BaseModel):