scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test(test_add_function_to_prompt.py): skip as anthropic now supports tool calling

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-04 18:31:18 -08:00
parent 71c98d3200
commit 18947f8c85
2 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/anthropic.py
View file

@ -227,7 +227,7 @@ def completion(
else:
text_content = completion_response["content"][0].get("text", None)
## TOOL CALLING - OUTPUT PARSE
if _is_function_call == True:
if text_content is not None and "invoke" in text_content:
function_name = extract_between_tags("tool_name", text_content)[0]
function_arguments_str = extract_between_tags("invoke", text_content)[
0

3
litellm/tests/test_add_function_to_prompt.py
View file

@ -41,10 +41,11 @@ def test_function_call_non_openai_model():
pass
test_function_call_non_openai_model()
# test_function_call_non_openai_model()
## case 2: add_function_to_prompt set
@pytest.mark.skip(reason="Anthropic now supports tool calling")
def test_function_call_non_openai_model_litellm_mod_set():
litellm.add_function_to_prompt = True
litellm.set_verbose = True