def test_live_run_with_guided_json_schema(self):
      json_schema = {
          "type": "object",
          "properties": {"title": {"type": "string"}, "rating": {"type": "number"}},
          "required": ["title", "rating"],
      }
      chat_messages = [
          ChatMessage.from_user(
              """
          Return the title and the rating based on the following movie review according
          to the provided json schema.
          Review: Inception is a really well made film. I rate it four stars out of five."""
          )
      ]

      component = NvidiaChatGenerator(
          model="meta/llama-3.1-70b-instruct",
          generation_kwargs={"extra_body": {"guided_json": json_schema}, "temperature": 0.0},
      )

      results = component.run(chat_messages)
      assert len(results["replies"]) == 1
      message = results["replies"][0].text
      output = json.loads(message)
      assert output["title"] == "Inception"
      assert "rating" in output