wca: allow AAP user without organization_id

ansible_wisdom/ai/api/model_client/tests/test_wca_client.py

@@ -240,35 +240,48 @@ def test_fatal_exception(self):
         b = WCAClient.fatal_exception(exc)
         self.assertTrue(b)
 
-    def test_playbook_gen(self):
+
+@override_settings(WCA_SECRET_BACKEND_TYPE='dummy')
+@override_settings(ANSIBLE_AI_MODEL_MESH_API_KEY=None)
+@override_settings(ANSIBLE_AI_MODEL_MESH_MODEL_NAME=None)
+class TestWCAClientExpGen(WisdomAppsBackendMocking, WisdomServiceLogAwareTestCase):
+    def setUp(self):
+        super().setUp()
         wca_client = WCAClient(inference_url='http://example.com/')
         wca_client.get_api_key = Mock(return_value="some-key")
         wca_client.get_token = Mock(return_value={"access_token": "a-token"})
         wca_client.get_model_id = Mock(return_value="a-random-model")
         wca_client.session = Mock()
         response = Mock
-        response.text = '{"playbook": "Oh!", "outline": "Ahh!"}'
+        response.text = '{"playbook": "Oh!", "outline": "Ahh!", "explanation": "!Óh¡"}'
         wca_client.session.post.return_value = response
+        self.wca_client = wca_client
+
+    def test_playbook_gen(self):
         request = Mock()
-        playbook, outline = wca_client.generate_playbook(
+        playbook, outline = self.wca_client.generate_playbook(
             request, text="Install Wordpress", create_outline=True
         )
         self.assertEqual(playbook, "Oh!")
         self.assertEqual(outline, "Ahh!")
 
     def test_playbook_exp(self):
-        wca_client = WCAClient(inference_url='http://example.com/')
-        wca_client.get_api_key = Mock(return_value="some-key")
-        wca_client.get_token = Mock(return_value={"access_token": "a-token"})
-        wca_client.get_model_id = Mock(return_value="a-random-model")
-        wca_client.session = Mock()
-        response = Mock
-        response.text = '{"explanation": "!Óh¡"}'
-        wca_client.session.post.return_value = response
         request = Mock()
-        explanation = wca_client.explain_playbook(request, content="Some playbook")
+        explanation = self.wca_client.explain_playbook(request, content="Some playbook")
         self.assertEqual(explanation, "!Óh¡")
 
+    def test_playbook_gen_no_org(self):
+        request = Mock()
+        request.user.organization = None
+        self.wca_client.generate_playbook(request, text="Install Wordpress")
+        self.wca_client.get_api_key.assert_called_with(None)
+
+    def test_playbook_exp_no_org(self):
+        request = Mock()
+        request.user.organization = None
+        self.wca_client.explain_playbook(request, content="Some playbook")
+        self.wca_client.get_api_key.assert_called_with(None)
+
 
 @override_settings(ANSIBLE_WCA_RETRY_COUNT=1)
 @override_settings(WCA_SECRET_BACKEND_TYPE="dummy")

ansible_wisdom/ai/api/model_client/wca_client.py

@@ -473,8 +473,9 @@ def self_test(self) -> HealthCheckSummary:
     def generate_playbook(
         self, request, text: str = "", create_outline: bool = False, outline: str = ""
     ) -> tuple[str, str]:
-        api_key = self.get_api_key(request.user.organization.id)
-        model_id = self.get_model_id(request.user.organization.id)
+        organization_id = request.user.organization.id if request.user.organization else None
+        api_key = self.get_api_key(organization_id)
+        model_id = self.get_model_id(organization_id)
 
         headers = self._get_base_headers(api_key)
         data = {
@@ -493,8 +494,9 @@ def generate_playbook(
         return response["playbook"], response["outline"]
 
     def explain_playbook(self, request, content: str) -> str:
-        api_key = self.get_api_key(request.user.organization.id)
-        model_id = self.get_model_id(request.user.organization.id)
+        organization_id = request.user.organization.id if request.user.organization else None
+        api_key = self.get_api_key(organization_id)
+        model_id = self.get_model_id(organization_id)
 
         headers = self._get_base_headers(api_key)
         data = {