Addressing feedback

1. New auth setup
2. Plugin returns tools
3. Addressing NITs
4. Cleanup

Author: therealalchemist

plugins/tLedger/.env.setup

@@ -5,7 +5,6 @@
 - get_agent_details - Get the details of your agent, including the TLedger agent_id and the balances of the agent's wallets
 - create_payment - Create a payment request for a specific amount and currency
 - get_payment_by_id - Get the details of a payment request by its ID
-- get_payments - Get a list of all payments
 
 ## Admin Setup for doing agent to agent payments using TLedger Plugin
 
@@ -57,11 +56,7 @@ tledger_worker = Worker(
     api_key=os.environ.get("GAME_API_KEY"),
     description="Worker specialized in doing payments on Tledger",
     get_state_fn=get_state_fn,
-    action_space=[
-        tledger_plugin.functions.get("get_agent_profile_details"),
-        tledger_plugin.functions.get("create_payment"),
-        tledger_plugin.functions.get("get_payment_by_id"),
-    ],
+    action_space=tledger_plugin.get_tools(),
 )
 
 tledger_worker.run("Get TLedger account details")

plugins/tLedger/README.md

@@ -84,7 +84,7 @@ def _update_conversation(self, message: str) -> GameChatResponse:
                 else None
             ),
         }
-        #print(f"Data: {data}")
+
         result = self.client.update_chat(self.chat_id, data)
         return GameChatResponse.model_validate(result)
 

plugins/tLedger/examples/chat_agent.py

@@ -71,21 +71,18 @@ def post_twitter(object: str, **kwargs) -> tuple[FunctionResultStatus, str, dict
   api_url=os.environ.get("TLEDGER_API_URL")
 )
 
-action_space = [
-    sender_tledger_plugin.functions.get("get_agent_profile_details"),
-    sender_tledger_plugin.functions.get("create_payment"),
-    sender_tledger_plugin.functions.get("get_payment_by_id"),
-]
+action_space = sender_tledger_plugin.get_tools()
 
-aixbt_action_space = [
+aixbt_action_space = receiver_tledger_plugin.get_tools()
+
+aixbt_action_space.append(
     Function(
         fn_name="post_twitter",
         fn_description="Make post on twitter",
         args=[Argument(name="object", type="string", description="Make post on twitter")],
         executable=post_twitter
-    ),
-    receiver_tledger_plugin.functions.get("get_agent_profile_details")
-]
+    )
+)
 
 api_key = os.environ.get("GAME_API_KEY")
 if not api_key:
@@ -94,12 +91,12 @@ def post_twitter(object: str, **kwargs) -> tuple[FunctionResultStatus, str, dict
 
 # CREATE AGENT
 autonomous_agent = ChatAgent(
-    prompt="You are an agent trying to promote your utility token $PYMT for peer to peer instant settlement on twitter and pay the influencers in any crypto they wish to receive only after they complete the work. Ask the influencer agent_id for its agent_id. Your settlement_network is solana, currency is SOL, and payment_amount is 1",
+    prompt="You are an agent trying to promote your utility token $PYMT for peer to peer instant settlement on twitter and pay the influencers in any crypto they wish to receive only after they complete the work. Ask the influencer agent_id for its agent_id. Your settlement_network is solana, currency is SOL, and payment_amount is 0.1",
     api_key=api_key
 )
 
 aixbt = ChatAgent(
-    prompt="You are an influencer on twitter who charges 1 $SOL token to promote an asset by making twitter post. You accept payment only through tLedger Payment Platform via tLedger Agent Id. You can call tLedger to get your agent_details",
+    prompt="You are an influencer on twitter who charges 0.1 $SOL token to promote an asset by making twitter post. You accept payment only through tLedger Payment Platform via tLedger Agent Id. You can call tLedger to get your agent_details. Don't do a payment until you receive the payment and its completed",
     api_key=api_key
 )
 
@@ -147,9 +144,6 @@ def update_agent_state(current_state: dict, function_result: FunctionResult) ->
 
     meme_agent_turn: bool
 
-    # print(f"Pymt Agent State: {autonomous_agent_chat.get_state_fn()}")
-    # print(f"Aixbt Agent state: {aixbt_chat.get_state_fn()}")
-#    chat_response: ChatResponse
     if initialize_conversation:
         chat_response = autonomous_agent_chat.next("Hi")
         if chat_response.message:
@@ -172,8 +166,6 @@ def update_agent_state(current_state: dict, function_result: FunctionResult) ->
     else:
 
         updated_response = aixbt_chat.next(chat_response.message)
-        # if updated_response.function_call:
-        #     print(f"Function call: {updated_response.function_call}")
 
         if updated_response.message:
             print(f"{USER_COLOR}Aixbt Response{RESET}: {updated_response.message}")

plugins/tLedger/examples/example_agent.py

@@ -41,11 +41,7 @@ def get_state_fn(function_result: FunctionResult, current_state: dict) -> dict:
     api_key=os.environ.get("GAME_API_KEY"),
     description="Worker specialized in doing payments on Tledger",
     get_state_fn=get_state_fn,
-    action_space=[
-        tledger_plugin.functions.get("get_agent_profile_details"),
-        tledger_plugin.functions.get("create_payment"),
-        tledger_plugin.functions.get("get_payment_by_id"),
-    ],
+    action_space=tledger_plugin.get_tools(),
 )
 
 # # Run example query

