fix(api): constrain shopping suggestion enums
This commit is contained in:
parent
cebea2ac86
commit
5d9d18bd05
2 changed files with 17 additions and 4 deletions
|
|
@ -121,6 +121,14 @@ def test_suggest_shopping(client, session):
|
|||
assert data["reasoning"] == "Test shopping"
|
||||
kwargs = mock_gemini.call_args.kwargs
|
||||
assert "USER PROFILE:" in kwargs["contents"]
|
||||
assert (
|
||||
'category: "cleanser" | "toner" | "essence"'
|
||||
in kwargs["config"].system_instruction
|
||||
)
|
||||
assert (
|
||||
'recommended_time: "am" | "pm" | "both"'
|
||||
in kwargs["config"].system_instruction
|
||||
)
|
||||
assert "function_handlers" in kwargs
|
||||
assert "get_product_details" in kwargs["function_handlers"]
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue