Spaces:
Running
Running
File size: 2,773 Bytes
2ca01e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
#!/usr/bin/env python3
"""
Test that source: mcp is added to API requests
"""
import os
from utils import A1DAPIClient, prepare_request_data
def test_source_field():
"""Test that source field is added to requests"""
print("🧪 Testing source field addition...")
print("=" * 50)
# Set API key from .env
os.environ['A1D_API_KEY'] = 'test_key_for_demo'
try:
client = A1DAPIClient()
# Prepare test data
test_data = prepare_request_data("remove_bg", image_url="https://example.com/test.jpg")
print(f"📋 Original data: {test_data}")
# The make_request method should add source: "mcp"
# We'll simulate this by checking what would be sent
request_data = {**test_data, "source": "mcp"}
print(f"📤 Request data with source: {request_data}")
# Verify source field is present
if "source" in request_data and request_data["source"] == "mcp":
print("✅ Source field correctly added!")
return True
else:
print("❌ Source field missing or incorrect!")
return False
except Exception as e:
print(f"❌ Error: {e}")
return False
def test_all_tools():
"""Test source field for all tools"""
print("\n🔧 Testing source field for all tools...")
from config import TOOLS_CONFIG
for tool_name, config in TOOLS_CONFIG.items():
print(f"\n📋 Testing {tool_name}...")
# Prepare sample data for each tool
if tool_name == "image_generator":
sample_data = {"prompt": "test prompt"}
elif "video" in tool_name:
sample_data = {"video_url": "https://example.com/test.mp4"}
else:
sample_data = {"image_url": "https://example.com/test.jpg"}
try:
test_data = prepare_request_data(tool_name, **sample_data)
request_data = {**test_data, "source": "mcp"}
if "source" in request_data and request_data["source"] == "mcp":
print(f" ✅ {tool_name}: Source field OK")
else:
print(f" ❌ {tool_name}: Source field missing")
return False
except Exception as e:
print(f" ❌ {tool_name}: Error - {e}")
return False
return True
if __name__ == "__main__":
print("🎯 Testing A1D MCP Server - Source Field")
print("=" * 60)
test1 = test_source_field()
test2 = test_all_tools()
if test1 and test2:
print("\n🎉 All tests passed!")
print("✅ Source field 'mcp' will be added to all API requests")
else:
print("\n❌ Some tests failed!")
|