Cleanup

test_mapping.py

@@ -1,41 +0,0 @@
-#!/usr/bin/env python3
-"""
-Test OpenRouter model mapping
-"""
-
-import sys
-
-sys.path.append("/Users/fahad/Developer/gemini-mcp-server")
-
-from simulator_tests.base_test import BaseSimulatorTest
-
-
-class MappingTest(BaseSimulatorTest):
-    def test_mapping(self):
-        """Test model alias mapping"""
-
-        # Test with 'flash' alias - should map to google/gemini-2.5-flash-preview-05-20
-        print("\nTesting 'flash' alias mapping...")
-
-        response, continuation_id = self.call_mcp_tool(
-            "chat",
-            {
-                "prompt": "Say 'Hello from Flash model!'",
-                "model": "flash",  # Should be mapped to google/gemini-2.5-flash-preview-05-20
-                "temperature": 0.1,
-            },
-        )
-
-        if response:
-            print("✅ Flash alias worked!")
-            print(f"Response: {response[:200]}...")
-            return True
-        else:
-            print("❌ Flash alias failed")
-            return False
-
-
-if __name__ == "__main__":
-    test = MappingTest(verbose=False)
-    success = test.test_mapping()
-    print(f"\nTest result: {'Success' if success else 'Failed'}")

test_model_mapping.py

@@ -1,113 +0,0 @@
-#!/usr/bin/env python3
-"""
-Simple test script to demonstrate model mapping through the MCP server.
-Tests how model aliases (flash, pro, o3) are mapped to OpenRouter models.
-"""
-
-import json
-import subprocess
-import sys
-from typing import Any
-
-
-def call_mcp_server(model: str, message: str = "Hello, which model are you?") -> dict[str, Any]:
-    """Call the MCP server with a specific model and return the response."""
-
-    # Prepare the request
-    request = {
-        "jsonrpc": "2.0",
-        "method": "completion",
-        "params": {"model": model, "messages": [{"role": "user", "content": message}], "max_tokens": 100},
-        "id": 1,
-    }
-
-    # Call the server
-    cmd = [sys.executable, "server.py"]
-
-    try:
-        # Send request to stdin and capture output
-        process = subprocess.Popen(
-            cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True
-        )
-
-        stdout, stderr = process.communicate(input=json.dumps(request))
-
-        if process.returncode != 0:
-            return {"error": f"Server returned non-zero exit code: {process.returncode}", "stderr": stderr}
-
-        # Parse the response
-        try:
-            response = json.loads(stdout)
-            return response
-        except json.JSONDecodeError:
-            return {"error": "Failed to parse JSON response", "stdout": stdout, "stderr": stderr}
-
-    except Exception as e:
-        return {"error": f"Failed to call server: {str(e)}"}
-
-
-def extract_model_info(response: dict[str, Any]) -> dict[str, str]:
-    """Extract model information from the response."""
-
-    if "error" in response:
-        return {"status": "error", "message": response.get("error", "Unknown error")}
-
-    # Look for result in the response
-    result = response.get("result", {})
-
-    # Extract relevant information
-    info = {"status": "success", "provider": "unknown", "model": "unknown"}
-
-    # Try to find provider and model info in the response
-    # This might be in metadata or debug info depending on server implementation
-    if "metadata" in result:
-        metadata = result["metadata"]
-        info["provider"] = metadata.get("provider", "unknown")
-        info["model"] = metadata.get("model", "unknown")
-
-    # Also check if the model info is in the response content itself
-    if "content" in result:
-        content = result["content"]
-        # Simple heuristic to detect OpenRouter models
-        if "openrouter" in content.lower() or any(x in content.lower() for x in ["claude", "gpt", "gemini"]):
-            info["provider"] = "openrouter"
-
-    return info
-
-
-def main():
-    """Test model mapping for different aliases."""
-
-    print("Model Mapping Test for MCP Server")
-    print("=" * 50)
-    print()
-
-    # Test models
-    test_models = ["flash", "pro", "o3"]
-
-    for model in test_models:
-        print(f"Testing model: '{model}'")
-        print("-" * 30)
-
-        response = call_mcp_server(model)
-        model_info = extract_model_info(response)
-
-        if model_info["status"] == "error":
-            print(f"  ❌ Error: {model_info['message']}")
-        else:
-            print(f"  ✓ Provider: {model_info['provider']}")
-            print(f"  ✓ Model: {model_info['model']}")
-
-        # Print raw response for debugging
-        if "--debug" in sys.argv:
-            print("\nDebug - Raw Response:")
-            print(json.dumps(response, indent=2))
-
-        print()
-
-    print("\nNote: This test assumes the MCP server is configured with OpenRouter.")
-    print("The actual model mappings depend on the server configuration.")
-
-
-if __name__ == "__main__":
-    main()