D-2 수정 작업: OpenAI API 우선 없으면 Ollama 적용

Last3Mon/64/workflow-lab/tests/test_openai_provider.py

@@ -0,0 +1,80 @@
+"""OpenAI provider, .env loading and the Ollama fallback, without network access."""
+
+import types
+
+import pytest
+
+from miniflow import envfile, llm
+
+
+def test_parse_env_lines():
+    values = envfile.parse('# comment\nexport A=1\nB="two words"\nC=3  # note\n\nbroken line\n')
+    assert values == {"A": "1", "B": "two words", "C": "3"}
+
+
+def test_load_keeps_shell_values(tmp_path, monkeypatch):
+    (tmp_path / ".env").write_text("MINIFLOW_TEST_A=from-file\nMINIFLOW_TEST_B=from-file\n", encoding="utf-8")
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setenv("MINIFLOW_DOTENV", "1")
+    monkeypatch.setenv("MINIFLOW_TEST_A", "from-shell")
+    monkeypatch.delenv("MINIFLOW_TEST_B", raising=False)
+    assert envfile.load() == tmp_path / ".env"
+    assert llm.os.environ["MINIFLOW_TEST_A"] == "from-shell"
+    assert llm.os.environ["MINIFLOW_TEST_B"] == "from-file"
+    monkeypatch.delenv("MINIFLOW_TEST_B")
+
+
+def test_openai_without_key(monkeypatch):
+    monkeypatch.setenv("MINIFLOW_LLM", "openai")
+    monkeypatch.delenv("OPENAI_API_KEY", raising=False)
+    monkeypatch.delenv("MINIFLOW_LLM_FALLBACK", raising=False)
+    with pytest.raises(llm.LLMError, match="OPENAI_API_KEY"):
+        llm.complete({}, "hello", "", {})
+
+
+def test_openai_request_shape(monkeypatch):
+    sent = {}
+
+    class FakeResponses:
+        def create(self, **kwargs):
+            sent.update(kwargs)
+            return types.SimpleNamespace(output_text='{"ok": true}', usage=None)
+
+    class FakeClient:
+        def __init__(self, **_):
+            self.responses = FakeResponses()
+
+    import openai
+
+    monkeypatch.setattr(openai, "OpenAI", FakeClient)
+    monkeypatch.setenv("MINIFLOW_LLM", "openai")
+    monkeypatch.setenv("OPENAI_API_KEY", "test-key")
+    monkeypatch.setenv("OPENAI_MODEL", "test-model")
+    schema = {"type": "object", "properties": {"ok": {"type": "boolean"}}}
+    reply, used = llm.complete({"temperature": 0, "json_schema": schema}, "hello", "be brief", {})
+    assert (reply, used) == ('{"ok": true}', "openai")
+    assert sent["model"] == "test-model" and sent["instructions"] == "be brief" and sent["temperature"] == 0
+    assert sent["text"]["format"]["schema"] == schema
+
+
+def test_fallback_to_second_provider(monkeypatch, capsys):
+    monkeypatch.setenv("MINIFLOW_LLM", "openai")
+    monkeypatch.delenv("OPENAI_API_KEY", raising=False)
+    monkeypatch.setenv("MINIFLOW_LLM_FALLBACK", "ollama")
+    monkeypatch.setattr(llm, "call_ollama", lambda *args, **kwargs: "from ollama")
+    reply, used = llm.complete({}, "hello", "", {})
+    assert (reply, used) == ("from ollama", "ollama")
+    assert "openai failed" in capsys.readouterr().err
+
+
+def test_both_providers_fail(monkeypatch):
+    monkeypatch.setenv("MINIFLOW_LLM", "openai")
+    monkeypatch.delenv("OPENAI_API_KEY", raising=False)
+    monkeypatch.setenv("MINIFLOW_LLM_FALLBACK", "ollama")
+
+    def broken(*args, **kwargs):
+        raise llm.LLMError("Cannot reach Ollama")
+
+    monkeypatch.setattr(llm, "call_ollama", broken)
+    with pytest.raises(llm.LLMError, match="OPENAI_API_KEY.*fallback ollama failed"):
+        llm.complete({}, "hello", "", {})