()
| 64 | |
| 65 | @pytest.fixture |
| 66 | def llm_request(): |
| 67 | return LlmRequest( |
| 68 | model=_TEST_MODEL_NAME, |
| 69 | contents=[ |
| 70 | Content( |
| 71 | role="user", |
| 72 | parts=[ |
| 73 | Part.from_text(text="What is the weather in San Francisco?") |
| 74 | ], |
| 75 | ) |
| 76 | ], |
| 77 | config=types.GenerateContentConfig( |
| 78 | temperature=0.1, |
| 79 | response_modalities=[types.Modality.TEXT], |
| 80 | system_instruction=_SYSTEM_PROMPT, |
| 81 | ), |
| 82 | ) |
| 83 | |
| 84 | |
| 85 | @pytest.mark.asyncio |