在 Elastic 中使用 GPU 推理进行语义搜索

PUT semantic-embeddings
{
 "mappings": {
   "properties": {
     "content": {
       "type": "semantic_text",
       "inference_id": ".elser-2-elastic"
     }
   }
 }
}

POST _reindex?wait_for_completion=false
{
 "source": {
   "index": "bm25-based-index",
   "size": 100
 },
 "dest": {
   "index": "semantic-embeddings"
 }
}

GET semantic-embeddings/_search
{
 "query": {
   "semantic": {
     "field": "content",
     "query": "Which park is most popular?"
   }
 }
}

{
 "took": 168,
 "timed_out": false,
 "_shards": {
   "total": 3,
   "successful": 3,
   "skipped": 0,
   "failed": 0
 },
 "hits": {
   "total": {
     "value": 3,
     "relation": "eq"
   },
   "max_score": 17.655077,
   "hits": [
     {
       "_index": "semantic-embeddings",
       "_id": "ldJ9C5oBYmL2EI9TGdMP",
       "_score": 17.655077,
       "_source": {
         "text": "Rocky Mountain National Park  is one of the most popular national parks in the United States. It receives over 4.5 million visitors annually, and is known for its mountainous terrain, including Longs Peak, which is the highest peak in the park. The park is home to a variety of wildlife, including elk, mule deer, moose, and bighorn sheep. The park is also home to a variety of ecosystems, including montane, subalpine, and alpine tundra. The park is a popular destination for hiking, camping, and wildlife viewing, and is a UNESCO World Heritage Site."
       }
     },
     {
       "_index": "semantic-embeddings",
       "_id": "k9J9C5oBYmL2EI9TGdMP",
       "_score": 12.741274,
       "_source": {
         "text": "Yellowstone National Park is one of the largest national parks in the United States. It ranges from the Wyoming to Montana and Idaho, and contains an area of 2,219,791 acress across three different states. Its most famous for hosting the geyser Old Faithful and is centered on the Yellowstone Caldera, the largest super volcano on the American continent. Yellowstone is host to hundreds of species of animal, many of which are endangered or threatened. Most notably, it contains free-ranging herds of bison and elk, alongside bears, cougars and wolves. The national park receives over 4.5 million visitors annually and is a UNESCO World Heritage Site."
       }
     },
     {
       "_index": "semantic-embeddings",
       "_id": "lNJ9C5oBYmL2EI9TGdMP",
       "_score": 11.813252,
       "_source": {
         "text": "Yosemite National Park is a United States National Park, covering over 750,000 acres of land in California. A UNESCO World Heritage Site, the park is best known for its granite cliffs, waterfalls and giant sequoia trees. Yosemite hosts over four million visitors in most years, with a peak of five million visitors in 2016. The park is home to a diverse range of wildlife, including mule deer, black bears, and the endangered Sierra Nevada bighorn sheep. The park has 1,200 square miles of wilderness, and is a popular destination for rock climbers, with over 3,000 feet of vertical granite to climb. Its most famous and cliff is the El Capitan, a 3,000 feet monolith along its tallest face."
       }
     }
   ]
 }
}