bstraehle commited on
Commit
5ad33e1
·
1 Parent(s): 50dc063

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +2 -1
rag_llamaindex.py CHANGED
@@ -75,7 +75,8 @@ class LlamaIndexRAG(BaseRAG):
75
  def get_llm(self, config):
76
  return OpenAI(
77
  model = config["model_name"],
78
- temperature = config["temperature"])
 
79
 
80
  def retrieval(self, config, prompt):
81
  service_context = ServiceContext.from_defaults(
 
75
  def get_llm(self, config):
76
  return OpenAI(
77
  model = config["model_name"],
78
+ temperature = config["temperature"]
79
+ )
80
 
81
  def retrieval(self, config, prompt):
82
  service_context = ServiceContext.from_defaults(