from anthropic import Anthropic
import os
from dotenv import load_dotenv

load_dotenv()

def ask_claude(question, system_prompt, model="claude-3-5-sonnet-20241022"):

    anthropic = Anthropic(
        api_key=os.environ['ANTHROPIC_API_KEY']
    )

    message = anthropic.messages.create(
        model=model,
        max_tokens=8000,
        system=system_prompt,
        messages=[
            {
                "role": "user",
                "content": question
            }
        ]
    )
    
    # 回答テキストのみを返す
    return message.content[0].text