Gemini Search Grounding

curl --request POST \
  'https://api.agtcloud.cn/v1beta/models/MODEL_ID:generateContent' \
  -H 'Content-Type: application/json' \
  -H 'Authorization: Bearer your-api-key' \
  --data '{
    "contents": [
      {
        "role": "user",
        "parts": [
          {
            "text": "Hôm nay có những tin trí tuệ nhân tạo nào đáng chú ý? Hãy cung cấp nguồn."
          }
        ]
      }
    ],
    "tools": [
      {
        "google_search": {}
      }
    ]
  }'

{
  "candidates": [
    {
      "content": {
        "role": "model",
        "parts": [
          {"text": "Theo các báo cáo gần đây, những tiến triển liên quan gồm…"}
        ]
      },
      "groundingMetadata": {
        "webSearchQueries": [
          "trí tuệ nhân tạo tin tức hôm nay"
        ],
        "groundingChunks": [
          {
            "web": {
              "uri": "https://example.com/article",
              "title": "Tiêu đề nguồn"
            }
          }
        ],
        "groundingSupports": [
          {
            "segment": {
              "startIndex": 0,
              "endIndex": 12,
              "text": "Theo các báo cáo gần đây"
            },
            "groundingChunkIndices": [0]
          }
        ]
      }
    }
  ]
}

function addCitations(text, supports = [], chunks = []) {
  const sorted = [...supports].sort(
    (a, b) => (b.segment?.endIndex ?? 0) - (a.segment?.endIndex ?? 0),
  );

  for (const support of sorted) {
    const endIndex = support.segment?.endIndex;
    if (endIndex == null) continue;

    const links = (support.groundingChunkIndices ?? [])
      .map((index) => {
        const source = chunks[index]?.web;
        return source?.uri ? `[${index + 1}](${source.uri})` : null;
      })
      .filter(Boolean);

    if (links.length > 0) {
      text = `${text.slice(0, endIndex)}${links.join('')}${text.slice(endIndex)}`;
    }
  }

  return text;
}