Update engine.py

engine.py

@@ -1,34 +1,33 @@
-from components.functions import NetworkResolver , HeliusAPI ,InfuraRPC , EthereumSigner ,SolanaSigner,EthereumPayment,SolanaPayment
-#from components.ucg_g1 import UGC_GRAPH_1
+from components.functions import NetworkResolver , HeliusAPI ,InfuraRPC , EthereumSigner ,SolanaSigner,EthereumPayment
+from components.ucg_g1 import UGC_GRAPH_1
 
 
 
+def PayloadGen(address):
+    payload=f"PAYLOAD:{address}"
+    return payload
+
 
 TOOLS = {
      
-    "NetworkResolver": lambda state: NetworkResolver(state["address"]),
-    "InfuraRPC":       lambda state: InfuraRPC(state["address"]),
-    "HeliusAPI":       lambda state: HeliusAPI(state["address"]),
-    "PayloadGenerator": lambda state: {
-        "payload": f"Sign for {state['address']} with balance {state.get('eth_balance', state.get('sol_balance',0))}"
-    },
-    "EthereumSigner":  lambda state: EthereumSigner(state["address"], state.get("payload","")),
-    "SolanaSigner":    lambda state: SolanaSigner(state["address"], state.get("payload","")),
-    "EthereumPayment":  lambda state:EthereumPayment(state["address"],state.get("to"),state.get("amount")),
-    "SolanaPayment":  lambda state:SolanaPayment(state["address"],state.get("to"),state.get("amount"))
-    
+    "NetworkResolver":  lambda state: NetworkResolver(state),
+    "InfuraRPC":        lambda state: InfuraRPC(state),
+    "HeliusAPI":        lambda state: HeliusAPI(state),
+    "PayloadGenerator": lambda state: PayloadGen(state),
+    "EthereumSigner":   lambda state: EthereumSigner(state),
+    "SolanaSigner":     lambda state: SolanaSigner(state),
+    "EthereumPayment":  lambda state: EthereumPayment(state)
 }
 
 
-
 class UCGEngine:
     def __init__(self, ugc_graph, tools): 
         self.ops = {op["operation"]: op for op in ugc_graph}
         self.tools = tools 
         self.executed = set()
 
-    def run(self, operation, address): 
-        self.state = address
+    def run(self, operation,state): 
+        self.state = state
         #initial facts 
         self.executed = set() 
         self._execute(operation)
@@ -52,9 +51,11 @@ class UCGEngine:
                 return
         
         if op_name in self.tools: 
+
+            new_facts=self.tools[op_name](self.state)
+
             new_facts =self.tools[op_name](self.state) 
+
             print(new_facts)
             self.state.update(new_facts)
-            self.executed.add(op_name)
-              
-        
+            self.executed.add(op_name)