OpenAIのEmbeddingの使い方 – Rainbow Engine

import openai import pandas as pd # OpenAI APIキーをセット openai.api_key = "xxxxxxxxxxxxxxxxxxxxx" # Modelの指定 model = 'text-embedding-ada-002' def get_embedding(text, model=model): text = text.replace("\n", " ") result = openai.Embedding.create( engine=model, input = [text] ) return result['data'][0]['embedding'] def main(): # 入力(prompt)のデータ(PandasのDataFrame)の生成 df = pd.DataFrame({'value': ['apple', 'b