Update api.py

api.py

@@ -307,12 +307,11 @@ def is_valid_response(response):
 
 
 def respond(input_text):
-    # 1) 사용자 입력 기억에 저장 (원하면)
     memory.process_input(input_text)
 
     if "이름" in input_text:
         response = "제 이름은 Flexi입니다."
-        memory.process_input(response)  # 답변도 기억에 추가 가능
+        memory.process_input(response)
         return response
 
     if "누구" in input_text:
@@ -320,21 +319,14 @@ def respond(input_text):
         memory.process_input(response)
         return response
 
-    # 기억에서 유사 문장 꺼내서 프롬프트 만들기
     related_memories = memory.retrieve(input_text, top_k=3)
-    merged_prompt = merge_prompt(input_text, related_memories)
-
-    # 모델로 응답 생성
-    response = generate_text_sample(model, merged_prompt)
+    merged_prompt = memory.merge_prompt(input_text, related_memories)
 
-    # 응답 검증, 안 맞으면 재생성
-    if not is_valid_response(response) or mismatch_tone(input_text, response):
+    for _ in range(3):  # 최대 3번 재시도
         response = generate_text_sample(model, merged_prompt)
-
-    # 최종 응답도 기억에 추가
-    memory.process_input(response)
-
-    return response
+        if is_valid_response(response) and not mismatch_tone(input_text, response):
+            memory.process_input(response)
+            return response
 
 
 @app.get("/generate", response_class=PlainTextResponse)