sanjaystarc commited on
Commit
ad31eba
·
verified ·
1 Parent(s): cd48b7e

Update core_agent.py

Browse files
Files changed (1) hide show
  1. core_agent.py +2 -2
core_agent.py CHANGED
@@ -14,8 +14,8 @@ from dotenv import load_dotenv
14
  from langchain_google_genai import ChatGoogleGenerativeAI
15
  from langchain_core.messages import HumanMessage, SystemMessage
16
  from langchain_core.tools import tool
17
- from langchain.agents import AgentExecutor, create_tool_calling_agent
18
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 
19
  from langchain_community.chat_message_histories import ChatMessageHistory
20
 
21
  warnings.filterwarnings("ignore")
@@ -405,4 +405,4 @@ def make_plotly_chart(chart_type, df, profile, x_col=None, y_col=None, color_col
405
  fig.update_layout(paper_bgcolor=DARK_BG, plot_bgcolor=CARD_BG,
406
  font=dict(family="DM Sans, sans-serif", color="#E0E0FF"),
407
  margin=dict(l=40, r=40, t=60, b=40))
408
- return fig
 
14
  from langchain_google_genai import ChatGoogleGenerativeAI
15
  from langchain_core.messages import HumanMessage, SystemMessage
16
  from langchain_core.tools import tool
 
17
  from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
18
+ from langchain.agents import AgentExecutor, create_tool_calling_agent
19
  from langchain_community.chat_message_histories import ChatMessageHistory
20
 
21
  warnings.filterwarnings("ignore")
 
405
  fig.update_layout(paper_bgcolor=DARK_BG, plot_bgcolor=CARD_BG,
406
  font=dict(family="DM Sans, sans-serif", color="#E0E0FF"),
407
  margin=dict(l=40, r=40, t=60, b=40))
408
+ return fig