Add support for loading api_key from config or env variable (#1421)

2024-06-13 23:49:54 +05:30
parent 08b67b4a78
commit 2855f1635b
10 changed files with 37 additions and 32 deletions
--- a/tests/llm/test_jina.py
+++ b/tests/llm/test_jina.py
@@ -74,5 +74,6 @@ def test_get_llm_model_answer_without_system_prompt(config, mocker):
    mocked_jinachat.assert_called_once_with(
        temperature=config.temperature,
        max_tokens=config.max_tokens,
+        jinachat_api_key=os.environ["JINACHAT_API_KEY"],
        model_kwargs={"top_p": config.top_p},
    )