Update components/agentcreation.py

components/agentcreation.py

@@ -31,14 +31,39 @@ async def create_keypair(agent):
         raise
 
 
-async def transfer_sol(from_keypair, to_pubkey, amount_sol):
+
+
+def load_keypair(agent):
+
+            document=coll.find_one({"owner":"system"})
+            keypair=None
+            agents=document.get("agents")
+            print(agents)
+    
+            for doc in agents:
+                if doc["agent_id"]==agent:
+                    keypair=doc["secret_key"]
+                    
+            secret_key_bytes =bytes.fromhex(keypair)
+            keypair = Keypair.from_bytes(secret_key_bytes)
+        
+            print(f"✅ Keypair loaded")
+            print(f"📍 Public Key: {keypair.pubkey()}")
+            publickey=str(keypair.pubkey())
+            return publickey
+
+    
+
+
+async def transfer_sol(agent_id, to_pubkey, amount_sol):
             """Transfer SOL from one account to another"""
             try:
+                from_publickey=load_keypair(agent_id)
                 lamports = int(amount_sol * 1e9)
             
                 transfer_ix = transfer(
                     TransferParams(
-                        from_pubkey=from_keypair.pubkey(),
+                        from_pubkey=from_publickey,
                         to_pubkey=to_pubkey,
                         lamports=lamports
                     ))
@@ -60,24 +85,7 @@ async def transfer_sol(from_keypair, to_pubkey, amount_sol):
 
 
 
-def load_keypair(agent):
 
-            document=coll.find_one({"owner":"system"})
-            keypair=None
-            agents=document.get("agents")
-            print(agents)
-    
-            for doc in agents:
-                if doc["agent_id"]==agent:
-                    keypair=doc["secret_key"]
-                    
-            secret_key_bytes =bytes.fromhex(keypair)
-            keypair = Keypair.from_bytes(secret_key_bytes)
-        
-            print(f"✅ Keypair loaded")
-            print(f"📍 Public Key: {keypair.pubkey()}")
-            publickey=str(keypair.pubkey())
-            return publickey
 
 
 
@@ -85,7 +93,7 @@ def load_keypair(agent):
 class VelocityPayableAgent():
 
     def __init__(self):
-        self.tools = [load_keypair,transfer_sol]
+        self.tools = [transfer_sol]
 
 
     
@@ -105,28 +113,8 @@ class VelocityPayableAgent():
     def Agent(self,user_input):
         client = Groq(api_key="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")
         available_functions= {
-            "load_keypair": load_keypair,
             "transfer_sol":transfer_sol,
         }
-        load_keypair_tool = {
-            "type": "function",
-            "function": {
-                "name": "load_keypair",
-                "description": "Load the Solana keypair for for the provided  agent id.",
-                "parameters": {
-                    "type": "object",
-                    "properties": {
-                        "agent":{
-                            "type":"string",
-                            "description":"agents id"
-                            
-                        }
-                    },
-                    "required": ["agent"]
-                }
-            }
-            }
-
         transfer_sol_tool = {
             "type": "function",
             "function": {
@@ -135,9 +123,9 @@ class VelocityPayableAgent():
                 "parameters": {
                 "type": "object",
                 "properties": {
-                    "from_keypair": {
+                    "agent_id": {
                         "type": "string",
-                        "description": "secret key of the agent"
+                        "description": "id of the agent"
                     },
                     "to_pubkey": {
                         "type": "string",
@@ -159,10 +147,9 @@ class VelocityPayableAgent():
              Rules:
                 1. You MUST call a tool for every query from the user.
                 2. You are NOT allowed to answer directly.
-                3. If multiple tools exist, choose the one that is closest to the user's intent.
-                4. Always return a tool call in the correct function-calling JSON format.
-                5. When unsure, still pick a tool instead of answering directly.
-                6. Do not apologize, do not explain, do not output reasoning—only valid tool calls.
+                3. Always return tool call in the correct function-calling JSON format.
+                4. When unsure, still pick a tool instead of answering directly.
+                5. Do not apologize, do not explain, do not output reasoning—only valid tool calls.
              """},
             {"role": "user", "content": user_input}]
     
@@ -191,7 +178,4 @@ class VelocityPayableAgent():
         print(final.choices[0].message.content)
         return final.choices[0].message.content
 
-        
-
-        
-    
+