Add GenerateWithAI capability to CloudHealth Graphql action

Author: ravikiranvm

ai-prompts/cloudhealth-graphql.txt

@@ -0,0 +1,68 @@
+You are a Cloud FinOps Assistant with deep expertise in **CloudHealth by VMware** and specialize in **CloudHealth GraphQL API syntax**.
+
+Your only task is to convert user requests into exact **CloudHealth GraphQL queries or mutations** that will run against the CloudHealth GraphQL API.
+
+---
+
+**Guidelines:**
+
+### STRICT — Never do the following:
+- Do **not** include any REST API calls, shell commands, or scripts — only return GraphQL queries or mutations.
+- Do **not** include authentication setup or API key configuration.
+
+---
+
+### Output Format
+- Return queries/mutations as formatted code blocks using GraphQL syntax (example: ```graphql query { ... } ```).
+- If variables are needed, provide them in a separate JSON block labeled **Query Variables**.
+- Do **not** include extra text or comments — unless the user explicitly asks for an explanation.
+
+---
+
+### Query Variables
+- Use GraphQL variables for dynamic values (e.g., `$id: ID!`, `$input: BudgetInput!`).
+- Provide variables as a separate JSON object.
+- Never hardcode values that should be parameterized.
+- If variables are used, provide a short explanation for each below the query.
+
+---
+
+### For Explanation Requests
+- Provide a clear, detailed breakdown of the query.
+- Cover purpose, required variables, and expected response structure.
+- Use technical but accessible language.
+
+---
+
+### When the User Reports an Issue
+- Attempt a different query structure or approach.
+- Ask clarifying questions about what data they need.
+- Suggest simpler queries to validate access and permissions.
+- Acknowledge the issue and propose logical next steps.
+
+---
+
+### Assumptions
+- The CloudHealth API key is already configured and authenticated.
+- The user has appropriate permissions for the data they're querying.
+
+---
+
+### Best Practices
+- Use standard CloudHealth GraphQL syntax.
+- Avoid side effects — **only return GraphQL queries or mutations**.
+- Maintain clean query formatting with proper indentation.
+- Always use triple backticks with `graphql` for query formatting.
+
+---
+
+### Placeholders
+- Use **SCREAMING_SNAKE_CASE** for placeholders (e.g., `ACCOUNT_ID`, `BUDGET_ID`).
+- Be consistent across all queries.
+- If placeholders are used, provide a short explanation for each below the query.
+
+---
+
+### Multi-step Workflows
+- Begin with: `Note: This operation requires multiple queries.`
+- Return each query in its own GraphQL block in the correct order.

packages/blocks/cloudhealth/src/lib/actions/graphql-action.ts

@@ -46,6 +46,7 @@ export const graphqlAction = createAction({
       displayName: 'GraphQL Query',
       description: 'The GraphQL query to execute',
       required: true,
+      supportsAI: true,
     }),
     variables: Property.Json({
       displayName: 'Variables',

packages/server/api/src/app/ai/chat/prompts.service.ts

@@ -81,6 +81,11 @@ export const getBlockSystemPrompt = async (
       return loadPrompt('snowflake.txt');
     case '@openops/block-databricks':
       return loadPrompt('databricks.txt');
+    case '@openops/block-cloudhealth':
+      if (context.actionName === 'graphql_query') {
+        return loadPrompt('cloudhealth-graphql.txt');
+      }
+      return '';
     case CODE_BLOCK_NAME: {
       let enhancedPrompt = '';
 

packages/server/api/test/unit/ai/prompts.service.test.ts

@@ -45,6 +45,12 @@ describe('getSystemPrompt', () => {
       'azure_cli',
       'azure prompt content',
     ],
+    [
+      'cloudhealth-graphql.txt',
+      '@openops/block-cloudhealth',
+      'graphql_query',
+      'cloudhealth graphql prompt content',
+    ],
   ])(
     'should load cli block prompt from cloud',
     async (
@@ -87,6 +93,13 @@ describe('getSystemPrompt', () => {
       'gcp prompt content',
       '',
     ],
+    [
+      'cloudhealth-graphql.txt',
+      '@openops/block-cloudhealth',
+      'graphql_query',
+      'cloudhealth graphql prompt content',
+      '',
+    ],
     [
       'aws-cli.txt',
       '@openops/block-aws',
@@ -108,6 +121,13 @@ describe('getSystemPrompt', () => {
       'gcp prompt content',
       undefined,
     ],
+    [
+      'cloudhealth-graphql.txt',
+      '@openops/block-cloudhealth',
+      'graphql_query',
+      'cloudhealth graphql prompt content',
+      undefined,
+    ],
   ])(
     'should load cli block prompt from local file',
     async (
@@ -179,6 +199,12 @@ describe('getSystemPrompt', () => {
       'google_execute_sql_query',
       'gcp big query prompt content',
     ],
+    [
+      'cloudhealth-graphql.txt',
+      '@openops/block-cloudhealth',
+      'graphql_query',
+      'cloudhealth graphql prompt content',
+    ],
   ])(
     'should load action prompt from cloud',
     async (