github/mealie
1
0
Fork 0
mirror of https://github.com/hay-kot/mealie.git synced 2025-08-21 14:03:32 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

fix(parser): add confidence calculation to BruteForceParser

Browse source
This commit is contained in:
Skye Samuels 2025-08-05 22:43:52 +00:00
commit 620ddbcfef
1 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

20
mealie/services/parser_services/ingredient_parser.py
View file

@ -31,8 +31,27 @@ class BruteForceParser(ABCIngredientParser):
async def parse_one(self, ingredient: str) -> ParsedIngredient: async def parse_one(self, ingredient: str) -> ParsedIngredient:
bfi = brute.parse(ingredient, self) bfi = brute.parse(ingredient, self)
confidences = []
qty_conf = 1.0 if bfi.amount else 0.0
unit_conf = 1.0 if bfi.unit else 0.0
food_conf = 1.0 if bfi.food else 0.0
note_conf = 1.0 if bfi.note else 0.0
for conf in [qty_conf, unit_conf, food_conf, note_conf]:
if conf > 0:
confidences.append(conf)
avg_conf = sum(confidences) / len(confidences) if confidences else 1.0
parsed_ingredient = ParsedIngredient( parsed_ingredient = ParsedIngredient(
input=ingredient, input=ingredient,
confidence=IngredientConfidence(
average=avg_conf,
quantity=qty_conf,
unit=unit_conf,
food=food_conf,
comment=note_conf,
),
ingredient=RecipeIngredient( ingredient=RecipeIngredient(
unit=CreateIngredientUnit(name=bfi.unit), unit=CreateIngredientUnit(name=bfi.unit),
food=CreateIngredientFood(name=bfi.food), food=CreateIngredientFood(name=bfi.food),
@ -125,7 +144,6 @@ class NLPParser(ABCIngredientParser):
food, food_conf = self._extract_food(ingredient) food, food_conf = self._extract_food(ingredient)
note, note_conf = self._extract_note(ingredient) note, note_conf = self._extract_note(ingredient)
# average confidence for components which were parsed
confidences: list[float] = [] confidences: list[float] = []
if qty: if qty:
confidences.append(qty_conf) confidences.append(qty_conf)