Elasticsearch 开放 inference API 增加了对 Azure OpenAI 嵌入的支持

from elasticsearch import Elasticsearch
from elasticsearch_inference.inference_modules import OpenAIEmbeddingModule
 
# 连接到Elasticsearch
es = Elasticsearch("https://your-elasticsearch-endpoint:443")
 
# 创建OpenAI嵌入模块实例
openai_embedding_module = OpenAIEmbeddingModule(
    es,
    model_name="text-embedding-ada-002",  # 使用OpenAI的text-embedding-ada-002模型
    field="content"  # 要嵌入的文本字段名
)
 
# 注册模块
es.inference.register(openai_embedding_module)
 
# 使用模块进行操作
es.inference.embedding(
    index="your_index",  # 指定索引
    document_type="_doc",  # 指定文档类型
    body={
        "content": "Elasticsearch is a distributed search and analytics engine."  # 待嵌入的文本
    }
)

这段代码展示了如何在Elasticsearch中使用OpenAI的嵌入模块。首先，我们创建了一个Elasticsearch客户端连接到Elasticsearch服务。然后，我们创建了一个OpenAI嵌入模块的实例，并指定了要使用的OpenAI模型和文本字段。接下来，我们将这个模块注册到Elasticsearch中，并使用它来对特定索引中的文档进行嵌入操作。