Update app.py

app.py

@@ -1,7 +1,8 @@
 import discord
 import logging
 import os
-from openai import OpenAI
+import requests
+import json
 import asyncio
 import subprocess
 
@@ -22,16 +23,17 @@ SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
 conversation_history = []
 
 # API 키 설정 및 정리
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-if not OPENAI_API_KEY:
+API_KEY = os.getenv("OPENAI_API_KEY")  # 기존 API 키를 그대로 사용
+if not API_KEY:
     # 환경 변수가 설정되지 않았을 경우, 여기에 API 키를 직접 입력하세요
-    OPENAI_API_KEY = "your_openai_api_key_here"  # 실제 키로 교체 필요
+    API_KEY = "your_api_key_here"  # 실제 키로 교체 필요
 else:
     # 환경 변수에서 가져온 API 키의 공백 및 줄바꿈 문자 제거
-    OPENAI_API_KEY = OPENAI_API_KEY.strip()
+    API_KEY = API_KEY.strip()
 
-# OpenAI 클라이언트 설정
-openai_client = OpenAI(api_key=OPENAI_API_KEY)
+# Fireworks API 설정
+FIREWORKS_API_URL = "https://api.fireworks.ai/inference/v1/chat/completions"
+FIREWORKS_MODEL = "accounts/fireworks/models/qwen3-30b-a3b"
 
 class MyClient(discord.Client):
     def __init__(self, *args, **kwargs):
@@ -80,7 +82,7 @@ async def generate_response(message):
     logging.debug(f'Conversation history updated: {conversation_history}')
     
     try:
-        # 시스템 메시지와 사용자 입력을 포함한 메시지 생성
+        # 메시지 형식 준비
         messages = [
             {
                 "role": "system", 
@@ -97,22 +99,48 @@ async def generate_response(message):
         
         logging.debug(f'Messages to be sent to the model: {messages}')
         
-        # OpenAI API 호출을 위한 비동기 처리
-        loop = asyncio.get_event_loop()
-        response = await loop.run_in_executor(None, lambda: openai_client.chat.completions.create(
-            model="gpt-4.1-mini",  # 또는 gpt-4.1-mini와 유사한 다른 사용 가능한 모델
-            messages=messages,
-            temperature=0.7,
-            max_tokens=1800,
-            top_p=0.85
-        ))
+        # Fireworks API 요청 페이로드 구성
+        payload = {
+            "model": FIREWORKS_MODEL,
+            "max_tokens": 1800,
+            "top_p": 0.85,
+            "top_k": 40,
+            "presence_penalty": 0,
+            "frequency_penalty": 0,
+            "temperature": 0.7,
+            "messages": messages
+        }
         
-        full_response_text = response.choices[0].message.content
-        logging.debug(f'Full model response: {full_response_text}')
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {API_KEY}"
+        }
         
-        conversation_history.append({"role": "assistant", "content": full_response_text})
+        # 비동기 방식으로 API 호출
+        loop = asyncio.get_event_loop()
+        response = await loop.run_in_executor(
+            None, 
+            lambda: requests.post(
+                FIREWORKS_API_URL, 
+                headers=headers, 
+                data=json.dumps(payload),
+                timeout=60
+            )
+        )
         
-        return f"{user_mention}, {full_response_text}"
+        # 응답 처리
+        if response.status_code == 200:
+            response_json = response.json()
+            full_response_text = response_json["choices"][0]["message"]["content"]
+            logging.debug(f'Full model response: {full_response_text}')
+            
+            conversation_history.append({"role": "assistant", "content": full_response_text})
+            
+            return f"{user_mention}, {full_response_text}"
+        else:
+            logging.error(f"API 응답 오류: {response.status_code} - {response.text}")
+            return f"{user_mention}, 죄송합니다. API 응답 중 오류가 발생했습니다 (상태 코드: {response.status_code})."
         
     except Exception as e:
         logging.error(f"Error in generate_response: {e}")