update response type for get_element (#37)

coffecup25 · web-flow · commit 34f8ddcb9e40 · 2024-09-10T21:45:38.000-07:00
diff --git a/dendrite_sdk/_api/response/get_element_response.py b/dendrite_sdk/_api/response/get_element_response.py
@@ -2,7 +2,10 @@
 
 from pydantic import BaseModel
 
+from dendrite_sdk._common.status import Status
+
 
 class GetElementResponse(BaseModel):
+    status: Status
     selectors: Optional[List[str]] = None
     message: str = ""
diff --git a/dendrite_sdk/_common/status.py b/dendrite_sdk/_common/status.py
@@ -1,4 +1,4 @@
 from typing import Literal
 
 
-Status = Literal["success", "failed"]
+Status = Literal["success", "failed", "wait_for"]

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`	`1`	`from typing import Literal`
`2`	`2`
`3`	`3`
`4`		`-Status = Literal["success", "failed"]`
	`4`	`+Status = Literal["success", "failed", "wait_for"]`