Update my_agent.py

my_agent.py

@@ -15,7 +15,7 @@ import os
 
 
 model = OpenAIServerModel(
-    model_id="GEMINI_15_PRO",  # Replace with the desired model ID
+    model_id="CLAUDE_3_SONNET_37",  # Replace with the desired model ID
     api_base=os.getenv("BASE_URL"),  # Base URL for the API
     api_key=os.getenv("OKTA_ACCESS_TOKEN"),  # API key for authentication
     custom_role_conversions=None,
@@ -23,10 +23,10 @@ model = OpenAIServerModel(
     client_kwargs={
         "default_headers": {"Subscription-Key": os.getenv("SUBSCRIPTION_KEY")},  # Additional headers,
     },
-    web_search_options={
-        "enabled": True,
-        "dynamic_threshold": 0.5
-    }
+    # web_search_options={
+    #     "enabled": True,
+    #     "dynamic_threshold": 0.5
+    # }
 )
 
 # Load system prompt from prompt.yaml file
@@ -47,6 +47,7 @@ agent = CodeAgent(
     name=None,
     description=None,
     prompt_templates=prompt_templates,
+    additional_authorized_imports=['pandas', 'numpy', 'openpyxl', 'xlrd'],
     model=model)