From d2497d54e8e03b31d2fdfa3887ee515ec35b0940 Mon Sep 17 00:00:00 2001 From: Vladimir Blagojevic Date: Fri, 9 Feb 2024 14:15:45 +0100 Subject: [PATCH] Update to use the default Secret.from_env_var(OPENAI_API_KEY) approach (#6941) --- e2e/pipelines/test_rag_pipelines_e2e.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/e2e/pipelines/test_rag_pipelines_e2e.py b/e2e/pipelines/test_rag_pipelines_e2e.py index fa3aeb8c0..d38053c96 100644 --- a/e2e/pipelines/test_rag_pipelines_e2e.py +++ b/e2e/pipelines/test_rag_pipelines_e2e.py @@ -30,7 +30,7 @@ def test_bm25_rag_pipeline(tmp_path): rag_pipeline = Pipeline() rag_pipeline.add_component(instance=InMemoryBM25Retriever(document_store=InMemoryDocumentStore()), name="retriever") rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name="prompt_builder") - rag_pipeline.add_component(instance=OpenAIGenerator(api_key=os.environ.get("OPENAI_API_KEY")), name="llm") + rag_pipeline.add_component(instance=OpenAIGenerator(), name="llm") rag_pipeline.add_component(instance=AnswerBuilder(), name="answer_builder") rag_pipeline.connect("retriever", "prompt_builder.documents") rag_pipeline.connect("prompt_builder", "llm") @@ -101,7 +101,7 @@ def test_embedding_retrieval_rag_pipeline(tmp_path): instance=InMemoryEmbeddingRetriever(document_store=InMemoryDocumentStore()), name="retriever" ) rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name="prompt_builder") - rag_pipeline.add_component(instance=OpenAIGenerator(api_key=os.environ.get("OPENAI_API_KEY")), name="llm") + rag_pipeline.add_component(instance=OpenAIGenerator(), name="llm") rag_pipeline.add_component(instance=AnswerBuilder(), name="answer_builder") rag_pipeline.connect("text_embedder", "retriever") rag_pipeline.connect("retriever", "prompt_builder.documents")