Update eb_agent_module.py

eb_agent_module.py

@@ -318,7 +318,7 @@ class AdvancedRAGSystem:
             logging.error(f"Error during RAG retrieval for query '{query[:50]}...': {e}", exc_info=True)
             return ""
 
-class EnhancedEmployerBrandingAgent:
+class EmployerBrandingAgent:
     def __init__(self,
                  all_dataframes: Optional[Dict[str, pd.DataFrame]] = None,
                  rag_documents_df: Optional[pd.DataFrame] = None,
@@ -808,12 +808,12 @@ class EnhancedEmployerBrandingAgent:
 
 # --- Helper Functions for External Integration ---
 def create_agent_instance(dataframes: Optional[Dict[str, pd.DataFrame]] = None,
-                          rag_docs: Optional[pd.DataFrame] = None) -> EnhancedEmployerBrandingAgent:
+                          rag_docs: Optional[pd.DataFrame] = None) -> EmployerBrandingAgent:
     """Factory function to create a new agent instance"""
     logging.info("Creating new EnhancedEmployerBrandingAgent instance via helper function.")
-    return EnhancedEmployerBrandingAgent(all_dataframes=dataframes, rag_documents_df=rag_docs)
+    return EmployerBrandingAgent(all_dataframes=dataframes, rag_documents_df=rag_docs)
 
-async def initialize_agent_async(agent: EnhancedEmployerBrandingAgent) -> bool:
+async def initialize_agent_async(agent: EmployerBrandingAgent) -> bool:
     """Async helper to initialize an agent instance"""
     logging.info("Initializing agent via async helper function.")
     return await agent.initialize()