petrov826 commited on
Commit
34b57fc
·
verified ·
1 Parent(s): 241588b

fix the way to set api_key

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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