Update agent.py
Browse files
agent.py
CHANGED
|
@@ -13,7 +13,7 @@ class BasicAgent:
|
|
| 13 |
## new agent code
|
| 14 |
#gemini_model = os.getenv(key="GEMINI_MODEL")
|
| 15 |
|
| 16 |
-
model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="
|
| 17 |
agent = CodeAgent(
|
| 18 |
tools=[youtube_transcription_tool, visit_webpage],
|
| 19 |
model=model,
|
|
|
|
| 13 |
## new agent code
|
| 14 |
#gemini_model = os.getenv(key="GEMINI_MODEL")
|
| 15 |
|
| 16 |
+
model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="MY_GEMINI_API_KEY"))
|
| 17 |
agent = CodeAgent(
|
| 18 |
tools=[youtube_transcription_tool, visit_webpage],
|
| 19 |
model=model,
|