|
1 | | -import os |
2 | 1 | from typing import Any |
3 | 2 | from unittest.mock import Mock, patch |
4 | 3 |
|
@@ -540,32 +539,31 @@ def test_llm_responses_ssl_verify_and_custom_provider(mock_responses): |
540 | 539 | assert kwargs.get("api_base") == "https://corporate-proxy.example.com/api" |
541 | 540 |
|
542 | 541 |
|
543 | | -def test_llm_ssl_verify_env_parsing(): |
544 | | - """Test that ssl_verify is correctly parsed from environment variables.""" |
| 542 | +def test_llm_ssl_verify_env_parsing(monkeypatch): |
| 543 | + """Test that ssl_verify is correctly parsed from environment variables. |
| 544 | +
|
| 545 | + Use pytest's monkeypatch to ensure environment is restored automatically. |
| 546 | + """ |
| 547 | + # Ensure model is set for all subsequent load_from_env calls |
| 548 | + monkeypatch.setenv("LLM_MODEL", "gpt-4") |
| 549 | + |
545 | 550 | # Test various false values |
546 | 551 | for value in ["false", "False", "FALSE", "0", "no", "off"]: |
547 | | - os.environ["LLM_SSL_VERIFY"] = value |
548 | | - os.environ["LLM_MODEL"] = "gpt-4" |
| 552 | + monkeypatch.setenv("LLM_SSL_VERIFY", value) |
549 | 553 | llm = LLM.load_from_env() |
550 | 554 | assert llm.ssl_verify is False, f"Failed for value: {value}" |
551 | 555 |
|
552 | 556 | # Test various true values |
553 | 557 | for value in ["true", "True", "TRUE", "1", "yes", "on"]: |
554 | | - os.environ["LLM_SSL_VERIFY"] = value |
| 558 | + monkeypatch.setenv("LLM_SSL_VERIFY", value) |
555 | 559 | llm = LLM.load_from_env() |
556 | 560 | assert llm.ssl_verify is True, f"Failed for value: {value}" |
557 | 561 |
|
558 | 562 | # Test certificate path (string value) |
559 | | - os.environ["LLM_SSL_VERIFY"] = "/path/to/cert.pem" |
| 563 | + monkeypatch.setenv("LLM_SSL_VERIFY", "/path/to/cert.pem") |
560 | 564 | llm = LLM.load_from_env() |
561 | 565 | assert llm.ssl_verify == "/path/to/cert.pem" |
562 | 566 |
|
563 | | - # Clean up |
564 | | - if "LLM_SSL_VERIFY" in os.environ: |
565 | | - del os.environ["LLM_SSL_VERIFY"] |
566 | | - if "LLM_MODEL" in os.environ: |
567 | | - del os.environ["LLM_MODEL"] |
568 | | - |
569 | 567 |
|
570 | 568 | def test_llm_vision_support(default_llm): |
571 | 569 | """Test LLM vision support detection.""" |
|
0 commit comments