-
Notifications
You must be signed in to change notification settings - Fork 28
Description
Hi LIDA team
When ever I choose a model other than gpt-3.5-turbo-1106, I get the an exception in the summarize object. For example if I choose gpt-4, I get the error;
File "/Users/indapa/miniconda3/envs/lida/lib/python3.11/site-packages/streamlit/runtime/scriptrunner/script_runner.py", line 534, in _run_script exec(code, module.__dict__) File "/Users/indapa/lida_indapa/main.py", line 165, in <module> st.session_state.summary = st.session_state.lida.summarize( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/Users/indapa/miniconda3/envs/lida/lib/python3.11/site-packages/lida/components/manager.py", line 131, in summarize return self.summarizer.summarize( ^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/Users/indapa/miniconda3/envs/lida/lib/python3.11/site-packages/lida/components/summarizer.py", line 144, in summarize data_summary = self.enrich( ^^^^^^^^^^^^ File "/Users/indapa/miniconda3/envs/lida/lib/python3.11/site-packages/lida/components/summarizer.py", line 115, in enrich raise ValueError(error_msg + "" + response.usage) ~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~
The relevant code is below. Throws exception when st_session_state.model is anything other than gpt-3.5-turbo-1106
st.session_state.textgen_config = TextGenerationConfig( n=num_visualizations, temperature=st.session_state.temp, model=st.session_state.model, use_cache=st.session_state.use_cache)
#make summary object and add to session state # **** lida.summarize ***** st.session_state.summary = st.session_state.lida.summarize( data=st.session_state.df, summary_method=selected_method, textgen_config=st.session_state.textgen_config)