跳到主要內容

【Azure OpenAI】o1 模型與 2024-09-01-preview API

距離上篇在 Early Access Playground 試用 o1 模型後又過了兩週,今天終於等到 API 開放使用啦!本篇將紀錄如何使用 Python SDK 存取 o1 模型。 系列文章 【Azure OpenAI】快速試用 o1 模型 模型佈署 在先前開放的 Early Access Playground 中使用 o1 是不需要另外佈署模型的,不過回到使用 API 來存取 o1 模型,就需要像之前的模型一樣先進行佈署才能使用,相信大家都很熟悉了。 使用 Python SDK 一樣使用熟悉的 openai 套件: 2024-09-01-preview 初始化的方式與先前模型都一樣,需要注意的是 o1 模型目前只能使用最新的 API 版本 2024-09-01-preview 來訪問。 Chat Completions 將 model 填入 o1-preview ,或是你的模型佈署名稱, messages 也一樣是歷史對話堆疊的 List。 回應如下: 查看 Token 使用量 內建 Chain of Thought 的 o1 比起過往的模型會消耗較多的 Token,因此我們特別把 Token 使用量拉出來看。 回應如下: 其中 prompt_tokens 、 completion_tokens 、 total_tokens 在先前的 API 就已經存在了,分別代表Token 的 Input、Output 與總使用量,而在新的 completion_tokens_details 中可以看到  reasoning_tokens 使用了 320 個 Tokens,居然佔了總輸出 Token 的 80% 以上! 控制 Token 成本 已往我們可以使用  max_tokens 參數來控制 Token 的用量,但在 o1 模型中棄用了 max_tokens ,取而代之的是使用  max_completion_tokens 參數,來看看這段程式碼: 回應如下: 沒東西?那再看一次 Token 量。 回應如下: Token 居然是有被使用的! 這表示 max_completion_tokens 並不像過往使用  max_tokens 這麼簡單,先前在回應遇到...

【Azure Cognitive Search】使用 Python SDK 在認知搜尋中進行向量搜尋(上)

認知搜尋 Azure Cognitive Search,在需要限縮 ChatGPT 回答內容的對話機器人情境中,Cognitive Search 可以作為其中最關鍵的「知識庫」角色,其 PaaS 服務的方便性加上後期與 Azure OpenAI 的整合,可說是 2023 年最受益於 ChatGPT 的 Azure 服務之一。

因應 Cognitive Search 在 10 月份對向量搜尋的最新版更新,本篇參考以下 GitHub 範例程式實作與心得分享;接續前篇由快速入門 Cognitive Search 的角度出發,一樣會使用 Python SDK 建立 Cognitive Search 的 Index ,而測試資料的向量會由 Azure OpenAI 的 Embedding API 產生後,與資料一同存入 Index 中,最後會在 Index 中進行向量搜尋。

完整程式碼於文末 GitHub 連結提供!

相關連結:https://github.com/Azure/cognitive-search-vector-pr/tree/main/demo-python

必要條件

  • 在任何區域或任何層級上的認知搜尋服務 Azure Cognitive Search,記得先複製好金鑰備用

  • Cognitive Search 的官方 Python SDK:azure-search-documents 的 11.4.0b11 版

以上項目如果不知道如何進行,建議可由下方連結的前文開始閱讀,參考如何建立需要的服務與環境。

相關連結:【Azure Cognitive Search】使用 Python SDK 在認知搜尋中進行全文檢索

測試資料

繼續借用以下官方放在 GitHub 上的範例資料,資料中包含了 108 項 Azure 服務的名稱與簡介,並存放於 Json 格式中。

相關連結:https://github.com/Azure/cognitive-search-vector-pr/blob/main/demo-python/data/text-sample.json

建立 Index

我們預計會對測試資料中的 title 與 content 兩欄位各自產生向量,所以在這次建立的 Index 中會新增 titleVector 與 contentVector 兩個欄位,用於存放這些向量。

首先使用以下程式與 Cognitive Search 建立連線,記得將 endpoint 與 credential 替換成自己的。

from azure.search.documents.indexes import SearchIndexClient
from azure.core.credentials import AzureKeyCredential

search_index_client = SearchIndexClient(
    endpoint="https://charlie-test.search.windows.net",
    credential=AzureKeyCredential("YOUR-SEARCH-API-KEY") # 你的 Cognitive Search 金鑰
)

建立 Index 的方式基本上與前文相同,一樣都是透過 Field 設定每個欄位:

