TeresaK commited on
Commit
a5718ea
·
verified ·
1 Parent(s): a01cae8

Update src/rag/pipeline.py

Browse files
Files changed (1) hide show
  1. src/rag/pipeline.py +1 -1
src/rag/pipeline.py CHANGED
@@ -86,7 +86,7 @@ class RAGPipeline:
86
  prompt_node = PromptNode(
87
  model_name_or_path="gpt-4",
88
  default_prompt_template=rag_prompt,
89
- api_key="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",
90
  max_length=3000,
91
  model_kwargs={"temperature": 0.2, "max_tokens": 4096},
92
  )
 
86
  prompt_node = PromptNode(
87
  model_name_or_path="gpt-4",
88
  default_prompt_template=rag_prompt,
89
+ api_key=OPENAI_API_KEY,
90
  max_length=3000,
91
  model_kwargs={"temperature": 0.2, "max_tokens": 4096},
92
  )