mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-08-06 23:51:15 +08:00
26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
|
import json
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
|
||
|
|
def test_mistral_ocr_factory_present_and_ocr_tagged():
|
||
|
|
repo_root = Path(__file__).resolve().parents[3]
|
||
|
|
data = json.loads((repo_root / "conf" / "llm_factories.json").read_text())
|
||
|
|
factories = {f["name"]: f for f in data["factory_llm_infos"]}
|
||
|
|
assert "Mistral OCR" in factories, "Mistral OCR factory missing"
|
||
|
|
fac = factories["Mistral OCR"]
|
||
|
|
assert "OCR" in fac["tags"]
|
||
|
|
# ships a default OCR model (like the other OCR factories) so it is usable
|
||
|
|
# without manually adding one through the model provider page; the llm_name
|
||
|
|
# is the real Mistral API id because the parser POSTs it verbatim to /v1/ocr.
|
||
|
|
models = {m["llm_name"]: m for m in fac["llm"]}
|
||
|
|
assert "mistral-ocr-latest" in models
|
||
|
|
assert models["mistral-ocr-latest"]["model_type"] == "ocr"
|
||
|
|
assert "OCR" in models["mistral-ocr-latest"]["tags"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_mistral_ocr_factory_distinct_from_mistral():
|
||
|
|
repo_root = Path(__file__).resolve().parents[3]
|
||
|
|
data = json.loads((repo_root / "conf" / "llm_factories.json").read_text())
|
||
|
|
names = [f["name"] for f in data["factory_llm_infos"]]
|
||
|
|
assert "Mistral" in names and "Mistral OCR" in names
|