phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(pattern_match_deployments.py): default to user input if unable to… (#6632)

Browse source 
* fix(pattern_match_deployments.py): default to user input if unable to map based on wildcards

* test: fix test

* test: reset test name

* test: update conftest to reload proxy server module between tests

* ci(config.yml): move langfuse out of local_testing

reduce ci/cd time

* ci(config.yml): cleanup langfuse ci/cd tests

* fix: update test to not use global proxy_server app module

* ci: move caching to a separate test pipeline

speed up ci pipeline

* test: update conftest to check if proxy_server attr exists before reloading

* build(conftest.py): don't block on inability to reload proxy_server

* ci(config.yml): update caching unit test filter to work on 'cache' keyword as well

* fix(encrypt_decrypt_utils.py): use function to get salt key

* test: mark flaky test

* test: handle anthropic overloaded errors

* refactor: create separate ci/cd pipeline for proxy unit tests

make ci/cd faster

* ci(config.yml): add litellm_proxy_unit_testing to build_and_test jobs

* ci(config.yml): generate prisma binaries for proxy unit tests

* test: readd vertex_key.json

* ci(config.yml): remove `-s` from proxy_unit_test cmd

speed up test

* ci: remove any 'debug' logging flag

speed up ci pipeline

* test: fix test

* test(test_braintrust.py): rerun

* test: add delay for braintrust test
This commit is contained in:
Krish Dholakia 2024-11-08 00:55:57 +05:30 • committed by GitHub
parent 44840d615d
commit 27e18358ab
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
77 changed files with 2861 additions and 76 deletions
Download patch file Download diff file Expand all files Collapse all files

45
tests/local_testing/test_proxy_encrypt_decrypt.py
View file

@ -1,45 +0,0 @@
import os
import sys
import pytest
from dotenv import load_dotenv
load_dotenv()
import io
import os
sys.path.insert(
0, os.path.abspath("../..")
) # Adds-the parent directory to the system path
from litellm.proxy import proxy_server
from litellm.proxy.common_utils.encrypt_decrypt_utils import (
decrypt_value_helper,
encrypt_value_helper,
)
def test_encrypt_decrypt_with_master_key():
setattr(proxy_server, "master_key", "sk-1234")
assert decrypt_value_helper(encrypt_value_helper("test")) == "test"
assert decrypt_value_helper(encrypt_value_helper(10)) == 10
assert decrypt_value_helper(encrypt_value_helper(True)) is True
assert decrypt_value_helper(encrypt_value_helper(None)) is None
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10})) == {"rpm": 10}
# encryption should actually occur for strings
assert encrypt_value_helper("test") != "test"
def test_encrypt_decrypt_with_salt_key():
os.environ["LITELLM_SALT_KEY"] = "sk-salt-key2222"
assert decrypt_value_helper(encrypt_value_helper("test")) == "test"
assert decrypt_value_helper(encrypt_value_helper(10)) == 10
assert decrypt_value_helper(encrypt_value_helper(True)) is True
assert decrypt_value_helper(encrypt_value_helper(None)) is None
assert decrypt_value_helper(encrypt_value_helper({"rpm": 10})) == {"rpm": 10}
# encryption should actually occur for strings
assert encrypt_value_helper("test") != "test"
os.environ.pop("LITELLM_SALT_KEY", None)