Spaces:
Restarting
Restarting
Update eb_agent_module.py
Browse files- eb_agent_module.py +4 -1
eb_agent_module.py
CHANGED
@@ -354,7 +354,10 @@ class EmployerBrandingAgent:
|
|
354 |
|
355 |
try:
|
356 |
# Configure LiteLLM with Gemini
|
357 |
-
llm = LiteLLM(
|
|
|
|
|
|
|
358 |
|
359 |
# Set PandasAI configuration
|
360 |
pai.config.set({
|
|
|
354 |
|
355 |
try:
|
356 |
# Configure LiteLLM with Gemini
|
357 |
+
llm = LiteLLM(
|
358 |
+
model="gemini/gemini-2.5-flash-preview-05-20", # Use gemini/ prefix for Gemini API
|
359 |
+
api_key=GEMINI_API_KEY
|
360 |
+
)
|
361 |
|
362 |
# Set PandasAI configuration
|
363 |
pai.config.set({
|