apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

disable test

Browse source
This commit is contained in:
co tran 2025-03-28 16:45:40 +00:00
parent 1f9e147999
commit 40f0c2bbed
1 changed files with 28 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

56
model_server/tests/core/test_function_calling.py
View file

@ -123,35 +123,35 @@ def get_greeting_data():
return req, False, False, False
@pytest.mark.asyncio
@pytest.mark.parametrize(
"get_data_func",
[
get_hallucination_data_complex,
get_complete_data,
get_irrelevant_data,
get_complete_data_2,
],
)
async def test_function_calling(get_data_func):
req, intent, hallucination, parameter_gathering = get_data_func()
# @pytest.mark.asyncio
# @pytest.mark.parametrize(
# "get_data_func",
# [
# get_hallucination_data_complex,
# get_complete_data,
# get_irrelevant_data,
# get_complete_data_2,
# ],
# )
# async def test_function_calling(get_data_func):
# req, intent, hallucination, parameter_gathering = get_data_func()
intent_response = await handler_map["Arch-Intent"].chat_completion(req)
# intent_response = await handler_map["Arch-Intent"].chat_completion(req)
assert handler_map["Arch-Intent"].detect_intent(intent_response) == intent
# assert handler_map["Arch-Intent"].detect_intent(intent_response) == intent
if intent:
function_calling_response = await handler_map["Arch-Function"].chat_completion(
req
)
assert (
handler_map["Arch-Function"].hallucination_state.hallucination
== hallucination
)
response_txt = function_calling_response.choices[0].message.content
# if intent:
# function_calling_response = await handler_map["Arch-Function"].chat_completion(
# req
# )
# assert (
# handler_map["Arch-Function"].hallucination_state.hallucination
# == hallucination
# )
# response_txt = function_calling_response.choices[0].message.content
if parameter_gathering:
prefill_prefix = handler_map["Arch-Function"].prefill_prefix
assert any(
response_txt.startswith(prefix) for prefix in prefill_prefix
), f"Response '{response_txt}' does not start with any of the prefixes: {prefill_prefix}"
# if parameter_gathering:
# prefill_prefix = handler_map["Arch-Function"].prefill_prefix
# assert any(
# response_txt.startswith(prefix) for prefix in prefill_prefix
# ), f"Response '{response_txt}' does not start with any of the prefixes: {prefill_prefix}"