phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

formatting

Browse source
This commit is contained in:
Dalton Flanagan 2024-08-14 17:03:43 -04:00
parent 069d877210
commit b311dcd143
22 changed files with 82 additions and 128 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_inference.py
View file

@ -10,20 +10,16 @@ from datetime import datetime
from llama_models.llama3_1.api.datatypes import (
BuiltinTool,
UserMessage,
StopReason,
SystemMessage,
ToolResponseMessage,
UserMessage,
)
from llama_toolchain.inference.api.datatypes import (
ChatCompletionResponseEventType,
)
from llama_toolchain.inference.meta_reference.inference import get_provider_impl
from llama_toolchain.inference.meta_reference.config import (
MetaReferenceImplConfig,
)
from llama_toolchain.inference.api.datatypes import ChatCompletionResponseEventType
from llama_toolchain.inference.api.endpoints import ChatCompletionRequest
from llama_toolchain.inference.meta_reference.config import MetaReferenceImplConfig
from llama_toolchain.inference.meta_reference.inference import get_provider_impl
MODEL = "Meta-Llama3.1-8B-Instruct"

8
tests/test_ollama_inference.py
View file

@ -4,16 +4,14 @@ from datetime import datetime
from llama_models.llama3_1.api.datatypes import (
BuiltinTool,
UserMessage,
StopReason,
SamplingParams,
SamplingStrategy,
StopReason,
SystemMessage,
ToolResponseMessage,
UserMessage,
)
from llama_toolchain.inference.api.datatypes import (
ChatCompletionResponseEventType,
)
from llama_toolchain.inference.api.datatypes import ChatCompletionResponseEventType
from llama_toolchain.inference.api.endpoints import ChatCompletionRequest
from llama_toolchain.inference.ollama.config import OllamaImplConfig
from llama_toolchain.inference.ollama.ollama import get_provider_impl