[Feature Request]: Robust block parsing in LLMExtractionStrategy.extract

[NiklasLue]

What needs to be done?

When LLMs return only one block of information, they do not put it in a list, i.e. they return a dictionary.
Once this block is parsed, the blocks are not a list as expected and an error is raised. One approach to make parsing more robust is to check if the blocks object is a dictionary (with the right structure). If so, put the single block into a list and continue.

What problem does this solve?

Errors raised due to faulty parsing.

Target users/beneficiaries

Developers

Current alternatives/workarounds

Patch the function manually

Proposed approach

In the response parsing of the extract method of the LLMExtractionStrategy add a check for dictionaries, e.g. like this (MVP)

        try:
            response_text = response.choices[0].message.content
            blocks = None

            if self.force_json_response:
                blocks = json.loads(response_text)
                if isinstance(blocks, dict):
                    # If it has only one key which calue is list then assign that to blocks, exampled: {"news": [..]}
                    if len(blocks) == 1 and isinstance(list(blocks.values())[0], list):
                        blocks = list(blocks.values())[0]
                    else:
                        # If it has only one key which value is not list then assign that to blocks, exampled: { "article_id": "1234", ... }
                        blocks = [blocks]
                elif isinstance(blocks, list):
                    # If it is a list then assign that to blocks
                    blocks = blocks
            else:
                # blocks = extract_xml_data(["blocks"], response.choices[0].message.content)["blocks"]
                blocks = extract_xml_data(["blocks"], response_text)["blocks"]
                blocks = json.loads(blocks)

            if isinstance(blocks, dict):
                blocks = [blocks]  # If it is a dict, convert it to a list with one item

            for block in blocks:
                block["error"] = False
        except Exception:
            parsed, unparsed = split_and_parse_json_objects(response.choices[0].message.content)
            blocks = parsed
            if unparsed:
                blocks.append({"index": 0, "error": True, "tags": ["error"], "content": unparsed})