apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-05 14:55:18 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix invoice_ocr

Browse source
This commit is contained in:
better629 2023-12-22 21:06:44 +08:00 • committed by geekan
parent 7a1252e356
commit 9607059392
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
metagpt/roles/invoice_ocr_assistant.py
View file

@ -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)