Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -180,7 +180,7 @@ class BasicAgent:
|
|
| 180 |
print("BasicAgent initialized.")
|
| 181 |
|
| 182 |
def __call__(self, question: str) -> str:
|
| 183 |
-
|
| 184 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 185 |
system_prompt = SystemMessage(
|
| 186 |
content="You are a general AI assistant. \
|
|
@@ -199,7 +199,8 @@ class BasicAgent:
|
|
| 199 |
message = HumanMessage(content=question)
|
| 200 |
print(message)
|
| 201 |
answer = None
|
| 202 |
-
|
|
|
|
| 203 |
try:
|
| 204 |
alfred = build_agent(chat1)
|
| 205 |
answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
|
|
|
|
| 180 |
print("BasicAgent initialized.")
|
| 181 |
|
| 182 |
def __call__(self, question: str) -> str:
|
| 183 |
+
|
| 184 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
| 185 |
system_prompt = SystemMessage(
|
| 186 |
content="You are a general AI assistant. \
|
|
|
|
| 199 |
message = HumanMessage(content=question)
|
| 200 |
print(message)
|
| 201 |
answer = None
|
| 202 |
+
wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
|
| 203 |
+
while not answer or answer in wrong_answers or answer.lower().startswith("error"):
|
| 204 |
try:
|
| 205 |
alfred = build_agent(chat1)
|
| 206 |
answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
|