hcltechaiforce/aiforce-rrs

AIForce - 检索与重排序服务 (RRS) 镜像

镜像概述

Retrieval and Reranking Service (RRS) 是为AIForce设计的微服务，核心功能是从OpenSearch检索相关文档，并通过高级重排序算法优化检索结果质量。该服务是提升RAG（检索增强生成）管道及其他AI驱动搜索工作流性能的关键组件。

核心功能和特性

• 文档检索：从OpenSearch高效获取与查询相关的文档
• 高级重排序：应用先进算法对初始检索结果进行二次排序，提升相关性
• 性能优化：专为增强RAG管道性能设计，减少无效信息干扰
• 兼容性：与AIForce生态系统无缝集成，支持各类AI驱动的搜索场景

使用场景和适用范围

• RAG管道：作为检索环节核心组件，为生成式AI提供高质量上下文
• 智能搜索系统：提升企业内部文档、知识库搜索的准确性
• AI辅助决策：为数据分析、研究等场景提供精准信息检索支持
• 内容推荐系统：优化推荐内容的相关性排序

使用方法和配置说明

环境变量配置

Docker部署示例

基础运行命令

bash
docker run -d \
  --name aiforce-rrs \
  -p 8080:8080 \
  -e OPENSEARCH_URL="http://opensearch:9200" \
  -e RERANKING_MODEL="cross-encoder/ms-marco-MiniLM-L-6-v2" \
  docker.xuanyuan.run/aiforce/rrs:latest

Docker Compose配置

yaml
version: '3.8'
services:
  rrs:
    image: docker.xuanyuan.run/aiforce/rrs:latest
    container_name: aiforce-rrs
    ports:
      - "8080:8080"
    environment:
      - OPENSEARCH_URL=http://opensearch:9200
      - OPENSEARCH_USER=admin
      - OPENSEARCH_PASSWORD=admin
      - TOP_K=150
      - RERANK_TOP_N=30
    depends_on:
      - opensearch
  opensearch:
    image: docker.xuanyuan.run/opensearchproject/opensearch:2.11.0
    container_name: opensearch
    environment:
      - discovery.type=single-node
      - OPENSEARCH_INITIAL_ADMIN_PASSWORD=admin
    ports:
      - "9200:9200"

接口调用示例

服务启动后，可通过HTTP POST请求调用检索重排序接口：

bash
curl -X POST http://localhost:8080/rerank \
  -H "Content-Type: application/json" \
  -d '{"query": "AIForce RRS使用方法", "index": "documents"}'

返回结果示例：

json
{
  "query": "AIForce RRS使用方法",
  "results": [
    {"id": "doc1", "content": "...", "score": 0.98},
    {"id": "doc3", "content": "...", "score": 0.92},
    ...
  ]
}