Spaces:
Running
Running
Commit
·
29212c7
1
Parent(s):
0eeea73
Modifying agent call
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ MODEL_API_KEY_MAP = {
|
|
| 13 |
}
|
| 14 |
|
| 15 |
|
| 16 |
-
def agent_fn(prompt, api_key=None):
|
| 17 |
env_key_name = MODEL_API_KEY_MAP.get(model_name)
|
| 18 |
if env_key_name:
|
| 19 |
if api_key:
|
|
|
|
| 13 |
}
|
| 14 |
|
| 15 |
|
| 16 |
+
def agent_fn(prompt, model_name, api_key=None):
|
| 17 |
env_key_name = MODEL_API_KEY_MAP.get(model_name)
|
| 18 |
if env_key_name:
|
| 19 |
if api_key:
|