🌠 솔라 프로2 모델 사용

API_KEY = os.getenv("UPSTAGE_API_KEY") 
API_URL = "https://api.upstage.ai/v1/solar/chat/completions"

def call_solar_pro(system_prompt, user_message):
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    payload = {
        "model": "solar-pro2",
        "messages": [
            {"role": "system", "content": system_prompt},
            {"role": "user", "content": user_message} 
        ],
        "temperature": 0.7,
        "max_tokens": 512 
    }
    response = requests.post(API_URL, headers=headers, data=json.dumps(payload))
    if response.status_code == 200:
        return response.json()
    else:
        print(f"Error: {response.status_code}")
        print(response.text)
        return None

gpt 모델에 비해서 답변이 조금 더 신선하다.