Firoj112 commited on
Commit
aa2777e
·
verified ·
1 Parent(s): b15d840

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -21,18 +21,26 @@ from tools.final_answer import FinalAnswerTool
21
  from GRADIO_UI import GradioUI
22
 
23
  # Set up logging
24
- logging.basicConfig(level=logging.INFO)
25
  logger = logging.getLogger(__name__)
26
 
 
 
 
27
  # Load environment variables
28
  load_dotenv()
29
  hf_token = os.getenv("HF_TOKEN")
30
  gemini_api_key = os.getenv("GOOGLE_API_KEY")
31
 
32
- if not hf_token:
33
- raise ValueError("HF_TOKEN environment variable not set.")
34
  if not gemini_api_key:
35
- raise ValueError("GEMINI_API_KEY environment variable not set.")
 
 
 
 
 
 
36
 
37
  login(hf_token, add_to_git_credential=False)
38
 
@@ -97,7 +105,7 @@ tools = [
97
  ]
98
 
99
  # Initialize model
100
- model = LiteLLMModel(model_name="gemini/gemini-2.0-flash")
101
 
102
  # Initialize agent
103
  agent = CodeAgent(
 
21
  from GRADIO_UI import GradioUI
22
 
23
  # Set up logging
24
+ logging.basicConfig(level=logging.DEBUG)
25
  logger = logging.getLogger(__name__)
26
 
27
+ litellm.set_verbose = True
28
+ logger.debug("Configuring litellm for gemini/gemini-2.0-flash")
29
+
30
  # Load environment variables
31
  load_dotenv()
32
  hf_token = os.getenv("HF_TOKEN")
33
  gemini_api_key = os.getenv("GOOGLE_API_KEY")
34
 
35
+ # Verify Gemini API key
 
36
  if not gemini_api_key:
37
+ raise ValueError("GEMINI_API_KEY not found in environment variables. Please set it in .env.")
38
+ logger.debug(f"GEMINI_API_KEY found: {bool(os.getenv('GEMINI_API_KEY'))}")
39
+
40
+ # Warn about Anthropic key
41
+ if os.getenv("ANTHROPIC_API_KEY"):
42
+ logger.warning("ANTHROPIC_API_KEY found in environment. This may cause conflicts.")
43
+
44
 
45
  login(hf_token, add_to_git_credential=False)
46
 
 
105
  ]
106
 
107
  # Initialize model
108
+ model = LiteLLMModel("gemini/gemini-2.0-flash")
109
 
110
  # Initialize agent
111
  agent = CodeAgent(