flag to return children or not, cattrs nested converter issues

Author: jalexanderII

parea/client.py

@@ -403,7 +403,7 @@ def _update_data_and_trace(self, data: Completion) -> Completion:
 
         return data
 
-    def get_trace_log(self, trace_id: str, return_children: bool = True) -> TraceLogTree:
+    def get_trace_log(self, trace_id: str, return_children: bool = True) -> Union[TraceLogTree, TraceLog]:
         response = self._client.request("GET", GET_TRACE_LOG_ENDPOINT.format(trace_id=trace_id))
         return structure_trace_log_from_api(response.json(), return_children)
 
@@ -423,10 +423,10 @@ def get_trace_log_scores(self, trace_id: str, check_context: bool = True, return
                 return scores
 
         response = self._client.request("GET", GET_TRACE_LOG_ENDPOINT.format(trace_id=trace_id))
-        tree: TraceLogTree = structure_trace_log_from_api(response.json(), return_children)
+        tree = structure_trace_log_from_api(response.json(), return_children)
         return extract_scores(tree)
 
-    async def aget_trace_log(self, trace_id: str, return_children: bool = True) -> TraceLogTree:
+    async def aget_trace_log(self, trace_id: str, return_children: bool = True) -> Union[TraceLogTree, TraceLog]:
         response = await self._client.request_async("GET", GET_TRACE_LOG_ENDPOINT.format(trace_id=trace_id))
         return structure_trace_log_from_api(response.json(), return_children)