From 9607059392cb964c7af1a0cfc8332db12daa65be Mon Sep 17 00:00:00 2001 From: better629 Date: Fri, 22 Dec 2023 21:06:44 +0800 Subject: [PATCH] fix invoice_ocr --- metagpt/roles/invoice_ocr_assistant.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/metagpt/roles/invoice_ocr_assistant.py b/metagpt/roles/invoice_ocr_assistant.py index bd60c43c8..84e354c0e 100644 --- a/metagpt/roles/invoice_ocr_assistant.py +++ b/metagpt/roles/invoice_ocr_assistant.py @@ -7,6 +7,7 @@ @File : invoice_ocr_assistant.py """ +import json from pathlib import Path from typing import Optional @@ -24,7 +25,7 @@ class InvoicePath(BaseModel): class OCRResults(BaseModel): - ocr_results: list = [] + ocr_result: str = "[]" class InvoiceData(BaseModel): @@ -88,10 +89,10 @@ class InvoiceOCRAssistant(Role): self._rc.todo = None content = INVOICE_OCR_SUCCESS - resp = OCRResults(ocr_results=resp) + resp = OCRResults(ocr_result=json.dumps(resp)) elif isinstance(todo, GenerateTable): ocr_results: OCRResults = msg.instruct_content - resp = await todo.run(ocr_results.ocr_results, self.filename) + resp = await todo.run(json.loads(ocr_results.ocr_result), self.filename) # Convert list to Markdown format string df = pd.DataFrame(resp)