Hariharan05 commited on
Commit
6b5a80d
·
verified ·
1 Parent(s): e67ef40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -6,16 +6,20 @@ import gradio as gr
6
  import os
7
  from dotenv import load_dotenv
8
  load_dotenv
 
 
 
 
9
  @tool
10
  def add_tool(a,b):
11
  """Add two numbers a and b"""
12
  return a+b
13
 
14
- web_search = TavilySearch(tavily_api_key=os.getenv("tavily_api_key"),
15
  max_results=5)
16
  llm = ChatGroq(
17
  model="llama-3.1-8b-instant",
18
- api_key=os.getenv("groq_api_key"),
19
 
20
  )
21
 
 
6
  import os
7
  from dotenv import load_dotenv
8
  load_dotenv
9
+
10
+ tavily_key = os.getenv("TAVILY_API_KEY")
11
+ groq_key = os.getenv("GROQ_API_KEY")
12
+
13
  @tool
14
  def add_tool(a,b):
15
  """Add two numbers a and b"""
16
  return a+b
17
 
18
+ web_search = TavilySearch(tavily_api_key=tavily_key,
19
  max_results=5)
20
  llm = ChatGroq(
21
  model="llama-3.1-8b-instant",
22
+ api_key=groq_key,
23
 
24
  )
25