GuglielmoTor commited on
Commit
d2218fc
·
verified ·
1 Parent(s): 205df41

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. eb_agent_module.py +1 -0
eb_agent_module.py CHANGED
@@ -541,6 +541,7 @@ class EmployerBrandingAgent:
541
  # For multi-df queries, you'd use pai.chat(query, df1, df2, ...)
542
  if len(self.pandas_dfs) == 1:
543
  df = list(self.pandas_dfs.values())[0]
 
544
  pandas_response = df.chat(query)
545
  else:
546
  # For multiple dataframes, use pai.chat with all dfs
 
541
  # For multi-df queries, you'd use pai.chat(query, df1, df2, ...)
542
  if len(self.pandas_dfs) == 1:
543
  df = list(self.pandas_dfs.values())[0]
544
+ logging.info(f"Using single DataFrame for query: {df}")
545
  pandas_response = df.chat(query)
546
  else:
547
  # For multiple dataframes, use pai.chat with all dfs