from azure.search.documents.indexes.models import SimpleField, SearchableField, SearchFieldDataType, SearchField
fields = [
    SimpleField(name="id", type=SearchFieldDataType.String, key=True, sortable=True, filterable=True, facetable=True),
    SearchableField(name="title", type=SearchFieldDataType.String),
    SearchableField(name="content", type=SearchFieldDataType.String),
    SearchableField(name="category", type=SearchFieldDataType.String, filterable=True),
    SearchField(name="titleVector", type=SearchFieldDataType.Collection(SearchFieldDataType.Single), vector_search_dimensions=1536, vector_search_profile="HnswProfile"),
    SearchField(name="contentVector", type=SearchFieldDataType.Collection(SearchFieldDataType.Single), vector_search_dimensions=1536, vector_search_profile="HnswProfile"),
]

前四個欄位與之前進行全文檢索時差不多,而 titleVector 與 contentVector 兩個新欄位說明一下:

  • 向量欄位的型態設定必須固定為type=SearchFieldDataType.Collection(SearchFieldDataType.Single),即單精度的浮點數向量

  • 使用 vector_search_dimensions 設定存放的向量維度,Azure OpenAI 的 text-embedding-ada-002 模型所產生的向量維度即為 1536

  • vector_search_profile 指定要由哪一種演算法來編製 Index,組態名必須要跟後續提供的設定一致

再來還需要一些向量搜尋的設定檔,因為比較繁雜一點,這邊分為三部份說明:

Algorithm

from azure.search.documents.indexes.models import (
    HnswVectorSearchAlgorithmConfiguration, VectorSearchAlgorithmKind, HnswParameters, ExhaustiveKnnVectorSearchAlgorithmConfiguration, ExhaustiveKnnParameters
)
algorithms=[
    HnswVectorSearchAlgorithmConfiguration(
        name="Hnsw",
        kind=VectorSearchAlgorithmKind.HNSW,
        parameters=HnswParameters(
            m=4,
            ef_construction=400,
            ef_search=500,
            metric="cosine",
        )
    ),
    ExhaustiveKnnVectorSearchAlgorithmConfiguration(
        name="ExhaustiveKnn",
        kind=VectorSearchAlgorithmKind.EXHAUSTIVE_KNN,
        parameters=ExhaustiveKnnParameters(
            metric="cosine",
        )
    )
]

algorithms 由多個演算法組態組成,目前在這版本中 Cognitive Search 提供 HNSW 與 Exhaustive KNN 兩種演算法:

  • HNSW:ANN 算法的其中一種,在編制 Index 時會產生額外的資料結構來加快搜尋速度,希望精確度與計算效率之間取得平衡,適用於大部分情況。

  • Exhaustive KNN:基本上是一種暴力演算法,可以找出真正最鄰近的向量,適用於願意以搜尋效能換取高準確度且資料集較小的情境。
