EXPERIMENT CODE UPDATED

src/playground/main_chat.py

@@ -1,3 +1,4 @@
+"""
 from openai import OpenAI
 import random
 import common.info
@@ -8,6 +9,7 @@
 PROMPT = common.info.getPrompt("playground_chat")
 KEY = getKey('OPENAI')
 MODEL = common.info.getModelName("Chat")
+MODEL_4o = common.info.getModelName("Chat_4o")
 NAME = "홍길동"
 EMERGENCY = ["요금체납", "주거위기", "고용위기", "급여/서비스 탈락", "긴급상황 위기", "건강위기", "에너지위기"]
 END_SIGNAL = "종료하겠습니다"
@@ -54,4 +56,102 @@
     sheetname = names[name]  # + ("_Good" if gb == 1 else "_Bad")
     applyChat("Chat model", sheetname, conversation_history)
     applyChat("Chat model(Backup)", sheetname, conversation_history)
-    # makeAssistFile(PROMPT, conversation_history)
+    # makeAssistFile(PROMPT, conversation_history)
+"""
+# TEMP CODE FOR EXPERIMENT
+from openai import OpenAI
+import random
+import common.info
+from common.auth_ import getKey
+from saveFile import makeAssistFile, applyChat
+from client import send_request_with_history
+
+PROMPT = common.info.getPrompt("playground_chat")
+KEY = getKey('OPENAI')
+MODEL_mini = common.info.getModelName("Chat")
+MODEL_4o = common.info.getModelName("Chat_4o")
+MODEL_base = 'gpt-4o-mini'
+NAME = "홍길동"
+EMERGENCY = ["요금체납", "주거위기", "고용위기", "급여/서비스 탈락", "긴급상황 위기", "건강위기", "에너지위기"]
+END_SIGNAL = "종료하겠습니다"
+
+client = OpenAI(api_key=KEY)
+
+# 위기유형 랜덤 선택 (한 번만 수행)
+numbers = [1, 2, 3, 3, 3, 3, 3, 3, 3, 4]
+EMERGENCY_COUNT = random.sample(numbers, 1)[0]
+EMERGENCY_LIST = random.sample(EMERGENCY, EMERGENCY_COUNT)
+
+# 초기 입력 생성
+first_ment = f"{NAME}: " + ", ".join(EMERGENCY_LIST)
+
+# mini와의 대화
+conversation_history_model = [
+    {
+        "role": "system",
+        "content": PROMPT
+    }
+]
+
+print("아무것도 입력하지 않고 전송 시 종료됩니다.\n")
+
+print("=== mini와의 대화 ===")
+print(f"You: {first_ment}")
+
+response_model = send_request_with_history(client, conversation_history_model, first_ment, MODEL_mini)
+print("mini:", response_model)
+
+while True:
+    user_input = input("You: ")
+    if user_input == '':
+        break
+    response_model = send_request_with_history(client, conversation_history_model, user_input, MODEL_mini)
+    print("mini:", response_model)
+    if END_SIGNAL in response_model:
+        break
+
+# 4o와의 대화
+conversation_history_model_4o = [
+    {
+        "role": "system",
+        "content": PROMPT
+    }
+]
+
+print("\n=== 4o와의 대화 ===")
+print(f"You: {first_ment}")
+
+response_model_4o = send_request_with_history(client, conversation_history_model_4o, first_ment, MODEL_4o)
+print("4o:", response_model_4o)
+
+while True:
+    user_input = input("You: ")
+    if user_input == '':
+        break
+    response_model_4o = send_request_with_history(client, conversation_history_model_4o, user_input, MODEL_4o)
+    print("4o:", response_model_4o)
+    if END_SIGNAL in response_model_4o:
+        break
+
+# base와의 대화
+conversation_history_model_base = [
+    {
+        "role": "system",
+        "content": PROMPT
+    }
+]
+
+print("\n=== base와의 대화 ===")
+print(f"You: {first_ment}")
+
+response_model_base = send_request_with_history(client, conversation_history_model_base, first_ment, MODEL_base)
+print("base:", response_model_base)
+
+while True:
+    user_input = input("You: ")
+    if user_input == '':
+        break
+    response_model_base = send_request_with_history(client, conversation_history_model_base, user_input, MODEL_base)
+    print("base:", response_model_base)
+    if END_SIGNAL in response_model_base:
+        break