Todd Deshane commited on
Commit
cb4043a
·
1 Parent(s): 8b28e7d

replace variable with correct value

Browse files
Files changed (1) hide show
  1. tools.py +1 -1
tools.py CHANGED
@@ -74,7 +74,7 @@ def _youtube_rag(prompt: str):
74
  docs = docsearch.get_relevant_documents(prompt)
75
  chat_model = ChatOpenAI(model_name="gpt-4-1106-preview")
76
  chain = load_qa_chain(llm=chat_model, chain_type="stuff")
77
- answer = chain.run(input_documents=docs, question=query)
78
  return answer
79
 
80
  def generate_image(prompt: str):
 
74
  docs = docsearch.get_relevant_documents(prompt)
75
  chat_model = ChatOpenAI(model_name="gpt-4-1106-preview")
76
  chain = load_qa_chain(llm=chat_model, chain_type="stuff")
77
+ answer = chain.run(input_documents=docs, question=prompt)
78
  return answer
79
 
80
  def generate_image(prompt: str):