Remove temporary test files

- Remove test_script.py (mistral model cost map updater)
- Remove test_url_encoding.py (URL encoding test)
This commit is contained in:
Cole McIntosh
2025-07-08 20:44:22 -06:00
parent d720b3d369
commit ae854e6d4e
2 changed files with 0 additions and 56 deletions
-25
View File
@@ -1,25 +0,0 @@
import json
mistral_model_cost_map = json.load(open("model_prices_and_context_window.json"))
for model, model_info in mistral_model_cost_map.items():
if (
"bedrock" in model_info.get("litellm_provider")
and model_info.get("mode") == "chat"
and any(
m in model
for m in [
"mistral.mistral-large-2402-v1:0",
"mistral.mistral-small-2402-v1:0",
]
)
):
"""
Update all mistral models to supports_response_schema
"""
del model_info["supports_tool_choice"]
print(f"Updated {model} to support response schema")
json.dump(
mistral_model_cost_map, open("model_prices_and_context_window.json", "w"), indent=4
)
-31
View File
@@ -1,31 +0,0 @@
# Test URL encoding handling for emails with + characters
import re
from urllib.parse import unquote
def test_user_id_parsing():
# Simulate the raw query string that would come from the URL
# When user calls: http://0.0.0.0:4000/user/info?user_id=machine-user+alp-air-admin-b58-b@tempus.com
# The query string would be: user_id=machine-user+alp-air-admin-b58-b@tempus.com
test_cases = [
"user_id=machine-user+alp-air-admin-b58-b@tempus.com",
"user_id=machine-user%2Balp-air-admin-b58-b@tempus.com", # URL encoded +
"user_id=regular@email.com",
"user_id=test-user@domain.com&other_param=value"
]
for query_string in test_cases:
print(f"\nTesting query string: {query_string}")
if 'user_id=' in query_string:
match = re.search(r'user_id=([^&]*)', query_string)
if match:
raw_user_id = unquote(match.group(1))
print(f"Extracted user_id: {raw_user_id}")
else:
print("No match found")
else:
print("user_id not found in query string")
if __name__ == "__main__":
test_user_id_parsing()