GuglielmoTor commited on
Commit
b3904a9
·
verified ·
1 Parent(s): ef2ba97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -45,7 +45,8 @@ try:
45
  GENERATION_CONFIG_PARAMS as EB_AGENT_GEN_CONFIG, # Rename to avoid conflict
46
  LLM_MODEL_NAME as EB_AGENT_LLM_MODEL, # Rename
47
  GEMINI_EMBEDDING_MODEL_NAME as EB_AGENT_EMBEDDING_MODEL, # Rename
48
- DEFAULT_SAFETY_SETTINGS as EB_AGENT_SAFETY_SETTINGS # Import safety settings
 
49
  )
50
  EB_AGENT_AVAILABLE = True
51
  logging.info("Successfully imported EmployerBrandingAgent module.")
 
45
  GENERATION_CONFIG_PARAMS as EB_AGENT_GEN_CONFIG, # Rename to avoid conflict
46
  LLM_MODEL_NAME as EB_AGENT_LLM_MODEL, # Rename
47
  GEMINI_EMBEDDING_MODEL_NAME as EB_AGENT_EMBEDDING_MODEL, # Rename
48
+ DEFAULT_SAFETY_SETTINGS as EB_AGENT_SAFETY_SETTINGS, # Import safety settings
49
+ get_all_schemas_representation
50
  )
51
  EB_AGENT_AVAILABLE = True
52
  logging.info("Successfully imported EmployerBrandingAgent module.")