mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
Revert "test_completion_mistral_api_mistral_large_function_call"
This reverts commit ef9177f0a8
.
This commit is contained in:
parent
0642a78abb
commit
b30e05b54f
3 changed files with 1 additions and 13 deletions
|
@ -27,10 +27,6 @@ def _ensure_extra_body_is_safe(extra_body: Optional[Dict]) -> Optional[Dict]:
|
||||||
if _prompt is not None and hasattr(_prompt, "__dict__"):
|
if _prompt is not None and hasattr(_prompt, "__dict__"):
|
||||||
extra_body["metadata"]["prompt"] = _prompt.__dict__
|
extra_body["metadata"]["prompt"] = _prompt.__dict__
|
||||||
|
|
||||||
if len(extra_body) == 0:
|
|
||||||
# don't send empty extra_body, providers like mistral will throw an error
|
|
||||||
return None
|
|
||||||
|
|
||||||
return extra_body
|
return extra_body
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -3470,10 +3470,6 @@ def get_optional_params( # noqa: PLR0915
|
||||||
optional_params["extra_body"] = _ensure_extra_body_is_safe(
|
optional_params["extra_body"] = _ensure_extra_body_is_safe(
|
||||||
extra_body=optional_params["extra_body"]
|
extra_body=optional_params["extra_body"]
|
||||||
)
|
)
|
||||||
|
|
||||||
if optional_params["extra_body"] is None:
|
|
||||||
# don't pass extra_body if it's empty/None
|
|
||||||
optional_params.pop("extra_body", None)
|
|
||||||
else:
|
else:
|
||||||
# if user passed in non-default kwargs for specific providers/models, pass them along
|
# if user passed in non-default kwargs for specific providers/models, pass them along
|
||||||
for k in passed_params.keys():
|
for k in passed_params.keys():
|
||||||
|
|
|
@ -16,7 +16,7 @@ sys.path.insert(
|
||||||
|
|
||||||
import os
|
import os
|
||||||
from unittest.mock import AsyncMock, MagicMock, patch
|
from unittest.mock import AsyncMock, MagicMock, patch
|
||||||
from litellm._logging import verbose_logger
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
import litellm
|
import litellm
|
||||||
|
@ -976,10 +976,6 @@ async def test_completion_codestral_chat_api():
|
||||||
|
|
||||||
def test_completion_mistral_api_mistral_large_function_call():
|
def test_completion_mistral_api_mistral_large_function_call():
|
||||||
litellm.set_verbose = True
|
litellm.set_verbose = True
|
||||||
import logging
|
|
||||||
|
|
||||||
logging.basicConfig(level=logging.DEBUG)
|
|
||||||
verbose_logger.setLevel(logging.DEBUG)
|
|
||||||
tools = [
|
tools = [
|
||||||
{
|
{
|
||||||
"type": "function",
|
"type": "function",
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue