import requests

def test_ollama_api(server_url, model_name, prompt):
    url = f"{server_url}/api/generate"
    headers = {"Content-Type": "application/json"}
    data = {
        "model": model_name,
        "prompt": prompt
    }

    try:
        response = requests.post(url, headers=headers, json=data)
        response.raise_for_status()  # HTTP 에러가 발생하면 예외 발생
        result = response.json()
        print("응답:", result["response"])
    except requests.exceptions.RequestException as e:
        print("API 요청 중 오류 발생:", e)

# 테스트 실행
server_url = "http://wrt08045.duckdns.org:11434"  # 올라마 API 서버의 URL
model_name = "Llama3.2-Koren"           # 사용할 모델 이름
prompt = "안녕하세요! 올라마 API를 테스트하고 있습니다."

test_ollama_api(server_url, model_name, prompt)