apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-28 10:26:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix invoice_ocr

Browse source
This commit is contained in:
better629 2023-12-22 20:25:19 +08:00 • committed by geekan
parent d02692e945
commit dc2a87ce12
4 changed files with 30 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/metagpt/roles/test_invoice_ocr_assistant.py
View file

@ -12,7 +12,7 @@ from pathlib import Path
import pandas as pd
import pytest
from metagpt.roles.invoice_ocr_assistant import InvoiceOCRAssistant
from metagpt.roles.invoice_ocr_assistant import InvoiceOCRAssistant, InvoicePath
from metagpt.schema import Message
@ -55,7 +55,7 @@ async def test_invoice_ocr_assistant(
):
invoice_path = Path.cwd() / invoice_path
role = InvoiceOCRAssistant()
await role.run(Message(content=query, instruct_content={"file_path": invoice_path}))
await role.run(Message(content=query, instruct_content=InvoicePath(file_path=invoice_path)))
invoice_table_path = Path.cwd() / invoice_table_path
df = pd.read_excel(invoice_table_path)
dict_result = df.to_dict(orient="records")