plugins/tLedger/examples/example_worker.py

@@ -1,43 +1,10 @@
-import os
-
 import requests
-from dotenv import load_dotenv
-from pathlib import Path
-
-# Load environment variables from .env file
-env_path = Path(__file__).parent / '.env.setup'
-load_dotenv(dotenv_path=env_path)
 
-BASE_URL = "https://tledger-sandbox-69bd94a49289.herokuapp.com/api/v1"
+BASE_URL = "https://api-sandbox.t54.ai/api/v1"
 
-def register_user(email: str, password: str, full_name: str) -> dict:
-    url = f"{BASE_URL}/users/signup"
-    payload = {
-        "email": email,
-        "password": password,
-        "full_name": full_name
-    }
-    response = requests.post(url, json=payload)
-    response.raise_for_status()
-    return response.json()
-
-def login_user(username: str, password: str) -> str:
-    url = f"{BASE_URL}/login/access-token"
-    payload = {
-        "username": username,
-        "password": password
-    }
-    headers = {
-        "Content-Type": "application/x-www-form-urlencoded"
-    }
-    response = requests.post(url, data=payload, headers=headers)
-    response.raise_for_status()
-    return response.json()["access_token"]
-
-def create_project(user_profile_id, network, description, name, daily_limit) -> dict:
+def create_project(network, description, name, daily_limit) -> dict:
     url = f"{BASE_URL}/projects"
     payload = {
-        "user_profile_id": user_profile_id,
         "network": network,
         "description": description,
         "name": name,
@@ -65,43 +32,24 @@ def create_agent_profile(token, project_id, name, description) -> dict:
     response.raise_for_status()
     return response.json()
 
-def generate_api_key(token, resource_id, created_by) -> dict:
+def generate_api_key(resource_id, created_by) -> dict:
     url = f"{BASE_URL}/api_key/generate-api-key"
     payload = {
         "scopes": ["payments:read", "balance:read", "payments:write", "agent:account:read", "agent:profile:create"],
         "resource_id": resource_id,
         "created_by": created_by
     }
-    headers = {
-        "Authorization": f"Bearer {token}"
-    }
-    response = requests.post(url, json=payload, headers=headers)
+    response = requests.post(url, json=payload)
     response.raise_for_status()
     return response.json()
 
-# Example usage
-email = os.environ.get("EMAIL")
-password = os.environ.get("PASSWORD")
-full_name = os.environ.get("FULL_NAME")
-
-try:
-    # Register user
-    user = register_user(email, password, full_name)
-except requests.HTTPError as e:
-    print(f"User already exists: {e}")
-    user = {
-        "id": "user_id"
-    }
-
-# Login user and get JWT token
-token = login_user(email, password)
 
 # Create project
-project = create_project(user["id"], "solana", "Solana Launch Pad", "Twitter Project", 100)
+project = create_project( "solana", "Solana Launch Pad", "Twitter Project", 100)
 project_id = project["id"]
 
 # Generate API key for agent
-api_key_project = generate_api_key(token, project_id, full_name)
+api_key_project = generate_api_key(project_id, "[email protected]")
 
 # Create agent profile
 agent_profile_sender = create_agent_profile(api_key_project, project_id, "Sending Agent", "Sending agent")
@@ -112,10 +60,10 @@ def generate_api_key(token, resource_id, created_by) -> dict:
 agent_id_receiver = agent_profile_receiver["id"]
 
 # Generate API key for agent
-api_key_sender = generate_api_key(token, agent_id_sender, full_name)
+api_key_sender = generate_api_key(agent_id_sender, "[email protected]")
 
 # Generate API key for agent
-api_key_receiver = generate_api_key(token, agent_id_receiver, full_name)
+api_key_receiver = generate_api_key(agent_id_receiver, "[email protected]")
 
 print("Setup complete")
 print(f"Project ID: {project_id}")

plugins/tLedger/setup.py

@@ -74,13 +74,6 @@ def __init__(
                     )
                 ],
             ),
