Skip to content

Commit d4bc012

Browse files
authored
Support dev version of granite-io (#742)
1 parent 3702571 commit d4bc012

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/pdl/pdl_granite_io.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,10 @@ async def async_generate_text(
7878
io_processor = GraniteioModel.processor_of_block(block)
7979
inputs = GraniteioModel.build_message(messages, block.parameters)
8080
result = io_processor.create_chat_completion(inputs) # pyright: ignore
81-
message = result.next_message.model_dump()
81+
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()
8285
raw_result = result.model_dump()
8386
return (
8487
message,

0 commit comments

Comments
 (0)