refactor(chain): extract ebbinghaus

Author: codeacme17

server/apis/question.py

@@ -16,7 +16,7 @@ def examine_question(data: types.AnswerQuestion):
     )
 
     return chain.aexamine_answer(
-        id=data.id,
+        quesiton_id=data.id,
         context=document_info["document"],
         question=question_info["content"],
         role=question_info["designated_role"],

server/llm_services/langchain_chain.py

@@ -10,6 +10,7 @@
 
 import db_services as _dbs_
 
+from utils.ebbinghaus import handle_ebbinghaus_memory
 from .langchain_llm import LLM
 from prompts import choose_prompt
 
@@ -80,15 +81,15 @@ async def _agenerate_questions(
 
     async def aexamine_answer(
         self,
-        id: int,
+        quesiton_id: int,
         context: str,
         question: str,
         answer: str,
         role: str,
         question_type: str,
     ):
         llm_chain = self._init_llm_chain(60, role, question_type)
-        coroutine = wait_done(llm_chain.apredict(
+        coroutine = _wait_done(llm_chain.apredict(
             context=context,
             question=question,
             answer=answer,
@@ -110,7 +111,7 @@ async def aexamine_answer(
         score = _extract_score(exmine)
         push_date = _get_push_date(score)
         await _dbs_.question.update_question_state(
-            id=id,
+            id=quesiton_id,
             answer=f"{answer} ||| {exmine}",
             score=score,
             push_date=push_date
@@ -138,7 +139,7 @@ def _init_llm_chain(self, timeout: int, role: str, question_type: str):
         )
 
 
-async def wait_done(
+async def _wait_done(
     fn: Awaitable,
     event: asyncio.Event
 ):
@@ -209,15 +210,5 @@ def _extract_score(anwser: str):
 
 def _get_push_date(score: int):
     now = datetime.datetime.now()
-    days = 0
-
-    if (0 <= score <= 3):
-        days = 1
-    if (4 <= score <= 6):
-        days = 3
-    if (7 <= score <= 9):
-        days = 7
-    if (10 == score):
-        days = 14
-
+    days = handle_ebbinghaus_memory(score)
     return ((now+datetime.timedelta(days)).strftime("%Y-%m-%d"))

server/utils/__init__.py

@@ -1,5 +1,5 @@
 from .file_handler import upload_file
 
 __all__ = [
-    "upload_file"
+    "upload_file",
 ]

server/utils/ebbinghaus.py

@@ -0,0 +1,11 @@
+def handle_ebbinghaus_memory(score: int):
+    if (0 <= score <= 3):
+        days = 1
+    if (4 <= score <= 6):
+        days = 3
+    if (7 <= score <= 9):
+        days = 7
+    if (10 == score):
+        days = 14
+
+    return days