Remove redundant check

Author: jcbwndsr

src/axiomatic/magic.py

@@ -26,54 +26,49 @@ def ax_api(self, *args, **kwargs):
         pass
 
     def axquery(self, query, cell=None):
-        if self.api_key:
-            if cell:
-                # REFINE
-                try:
-                    exec(cell)
-                    feedback = "Code executed successfully."
-                except Exception as e:
-                    feedback = f"Errors:\n{e}"
-                print(feedback)
-                current_query = f"{self.query}\n{query}"
-                result = self.client.pic.refine(
-                    query=current_query, code=cell, feedback=feedback
+        if cell:
+            # REFINE
+            try:
+                exec(cell)
+                feedback = "Code executed successfully."
+            except Exception as e:
+                feedback = f"Errors:\n{e}"
+            print(feedback)
+            current_query = f"{self.query}\n{query}"
+            result = self.client.pic.refine(
+                query=current_query, code=cell, feedback=feedback
+            )
+        else:
+            # GENERATE FROM SCRATCH
+            self.query = query
+            result = self.client.pic.generate(query=query)
+
+        # Process output
+        pre_thought = result.raw_content.split("<thought>")[0]
+        thought = result.thought_text.replace("\n", "<br>")
+        if not thought:
+            output = result.raw_content.replace("\n", "<br>")
+        else:
+            output = pre_thought + "<br>" + thought
+        html_content = f"""<div style='font-family: Arial, sans-serif; line-height: 1.5;'>"
+<div style='color: #6EB700;'><strong>AX:</strong> {output}</div>"""
+        display(HTML(html_content))
+
+        # Process code
+        # remove last three lines (saving file)
+        if result.code:
+            code = "\n".join(result.code.split("\n")[:-3] + ["c"])
+            if "google.colab" in sys.modules:
+                # When running in colab
+                from google.colab import _frontend  # type: ignore
+
+                _frontend.create_scratch_cell(
+                    f"""# {query}\n# %%ax_fix\n{code}""", bottom_pane=True
                 )
             else:
-                # GENERATE FROM SCRATCH
-                self.query = query
-                result = self.client.pic.generate(query=query)
-
-            # Process output
-            pre_thought = result.raw_content.split("<thought>")[0]
-            thought = result.thought_text.replace("\n", "<br>")
-            if not thought:
-                output = result.raw_content.replace("\n", "<br>")
-            else:
-                output = pre_thought + "<br>" + thought
-            html_content = f"""<div style='font-family: Arial, sans-serif; line-height: 1.5;'>"
-<div style='color: #6EB700;'><strong>AX:</strong> {output}</div>"""
-            display(HTML(html_content))
-
-            # Process code
-            # remove last three lines (saving file)
-            if result.code:
-                code = "\n".join(result.code.split("\n")[:-3] + ["c"])
-                if "google.colab" in sys.modules:
-                    # When running in colab
-                    from google.colab import _frontend  # type: ignore
-
-                    _frontend.create_scratch_cell(
-                        f"""# {query}\n# %%ax_fix\n{code}""", bottom_pane=True
-                    )
-                else:
-                    # When running in jupyter
-                    get_ipython().set_next_input(f"# %%ax_fix\n{code}", replace=False)
+                # When running in jupyter
+                get_ipython().set_next_input(f"# %%ax_fix\n{code}", replace=False)
 
-        else:
-            print(
-                "Please set your Axiomatic API key first with the command %ax_api API_KEY and restart the kernel. Request the api key at our Customer Service."
-            )
 
     def ax_query(self, query, cell=None):
         # Updates the target circuit query