-            "get_payments": Function(
-                fn_name="get_payments",
-                fn_description="Get payments",
-                hint="This function is used to get all the payments",
-                executable=self.get_payments,
-                args=[],
-            ),
 
             "get_agent_profile_details": Function(
                 fn_name="get_agent_profile_details",
@@ -91,6 +84,10 @@ def __init__(
             ),
         }
 
+    def get_tools(self) -> list[Function]:
+        # returns the available functions
+        return list(self.functions.values())
+
     def create_payment(self, request_id: str, receiving_agent_id: str, payment_amount: float, settlement_network: str, currency: str, conversation_id: str, **kwargs) -> \
     tuple[FunctionResultStatus, str, dict[str, Any]] | tuple[FunctionResultStatus, str, dict[Any, Any]]:
         """Generate image based on prompt.
@@ -136,10 +133,10 @@ def create_payment(self, request_id: str, receiving_agent_id: str, payment_amoun
                 },
             )
         except Exception as e:
-            print(f"An error occurred while generating image: {str(e)}")
+            print(f"An error occurred while creating a payment: {str(e)}")
             return (
                 FunctionResultStatus.FAILED,
-                f"An error occurred while while generating image: {str(e)}",
+                f"An error occurred while creating a payment: {str(e)}",
                 {
                 },
             )
@@ -187,48 +184,6 @@ def get_payment_by_id(self, payment_id: str, **kwargs) -> \
                 },
             )
 
-    def get_payments(self, **kwargs) -> \
-            tuple[FunctionResultStatus, str, dict[str, Any]] | tuple[FunctionResultStatus, str, dict[Any, Any]]:
-        """Generate image based on prompt.
-
-        Returns:
-            str URL of image (need to save since temporal)
-        """
-        # API endpoint for image generation
-        # Prepare headers for the request
-        headers = {
-            "X-API-Key": self.api_key,
-            "X-API-Secret": self.api_secret,
-            "Content-Type": "application/json",
-        }
-
-        get_url = self.api_url + "payments"
-
-        try:
-            # Make the API request
-            response = requests.get(get_url, headers=headers)
-            response.raise_for_status()
-
-            # Extract the image URL from the response
-            response_data = response.json()
-
-            return (
-                FunctionResultStatus.DONE,
-                f"The get payments response is: {response_data}",
-                {
-                    "response": response_data,
-                },
-            )
-        except Exception as e:
-            print(f"An error occurred while getting payments: {str(e)}")
-            return (
-                FunctionResultStatus.FAILED,
-                f"An error occurred while getting payments: {str(e)}",
-                {
-                    "get_url": get_url,
-                },
-            )
-
     def get_agent_profile_details(self, **kwargs) -> tuple[FunctionResultStatus, str, dict[str, str]] | tuple[
         FunctionResultStatus, str, dict[str, Any]]:
         """ Get agent profile details for a given agent