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