AakashJammula commited on
Commit
86c6189
·
verified ·
1 Parent(s): 19586a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -14,12 +14,10 @@ import os
14
  from langchain_google_genai import ChatGoogleGenerativeAI
15
 
16
  app = FastAPI()
17
- TAVILY_KEY = os.getenv("TAVILY_API_KEY")
18
- GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
19
 
20
- llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0.5, google_api_key=GOOGLE_API_KEY)
21
  # Tavily Web Search Tool
22
- search_tool = TavilySearch(apiKey=TAVILY_KEY)
23
 
24
  # Calculator Tool (simple math)
25
  calculator_tool = Tool.from_function(
 
14
  from langchain_google_genai import ChatGoogleGenerativeAI
15
 
16
  app = FastAPI()
 
 
17
 
18
+ llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0.5)
19
  # Tavily Web Search Tool
20
+ search_tool = TavilySearch()
21
 
22
  # Calculator Tool (simple math)
23
  calculator_tool = Tool.from_function(