ameglei-external commited on
Commit
eb37598
·
verified ·
1 Parent(s): c3243f8

Adjust prompts and increase a search articles amount returning

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -142,7 +142,7 @@ class BasicAgent:
142
  def search_tool(question: str, max_length: int = 100000) -> str:
143
  print(f"\nCalling search tool with: {question}, max_lentgh: {max_length}")
144
  search_ = TavilySearch(
145
- max_results=3,
146
  topic="general",
147
  )
148
  info = search_.invoke({"query": question})
@@ -202,7 +202,7 @@ class BasicAgent:
202
  }
203
  ])
204
 
205
- response = model.invoke([SystemMessage(content="Analyze the image."), msg])
206
  result = response.content
207
  print("Result:", result)
208
  return result
 
142
  def search_tool(question: str, max_length: int = 100000) -> str:
143
  print(f"\nCalling search tool with: {question}, max_lentgh: {max_length}")
144
  search_ = TavilySearch(
145
+ max_results=4,
146
  topic="general",
147
  )
148
  info = search_.invoke({"query": question})
 
202
  }
203
  ])
204
 
205
+ response = model.invoke([SystemMessage(content="Analyze the image and answer the question."), msg])
206
  result = response.content
207
  print("Result:", result)
208
  return result