Yuchan5386 commited on
Commit
3a8378f
Β·
verified Β·
1 Parent(s): 87ce794

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +4 -7
api.py CHANGED
@@ -328,6 +328,7 @@ def parse_math_question(text):
328
  except:
329
  return "계산할 수 μ—†λŠ” μˆ˜μ‹μ΄μ—μš”. λ‹€μ‹œ ν•œλ²ˆ 확인해 μ£Όμ„Έμš”!"
330
 
 
331
  def respond(input_text):
332
  intent = simple_intent_classifier(input_text)
333
 
@@ -347,16 +348,12 @@ def respond(input_text):
347
  summary = summarize_from_wikipedia(keyword)
348
  return f"{summary}\nλ‹€λ₯Έ κΆκΈˆν•œ 점 μžˆμœΌμ‹ κ°€μš”?"
349
 
350
- # βœ… κΈ°μ–΅ 기반 병합 μΆ”κ°€
351
- related_memories = memory.retrieve(input_text, top_k=3)
352
- merged_prompt = merge_prompt_with_memory(input_text, related_memories)
353
-
354
- response = generate_text_sample(model, merged_prompt)
355
  if not is_valid_response(response) or mismatch_tone(input_text, response):
356
- response = generate_text_sample(model, merged_prompt)
357
  return response
358
 
359
-
360
  @app.get("/generate", response_class=PlainTextResponse)
361
  async def generate(request: Request):
362
  prompt = request.query_params.get("prompt", "μ•ˆλ…•ν•˜μ„Έμš”")
 
328
  except:
329
  return "계산할 수 μ—†λŠ” μˆ˜μ‹μ΄μ—μš”. λ‹€μ‹œ ν•œλ²ˆ 확인해 μ£Όμ„Έμš”!"
330
 
331
+ # μ΅œμ’… 응닡 ν•¨μˆ˜
332
  def respond(input_text):
333
  intent = simple_intent_classifier(input_text)
334
 
 
348
  summary = summarize_from_wikipedia(keyword)
349
  return f"{summary}\nλ‹€λ₯Έ κΆκΈˆν•œ 점 μžˆμœΌμ‹ κ°€μš”?"
350
 
351
+ # 일상 λŒ€ν™”: μƒ˜ν”Œλ§ + fallback
352
+ response = generate_text_sample(model, input_text)
 
 
 
353
  if not is_valid_response(response) or mismatch_tone(input_text, response):
354
+ response = generate_text_sample(model, input_text)
355
  return response
356
 
 
357
  @app.get("/generate", response_class=PlainTextResponse)
358
  async def generate(request: Request):
359
  prompt = request.query_params.get("prompt", "μ•ˆλ…•ν•˜μ„Έμš”")