Spaces:
Runtime error
Runtime error
Update my_agent.py
Browse files- my_agent.py +2 -2
my_agent.py
CHANGED
|
@@ -15,7 +15,7 @@ import os
|
|
| 15 |
|
| 16 |
|
| 17 |
model = OpenAIServerModel(
|
| 18 |
-
model_id="
|
| 19 |
api_base=os.getenv("BASE_URL"), # Base URL for the API
|
| 20 |
api_key=os.getenv("OKTA_ACCESS_TOKEN"), # API key for authentication
|
| 21 |
custom_role_conversions=None,
|
|
@@ -47,7 +47,7 @@ agent = CodeAgent(
|
|
| 47 |
name=None,
|
| 48 |
description=None,
|
| 49 |
prompt_templates=prompt_templates,
|
| 50 |
-
additional_authorized_imports=['pandas', 'numpy', 'openpyxl', 'xlrd'],
|
| 51 |
model=model)
|
| 52 |
|
| 53 |
|
|
|
|
| 15 |
|
| 16 |
|
| 17 |
model = OpenAIServerModel(
|
| 18 |
+
model_id="CLAUDE_3_SONNET_35", # Replace with the desired model ID
|
| 19 |
api_base=os.getenv("BASE_URL"), # Base URL for the API
|
| 20 |
api_key=os.getenv("OKTA_ACCESS_TOKEN"), # API key for authentication
|
| 21 |
custom_role_conversions=None,
|
|
|
|
| 47 |
name=None,
|
| 48 |
description=None,
|
| 49 |
prompt_templates=prompt_templates,
|
| 50 |
+
additional_authorized_imports=['pandas', 'numpy', 'openpyxl', 'xlrd', 'os'],
|
| 51 |
model=model)
|
| 52 |
|
| 53 |
|