mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-15 20:13:07 +00:00
feat(tests): make inference_recorder into api_recorder (include tool_invoke)
This commit is contained in:
parent
b96640eca3
commit
9205731cd6
19 changed files with 849 additions and 666 deletions
|
|
@ -39,7 +39,7 @@ class Setup(BaseModel):
|
|||
|
||||
name: str
|
||||
description: str
|
||||
defaults: dict[str, str] = Field(default_factory=dict)
|
||||
defaults: dict[str, str | int] = Field(default_factory=dict)
|
||||
env: dict[str, str] = Field(default_factory=dict)
|
||||
|
||||
|
||||
|
|
@ -88,6 +88,7 @@ SETUP_DEFINITIONS: dict[str, Setup] = {
|
|||
defaults={
|
||||
"text_model": "openai/gpt-4o",
|
||||
"embedding_model": "openai/text-embedding-3-small",
|
||||
"embedding_dimension": 1536,
|
||||
},
|
||||
),
|
||||
"tgi": Setup(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue