Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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 |
-
|
33 |
-
raise ValueError("HF_TOKEN environment variable not set.")
|
34 |
if not gemini_api_key:
|
35 |
-
raise ValueError("GEMINI_API_KEY environment
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
|
37 |
login(hf_token, add_to_git_credential=False)
|
38 |
|
@@ -97,7 +105,7 @@ tools = [
|
|
97 |
]
|
98 |
|
99 |
# Initialize model
|
100 |
-
model = LiteLLMModel(
|
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(
|