JabrilJacobs commited on
Commit
30787d3
·
verified ·
1 Parent(s): d8ed545

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ class BasicAgent:
33
  class NewAgent:
34
  def __init__(self):
35
  print("NewAgent initialized.")
36
- def __call__(self, question: str) -> str:
37
  print(f"Agent received question (first 50 chars): {question[:50]}...")
38
 
39
  # Initialize the web search tool
@@ -61,7 +61,7 @@ class NewAgent:
61
  sys_msg = SystemMessage(
62
  content=f"""
63
  You are a general AI assistant. I will ask you a question.
64
- If a file_name is provided, it indicates there's an associated file you may need to analyze:
65
  If you cannot find an answer, you may report your thoughts.
66
  If you find an answer, your response should only contain your final answer. Report nothing before or after this answer.
67
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
 
33
  class NewAgent:
34
  def __init__(self):
35
  print("NewAgent initialized.")
36
+ def __call__(self, question: str, file_name: str = "") -> str:
37
  print(f"Agent received question (first 50 chars): {question[:50]}...")
38
 
39
  # Initialize the web search tool
 
61
  sys_msg = SystemMessage(
62
  content=f"""
63
  You are a general AI assistant. I will ask you a question.
64
+ If a file_name is provided, it indicates there's an associated file you may need to analyze.
65
  If you cannot find an answer, you may report your thoughts.
66
  If you find an answer, your response should only contain your final answer. Report nothing before or after this answer.
67
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.