GuglielmoTor commited on
Commit
eb80e2d
·
verified ·
1 Parent(s): 573c97f

Update insight_and_tasks/agents/follower_agent.py

Browse files
insight_and_tasks/agents/follower_agent.py CHANGED
@@ -16,6 +16,9 @@ logger = logging.getLogger(__name__)
16
  # Define the model globally or pass it as a parameter. For now, using a constant.
17
  # Consider moving this to a shared config or environment variable.
18
  DEFAULT_AGENT_MODEL = "gemini-2.5-flash-preview-05-20" # Or your specific model like "gemini-1.5-flash-preview-05-20"
 
 
 
19
 
20
  class EnhancedFollowerAnalysisAgent:
21
  """
 
16
  # Define the model globally or pass it as a parameter. For now, using a constant.
17
  # Consider moving this to a shared config or environment variable.
18
  DEFAULT_AGENT_MODEL = "gemini-2.5-flash-preview-05-20" # Or your specific model like "gemini-1.5-flash-preview-05-20"
19
+ os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "False"
20
+ GOOGLE_API_KEY = os.environ.get("GEMINI_API_KEY")
21
+ os.environ["GOOGLE_API_KEY"] = GOOGLE_API_KEY
22
 
23
  class EnhancedFollowerAnalysisAgent:
24
  """