| 74 | |
| 75 | @fixture(autouse=True) |
| 76 | def llm_backend(request: FixtureRequest): |
| 77 | # Set backend environment value. |
| 78 | original_val = os.environ.get('GOOGLE_GENAI_USE_ENTERPRISE') |
| 79 | backend_type = request.param |
| 80 | if backend_type == 'GOOGLE_AI': |
| 81 | os.environ['GOOGLE_GENAI_USE_ENTERPRISE'] = '0' |
| 82 | else: |
| 83 | os.environ['GOOGLE_GENAI_USE_ENTERPRISE'] = '1' |
| 84 | |
| 85 | yield # Run the test |
| 86 | |
| 87 | # Restore the environment |
| 88 | if original_val is None: |
| 89 | os.environ.pop('GOOGLE_GENAI_USE_ENTERPRISE', None) |
| 90 | else: |
| 91 | os.environ['GOOGLE_GENAI_USE_ENTERPRISE'] = original_val |
| 92 | |
| 93 | |
| 94 | @hookimpl(tryfirst=True) |