WIP
- OpenRouter model configuration registry - Model definition file for users to be able to control - Update instructions
This commit is contained in:
38
test_mapping.py
Normal file
38
test_mapping.py
Normal file
@@ -0,0 +1,38 @@
|
||||
#!/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'}")
|
||||
Reference in New Issue
Block a user