Skip to content

Commit 9707d12

Browse files
committed
Fix test
1 parent ebeeead commit 9707d12

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

tests/e2e/data/config_files/simple_kg_pipeline_config.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"extras": {
55
"openai_api_key": {
66
"resolver_": "ENV",
7-
"key_": "MY_OPENAI_KEY"
7+
"var_": "MY_OPENAI_KEY"
88
}
99
},
1010
"neo4j_config": {

tests/e2e/experimental/pipeline/config/test_pipeline_runner_e2e.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -140,12 +140,12 @@ async def test_simple_kg_pipeline_from_json_config(
140140
assert config is not None
141141
# extras must be resolved:
142142
assert config._global_data["extras"] == {"openai_api_key": "my-openai-key"}
143-
# API key for LLM is read from "extras" (see config file)
143+
# API key for LLM is read from env vars (see config file)
144144
default_llm = config._global_data["llm_config"]["default"]
145-
assert default_llm.client.api_key == "my-openai-key"
146-
# API key for embedder is read from env vars (see config file)
145+
assert default_llm.client.api_key == "sk-my-secret-key"
146+
# API key for embedder is read from extras (see config file)
147147
default_embedder = config._global_data["embedder_config"]["default"]
148-
assert default_embedder.client.api_key == "sk-my-secret-key" # read from env vaf
148+
assert default_embedder.client.api_key == "my-openai-key"
149149

150150
# then run pipeline and check results
151151
res = await runner.run({"file_path": "tests/e2e/data/documents/harry_potter.pdf"})

0 commit comments

Comments
 (0)