phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Added tests for persistence (#274)

Browse source
This commit is contained in:
Sarthak Deshpande 2024-10-23 08:11:46 +05:30 • committed by GitHub
parent 821810657f
commit 2e5e46d896
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 152 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/tests/inference/test_prompt_adapter.py
View file

@ -7,8 +7,10 @@
import unittest
from llama_models.llama3.api import * # noqa: F403
from llama_stack.inference.api import * # noqa: F403
from llama_stack.inference.prompt_adapter import chat_completion_request_to_messages
from llama_stack.apis.inference.inference import * # noqa: F403
from llama_stack.providers.utils.inference.prompt_adapter import (
chat_completion_request_to_messages,
)
MODEL = "Llama3.1-8B-Instruct"