fix: empty hash

schema.graphql

@@ -127,6 +127,6 @@ type UpdateAgentHash @entity(immutable: true) {
 
 type MechAgent @entity {
   id: ID! # uint256
-  mech: Bytes! # address
+  mech: Bytes # address
   agentHash: Bytes # bytes32
 }

src/agent-factory.ts

@@ -22,9 +22,19 @@ export function handleCreateMech(event: CreateMechEvent): void {
 
   entity.save()
 
-  let mechAgent = new MechAgent(event.params.agentId.toHexString());
-  mechAgent.mech = event.params.mech;
-  mechAgent.save()
+  let mechAgent = MechAgent.load(event.params.agentId.toHexString());
+
+  // Mech is created after agent, which already handles mechAgent creation
+  // add this check just in case
+  if (mechAgent !== null) {
+    mechAgent.mech = event.params.mech;
+    mechAgent.save()
+  } else {
+    mechAgent = new MechAgent(event.params.agentId.toHexString());
+    mechAgent.mech = event.params.mech;
+    mechAgent.save()
+  }
+
 
   AgentMech.create(event.params.mech);
 }

src/agent-registry.ts

@@ -76,6 +76,17 @@ export function handleCreateAgent(event: CreateAgentEvent): void {
   entity.transactionHash = event.transaction.hash
 
   entity.save()
+
+  let mechAgent = MechAgent.load(event.params.agentId.toHexString());
+
+  if (mechAgent !== null) {
+    mechAgent.agentHash = event.params.agentHash;
+    mechAgent.save()
+  } else {
+    mechAgent = new MechAgent(event.params.agentId.toHexString());
+    mechAgent.agentHash = event.params.agentHash;
+    mechAgent.save()
+  }
 }
 
 export function handleManagerUpdated(event: ManagerUpdatedEvent): void {