Spaces:
Running
Running
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 |
"""
|