external_api_engine_hf.py

import requests
import os  # ← 環境変数の読み込みに必要

API_TOKEN = os.getenv("neko")  # ← Secretsに登録した名前
API_URL = "https://api-inference.huggingface.co/models/rinna/japanese-gpt-neox-3.6b"

headers = {
    "Authorization": f"Bearer {API_TOKEN}"
}

def call_external_api(input_text):
    try:
        payload = {
            "inputs": input_text,
            "parameters": {
                "max_new_tokens": 100,
                "temperature": 0.8,
                "do_sample": True
            }
        }
        response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
        response.raise_for_status()
        generated = response.json()

        if isinstance(generated, list) and "generated_text" in generated[0]:
            return generated[0]["generated_text"]
        else:
            return "[API応答形式が不明です]"
    except Exception as e:
        return f"[Hugging Face APIエラー] {str(e)}"