def set_form_schema()

in aidial_sdk/chat_completion/choice.py [0:0]

• 13 lines of code
• 4 McCabe index (conditional complexity)

    def set_form_schema(self, form_schema: Any) -> None:
        if self._schema_submitted:
            raise runtime_error("Trying to set form schema twice")

        if not self._opened:
            raise runtime_error(
                "Trying to append form schema to an unopened choice"
            )
        if self._closed:
            raise runtime_error(
                "Trying to append form schema to a closed choice"
            )

        self._schema_submitted = True
        self.send_chunk(FormSchemaChunk(self._index, form_schema))