eaglelandsonce commited on
Commit
fd234e5
1 Parent(s): 160e47e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  import gradio as gr
3
  from textwrap import dedent
4
-
5
 
6
  # Tool import
7
  from crewai.tools.gemini_tools import GeminiSearchTools
@@ -12,6 +12,13 @@ from crewai.tools.sec_tools import SECTools
12
  # Google Langchain
13
  from langchain_google_genai import ChatGoogleGenerativeAI
14
 
 
 
 
 
 
 
 
15
  from crewai import Agent, Task, Crew, Process
16
 
17
  # os.environ["OPENAI_API_KEY"] = "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"
 
1
  import os
2
  import gradio as gr
3
  from textwrap import dedent
4
+ import google.generativeai as genai
5
 
6
  # Tool import
7
  from crewai.tools.gemini_tools import GeminiSearchTools
 
12
  # Google Langchain
13
  from langchain_google_genai import ChatGoogleGenerativeAI
14
 
15
+ # Retrieve API Key from Environment Variable
16
+ GOOGLE_AI_STUDIO = os.environ.get('GOOGLE_AI_STUDIO2')
17
+
18
+ # Ensure the API key is available
19
+ if not GOOGLE_AI_STUDIO:
20
+ raise ValueError("API key not found. Please set the GOOGLE_AI_STUDIO2 environment variable.")
21
+
22
  from crewai import Agent, Task, Crew, Process
23
 
24
  # os.environ["OPENAI_API_KEY"] = "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"