cd1105b741
- OpenRouter model configuration registry - Model definition file for users to be able to control - Update instructions
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
#!/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-flash-1.5-8b
|
|
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-flash-1.5-8b
|
|
"temperature": 0.1
|
|
}
|
|
)
|
|
|
|
if response:
|
|
print(f"✅ 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'}") |