Azure OpenAI embeddings API確認

maito17s

こちらを試してみる。

Visual Studio CodeのRest Clientにて
リスエストと応答一部

@api_key = xxx
@endpoint = https://openai-xxxx.openai.azure.com
@api_version = 2023-08-01-preview
@deployment_name = xxxx

### 埋込みAPI（Chat） - "/embeddings"
POST {{endpoint}}/openai/deployments/{{deployment_name}}/embeddings?api-version={{api_version}}
api-key: {{api_key}}
Content-Type: application/json

{"input": "こんちわ"}

// https://learn.microsoft.com/ja-jp/azure/ai-services/openai/reference#embeddings
HTTP/1.1 20
0 OK
Content-Length: 33432
Content-Type: application/json
access-control-allow-origin: *
x-content-type-options: nosniff
apim-request-id: ea588166-5908-4323-944c-fb1c8e7f4a88
x-request-id: 7e475583-08ec-4a9c-a473-7de5938f1691
x-ms-client-request-id: ea588166-5908-4323-944c-fb1c8e7f4a88
x-ms-region: Japan East
Strict-Transport-Security: max-age=31536000; includeSubDomains; preload
Date: Fri, 03 Nov 2023 11:07:22 GMT
Connection: close

{
  "object": "list",
  "data": [
    {
      "object": "embedding",
      "index": 0,
      "embedding": [
        -0.021363307,
        0.013723982,
        0.0064967778,
        -0.028546954
      ]
    }
  ],
  "model": "ada",
  "usage": {
    "prompt_tokens": 3,
    "total_tokens": 3
  }
}

Pythonコード実装

import openai

OPENAI_API_TYPE = "azure"
OPENAI_API_KEY = ""
OPENAI_SERVICE_NAME = ""
OPENAI_API_VERSION = "2023-07-01-preview"
OPENAI_BASE_URL = f"https://{OPENAI_SERVICE_NAME}.openai.azure.com/"
OPENAI_EMBEDDING_MODEL_NAME = "text-embedding-ada-002"
OPENAI_EMBEDDING_MODEL_DEPLOYMENT = ""

openai.api_type = OPENAI_API_TYPE
openai.api_base = OPENAI_BASE_URL
openai.api_version = OPENAI_API_VERSION
openai.api_key = OPENAI_API_KEY

inputs = ["A", "B", "C"] #max  array size=16

embedding = openai.Embedding.create(
  input=inputs,
  deployment_id=OPENAI_EMBEDDING_MODEL_DEPLOYMENT
  #engine ="text-embedding-ada-002"
)

print(embedding)
{
  "object": "list",
  "data": [
    {
      "object": "embedding",
      "index": 0,
      "embedding": [
        0.015408080071210861,
        -0.013772061094641685,
        0.02419162355363369,
        0.0016184182604774833,
        -0.03226443752646446,
        0.007174345199018717,
        -0.0174598079174757,
        -0.020611155778169632,
        -0.003516770200803876,
        -0.01133144274353981,
        0.005947330966591835,
        0.017526857554912567,
        0.007442545145750046,
        -0.011854431591928005,
        -0.008884118869900703,
        -0.0049717542715370655,
        0.03540237620472908,
        0.005990913603454828,
...
    "prompt_tokens": 3,
    "total_tokens": 3
  }
}

この記事が気に入ったらサポートをしてみませんか？