Skip to content

Commit

Permalink
add response sample to the prompt message
Browse files Browse the repository at this point in the history
  • Loading branch information
kota-yata committed Dec 6, 2023
1 parent b7b59f3 commit 22a837e
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion scripts/add_topic_poc.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,12 @@ def get_topic(client: OpenAI, tweet: str, note: str) -> Dict[str, List[str]]:
```
このセットに対してのトピックは「{" ".join(fewshot_sample["topics"])}」です。
これを踏まえて、以下のツイートとコミュニティノートに対して同じ粒度で複数のトピックを提示してください。形式はJSONで、キーをtopicsとして値にトピックを配列で格納してください。
レスポンスの例:
```
{
"topics": ["医療", "福祉", ...]
}
```
""",
},
{
Expand All @@ -41,7 +47,7 @@ def get_topic(client: OpenAI, tweet: str, note: str) -> Dict[str, List[str]]:
""",
},
],
model="gpt-4",
model="gpt-3.5-turbo",
temperature=0.0,
)

Expand Down

0 comments on commit 22a837e

Please sign in to comment.