VscodeでPrompt Flowを作成する-AOAI接続方法

やること

VscodeでPrompt Flowを構築した。
構築したPrompt FlowをAOAIとの接続を行う。

前提

• VscodeでPromtp Flowのフローを構築済みであること

https://zenn.dev/headwaters/articles/453ee41ab56776

• AOAIを構築済みであること

手順

1. VscodeでPromtp Flowのフローを開く

https://zenn.dev/headwaters/articles/453ee41ab56776

2. 任意のディレクトリ上でyamlファイルを作成する
   ※ファイル名はなんでも良いです。
3. yamlファイルに下記の内容を書く

$schema:https://azuremlschemas.azureedge.net/promptflow/latest/AzureOpenAIConnection.schema.json
name: aoai # なんでもよい
type: azure_open_ai
api_key: "<API_KEY>" # Azure OpenAIリソースのAPIキー
api_base: "<API_BASE>" # Azure OpenAIリソースのベース（エンドポイントURL）
api_type: "azure"
api_version: "2023-05-15"

4. 下記のコマンドを実行

pf connection create -f <path-to-azure-open-ai-connection>

5. AOAIとの接続されたことを確認
   
6. 「flow.dag.yaml」をクリック
7. 「＋」をクリック
   
8. 「LLM」をクリック
   
9. 名前を入力し、Enterを押す
   
10. 「new file」をクリック
    
11. フローに追加されたことを確認
12. connection先を4で作成した接続先に変更
    
13. flow.dag.yamlを下記のように書き換える
    

$schema: https://azuremlschemas.azureedge.net/promptflow/latest/Flow.schema.json
environment:
  python_requirements_txt: requirements.txt
inputs:
  text:
    type: string
    default: Hello
outputs:
  output_prompt:
    type: string
    reference: ${echo_my_prompt.output}
nodes:
- name: hello_prompt
  type: prompt
  source:
    type: code
    path: hello.jinja2
  inputs:
    text: ${inputs.text}
- name: aoai
  type: llm
  source:
    type: code
    path: aoai.jinja2
  inputs:
    connection: aoai
    api: chat
    prompt: ${hello_prompt.output}
    deployment_name: # deploy名を書く
    temperature: 0.5
  connection: aoai
  api: chat
- name: echo_my_prompt
  type: python
  source:
    type: code
    path: hello.py
  inputs:
    input1: ${aoai.output}

14. 下記のフローが作成され、AOAIとも接続ができている
    
15. 「開始ボタン」をクリック
    
16. 結果が返ってくることを確認