Would like have the ability to fix json output when streaming from a structure

[shhlife]

When streaming output from a structure that's using structured output, it would be helpful to ensure the output data is valid.

For example:

from griptape.structures import Agent
from griptape.utils import Stream
import schema
agent = Agent(stream=True, 
              output_schema=schema.Schema({"item": str, "description": str})
              )
full_result = ""
for artifact in Stream(agent).run("give me a cookie name and its description"):
    full_result += artifact.value
    print(full_result)

will print out partial strings as the chunks come back

{"
{"item
{"item":"
{"item":"**
{"item":"**C
{"item":"**Cinnamon
{"item":"**Cinnamon Sw
{"item":"**Cinnamon Swirl
{"item":"**Cinnamon Swirl Delight
{"item":"**Cinnamon Swirl Delight**
{"item":"**Cinnamon Swirl Delight**","
{"item":"**Cinnamon Swirl Delight**","description
{"item":"**Cinnamon Swirl Delight**","description":"
{"item":"**Cinnamon Swirl Delight**","description":"A
{"item":"**Cinnamon Swirl Delight**","description":"A delightful

it's cool seeing this, but unfortunately the results aren't clean json.

If we integrate the library json-repair, we can make it output cleaner results..

from griptape.structures import Agent
from griptape.utils import Stream
from json_repair import repair_json # json_repair
import schema
agent = Agent(stream=True, 
              output_schema=schema.Schema({"item": str, "description": str})
              )
full_result = ""
for artifact in Stream(agent).run("give me a cookie name and a short description"):
    full_result += artifact.value
    print(repair_json(full_result)) # repair it

and we get these results:

""
{}
{}
{"item": ""}
{"item": "Moon"}
{"item": "Moonlit"}
{"item": "Moonlit Almond"}
{"item": "Moonlit Almond Delight"}
{"item": "Moonlit Almond Delight"}
{"item": "Moonlit Almond Delight"}
{"item": "Moonlit Almond Delight", "description": ""}
{"item": "Moonlit Almond Delight", "description": "A"}

can we add this as part of Stream?

[collindutter]

Once Loaders are Driver-based, this library would fit great into JsonParser.parse.