fix: add types for backwards compatibility

Author: saunack-shapley

src/parallel/types/parsed_task_run_result.py

@@ -3,7 +3,9 @@
 from pydantic import BaseModel
 
 from .._models import GenericModel
-from .task_run_result import TaskRunResult, OutputTaskRunJsonOutput, OutputTaskRunTextOutput
+from .task_run_result import TaskRunResult
+from .task_run_json_output import TaskRunJsonOutput
+from .task_run_text_output import TaskRunTextOutput
 
 ContentType = TypeVar("ContentType", bound=BaseModel)
 
@@ -14,16 +16,23 @@
 # pyright: reportIncompatibleVariableOverride=false
 
 
-class ParsedOutputTaskRunTextOutput(OutputTaskRunTextOutput, GenericModel, Generic[ContentType]):
+class ParsedTaskRunTextOutput(TaskRunTextOutput, GenericModel, Generic[ContentType]):
     parsed: None
     """The parsed output from the task run."""
 
 
-class ParsedOutputTaskRunJsonOutput(OutputTaskRunJsonOutput, GenericModel, Generic[ContentType]):
+class ParsedTaskRunJsonOutput(TaskRunJsonOutput, GenericModel, Generic[ContentType]):
     parsed: Optional[ContentType] = None
     """The parsed output from the task run."""
 
 
 class ParsedTaskRunResult(TaskRunResult, GenericModel, Generic[ContentType]):
-    output: Union[ParsedOutputTaskRunTextOutput[ContentType], ParsedOutputTaskRunJsonOutput[ContentType]]  # type: ignore[assignment]
+    output: Union[ParsedTaskRunTextOutput[ContentType], ParsedTaskRunJsonOutput[ContentType]]  # type: ignore[assignment]
     """The parsed output from the task run."""
+
+
+ParsedOutputTaskRunTextOutput = ParsedTaskRunTextOutput  # for backwards compatibility with v0.1.3
+"""This is deprecated, `ParsedTaskRunTextOutput` should be used instead"""
+
+ParsedOutputTaskRunJsonOutput = ParsedTaskRunJsonOutput  # for backwards compatibility with v0.1.3
+"""This is deprecated, `ParsedTaskRunJsonOutput` should be used instead"""

src/parallel/types/task_run.py

@@ -11,7 +11,7 @@
 
 __all__ = [
     "TaskRun",
-# for backwards compatibility with v0.1.3
+    "Warning",  # for backwards compatibility with v0.1.3
 ]
 
 

src/parallel/types/task_run_result.py

@@ -14,19 +14,16 @@
 __all__ = [
     "TaskRunResult",
     "Output",
-    "OutputTaskRunTextOutput",
-    "OutputTaskRunTextOutputBasis",
-    "OutputTaskRunTextOutputBasisCitation",
     "OutputTaskRunJsonOutput",
     "OutputTaskRunJsonOutputBasis",
     "OutputTaskRunJsonOutputBasisCitation",
+    "OutputTaskRunTextOutput",
+    "OutputTaskRunTextOutputBasis",
+    "OutputTaskRunTextOutputBasisCitation",
 ]
 
-OutputTaskRunTextOutputBasis = FieldBasis  # for backwards compatibility with v0.1.3
-"""This is deprecated, `FieldBasis` should be used instead"""
-
-OutputTaskRunTextOutputBasisCitation = Citation  # for backwards compatibility with v0.1.3
-"""This is deprecated, `Citation` should be used instead"""
+OutputTaskRunJsonOutput = TaskRunJsonOutput  # for backwards compatibility with v0.1.3
+"""This is deprecated, `TaskRunJsonOutput` should be used instead"""
 
 OutputTaskRunJsonOutputBasis = FieldBasis  # for backwards compatibility with v0.1.3
 """This is deprecated, `FieldBasis` should be used instead"""
@@ -37,8 +34,11 @@
 OutputTaskRunTextOutput = TaskRunTextOutput  # for backwards compatibility with v0.1.3
 """This is deprecated, `TaskRunTextOutput` should be used instead"""
 
-OutputTaskRunJsonOutput = TaskRunJsonOutput  # for backwards compatibility with v0.1.3
-"""This is deprecated, `TaskRunJsonOutput` should be used instead"""
+OutputTaskRunTextOutputBasis = FieldBasis  # for backwards compatibility with v0.1.3
+"""This is deprecated, `FieldBasis` should be used instead"""
+
+OutputTaskRunTextOutputBasisCitation = Citation  # for backwards compatibility with v0.1.3
+"""This is deprecated, `Citation` should be used instead"""
 
 
 Output: TypeAlias = Annotated[Union[TaskRunTextOutput, TaskRunJsonOutput], PropertyInfo(discriminator="type")]