import requests
import json

def call_bedrock_claude_api(message, api_endpoint="https://frv9cm6qs7.execute-api.us-east-1.amazonaws.com/api", 
                           api_key="QZsgWWJjD47lUFkxO1X3d7GkdYtILigJ9Xgb5zW8"):
    try:
        headers = {
            'x-api-key': api_key,
            'Content-Type': 'application/json'
        }
        
        payload = {
            'message': message
        }

        response = requests.post(api_endpoint, headers=headers, json=payload)
        response.raise_for_status()  # Raises an HTTPError for bad responses (4xx, 5xx)
        
        return response.json()

    except requests.exceptions.RequestException as e:
        print(f"API呼び出しエラー: {e}")
        return None

def main():
    # テスト用メッセージ
    test_message = "こんにちは、今日の天気を教えてください。"
    
    result = call_bedrock_claude_api(test_message)
    if result:
        print("API応答:", json.dumps(result, indent=2, ensure_ascii=False))

if __name__ == "__main__":
    main()

"""
curlコマンドでの接続テスト:

curl -X POST \
  'https://frv9cm6qs7.execute-api.us-east-1.amazonaws.com/api' \
  -H 'x-api-key: QZsgWWJjD47lUFkxO1X3d7GkdYtILigJ9Xgb5zW8' \
  -H 'Content-Type: application/json' \
  -d '{"message": "こんにちは、今日の天気を教えてください。"}'
"""