File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
experimental/pipeline/config Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change 44 "extras" : {
55 "openai_api_key" : {
66 "resolver_" : " ENV" ,
7- "key_ " : " MY_OPENAI_KEY"
7+ "var_ " : " MY_OPENAI_KEY"
88 }
99 },
1010 "neo4j_config" : {
Original file line number Diff line number Diff 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" })
You can’t perform that action at this time.
0 commit comments