Update app_layer_agent.py

app_layer_agent.py

@@ -11,35 +11,61 @@ runner=UCGEngine(UGC_GRAPH_1,TOOLS)
 
 
 
-def UCGExecutor(operation: str,address: str, ):
-    """
-    Execute a UGC graph operation for a given address.
-    
-    Args:
-        
-        operation (str): The UGC operation to run (goal node in the graph).
-        address (str): Ethereum or Solana wallet address.
-        
-    Returns:
-        dict: The final state returned by the UGC engine.
-    """
-    inputs = {"address": address}
-
-    # Assuming `runner` is an instance of UGCRunner with UGC_GRAPH_1 and TOOLS
-    result = runner.run(operation,inputs)
-    return result
 
 
 
 
 available_functions = {
-    "UCGExecutor":UCGExecutor
+    "UCGEngine":runner.run
   
 }
 
 
 
 
+system='''
+You are an autonomous agent with access to the UGC_Execution_Tool.
+
+The tool has 2 parameters :
+    Operation : The operation to perform from Supported Operations inferred from the user request
+    State: The required parameters for that operation inferred from the user request
+
+
+
+Supported Operations:
+
+1. NetworkResolver
+   Inputs:
+     - address (string)
+
+2. InfuraRPC
+   Inputs:
+     - address (string)
+
+3. HeliusAPI
+   Inputs:
+     - address (string)
+
+4. EthereumSigner
+   Inputs:
+     - address (string)
+     - payload (hex string)
+
+5. SolanaSigner
+   Inputs:
+     - address (string)
+     - payload (hex string)
+
+
+7. SolanaPayment
+   Inputs:
+     - address (string)
+     - to (string)
+     - amount (string)
+
+'''
+
+
 def run_conversation(user_prompt):
     """Run a conversation with tool calling"""
     
@@ -47,25 +73,7 @@ def run_conversation(user_prompt):
 
         {
             "role":"system",
-            "content":'''
-            
-You are an AI assistant for the UGC Engine. Your job is to convert a user's natural language query into a valid tool call for the UGCExecutor.
-
-The tool accepts only two fields:
-1. "address": The wallet address (Ethereum hex or Solana Base58) to operate on.
-2. "operation": The UGC operation/tool to run. 
-
-Valid operations/tools available:
-- InfuraRPC: Fetch ETH balance for an Ethereum address.
-- HeliusAPI: Fetch SOL balance for a Solana address.
-- EthereumSigner: Produce a cryptographic signature for a given Ethereum address.
-- SolanaSigner: Produce a cryptographic signature for a given Solana address.
-
-Analyze the user query to determine:
-- Which wallet address is mentioned.
-- Which operation/tool the query implies.
-Then call the tool with the required inputs
-'''
+            "content":system
         },
         {
             "role": "user",
@@ -75,33 +83,31 @@ Then call the tool with the required inputs
     
     # Define the tool schema
     tools = [
-        
-    {
-    "type": "function",
-    "function": {
-        "name": "UCGExecutor",
-        "description": "Execute a Unified Capabilities Graph operation for a given wallet address. The AI specifies the target operation, and the engine resolves the workflow automatically.",
-        "parameters": {
-            "type": "object",
-            "properties": {
-
-                 "operation": {
-                    "type": "string",
-                    "description": "The name of the target operation/goal to run in the UGC graph (e.g., EthereumSigner, SolanaSigner, InfuraRPC, HeliusAPI)."
-                },
-                "address": {
-                    "type": "string",
-                    "description": "The wallet address to operate on (Ethereum hex or Solana Base58)."
-                },
-               
-            },
-            "required": ["address", "operation"],
-            "additionalProperties": False
+
+        {
+      "type": "function",
+      "function": {
+            "name": "UGC_Execution_Tool",
+            "description": "Execute a specified operation with the provided state using the UGCEngine",
+            "parameters": 
+              "type": "object",
+                  "properties": {
+                  "operation": {
+                      "type": "string",
+                      "description": "The name of the operation to execute. Must correspond to a valid operation in Supported Operations."
+                    },
+                "state": {
+                  "type": "object",
+                "description": "Dictionary of input values for the operation inferred from the user query for the Supported Operations",
         }
+      },
+      "required": ["operation", "state"],
+      "additionalProperties": false
     }
+  }
 }
 
-
+        
     ]
     
     # Step 1: Make initial API call
@@ -128,7 +134,7 @@ Then call the tool with the required inputs
             function_args = json.loads(tool_call.function.arguments)
             function_response = function_to_call(
                 operation=function_args.get("operation"),
-                address=function_args.get("address")
+                address=function_args.get("state")
             )
             
             # Add tool response to conversation