Update app.py
Browse files
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.
|