Coverage for opt/mealie/lib/python3.12/site-packages/mealie/schema/openai/_base.py: 52%

1import re 

2from typing import Self 

3 

4from pydantic import BaseModel 

5 

6from mealie.core.root_logger import get_logger 

7 

8RE_NULLS = re.compile(r"[\x00\u0000]|\\u0000") 

9 

10logger = get_logger() 

11 

12 

13class OpenAIBase(BaseModel): 

14 """ 

15 This class defines the JSON schema sent to OpenAI. Its schema is 

16 injected directly into the OpenAI prompt. 

17 """ 

18 

19 __doc__ = "" # we don't want to include the docstring in the JSON schema 

20 

21 @classmethod 

22 def _preprocess_response(cls, response: str | None) -> str: 

23 if not response: 

24 return "" 

25 

26 response = re.sub(RE_NULLS, "", response) 

27 return response 

28 

29 @classmethod 

30 def _process_response(cls, response: str) -> Self: 

31 try: 

32 return cls.model_validate_json(response) 

33 except Exception: 

34 logger.debug(f"Failed to parse OpenAI response as {cls}. Response: {response}") 

35 raise 

36 

37 @classmethod 

38 def parse_openai_response(cls, response: str | None) -> Self: 

39 """ 

40 Parse the OpenAI response into a class instance. 

41 """ 

42 

43 response = cls._preprocess_response(response) 

44 return cls._process_response(response)