feat(schema): add new fields to task query result

Author: BobAnkh

README.md

@@ -54,6 +54,7 @@ c.download_artifact(
 
 | **Mitosis** | **pynetmito** |
 | :---------: | :-----------: |
+|    0.6.1    |     0.2.2     |
 |    0.6.0    |     0.2.1     |
 |    0.6.0    |     0.2.0     |
 |    0.5.3    |     0.1.3     |

src/pynetmito/schemas.py

@@ -241,6 +241,7 @@ class TaskResultMessage(str, Enum):
     RESOURCE_FORBIDDEN = "ResourceForbidden"
     WATCH_TIMEOUT = "WatchTimeout"
     USER_CANCELLATION = "UserCancellation"
+    SUBMIT_NEW_TASK_FAILED = "SubmitNewTaskFailed"
 
 
 class RemoteResourceArtifact(BaseAPIModel):
@@ -432,6 +433,8 @@ class TaskQueryInfo(BaseAPIModel):
     priority: int
     spec: dict
     result: Optional[dict] = Field(default=None)
+    upstream_task_uuid: Optional[UUID4] = Field(default=None)
+    downstream_task_uuid: Optional[UUID4] = Field(default=None)
 
     @field_validator("created_at", mode="before")
     @classmethod
@@ -471,6 +474,8 @@ class ParsedTaskQueryInfo(BaseAPIModel):
     priority: int
     spec: TaskSpec
     result: Optional[TaskResultSpec] = Field(default=None)
+    upstream_task_uuid: Optional[UUID4] = Field(default=None)
+    downstream_task_uuid: Optional[UUID4] = Field(default=None)
 
     @field_validator("created_at", mode="before")
     @classmethod