Scott Cogan commited on
Commit ·
9aa8d3a
1
Parent(s): 5f1fcc8
requirements update for llm compat
Browse files
app.py
CHANGED
|
@@ -320,7 +320,7 @@ class BasicAgent:
|
|
| 320 |
if "429" in error_str:
|
| 321 |
if "GenerateRequestsPerDayPerProjectPerModel-FreeTier" in error_str:
|
| 322 |
logger.warning("Daily quota limit reached for primary LLM, trying fallback")
|
| 323 |
-
if self.fallback_llm:
|
| 324 |
try:
|
| 325 |
response = self.fallback_llm.invoke(
|
| 326 |
[self.sys_msg] + messages,
|
|
@@ -398,7 +398,7 @@ class BasicAgent:
|
|
| 398 |
if "429" in error_str:
|
| 399 |
if "GenerateRequestsPerDayPerProjectPerModel-FreeTier" in error_str:
|
| 400 |
logger.warning("Daily quota limit reached, trying fallback")
|
| 401 |
-
if self.fallback_llm:
|
| 402 |
try:
|
| 403 |
response = self.fallback_llm.invoke(
|
| 404 |
[self.sys_msg] + messages,
|
|
@@ -426,7 +426,8 @@ class BasicAgent:
|
|
| 426 |
else:
|
| 427 |
logger.warning("Rate limit hit, waiting before retry...")
|
| 428 |
time.sleep(60) # Wait for 60 seconds before retry
|
| 429 |
-
|
|
|
|
| 430 |
|
| 431 |
def call_tools(self, state: AgentState) -> AgentState:
|
| 432 |
"""Call the tools based on the model's response."""
|
|
|
|
| 320 |
if "429" in error_str:
|
| 321 |
if "GenerateRequestsPerDayPerProjectPerModel-FreeTier" in error_str:
|
| 322 |
logger.warning("Daily quota limit reached for primary LLM, trying fallback")
|
| 323 |
+
if hasattr(self, 'fallback_llm') and self.fallback_llm is not None:
|
| 324 |
try:
|
| 325 |
response = self.fallback_llm.invoke(
|
| 326 |
[self.sys_msg] + messages,
|
|
|
|
| 398 |
if "429" in error_str:
|
| 399 |
if "GenerateRequestsPerDayPerProjectPerModel-FreeTier" in error_str:
|
| 400 |
logger.warning("Daily quota limit reached, trying fallback")
|
| 401 |
+
if hasattr(self, 'fallback_llm') and self.fallback_llm is not None:
|
| 402 |
try:
|
| 403 |
response = self.fallback_llm.invoke(
|
| 404 |
[self.sys_msg] + messages,
|
|
|
|
| 426 |
else:
|
| 427 |
logger.warning("Rate limit hit, waiting before retry...")
|
| 428 |
time.sleep(60) # Wait for 60 seconds before retry
|
| 429 |
+
raise # Re-raise to trigger retry
|
| 430 |
+
raise # Re-raise other exceptions to trigger retry
|
| 431 |
|
| 432 |
def call_tools(self, state: AgentState) -> AgentState:
|
| 433 |
"""Call the tools based on the model's response."""
|