Spaces:
Sleeping
Sleeping
fix the way to set api_key
Browse files
app.py
CHANGED
|
@@ -52,7 +52,6 @@ class BasicAgent:
|
|
| 52 |
"model_name": "gemini-flash-family",
|
| 53 |
"litellm_params": {
|
| 54 |
"model": "gemini/gemini-2.5-flash",
|
| 55 |
-
"api_key": os.getenv("API_KEY"),
|
| 56 |
"rpm": 10,
|
| 57 |
"tpm": 250_000,
|
| 58 |
},
|
|
@@ -62,7 +61,6 @@ class BasicAgent:
|
|
| 62 |
"model_name": "gemini-flash-family",
|
| 63 |
"litellm_params": {
|
| 64 |
"model": "gemini/gemini-2.0-flash",
|
| 65 |
-
"api_key": os.getenv("API_KEY"),
|
| 66 |
"rpm": 15,
|
| 67 |
"tpm": 1_000_000,
|
| 68 |
},
|
|
@@ -72,7 +70,6 @@ class BasicAgent:
|
|
| 72 |
"model_name": "gemini-flash-family",
|
| 73 |
"litellm_params": {
|
| 74 |
"model": "gemini/gemini-2.0-flash-lite",
|
| 75 |
-
"api_key": os.getenv("API_KEY"),
|
| 76 |
"rpm": 30,
|
| 77 |
"tpm": 1_000_000,
|
| 78 |
},
|
|
@@ -93,6 +90,7 @@ class BasicAgent:
|
|
| 93 |
AssemblyAIAudioTranscriptTool(),
|
| 94 |
],
|
| 95 |
model=model,
|
|
|
|
| 96 |
# step_callbacks=[rate_limiter.increment_and_sleep_if_needed],
|
| 97 |
additional_authorized_imports=["pandas", "openpyxl", "requests"],
|
| 98 |
instructions=CUSTOM_PROMPT
|
|
|
|
| 52 |
"model_name": "gemini-flash-family",
|
| 53 |
"litellm_params": {
|
| 54 |
"model": "gemini/gemini-2.5-flash",
|
|
|
|
| 55 |
"rpm": 10,
|
| 56 |
"tpm": 250_000,
|
| 57 |
},
|
|
|
|
| 61 |
"model_name": "gemini-flash-family",
|
| 62 |
"litellm_params": {
|
| 63 |
"model": "gemini/gemini-2.0-flash",
|
|
|
|
| 64 |
"rpm": 15,
|
| 65 |
"tpm": 1_000_000,
|
| 66 |
},
|
|
|
|
| 70 |
"model_name": "gemini-flash-family",
|
| 71 |
"litellm_params": {
|
| 72 |
"model": "gemini/gemini-2.0-flash-lite",
|
|
|
|
| 73 |
"rpm": 30,
|
| 74 |
"tpm": 1_000_000,
|
| 75 |
},
|
|
|
|
| 90 |
AssemblyAIAudioTranscriptTool(),
|
| 91 |
],
|
| 92 |
model=model,
|
| 93 |
+
api_key=os.getenv("API_KEY"),
|
| 94 |
# step_callbacks=[rate_limiter.increment_and_sleep_if_needed],
|
| 95 |
additional_authorized_imports=["pandas", "openpyxl", "requests"],
|
| 96 |
instructions=CUSTOM_PROMPT
|