GuglielmoTor commited on
Commit
eea7141
·
verified ·
1 Parent(s): 3ae9352

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -44,8 +44,7 @@ try:
44
  EmployerBrandingAgent,
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_RAG_DOCUMENTS as eb_agent_default_rag_docs,
49
  DEFAULT_SAFETY_SETTINGS as EB_AGENT_SAFETY_SETTINGS # Import safety settings
50
  )
51
  EB_AGENT_AVAILABLE = True
@@ -59,7 +58,7 @@ except ImportError as e:
59
  async def process_query(self, query, **kwargs): return "# Error: Employer Branding Agent module not loaded."
60
  def update_dataframes(self, dfs): pass
61
  def clear_chat_history(self): pass
62
- EB_AGENT_GEN_CONFIG, EB_AGENT_LLM_MODEL, EB_AGENT_EMBEDDING_MODEL, eb_agent_default_rag_docs, EB_AGENT_SAFETY_SETTINGS = {}, None, None, pd.DataFrame(), {}
63
 
64
 
65
  # Configure logging
@@ -698,7 +697,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
698
  generation_config_dict=EB_AGENT_GEN_CONFIG,
699
  safety_settings_list=EB_AGENT_SAFETY_SETTINGS,
700
  all_dataframes=dataframes_for_agent,
701
- rag_documents_df=eb_agent_default_rag_docs.copy(),
702
  embedding_model_name=EB_AGENT_EMBEDDING_MODEL
703
  )
704
 
 
44
  EmployerBrandingAgent,
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
 
58
  async def process_query(self, query, **kwargs): return "# Error: Employer Branding Agent module not loaded."
59
  def update_dataframes(self, dfs): pass
60
  def clear_chat_history(self): pass
61
+ EB_AGENT_GEN_CONFIG, EB_AGENT_LLM_MODEL, EB_AGENT_EMBEDDING_MODEL, EB_AGENT_SAFETY_SETTINGS = {}, None, None, pd.DataFrame(), {}
62
 
63
 
64
  # Configure logging
 
697
  generation_config_dict=EB_AGENT_GEN_CONFIG,
698
  safety_settings_list=EB_AGENT_SAFETY_SETTINGS,
699
  all_dataframes=dataframes_for_agent,
 
700
  embedding_model_name=EB_AGENT_EMBEDDING_MODEL
701
  )
702