TeresaK commited on
Commit
1357bc1
1 Parent(s): a5718ea

Update src/rag/pipeline.py

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