Vs code extension not working with graphql code gen

I am following this guide to configuration type-safe Apollo GraphQL on my react app

However, the code below won’t be detected as GraphQL in vscode

import { gql } from "../gql/generated/gql";

const GET_BOOK = gql(/* GraphQL */ `
    query getBook {
        book
    }
`);

If I use the default function, it works

import { gql } from "@apollo/client";

const GET_BOOK = gql`
    query getBook {
        book
    }
`;

Have I missed anything?

Hi @SonMooSans , do you have a GraphQL extension installed in VS Code? If you install the Apollo extension, it should show up like this: