Scott Cogan commited on
Commit
9aa8d3a
·
1 Parent(s): 5f1fcc8

requirements update for llm compat

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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
- raise
 
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."""