Prat0 commited on
Commit
73e785d
·
verified ·
1 Parent(s): f1ddf07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -87,7 +87,7 @@ async def process_query(input_data: AgentInput):
87
 
88
  # Set up ComposioToolSet
89
  composio_toolset = ComposioToolSet(api_key=os.getenv("COMPOSIO_API_KEY"), output_dir=Path("./plots/"))
90
- tools = composio_toolset.get_tools(apps=[App.GOOGLESHEETS])
91
 
92
  # Define prefix messages
93
  prefix_messages = [
@@ -95,7 +95,8 @@ async def process_query(input_data: AgentInput):
95
  role="system",
96
  content=(
97
  "You are an AI assistant who is an expert at Google Sheets. "
98
- "Use Google Sheets Tool and perform the necessary operations based on query"
 
99
  )
100
  )
101
  ]
 
87
 
88
  # Set up ComposioToolSet
89
  composio_toolset = ComposioToolSet(api_key=os.getenv("COMPOSIO_API_KEY"), output_dir=Path("./plots/"))
90
+ tools = composio_toolset.get_tools(apps=[App.GOOGLESHEETS, App.CODEINTERPRETER])
91
 
92
  # Define prefix messages
93
  prefix_messages = [
 
95
  role="system",
96
  content=(
97
  "You are an AI assistant who is an expert at Google Sheets. "
98
+ "Use Google Sheets Tool and perform the necessary operations based on query, use code interpreter to plot graphs"
99
+ "return the plotted graph images in markdown format."
100
  )
101
  )
102
  ]