【技術解説】AI API プログラミング【キャラAIバトルロイヤル】

########################################################################
# Claude用APIサンプルプログラム (Python標準ライブラリ版)
########################################################################

########################################################################
# インポート
########################################################################
import urllib.request
import json
import re

########################################################################
# 設定（APIキーとモデル）
########################################################################
sAPI_KEY    = "あなたのAPIキーをここに貼り付け"
sMODEL_NAME = "claude-sonnet-4-5" # AIモデル名
sURL        = "https://api.anthropic.com/v1/messages"

########################################################################
# AIにプロンプトを渡して回答文字列を取得
#   arguments:
#     sPromptText .. AIへの指示テキスト文字列
#   return:
#     回答文字列
########################################################################
def getAIResponse( sPromptText:str ):

	# 1. ヘッダーの作成
	headers = {
		"Content-Type":"application/json",
		"x-api-key":sAPI_KEY,
		"anthropic-version": "2023-06-01"
	}

	# 2. 送信データ(JSON)の作成
	data = {
		"model": sMODEL_NAME,
		"max_tokens": 1024,
		"messages": [
		        {
				"role":"user",
				"content":sPromptText
			}
		]
	}

	# 3. リクエストの送信
	oRequest = urllib.request.Request(
		sURL,
		headers=headers,
		data=json.dumps(data).encode('utf-8')
	)
	try:
		with urllib.request.urlopen(oRequest) as oResponse:
			sResponseBody = oResponse.read().decode('utf-8')
			oResponseJson = json.loads(sResponseBody)
		# コンテンツの取り出し
		sContent = oResponseJson["content"][0]["text"]
		return sContent
	except urllib.error.HTTPError as e:
		print( f"Error: {e.read().decode('utf-8')}" )
		return None

########################################################################
# AIの回答文字列からJSON文字列を抽出
#   arguments:
#     sContent .. AI回答文字列
#   return:
#     JSON文字列
########################################################################
def extractJson( sContent:str ) -> str:
	sContent = re.sub( r'^.*?```json\s*' , '' , sContent , flags=re.DOTALL )
	sContent = re.sub( r'\s*```.*$' , '' , sContent , flags=re.DOTALL )
	return sContent

########################################################################
# 実行
########################################################################
if __name__ == "__main__":
	sPrompt = "RPGの敵キャラを1体考えて。結果はJSON形式だけで返して。"

	# APIを呼ぶ
	sRawText = getAIResponse(sPrompt)

	if sRawText:
		print("--- AIの生の回答 ---")
		print(sRawText)
		print("\n--- JSON抽出後 ---")
		sJsonText = extractJson(sRawText)
		print(sJsonText)