phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-10 04:08:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix prompt adapter, report

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-13 22:12:30 -08:00
parent 4a7273581c
commit 2efbe9f73b
2 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/providers/utils/inference/prompt_adapter.py
View file

@ -14,13 +14,6 @@ from typing import List, Optional, Tuple, Union
import httpx import httpx
from llama_models.llama3.api.chat_format import ChatFormat from llama_models.llama3.api.chat_format import ChatFormat
from llama_models.llama3.prompt_templates import (
BuiltinToolGenerator,
FunctionTagCustomToolGenerator,
JsonCustomToolGenerator,
PythonListCustomToolGenerator,
SystemDefaultGenerator,
)
from PIL import Image as PIL_Image from PIL import Image as PIL_Image
from llama_stack.apis.common.content_types import ( from llama_stack.apis.common.content_types import (
@ -51,6 +44,13 @@ from llama_stack.models.llama.datatypes import (
ToolPromptFormat, ToolPromptFormat,
is_multimodal, is_multimodal,
) )
from llama_stack.models.llama.llama3.prompt_templates import (
BuiltinToolGenerator,
FunctionTagCustomToolGenerator,
JsonCustomToolGenerator,
PythonListCustomToolGenerator,
SystemDefaultGenerator,
)
from llama_stack.models.llama.sku_list import resolve_model from llama_stack.models.llama.sku_list import resolve_model
from llama_stack.providers.utils.inference import supported_inference_models from llama_stack.providers.utils.inference import supported_inference_models

12
tests/client-sdk/report.py
View file

@ -13,8 +13,12 @@ from typing import Optional
from urllib.parse import urlparse from urllib.parse import urlparse
import pytest import pytest
from llama_models.datatypes import CoreModelId from metadata import API_MAPS
from llama_models.sku_list import ( from pytest import CollectReport
from termcolor import cprint
from llama_stack.models.llama.datatypes import CoreModelId
from llama_stack.models.llama.sku_list import (
all_registered_models, all_registered_models,
llama3_1_instruct_models, llama3_1_instruct_models,
llama3_2_instruct_models, llama3_2_instruct_models,
@ -22,10 +26,6 @@ from llama_models.sku_list import (
llama3_instruct_models, llama3_instruct_models,
safety_models, safety_models,
) )
from metadata import API_MAPS
from pytest import CollectReport
from termcolor import cprint
from llama_stack.providers.datatypes import Api from llama_stack.providers.datatypes import Api
from llama_stack.providers.tests.env import get_env_or_fail from llama_stack.providers.tests.env import get_env_or_fail