[feat][abaoxomtieu]: grammar check route

apis/V1/prompts/prompt.py

@@ -100,8 +100,44 @@ def summarizeNclassify_doc_prompt(docs):
 
     You must return in JSON format, field is mandatory:
     {{
-        summary: a concise summary about the document/paper
-        type: must be "Business" or "RnD" based on document content
+        "summary": a concise summary about the document/paper
+        "type": must be "Business" or "RnD" based on document content
+    }}
+"""
+    return prompt
+
+
+def grammar_check_prompt(sentence):
+    prompt = f"""
+    # Introduction: You are an expert in English grammar. You are given a text to check grammar.
+    # System instructions: 
+    0. No yapping
+    1. Read the text carefully.
+    2. Correct the grammar mistakes in the text.
+    3. Return the corrected text.
+    Examples:
+
+    1.
+    Sentence: "She don't know nothing about the new project, and there is many details that needs to be explained."
+    Incorrect: [don't know nothing, is, needs]
+    Correct: [doesn't know anything, are, need]
+    Corrected sentence: "She doesn't know anything about the new project, and there are many details that need to be explained."
+
+    2.
+    Sentence: "He go to the store every day, but he never buy nothing."
+    Incorrect: [go, buy nothing]
+    Correct: [goes, buys anything]
+    Corrected sentence: "He goes to the store every day, but he never buys anything."
+    
+    
+    ***My sentence: {sentence}
+
+    You must return in JSON format, field is mandatory:
+    {{
+        "sentence": my origin sentence, 
+        "incorrect": [list of incorrect words or phrases, it can be repeated], 
+        "correct": [list of correct words or phrases, it can be repeated], 
+        "corrected_sentence": corrected sentence
     }}
 """
     return prompt

apis/V1/routes/chat.py

@@ -1,20 +1,18 @@
 from fastapi import APIRouter, status
 from ..utils.response_fmt import jsonResponseFmt
 from ..configs.thread_config import run_in_thread
-from ..prompts.document_type_clf import chain_type_classify
-from ..prompts.summarize_doc import chain_summarize
-from ..prompts.translate_abstract_BIZ import chain_BIZ
-from ..prompts.translate_abstract_RnD import chain_RnD
 from ..configs.llm_config import llm
 from ..prompts.prompt import (
     classify_document_prompt,
     summarizeNclassify_doc_prompt,
     translate_BIZ_prompt,
     translate_RnD_prompt,
+    grammar_check_prompt,
 )
 from pydantic import BaseModel, Field
 from langchain_core.output_parsers import JsonOutputParser
 import asyncio
+
 router = APIRouter(prefix="/llm", tags=["LLM"])
 
 # @router.post("/classify_document_type", status_code=status.HTTP_200_OK)
@@ -25,8 +23,11 @@ router = APIRouter(prefix="/llm", tags=["LLM"])
 #     result = await run_in_thread(type_classify, text)
 #     return jsonResponseFmt(result)
 
+
 class SummaryNClassifyDocument(BaseModel):
     text: str
+
+
 @router.post("/summary", status_code=status.HTTP_200_OK)
 async def summarize_doc(body: SummaryNClassifyDocument):
     prompt = summarizeNclassify_doc_prompt(body.text)
@@ -43,13 +44,9 @@ class TranslateAbstract(BaseModel):
 
 @router.post("/translate_RnD", status_code=status.HTTP_200_OK)
 async def translate_RnD(body: TranslateAbstract):
-    print("Đã vào translate_RnD")
-    await asyncio.sleep(15)
     result = await run_in_thread(
         llm.invoke, translate_RnD_prompt(body.abstract, body.sentence, body.word)
     )
-    # result = JsonOutputParser().parse(result)
-    print(result)
     return jsonResponseFmt(result)
 
 
@@ -59,7 +56,6 @@ async def translate_BIZ(body: TranslateAbstract):
     result = await run_in_thread(
         translate_BIZ_prompt, body.abstract, body.sentence, body.word
     )
-    # result = JsonOutputParser().parse(result)
     print(result)
     return jsonResponseFmt(result)
 
@@ -77,3 +73,14 @@ async def classify_document_type(body: ClassifyDocumentType):
     result = await run_in_thread(llm.invoke, summarize_doc(body.text))
     result = JsonOutputParser().parse(result)
     return jsonResponseFmt(result)
+
+
+class GrammarCheck(BaseModel):
+    sentence: str = Field(description="Sentence to check", min_length=1)
+
+
+@router.post("grammar_check", status_code=status.HTTP_200_OK)
+async def grammar_check(body: GrammarCheck):
+    result = await run_in_thread(llm.invoke, grammar_check_prompt(body.sentence))
+    result = JsonOutputParser().parse(result)
+    return jsonResponseFmt(result)