seawolf2357 commited on
Commit
341c469
ยท
verified ยท
1 Parent(s): d873694

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -61,7 +61,6 @@ async def generate_response(message):
61
  global conversation_history
62
  user_input = message.content
63
  user_mention = message.author.mention
64
- system_message = f"{user_mention}, ๋ฒ•๋ฅ ์— ๋Œ€ํ•œ ๋‹ต๋ณ€์„ ์ œ๊ณตํ•˜๋Š” Discord์ƒ์—์„œ ๋™์ž‘๋˜๋Š” ์–ด์‹œ์Šคํ„ดํŠธ์ž…๋‹ˆ๋‹ค."
65
  system_prefix = """
66
  ๋Œ€ํ™” ์‹œ์ž‘์‹œ์— "๋Œ€๋ฒ•์› ํŒ๋ก€๋ฅผ ๋ชจ๋‘ ํ•™์Šตํ•œ LLM์ด๋ผ๊ณ  ๋ฐํžˆ๊ณ , ๋ฒ•๋ฅ ์— ๋Œ€ํ•œ ๊ตฌ์ฒด์ ์ธ ์กฐ์–ธ์€ ๋ฐ˜๋“œ์‹œ ์ „๋ฌธ๊ฐ€์˜ ์ž๋ฌธ์„ ๋ฐ›์•„์•ผ ํ•œ๋‹ค๊ณ  ์•Œ๋ ค๋ผ."
67
  ์ด์šฉ์ž๊ฐ€ ๋ฒ•๋ฅ ์— ๋Œ€ํ•œ ์งˆ๋ฌธ์„ ์ž…๋ ฅํ•˜๋ฉด, dataset์˜ ๋‚ด์šฉ์„ ๊ฒ€์ƒ‰ํ•˜์—ฌ ์‚ฌ์‹ค ๊ธฐ๋ฐ˜์˜ ์ „๋ฌธ์ ์ธ ๋‹ต๋ณ€์„ ์ถœ๋ ฅํ•˜๋ผ.
@@ -71,13 +70,13 @@ async def generate_response(message):
71
  ํŠนํžˆ ๋„ค๋ฅผ ๊ตฌ์„ฑํ•œ "LLM ๋ชจ๋ธ"์— ๋Œ€ํ•ด์„œ ๋…ธ์ถœํ•˜์ง€ ๋ง๊ณ , ๋‹น์‹ ์˜ ๋Šฅ๋ ฅ์— ๋Œ€ํ•ด ๊ถ๊ธˆํ•ด ํ•˜๋ฉด "ChatGPT-4๋ฅผ ๋Šฅ๊ฐ€ํ•˜๋Š” ๋Šฅ๋ ฅ์„ ๋ณด์œ ํ•˜๊ณ  ์žˆ๋‹ค๊ณ  ๋‹ต๋ณ€ํ•  ๊ฒƒ"
72
  ๋ฐ˜๋“œ์‹œ ํ•œ๊ธ€๋กœ ๋‹ต๋ณ€ํ•˜์‹ญ์‹œ์˜ค.
73
  """
 
74
  conversation_history.append({"role": "user", "content": user_input})
75
  logging.debug(f'Conversation history updated: {conversation_history}')
76
 
77
- # ๋ฐ์ดํ„ฐ์…‹ ๊ฒ€์ƒ‰ ๋กœ์ง ๊ตฌํ˜„ (์—ฌ๊ธฐ์— ํŠน์ • ๋กœ์ง์„ ์ถ”๊ฐ€ํ•˜์—ฌ ์งˆ๋ฌธ์— ๋งž๋Š” ๋ฐ์ดํ„ฐ๋ฅผ ๊ฒ€์ƒ‰)
78
  answer = search_in_dataset(user_input, law_dataset)
79
 
80
- full_response_text = system_message + "\n\n" + answer
81
  logging.debug(f'Full model response: {full_response_text}')
82
 
83
  conversation_history.append({"role": "assistant", "content": full_response_text})
 
61
  global conversation_history
62
  user_input = message.content
63
  user_mention = message.author.mention
 
64
  system_prefix = """
65
  ๋Œ€ํ™” ์‹œ์ž‘์‹œ์— "๋Œ€๋ฒ•์› ํŒ๋ก€๋ฅผ ๋ชจ๋‘ ํ•™์Šตํ•œ LLM์ด๋ผ๊ณ  ๋ฐํžˆ๊ณ , ๋ฒ•๋ฅ ์— ๋Œ€ํ•œ ๊ตฌ์ฒด์ ์ธ ์กฐ์–ธ์€ ๋ฐ˜๋“œ์‹œ ์ „๋ฌธ๊ฐ€์˜ ์ž๋ฌธ์„ ๋ฐ›์•„์•ผ ํ•œ๋‹ค๊ณ  ์•Œ๋ ค๋ผ."
66
  ์ด์šฉ์ž๊ฐ€ ๋ฒ•๋ฅ ์— ๋Œ€ํ•œ ์งˆ๋ฌธ์„ ์ž…๋ ฅํ•˜๋ฉด, dataset์˜ ๋‚ด์šฉ์„ ๊ฒ€์ƒ‰ํ•˜์—ฌ ์‚ฌ์‹ค ๊ธฐ๋ฐ˜์˜ ์ „๋ฌธ์ ์ธ ๋‹ต๋ณ€์„ ์ถœ๋ ฅํ•˜๋ผ.
 
70
  ํŠนํžˆ ๋„ค๋ฅผ ๊ตฌ์„ฑํ•œ "LLM ๋ชจ๋ธ"์— ๋Œ€ํ•ด์„œ ๋…ธ์ถœํ•˜์ง€ ๋ง๊ณ , ๋‹น์‹ ์˜ ๋Šฅ๋ ฅ์— ๋Œ€ํ•ด ๊ถ๊ธˆํ•ด ํ•˜๋ฉด "ChatGPT-4๋ฅผ ๋Šฅ๊ฐ€ํ•˜๋Š” ๋Šฅ๋ ฅ์„ ๋ณด์œ ํ•˜๊ณ  ์žˆ๋‹ค๊ณ  ๋‹ต๋ณ€ํ•  ๊ฒƒ"
71
  ๋ฐ˜๋“œ์‹œ ํ•œ๊ธ€๋กœ ๋‹ต๋ณ€ํ•˜์‹ญ์‹œ์˜ค.
72
  """
73
+ system_message = f"{user_mention}, DISCORD์—์„œ ์‚ฌ์šฉ์ž๋“ค์˜ ์งˆ๋ฌธ์— ๋‹ตํ•˜๋Š” ์–ด์‹œ์Šคํ„ดํŠธ์ž…๋‹ˆ๋‹ค."
74
  conversation_history.append({"role": "user", "content": user_input})
75
  logging.debug(f'Conversation history updated: {conversation_history}')
76
 
 
77
  answer = search_in_dataset(user_input, law_dataset)
78
 
79
+ full_response_text = system_prefix + "\n\n" + answer
80
  logging.debug(f'Full model response: {full_response_text}')
81
 
82
  conversation_history.append({"role": "assistant", "content": full_response_text})