Skip to content

Commit 3e9789e

Browse files
fix(serialization): resolve BaseStructureRunDriver to allow for serialization (#1914)
Co-authored-by: Collin Dutter <[email protected]>
1 parent 366af16 commit 3e9789e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

griptape/schemas/base_schema.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,6 +250,7 @@ def _resolve_types(cls, attrs_cls: type, types_override: Optional[dict[str, type
250250
from griptape.drivers.prompt.base_prompt_driver import StructuredOutputStrategy
251251
from griptape.drivers.ruleset import BaseRulesetDriver
252252
from griptape.drivers.sql import BaseSqlDriver
253+
from griptape.drivers.structure_run import BaseStructureRunDriver
253254
from griptape.drivers.text_to_speech import BaseTextToSpeechDriver
254255
from griptape.drivers.vector import BaseVectorStoreDriver
255256
from griptape.drivers.web_scraper import BaseWebScraperDriver
@@ -291,6 +292,7 @@ def _resolve_types(cls, attrs_cls: type, types_override: Optional[dict[str, type
291292
"BaseSqlDriver": BaseSqlDriver,
292293
"BaseObservabilityDriver": BaseObservabilityDriver,
293294
"BaseAssistantDriver": BaseAssistantDriver,
295+
"BaseStructureRunDriver": BaseStructureRunDriver,
294296
"BaseArtifact": BaseArtifact,
295297
"BaseMetaEntry": BaseMetaEntry,
296298
"PromptStack": PromptStack,

0 commit comments

Comments
 (0)