diff --git a/litellm/__init__.py b/litellm/__init__.py index d364335aba..ef7bfe91d6 100644 --- a/litellm/__init__.py +++ b/litellm/__init__.py @@ -101,7 +101,8 @@ for key, value in model_cost.items(): elif value.get('litellm_provider') == 'anthropic': anthropic_models.append(key) elif value.get('litellm_provider') == 'openrouter': - openrouter_models.append(key) + split_string = key.split('/', 1) + openrouter_models.append(split_string[1]) elif value.get('litellm_provider') == 'vertex_ai-text-models': vertex_text_models.append(key) elif value.get('litellm_provider') == 'vertex_ai-code-text-models': diff --git a/litellm/__pycache__/__init__.cpython-311.pyc b/litellm/__pycache__/__init__.cpython-311.pyc index f98964ab77..77b9f46da5 100644 Binary files a/litellm/__pycache__/__init__.cpython-311.pyc and b/litellm/__pycache__/__init__.cpython-311.pyc differ diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc index 9f7c5169ee..24ef31d5f0 100644 Binary files a/litellm/__pycache__/main.cpython-311.pyc and b/litellm/__pycache__/main.cpython-311.pyc differ diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc index ae0db3f8d0..dbb232368b 100644 Binary files a/litellm/__pycache__/utils.cpython-311.pyc and b/litellm/__pycache__/utils.cpython-311.pyc differ diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index fe89153724..4d418ccbdf 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -336,6 +336,8 @@ def test_completion_openai_litellm_key(): # except Exception as e: # pytest.fail(f"Error occurred: {e}") +# test_completion_openrouter() + def test_completion_openai_with_more_optional_params(): try: response = completion( @@ -816,7 +818,7 @@ def test_completion_ai21(): except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_ai21() +test_completion_ai21() # test config file with completion # # def test_completion_openai_config(): # try: