Spaces:
Running
Running
Update app.py
Browse files
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,
|
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 |
|