Fix issues with API docs

Author: patrikbraborec

docusaurus.config.js

@@ -1,7 +1,7 @@
 const { join, resolve } = require('node:path');
 
 const clsx = require('clsx');
-const { createApiPageMD } = require('docusaurus-plugin-openapi-docs/lib/markdown');
+const { createApiPageMD, createInfoPageMD } = require('docusaurus-plugin-openapi-docs/lib/markdown');
 
 const { config } = require('./apify-docs-theme');
 const { collectSlugs } = require('./tools/utils/collectSlugs');
@@ -196,6 +196,38 @@ module.exports = {
                                     md = md.replace('-->', '-->');
                                 }
 
+                                // Add LLMButtons import and component
+                                if (!md.includes('import LLMButtons')) {
+                                    // Find the first import statement and add LLMButtons import after it
+                                    const firstImportMatch = md.match(/^import\s+.*?from\s+["'][^"']*["'];?\s*$/m);
+                                    if (firstImportMatch) {
+                                        const importEnd = md.indexOf(firstImportMatch[0]) + firstImportMatch[0].length;
+                                        const llmButtonsImport = '\nimport LLMButtons from "@site/src/components/LLMButtons";';
+                                        md = md.slice(0, importEnd) + llmButtonsImport + md.slice(importEnd);
+                                    }
+                                }
+
+                                // Find the first Heading h1 and add LLMButtons after it
+                                // eslint-disable-next-line max-len
+                                const headingRegex = /(<Heading[^>]*as=\{"h1"\}[^>]*className=\{"openapi__heading"\}[^>]*children=\{[^}]*\}[^>]*>\s*<\/Heading>)/;
+                                md = md.replace(headingRegex, '$1\n\n<LLMButtons />\n');
+
+                                return md;
+                            },
+                            createInfoPageMD: (pageData) => {
+                                let md = createInfoPageMD(pageData);
+
+                                // Add LLMButtons import and component
+                                if (!md.includes('import LLMButtons')) {
+                                    // eslint-disable-next-line max-len
+                                    md = md.replace('import Heading from "@theme/Heading";', 'import Heading from "@theme/Heading";\nimport LLMButtons from "@site/src/components/LLMButtons";');
+                                }
+
+                                // Find the first Heading h1 and add LLMButtons after it
+                                // eslint-disable-next-line max-len
+                                const headingRegex = /(<Heading[^>]*as=\{"h1"\}[^>]*className=\{"openapi__heading"\}[^>]*children=\{[^}]*\}[^>]*>\s*<\/Heading>)/;
+                                md = md.replace(headingRegex, '$1\n\n<LLMButtons />\n');
+
                                 return md;
                             },
                         },

src/theme/DocItem/Content/index.js

@@ -22,15 +22,42 @@ function useSyntheticTitle() {
 export default function DocItemContent({ children }) {
   const syntheticTitle = useSyntheticTitle();
   const location = useLocation();
-  const shouldShowLLMButtons = !location.pathname.startsWith('/legal');
+
+  // Define the allowed API v2 paths that should show LLMButtons (tag/info pages)
+  // The logic is handled here, and also in docusaurus.config.js (see docusaurus-plugin-openapi-docs)
+  const allowedApiV2Paths = [
+    '/api/v2/getting-started',
+    '/api/v2/actors',
+    '/api/v2/actors-actor-versions',
+    '/api/v2/actors-actor-builds',
+    '/api/v2/actors-actor-runs',
+    '/api/v2/actors-webhook-collection',
+    '/api/v2/actor-builds',
+    '/api/v2/actor-runs',
+    '/api/v2/actor-tasks',
+    '/api/v2/storage-datasets',
+    '/api/v2/storage-key-value-stores',
+    '/api/v2/storage-request-queues',
+    '/api/v2/storage-request-queues-requests',
+    '/api/v2/storage-request-queues-requests-locks',
+    '/api/v2/webhooks-webhooks',
+    '/api/v2/webhooks-webhook-dispatches',
+    '/api/v2/schedules',
+    '/api/v2/store',
+    '/api/v2/logs',
+    '/api/v2/users',
+    '/platform',
+  ];
+
+  const shouldShowLLMButtons = allowedApiV2Paths.some((path) => location.pathname.startsWith(path));
 
   return (
     <div className={clsx(ThemeClassNames.docs.docMarkdown, 'markdown')}>
       <header>
-          <Heading as="h1">{syntheticTitle}</Heading>
-          {shouldShowLLMButtons && <LLMButtons />}
-          <MDXContent>{children}</MDXContent>
-        </header>
+        <Heading as="h1">{syntheticTitle}</Heading>
+        {shouldShowLLMButtons && <LLMButtons />}
+        <MDXContent>{children}</MDXContent>
+      </header>
     </div>
   );
 }