pratikshahp commited on
Commit
74ef341
·
verified ·
1 Parent(s): 794d2e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def perform_rag(repo_data, prompt):
70
  # Perform generation using OpenAI
71
  llm = OpenAI(api_key=openai_api_key, model="gpt-4o-mini", temperature=0.7, max_tokens=500)
72
  question_with_context = f"Context: {text}\n\nQuestion: {prompt}\n\nAnswer:"
73
- response = llm.generate(question_with_context)
74
 
75
  return response
76
  else:
 
70
  # Perform generation using OpenAI
71
  llm = OpenAI(api_key=openai_api_key, model="gpt-4o-mini", temperature=0.7, max_tokens=500)
72
  question_with_context = f"Context: {text}\n\nQuestion: {prompt}\n\nAnswer:"
73
+ response = llm.generate([question_with_context])
74
 
75
  return response
76
  else: