seawolf2357 commited on
Commit
dd6eadc
Β·
verified Β·
1 Parent(s): 51ebe4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -30,7 +30,6 @@ class MyClient(discord.Client):
30
  response = await generate_response(message.content)
31
  await message.channel.send(response)
32
 
33
- # Hugging Face μΆ”λ‘  APIλ₯Ό μ‚¬μš©ν•˜μ—¬ 응닡 생성
34
  async def generate_response(user_input):
35
  system_message = "DISCORDμ—μ„œ μ‚¬μš©μžλ“€μ˜ μ§ˆλ¬Έμ— λ‹΅ν•˜λŠ” μ „λ¬Έ AI μ–΄μ‹œμŠ€ν„΄νŠΈ μ—­ν• μž…λ‹ˆλ‹€."
36
  system_prefix = """
@@ -44,12 +43,12 @@ async def generate_response(user_input):
44
 
45
  # 동기 ν•¨μˆ˜λ₯Ό λΉ„λ™κΈ°λ‘œ μ²˜λ¦¬ν•˜κΈ° μœ„ν•œ 래퍼 μ‚¬μš©
46
  loop = asyncio.get_event_loop()
47
- response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(messages, max_tokens=1000, stream=False, temperature=0.7, top_p=0.95))
48
- logging.debug(f'API response: {response}') # API 응닡 λ‘œκΉ…
49
 
50
- # API 응닡 λ‘œκ·Έμ— 따라 content 속성 μ‚¬μš©
51
  return response.choices[0].message.content.strip()
52
 
 
53
  # λ””μŠ€μ½”λ“œ 봇 μΈμŠ€ν„΄μŠ€ 생성 및 μ‹€ν–‰
54
  discord_client = MyClient(intents=intents)
55
  discord_client.run(os.getenv('DISCORD_TOKEN'))
 
30
  response = await generate_response(message.content)
31
  await message.channel.send(response)
32
 
 
33
  async def generate_response(user_input):
34
  system_message = "DISCORDμ—μ„œ μ‚¬μš©μžλ“€μ˜ μ§ˆλ¬Έμ— λ‹΅ν•˜λŠ” μ „λ¬Έ AI μ–΄μ‹œμŠ€ν„΄νŠΈ μ—­ν• μž…λ‹ˆλ‹€."
35
  system_prefix = """
 
43
 
44
  # 동기 ν•¨μˆ˜λ₯Ό λΉ„λ™κΈ°λ‘œ μ²˜λ¦¬ν•˜κΈ° μœ„ν•œ 래퍼 μ‚¬μš©
45
  loop = asyncio.get_event_loop()
46
+ response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(
47
+ messages, max_tokens=1000, stream=False, temperature=1.0, top_p=0.9)) # νŒŒλΌλ―Έν„° μ‘°μ •
48
 
 
49
  return response.choices[0].message.content.strip()
50
 
51
+
52
  # λ””μŠ€μ½”λ“œ 봇 μΈμŠ€ν„΄μŠ€ 생성 및 μ‹€ν–‰
53
  discord_client = MyClient(intents=intents)
54
  discord_client.run(os.getenv('DISCORD_TOKEN'))