devasurya commited on
Commit
d89bf92
·
verified ·
1 Parent(s): c0001f6

Update my_agent.py

Browse files
Files changed (1) hide show
  1. my_agent.py +2 -2
my_agent.py CHANGED
@@ -15,7 +15,7 @@ import os
15
 
16
 
17
  model = OpenAIServerModel(
18
- model_id="CLAUDE_3_SONNET_37", # 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,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