openai embedding base search example

openai-embedding-base-search-example

基於OpenAI 的embedding 接口和GPT3.5接口實現的文章搜索實例（當然，大部分代碼是通過ChatGPT 給的，我只做了一些改動封裝）

PS.代碼只是簡單示例，實際使用應該用向量數據庫存儲文章的embedding 數據

流程

• 為所有文章生成embedding 數據（只需要做一次）
• 為用戶的輸入生成embedding 數據
• 通過相似度算法，找出與用戶輸入相似度高的前3 篇文章
• 分割這3 篇文章的所有段落
• 分別把用戶輸入和文章段落給到GPT 模型，如果模型給出肯定回答，則把段落返回輸出

運行

• 需要安裝nodejs v18+
• 在本項目根目錄創建.env文件，設置OpenAPI API KEY 環境變量

OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
#如果需要使用代理访问，请提供下面的环境变量
# https_proxy=http://127.0.0.1:1080

• 安裝依賴npm install
• 運行npm run dev

更多參考

• 基於embedding的領域知識問答：https://github.com/openai/openai-cookbook/blob/main/examples/Question_answering_using_embeddings.ipynb
• 生成OpenAI API KEY：https://platform.openai.com/account/api-keys