GuglielmoTor commited on
Commit
caf3e08
·
verified ·
1 Parent(s): 22dcfa0

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. eb_agent_module.py +3 -3
eb_agent_module.py CHANGED
@@ -509,11 +509,11 @@ class EmployerBrandingAgent:
509
 
510
  return response_text.strip()
511
 
512
- except types.generation_types.BlockedPromptException as bpe: # Specific exception for google-generativeai
513
- logging.error(f"BlockedPromptException from LLM: {bpe}", exc_info=True)
514
  return f"I'm sorry, your request was blocked by the safety filter. Please rephrase your query. Details: {bpe}"
515
  except Exception as e:
516
- logging.error(f"Error in _generate_response: {e}", exc_info=True)
517
  return f"I encountered an error while processing your request: {type(e).__name__} - {str(e)}"
518
 
519
 
 
509
 
510
  return response_text.strip()
511
 
512
+ except types.BlockedPromptException as bpe: # <--- CORRECTED LINE
513
+ logging.error(f"BlockedPromptException from LLM: {bpe}", exc_info=True) # Keep exc_info=True for full traceback in logs
514
  return f"I'm sorry, your request was blocked by the safety filter. Please rephrase your query. Details: {bpe}"
515
  except Exception as e:
516
+ logging.error(f"Error in _generate_response: {e}", exc_info=True) # Keep exc_info=True
517
  return f"I encountered an error while processing your request: {type(e).__name__} - {str(e)}"
518
 
519