Merge branch 'meta-llama:main' into main

This commit is contained in:
Chacksu 2024-11-21 18:07:53 -05:00 committed by GitHub
commit 94bfd9a1d1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 75 additions and 128 deletions

View file

@ -524,10 +524,10 @@ class ChatAgent(ShieldRunnerMixin):
message.content = [message.content] + attachments
yield message
else:
log.info(f"Partial message: {str(message)}", color="green")
log.info(f"Partial message: {str(message)}")
input_messages = input_messages + [message]
else:
log.info(f"{str(message)}", color="green")
log.info(f"{str(message)}")
try:
tool_call = message.tool_calls[0]

View file

@ -89,7 +89,7 @@ class TestInference:
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"meta-reference",
"inline::meta-reference",
"remote::ollama",
"remote::tgi",
"remote::together",
@ -135,7 +135,7 @@ class TestInference:
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"meta-reference",
"inline::meta-reference",
"remote::tgi",
"remote::together",
"remote::fireworks",
@ -194,7 +194,7 @@ class TestInference:
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"meta-reference",
"inline::meta-reference",
"remote::fireworks",
"remote::tgi",
"remote::together",

View file

@ -44,7 +44,7 @@ class TestVisionModelInference:
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"meta-reference",
"inline::meta-reference",
"remote::together",
"remote::fireworks",
"remote::ollama",
@ -78,7 +78,7 @@ class TestVisionModelInference:
provider = inference_impl.routing_table.get_provider_impl(inference_model)
if provider.__provider_spec__.provider_type not in (
"meta-reference",
"inline::meta-reference",
"remote::together",
"remote::fireworks",
"remote::ollama",