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