From ae854e6d4ec8de23f5a67b1ecaae7b36fdfbc840 Mon Sep 17 00:00:00 2001 From: Cole McIntosh Date: Tue, 8 Jul 2025 20:44:22 -0600 Subject: [PATCH] Remove temporary test files - Remove test_script.py (mistral model cost map updater) - Remove test_url_encoding.py (URL encoding test) --- test_script.py | 25 ------------------------- test_url_encoding.py | 31 ------------------------------- 2 files changed, 56 deletions(-) delete mode 100644 test_script.py delete mode 100644 test_url_encoding.py diff --git a/test_script.py b/test_script.py deleted file mode 100644 index be44d11801..0000000000 --- a/test_script.py +++ /dev/null @@ -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 -) diff --git a/test_url_encoding.py b/test_url_encoding.py deleted file mode 100644 index c3b1a6e205..0000000000 --- a/test_url_encoding.py +++ /dev/null @@ -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() \ No newline at end of file