mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 19:24:27 +00:00
Litellm dev 12 12 2024 (#7203)
* fix(azure/): support passing headers to azure openai endpoints
Fixes https://github.com/BerriAI/litellm/issues/6217
* fix(utils.py): move default tokenizer to just openai
hf tokenizer makes network calls when trying to get the tokenizer - this slows down execution time calls
* fix(router.py): fix pattern matching router - add generic "*" to it as well
Fixes issue where generic "*" model access group wouldn't show up
* fix(pattern_match_deployments.py): match to more specific pattern
match to more specific pattern
allows setting generic wildcard model access group and excluding specific models more easily
* fix(proxy_server.py): fix _delete_deployment to handle base case where db_model list is empty
don't delete all router models b/c of empty list
Fixes https://github.com/BerriAI/litellm/issues/7196
* fix(anthropic/): fix handling response_format for anthropic messages with anthropic api
* fix(fireworks_ai/): support passing response_format + tool call in same message
Addresses https://github.com/BerriAI/litellm/issues/7135
* Revert "fix(fireworks_ai/): support passing response_format + tool call in same message"
This reverts commit 6a30dc6929
.
* test: fix test
* fix(replicate/): fix replicate default retry/polling logic
* test: add unit testing for router pattern matching
* test: update test to use default oai tokenizer
* test: mark flaky test
* test: skip flaky test
This commit is contained in:
parent
e65f990319
commit
a42f008cd0
19 changed files with 496 additions and 103 deletions
|
@ -113,7 +113,14 @@ import os
|
|||
({"prompt": "Hello world"}, "image_generation"),
|
||||
],
|
||||
)
|
||||
def test_azure_extra_headers(input, call_type):
|
||||
@pytest.mark.parametrize(
|
||||
"header_value",
|
||||
[
|
||||
"headers",
|
||||
"extra_headers",
|
||||
],
|
||||
)
|
||||
def test_azure_extra_headers(input, call_type, header_value):
|
||||
from litellm import embedding, image_generation
|
||||
|
||||
http_client = Client()
|
||||
|
@ -128,18 +135,21 @@ def test_azure_extra_headers(input, call_type):
|
|||
func = embedding
|
||||
elif call_type == "image_generation":
|
||||
func = image_generation
|
||||
response = func(
|
||||
model="azure/chatgpt-v-2",
|
||||
api_base="https://openai-gpt-4-test-v-1.openai.azure.com",
|
||||
api_version="2023-07-01-preview",
|
||||
api_key="my-azure-api-key",
|
||||
extra_headers={
|
||||
|
||||
data = {
|
||||
"model": "azure/chatgpt-v-2",
|
||||
"api_base": "https://openai-gpt-4-test-v-1.openai.azure.com",
|
||||
"api_version": "2023-07-01-preview",
|
||||
"api_key": "my-azure-api-key",
|
||||
header_value: {
|
||||
"Authorization": "my-bad-key",
|
||||
"Ocp-Apim-Subscription-Key": "hello-world-testing",
|
||||
},
|
||||
**input,
|
||||
)
|
||||
}
|
||||
response = func(**data)
|
||||
print(response)
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue