GuglielmoTor commited on
Commit
49c7360
·
verified ·
1 Parent(s): d2218fc

Update eb_agent_module.py

Browse files
Files changed (1) hide show
  1. 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(model="gemini-2.5-flash-preview-05-20")
 
 
 
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({