Todd Deshane commited on
Commit
6824b31
·
1 Parent(s): f1a7d41

fix syntax error

Browse files
Files changed (1) hide show
  1. tools.py +1 -1
tools.py CHANGED
@@ -75,7 +75,7 @@ def _youtube_rag(prompt: str):
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):
81
  image_name = _generate_image(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):
81
  image_name = _generate_image(prompt)