()
| 46 | |
| 47 | @pytest.fixture |
| 48 | def llm_request(): |
| 49 | return LlmRequest( |
| 50 | model=_TEST_MODEL_NAME, |
| 51 | contents=[Content(role="user", parts=[Part.from_text(text="hello")])], |
| 52 | config=types.GenerateContentConfig( |
| 53 | temperature=0.1, |
| 54 | response_modalities=[types.Modality.TEXT], |
| 55 | system_instruction=_SYSTEM_PROMPT, |
| 56 | ), |
| 57 | ) |
| 58 | |
| 59 | |
| 60 | @pytest.fixture |