fix specs

Author: SamSaffron

lib/completions/endpoints/open_ai.rb

@@ -27,6 +27,19 @@ def provider_id
           AiApiAuditLog::Provider::OpenAI
         end
 
+        def perform_completion!(
+          dialect,
+          user,
+          model_params = {},
+          feature_name: nil,
+          feature_context: nil,
+          partial_tool_calls: false,
+          &blk
+        )
+          @disable_native_tools = dialect.disable_native_tools?
+          super
+        end
+
         private
 
         def model_uri

spec/lib/modules/ai_bot/personas/persona_spec.rb

@@ -46,6 +46,7 @@ def system_prompt
     }
   end
 
+  fab!(:admin)
   fab!(:user)
   fab!(:upload)
 
@@ -238,6 +239,20 @@ def system_prompt
 
       # should be ordered by priority and then alpha
       expect(DiscourseAi::AiBot::Personas::Persona.all(user: user)).to eq(
+        [
+          DiscourseAi::AiBot::Personas::General,
+          DiscourseAi::AiBot::Personas::Artist,
+          DiscourseAi::AiBot::Personas::Creative,
+          DiscourseAi::AiBot::Personas::DiscourseHelper,
+          DiscourseAi::AiBot::Personas::GithubHelper,
+          DiscourseAi::AiBot::Personas::Researcher,
+          DiscourseAi::AiBot::Personas::SettingsExplorer,
+          DiscourseAi::AiBot::Personas::SqlHelper,
+        ],
+      )
+
+      # it should allow staff access to WebArtifactCreator
+      expect(DiscourseAi::AiBot::Personas::Persona.all(user: admin)).to eq(
         [
           DiscourseAi::AiBot::Personas::General,
           DiscourseAi::AiBot::Personas::Artist,
@@ -256,7 +271,7 @@ def system_prompt
       SiteSetting.ai_google_custom_search_api_key = ""
       SiteSetting.ai_artifact_security = "disabled"
 
-      expect(DiscourseAi::AiBot::Personas::Persona.all(user: user)).to contain_exactly(
+      expect(DiscourseAi::AiBot::Personas::Persona.all(user: admin)).to contain_exactly(
         DiscourseAi::AiBot::Personas::General,
         DiscourseAi::AiBot::Personas::SqlHelper,
         DiscourseAi::AiBot::Personas::SettingsExplorer,