GuglielmoTor commited on
Commit
a05b048
·
verified ·
1 Parent(s): 67e5b47

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. eb_agent_module.py +5 -2
eb_agent_module.py CHANGED
@@ -6,7 +6,7 @@ import logging
6
  import numpy as np
7
  import textwrap
8
  from datetime import datetime
9
- from typing import Dict, List, Optional, Union, Any
10
  import traceback
11
  import pandasai as pai
12
  from pandasai_litellm import LiteLLM
@@ -877,7 +877,10 @@ class EmployerBrandingAgent:
877
  "num_rag_documents": len(self.rag_system.documents_df) if self.rag_system.documents_df is not None else 0,
878
  "llm_model_name": self.llm_model_name,
879
  "embedding_model_name": self.rag_system.embedding_model_name,
880
- "chat_history_length": len(self.chat_history)
 
 
 
881
  }
882
 
883
  def get_available_analyses(self) -> List[str]:
 
6
  import numpy as np
7
  import textwrap
8
  from datetime import datetime
9
+ from typing import Dict, List, Optional, Union, Any, Tuple
10
  import traceback
11
  import pandasai as pai
12
  from pandasai_litellm import LiteLLM
 
877
  "num_rag_documents": len(self.rag_system.documents_df) if self.rag_system.documents_df is not None else 0,
878
  "llm_model_name": self.llm_model_name,
879
  "embedding_model_name": self.rag_system.embedding_model_name,
880
+ "chat_history_length": len(self.chat_history),
881
+ "charts_save_path_pandasai": pai.config.save_charts_path if pai.config.llm else "PandasAI not configured"
882
+ }
883
+
884
  }
885
 
886
  def get_available_analyses(self) -> List[str]: