fix other mypy madness

Author: BeArchiTek

infrahub_sdk/node/node.py

@@ -580,8 +580,9 @@ async def artifact_generate(self, name: str) -> None:
         self._validate_artifact_support(ARTIFACT_GENERATE_FEATURE_NOT_SUPPORTED_MESSAGE)
 
         artifact = await self._client.get(kind="CoreArtifact", name__value=name, object__ids=[self.id])
-        await artifact._get_relationship_one(name="definition").fetch()
-        await artifact._get_relationship_one(name="definition").peer.generate([artifact.id])
+        if artifact.id:
+            await artifact._get_relationship_one(name="definition").fetch()
+            await artifact._get_relationship_one(name="definition").peer.generate([artifact.id])
 
     async def artifact_fetch(self, name: str) -> str | dict[str, Any]:
         self._validate_artifact_support(ARTIFACT_GENERATE_FEATURE_NOT_SUPPORTED_MESSAGE)
@@ -1172,8 +1173,9 @@ def generate(self, nodes: list[str] | None = None) -> None:
     def artifact_generate(self, name: str) -> None:
         self._validate_artifact_support(ARTIFACT_GENERATE_FEATURE_NOT_SUPPORTED_MESSAGE)
         artifact = self._client.get(kind="CoreArtifact", name__value=name, object__ids=[self.id])
-        artifact._get_relationship_one(name="definition").fetch()
-        artifact._get_relationship_one(name="definition").peer.generate([artifact.id])
+        if artifact.id:
+            artifact._get_relationship_one(name="definition").fetch()
+            artifact._get_relationship_one(name="definition").peer.generate([artifact.id])
 
     def artifact_fetch(self, name: str) -> str | dict[str, Any]:
         self._validate_artifact_support(ARTIFACT_FETCH_FEATURE_NOT_SUPPORTED_MESSAGE)

infrahub_sdk/node/related_node.py

@@ -34,7 +34,7 @@ def __init__(self, branch: str, schema: RelationshipSchemaAPI, data: Any | dict,
         self._properties_object = PROPERTIES_OBJECT
         self._properties = self._properties_flag + self._properties_object
 
-        self._peer = None
+        self._peer: CoreNodeBase | InfrahubNode | InfrahubNodeSync | None = None
         self._id: str | None = None
         self._hfid: list[str] | None = None
         self._display_label: str | None = None