其中各自都包含演算法的超參數設定,metric 參數指定計算距離的方式,最常見的設定是餘弦相似度,而其他細項都與演算法本身理論有關,這邊只要儘管用就對了(絕對不是我不會

兩種演算法各有優缺點,大致上就是在速度與準確度上做取捨,兩者也能同時放入設定檔並在執行搜尋時選擇不同演算法,但在編制 Index 時只能選擇其中一種,選擇時有兩點需要注意:

  • 演算法本身是不收費的,但 HNSW 因為有產生額外的資料結構,儲存時會占用較多的空間,也代表可能產生更多的費用,詳細可以了解 Cognitive Search 的計價方式

  • 選擇 HNSW 編制 Index 後兩種演算法都能選擇,但如果選擇了 Exhaustive KNN ,在搜尋時就不能使用 HNSW 了

Vectorizer

from azure.search.documents.indexes.models import AzureOpenAIVectorizer, AzureOpenAIParameters
vectorizers=[
    AzureOpenAIVectorizer(
        name="AzureOpenAI",
        kind="azureOpenAI",
        azure_open_ai_parameters=AzureOpenAIParameters(
            resource_uri="https://xxxxx.openai.azure.com/",  # 你的 Azure OpenAI 端點
            deployment_id="text-embedding-ada-002",
            api_key="OPENAI_API_KEY" # 你的 Azure OpenAI 金鑰
        )
    )  
]

這版的 Cognitive Search 能自動將輸入的文字轉換為向量,在使用上可以少一段呼叫 Embedding 的程式,所以必須在設定中先放入 Azure OpenAI 的各項資訊。

Profile

from azure.search.documents.indexes.models import VectorSearchProfile
profiles=[
    VectorSearchProfile(
        name="HnswProfile",
        algorithm="Hnsw",
        vectorizer="AzureOpenAI"
    ),
    VectorSearchProfile(
        name="ExhaustiveKnnProfile",
        algorithm="ExhaustiveKnn",
        vectorizer="AzureOpenAI"
    )
]

在 Profile 中把各項設定綁定在一起,首先是參數 algorithm 與 vectorizer,對應到上述兩個設定,而 name 對應到設定 Index 欄位時使用的 vector_search_profile,記得檢查互相的命名有沒有一樣。

最後就照著格式把設定通通丟進去來建立 Index:

from azure.search.documents.indexes.models import VectorSearch, SearchIndex

vector_search = VectorSearch(profiles=profiles, algorithms=algorithms, vectorizers=vectorizers)
index = SearchIndex(name="text-sample-vertor", fields=fields, vector_search=vector_search)
result = search_index_client.create_or_update_index(index)

其中 SearchIndex 的 name 可以自行設定 Index 的名稱。

建立完成後就能在 Portal 上看到該 Index,切換到 Cognitive Search 畫面左邊的「索引」頁籤,其中可以看到所有欄位與相關設定;兩個儲存向量的欄位與其他欄位略有不同,但都符合上述的設定。

由 Portal 查看建立完成的 Index

建立向量

回到測試資料中,我們必須要為每一筆資料建立對應的向量,才能開始使用向量搜尋,其中我們會對標題與內容都各自生成向量,最簡單快速的方法就是使用 text-embedding-ada-002,作法可以參考以下的 Azure OpenAI Embedding 使用方式。

而另一個更快的方式,就是直接使用現成的向量😍,參考以下第二個連結。

相關連結:【Azure OpenAI】Azure OpenAI Python SDK 使用方式:Embedding

相關連結:https://github.com/Azure/cognitive-search-vector-pr/blob/main/demo-python/output/docVectors.json

在 Index 中寫入資料

如果選擇了上述更快的方式,基本上已經把寫入 Index 所需要的 Json 格式都整理好了,直接檔案打開後就能用了。

import json
from azure.search.documents import SearchClient
from azure.core.credentials import AzureKeyCredential

with open('docVectors.json', 'r') as file:  
    documents = json.load(file)

search_client = SearchClient(
    endpoint="https://charlie-test.search.windows.net", 
    index_name="text-sample-vertor",
    # credential=AzureKeyCredential("YOUR-SEARCH-API-KEY") # 你的 Cognitive Search 金鑰
)
result = search_client.upload_documents(documents)

寫入完成,一樣回到 Cognitive Search 畫面左邊的「索引」頁籤,稍等幾分鐘後重新整理畫面,就能看到 108 筆資料成功被載入 Index 了!

108 筆資料成功被載入 Index

總結

到此,我們建立了一個可以儲存向量的 Index,並寫入了包含向量欄位在內的 108 筆資料。

下一篇中會在這個 Index 上開始真正執行向量搜尋,在搜尋時快速地切換兩種演算法,還有與全文檢索的混和搜尋方式,都會在下一篇中詳細介紹,請繼續收看!

系列文章

完整程式碼

留言

這個網誌中的熱門文章

【Azure OpenAI】購買 PTU 時微軟不會告訴你的事

Provisioned Throughput Units 一直是目前在 Azure OpenAI 對於延遲問題的最有效解法,同時也是官方最推薦的方案。有別於基本的隨付即用,PTU 具有穩定、可預測的延遲等優勢,適合用於正式上線的生產環境。 但 PTU 的成本是一個不可忽視的問題,儘管選購最小單位量的 PTU,也是需要應用到達一定規模後才看得出使用效益。在確認是否購買 PTU 時,除了詳細閱讀官方文件並使用官方推出的計算機規劃額度外,以下幾點或許也是你該注意的。 不可自行購買 PTU 首先,是的,截自撰文當日 (2024/07/09) PTU 只能透過微軟業務窗口洽詢購買細節,這可能對於多數用戶是不友善的,但我相信這個過程很快就能得到優化。 相關連結: https://learn.microsoft.com/en-us/azure/ai-services/openai/concepts/provisioned-throughput#how-do-i-get-access-to-provisioned PTU 的可用區域僅供參考 在官方文件中記錄了下述表格,其中詳細的呈現各種模型的 PTU 在不同地區的可用性,但這張表只是一個參考,因為當你洽詢業務窗口時你會得到另一張不同的表格。 其中對於台灣用戶可能最有影響的,是我們沒辦法在日本東部購買 gpt-4o 模型的 PTU,對於想透過購買 PTU 以降低模型延遲的用戶來說這是一個矛盾的選擇,當然更不用提隨之產生的跨區傳輸量成本。 一樣截至撰文為止,為何在打勾區域✅無法購買的問題,官方並沒有給出任何理由,或許是我們採購量沒有達到官方需要解釋的程度😔 相關連結: https://learn.microsoft.com/en-us/azure/ai-services/openai/concepts/provisioned-throughput#what-models-and-regions-are-available-for-provisioned-throughput 有別於你認知的定價策略 承如前述,PTU 的成本絕對是導入時的重要考量點。 PTU 的售價到底是多少 事實上 PTU 的官方售價早就是公開的秘密,稱之為秘密是因為 PTU 的售價截至目前並沒有被列在官方文件或定價計算機中,但在最新的 Azure OpenAI S...

【Azure OpenAI】o1 模型與 2024-09-01-preview API

距離上篇在 Early Access Playground 試用 o1 模型後又過了兩週,今天終於等到 API 開放使用啦!本篇將紀錄如何使用 Python SDK 存取 o1 模型。 系列文章 【Azure OpenAI】快速試用 o1 模型 模型佈署 在先前開放的 Early Access Playground 中使用 o1 是不需要另外佈署模型的,不過回到使用 API 來存取 o1 模型,就需要像之前的模型一樣先進行佈署才能使用,相信大家都很熟悉了。 使用 Python SDK 一樣使用熟悉的 openai 套件: 2024-09-01-preview 初始化的方式與先前模型都一樣,需要注意的是 o1 模型目前只能使用最新的 API 版本 2024-09-01-preview 來訪問。 Chat Completions 將 model 填入 o1-preview ,或是你的模型佈署名稱, messages 也一樣是歷史對話堆疊的 List。 回應如下: 查看 Token 使用量 內建 Chain of Thought 的 o1 比起過往的模型會消耗較多的 Token,因此我們特別把 Token 使用量拉出來看。 回應如下: 其中 prompt_tokens 、 completion_tokens 、 total_tokens 在先前的 API 就已經存在了,分別代表Token 的 Input、Output 與總使用量,而在新的 completion_tokens_details 中可以看到  reasoning_tokens 使用了 320 個 Tokens,居然佔了總輸出 Token 的 80% 以上! 控制 Token 成本 已往我們可以使用  max_tokens 參數來控制 Token 的用量,但在 o1 模型中棄用了 max_tokens ,取而代之的是使用  max_completion_tokens 參數,來看看這段程式碼: 回應如下: 沒東西?那再看一次 Token 量。 回應如下: Token 居然是有被使用的! 這表示 max_completion_tokens 並不像過往使用  max_tokens 這麼簡單,先前在回應遇到...

【Azure OpenAI】快速試用 o1 模型

在 OpenAI 與 Azure OpenAI 同時發佈 o1 系列模型的一週後,我也順利通過 Azure OpenAI 的使用申請啦!本篇就來快速試用一下最新的o1 系列模型。 提出申請 目前如果要使用 o1 系列模型都需要經過微軟的資格審查,申請表單可以參考以下連結,表單只需要填寫一份,申請通過後 o1-preview 和 o1-mini 兩個模型都能使用。 相關連結: https://aka.ms/oai/modelaccess 使用 AI Studio 首先你必須要有一個位於 美東 2 地區的 Azure OpenAI 資源,不管是原有的或是新建立的資源都可以。 因為目前 o1 系列模型還處於早期訪問階段,資源中不需要自行佈署模型,取而代之的是需要透過 Early Access Playground 才能使用到 o1 系列模型。 而這次比較特別的是只能使用 AI Studio 的 Playground,看得出來微軟要慢慢整併掉 Azure OpenAI Studio 了。 草莓問題 這次就拿近期已經被大家玩爛的草莓問題來測試,在這個問題中我們會詢問 GPT 在「Strawberry」這個單字裡包含了多少個字母「r」,沒錯,這個草莓問題就是這麼簡單無聊,但結果卻出乎意料。 gpt-4o:兩次 gpt-4o 會有非常高的機率回答:兩次,看似如此簡單的問題又能讓 gpt-4o 屢屢回答錯誤,這就是草莓問題出名的原因,大家也可以自己嘗試看看。 o1-preview:三次 反觀加入 Chain of Thought 概念的 o1-preview 就輕鬆解決了這個草莓問題 😂 總結 根據官方資訊,具有 Chain of Thought 的 o1 模型犧牲了回應的即時性,但大幅改善在邏輯與推理類型問題中的表現,同時成本方面 o1-preview 相較 gpt-4o-0806 貴了 6 倍,對於企業來說就需要好好思考是否有適用的情境了,不過現階段還是繼續期待 API 可用的那天。