We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3702571 commit d4bc012Copy full SHA for d4bc012
src/pdl/pdl_granite_io.py
@@ -78,7 +78,10 @@ async def async_generate_text(
78
io_processor = GraniteioModel.processor_of_block(block)
79
inputs = GraniteioModel.build_message(messages, block.parameters)
80
result = io_processor.create_chat_completion(inputs) # pyright: ignore
81
- message = result.next_message.model_dump()
+ try: # TODO: update when new version of granite-io is released
82
+ message = result.next_message.model_dump()
83
+ except AttributeError:
84
+ message = result.results[0].next_message.model_dump()
85
raw_result = result.model_dump()
86
return (
87
message,
0 commit comments