refactor: Read token values using Secret util

prosto · Feb 8, 2024 · f75fa32 · f75fa32
1 parent e4ea636
commit f75fa32
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/examples/rag_pipeline.py b/examples/rag_pipeline.py
@@ -1,15 +1,14 @@
-import os
-
 from haystack import GeneratedAnswer, Pipeline
 from haystack.components.builders.answer_builder import AnswerBuilder
 from haystack.components.builders.prompt_builder import PromptBuilder
 from haystack.components.embedders import SentenceTransformersTextEmbedder
 from haystack.components.generators import HuggingFaceTGIGenerator
+from haystack.utils import Secret
 
 from neo4j_haystack import Neo4jDocumentStore, Neo4jEmbeddingRetriever
 
 # Load HF Token from environment variables.
-HF_TOKEN = os.environ.get("HF_TOKEN")
+HF_TOKEN = Secret.from_env_var("HF_API_TOKEN")
 
 # Make sure you have a running Neo4j database with indexed documents available (see `indexing_pipeline.py`),
 # e.g. with Docker:

diff --git a/examples/rag_pipeline_cypher.py b/examples/rag_pipeline_cypher.py
@@ -1,15 +1,14 @@
-import os
-
 from haystack import GeneratedAnswer, Pipeline
 from haystack.components.builders.answer_builder import AnswerBuilder
 from haystack.components.builders.prompt_builder import PromptBuilder
 from haystack.components.embedders import SentenceTransformersTextEmbedder
 from haystack.components.generators import HuggingFaceTGIGenerator
+from haystack.utils import Secret
 
 from neo4j_haystack import Neo4jClientConfig, Neo4jDynamicDocumentRetriever
 
 # Load HF Token from environment variables.
-HF_TOKEN = os.environ.get("HF_TOKEN")
+HF_TOKEN = Secret.from_env_var("HF_API_TOKEN")
 
 # Make sure you have a running Neo4j database with indexed documents available (see `indexing_pipeline.py`),
 # e.g. with Docker: