()
| 30 | |
| 31 | @pytest.fixture |
| 32 | def gemma_request(): |
| 33 | return LlmRequest( |
| 34 | model=DEFAULT_GEMMA_MODEL, |
| 35 | contents=[ |
| 36 | Content( |
| 37 | role="user", |
| 38 | parts=[ |
| 39 | Part.from_text(text="You are a helpful assistant."), |
| 40 | Part.from_text(text="Hello!"), |
| 41 | ], |
| 42 | ) |
| 43 | ], |
| 44 | config=types.GenerateContentConfig( |
| 45 | temperature=0.1, |
| 46 | response_modalities=[types.Modality.TEXT], |
| 47 | system_instruction="Talk like a pirate.", |
| 48 | ), |
| 49 | ) |
| 50 | |
| 51 | |
| 52 | @pytest.mark.asyncio |