SurabhiT commited on
Commit
082148d
·
verified ·
1 Parent(s): 09870ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -8
app.py CHANGED
@@ -13,8 +13,11 @@ from crewai.process import Process
13
  import gradio as gr
14
  import numpy as np
15
 
 
 
 
16
  os.environ["OPENAI_API_KEY"] = "NA"
17
- os.environ["GOOGLE_API_KEY"] = "AIzaSyD1Gxk5OZMnqAlzWoYjAoDBeu3Z5l2TS7U"
18
  os.environ["GROQ_API_KEY"] = "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"
19
 
20
  from crewai_tools import PDFSearchTool
@@ -23,18 +26,13 @@ from crewai_tools import DOCXSearchTool
23
  from crewai_tools import TXTSearchTool
24
  from crewai_tools import CSVSearchTool
25
 
26
- # llm = ChatGroq(
27
- # model="llama3-70b-8192",
28
- # verbose=True,
29
- # temperature=0.8,
30
- # api_key=os.environ["GROQ_API_KEY"]
31
- # )
32
 
33
  llm = ChatGoogleGenerativeAI(
34
  model="gemini-1.5-flash-latest",
35
  verbose=True,
36
  temperature=0.6,
37
- google_api_key=os.environ["GOOGLE_API_KEY"]
38
  )
39
 
40
 
 
13
  import gradio as gr
14
  import numpy as np
15
 
16
+ api_key=os.getenv("GOOGLE_API_KEY")
17
+
18
+
19
  os.environ["OPENAI_API_KEY"] = "NA"
20
+ os.environ["GOOGLE_API_KEY"] = api_key
21
  os.environ["GROQ_API_KEY"] = "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"
22
 
23
  from crewai_tools import PDFSearchTool
 
26
  from crewai_tools import TXTSearchTool
27
  from crewai_tools import CSVSearchTool
28
 
29
+
 
 
 
 
 
30
 
31
  llm = ChatGoogleGenerativeAI(
32
  model="gemini-1.5-flash-latest",
33
  verbose=True,
34
  temperature=0.6,
35
+ google_api_key=api_key
36
  )
37
 
38