simplified examples file

libs/python/agent/agent/proxy/examples.py

@@ -17,44 +17,6 @@ async def test_http_endpoint():
     anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
     assert isinstance(anthropic_api_key, str), "ANTHROPIC_API_KEY environment variable must be set"
 
-    # Example 1: Simple text request
-    simple_request = {
-        "model": "anthropic/claude-3-5-sonnet-20241022",
-        "input": "Tell me a three sentence bedtime story about a unicorn.",
-        "env": {
-            "ANTHROPIC_API_KEY": anthropic_api_key
-        }
-    }
-    
-    # Example 2: Multi-modal request with image
-    multimodal_request = {
-        "model": "anthropic/claude-3-5-sonnet-20241022",
-        "input": [
-            {
-                "role": "user",
-                "content": [
-                    {"type": "input_text", "text": "what is in this image?"},
-                    {
-                        "type": "input_image",
-                        "image_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
-                    }
-                ]
-            }
-        ],
-        "env": {
-            "ANTHROPIC_API_KEY": anthropic_api_key
-        }
-    }
-    
-    # Example 3: Request with custom agent and computer kwargs
-    custom_request = {
-        "model": "anthropic/claude-3-5-sonnet-20241022",
-        "input": "Take a screenshot and tell me what you see",
-        "env": {
-            "ANTHROPIC_API_KEY": anthropic_api_key
-        }
-    }
-    
     # Test requests
     base_url = "https://m-linux-96lcxd2c2k.containers.cloud.trycua.com:8443"
     # base_url = "http://localhost:8000"
@@ -63,9 +25,37 @@ async def test_http_endpoint():
     
     async with aiohttp.ClientSession() as session:
         for i, request_data in enumerate([
-            simple_request,
-            # multimodal_request,
-            custom_request
+            # ==== Request Body Examples ====
+
+            # Simple text request
+            {
+                "model": "anthropic/claude-3-5-sonnet-20241022",
+                "input": "Tell me a three sentence bedtime story about a unicorn.",
+                "env": {
+                    "ANTHROPIC_API_KEY": anthropic_api_key
+                }
+            },
+
+            # Multimodal request with image
+            # {
+            #     "model": "anthropic/claude-3-5-sonnet-20241022",
+            #     "input": [
+            #         {
+            #             "role": "user",
+            #             "content": [
+            #                 {"type": "input_text", "text": "what is in this image?"},
+            #                 {
+            #                     "type": "input_image",
+            #                     "image_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
+            #                 }
+            #             ]
+            #         }
+            #     ],
+            #     "env": {
+            #         "ANTHROPIC_API_KEY": anthropic_api_key
+            #     }
+            # }
+
         ], 1):
             print(f"\n--- Test {i} ---")
             print(f"Request: {json.dumps(request_data, indent=2)}")
@@ -85,55 +75,6 @@ async def test_http_endpoint():
                 print(f"Error: {e}")
 
 
-def curl_examples():
-    """Print curl command examples."""
-    
-    print("=== CURL Examples ===\n")
-    
-    print("1. Simple text request:")
-    print("""curl http://localhost:8000/responses \\
-  -H "Content-Type: application/json" \\
-  -d '{
-    "model": "anthropic/claude-3-5-sonnet-20241022",
-    "input": "Tell me a three sentence bedtime story about a unicorn."
-  }'""")
-    
-    print("\n2. Multi-modal request with image:")
-    print("""curl http://localhost:8000/responses \\
-  -H "Content-Type: application/json" \\
-  -d '{
-    "model": "anthropic/claude-3-5-sonnet-20241022",
-    "input": [
-      {
-        "role": "user",
-        "content": [
-          {"type": "input_text", "text": "what is in this image?"},
-          {
-            "type": "input_image",
-            "image_url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg"
-          }
-        ]
-      }
-    ]
-  }'""")
-    
-    print("\n3. Request with custom configuration:")
-    print("""curl http://localhost:8000/responses \\
-  -H "Content-Type: application/json" \\
-  -d '{
-    "model": "anthropic/claude-3-5-sonnet-20241022",
-    "input": "Take a screenshot and tell me what you see",
-    "agent_kwargs": {
-      "save_trajectory": true,
-      "verbosity": 20
-    },
-    "computer_kwargs": {
-      "os_type": "linux",
-      "provider_type": "cloud"
-    }
-  }'""")
-
-
 async def test_p2p_client():
     """Example P2P client using peerjs-python."""
     try:
@@ -184,9 +125,7 @@ async def test_p2p_client():
 if __name__ == "__main__":
     import sys
     
-    if len(sys.argv) > 1 and sys.argv[1] == "curl":
-        curl_examples()
-    elif len(sys.argv) > 1 and sys.argv[1] == "p2p":
+    if len(sys.argv) > 1 and sys.argv[1] == "p2p":
         asyncio.run(test_p2p_client())
     else:
         asyncio.run(test_http_endpoint())