mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-03 01:03:59 +00:00
Use int for year in test case
This commit is contained in:
parent
00affd1f02
commit
357d7ea9ea
2 changed files with 4 additions and 4 deletions
|
@ -199,8 +199,8 @@ def test_text_completion_structured_output(client_with_models, text_model_id, te
|
||||||
|
|
||||||
class AnswerFormat(BaseModel):
|
class AnswerFormat(BaseModel):
|
||||||
name: str
|
name: str
|
||||||
year_born: str
|
year_born: int
|
||||||
year_retired: str
|
year_retired: int
|
||||||
|
|
||||||
tc = TestCase(test_case)
|
tc = TestCase(test_case)
|
||||||
|
|
||||||
|
|
|
@ -40,8 +40,8 @@
|
||||||
"user_input": "Michael Jordan was born in 1963. He played basketball for the Chicago Bulls. He retired in 2003.",
|
"user_input": "Michael Jordan was born in 1963. He played basketball for the Chicago Bulls. He retired in 2003.",
|
||||||
"expected": {
|
"expected": {
|
||||||
"name": "Michael Jordan",
|
"name": "Michael Jordan",
|
||||||
"year_born": "1963",
|
"year_born": 1963,
|
||||||
"year_retired": "2003"
|
"year_retired": 2003
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue