mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-26 02:20:11 +00:00
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:
@@ -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
|
||||
)
|
||||
@@ -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()
|
||||
Reference in New Issue
Block a user