Merge pull request #48 from gomate-community/pipeline

Pipeline
gomate-community · Aug 29, 2024 · 3a8d98f · 3a8d98f
2 parents f25d7a1 + 648b354
commit 3a8d98f
Show file tree

Hide file tree

Showing 8 changed files with 478 additions and 184 deletions.
diff --git a/api/apps/core/citation/bodys.py b/api/apps/core/citation/bodys.py
@@ -33,3 +33,11 @@ class CitationBody(BaseModel):
         [1,2],
         description="文档对应的索引"
     )
+    show_code:bool=Field(
+        default=True,
+        description="是否显示引用代码块"
+    )
+    selected_docs:List[dict]=Field(
+        default=[]
+
+    )
diff --git a/api/apps/core/citation/views.py b/api/apps/core/citation/views.py
@@ -25,14 +25,18 @@
 async def citation(citation_body: CitationBody):
     response = citation_body.response
     evidences = citation_body.evidences
-    selected_idx=citation_body.selected_idx
+    selected_idx = citation_body.selected_idx
+    show_code = citation_body.show_code
+    selected_docs=citation_body.selected_docs
     # loguru.logger.info(response)
     # loguru.logger.info(evidences)
     citation_response = mc.ground_response(
         response=response,
         evidences=evidences,
         selected_idx=selected_idx,
-        markdown=True
+        markdown=True,
+        show_code=show_code,
+        selected_docs=selected_docs
     )
     loguru.logger.info(citation_response)
     return ApiResponse(citation_response, message="答案引用成功")
diff --git a/api/apps/core/parser/views.py b/api/apps/core/parser/views.py
@@ -60,8 +60,9 @@ async def parser(file: UploadFile = File(...),chunk_size=512):
             raise NotImplementedError(
                 "file type not supported yet(pdf, xlsx, doc, docx, txt supported)")
         contents = parser.parse(content)
-        loguru.logger.info(contents[0])
+        # loguru.logger.info(contents[0])
         contents = tc.chunk_sentences(contents, chunk_size=512)
+        loguru.logger.info(len(contents))
         # 返回成功响应
         return JSONResponse(content=contents, status_code=200)
     except Exception as e: