dygoo commited on
Commit
5600b50
·
verified ·
1 Parent(s): 7119427

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -235,18 +235,18 @@ class BasicAgent:
235
  # For search results
236
  # Split results into sentences and find most relevant
237
  sentences = search_results.split('. ')
238
- if len(sentences) <= 3:
239
- return search_results[:300]
240
 
241
  # Try to find sentences with keywords from question
242
- keywords = [w for w in question.lower().split() if len(w) > 3]
243
  relevant_sentences = [] # NEW LINE
244
 
245
  for sentence in sentences:
246
  sentence_lower = sentence.lower()
247
  if any(keyword in sentence_lower for keyword in keywords):
248
  relevant_sentences.append(sentence)
249
- if len(relevant_sentences) >= 3: # Get up to 3 relevant sentences
250
  break
251
 
252
  # If we found relevant sentences, use them
@@ -254,7 +254,7 @@ class BasicAgent:
254
  return '. '.join(relevant_sentences)
255
 
256
  # Fallback to first few sentences
257
- return '. '.join(sentences[:3])
258
 
259
  def _formulate_direct_answer(self, relevant_info, question):
260
 
 
235
  # For search results
236
  # Split results into sentences and find most relevant
237
  sentences = search_results.split('. ')
238
+ if len(sentences) <= 50:
239
+ return search_results[:3000]
240
 
241
  # Try to find sentences with keywords from question
242
+ keywords = [w for w in question.lower().split() if len(w) > 40]
243
  relevant_sentences = [] # NEW LINE
244
 
245
  for sentence in sentences:
246
  sentence_lower = sentence.lower()
247
  if any(keyword in sentence_lower for keyword in keywords):
248
  relevant_sentences.append(sentence)
249
+ if len(relevant_sentences) >= 40:
250
  break
251
 
252
  # If we found relevant sentences, use them
 
254
  return '. '.join(relevant_sentences)
255
 
256
  # Fallback to first few sentences
257
+ return '. '.join(sentences[:40])
258
 
259
  def _formulate_direct_answer(self, relevant_info, question):
260