Method create

tests/unittests/workflow/testing_utils.py:390–431

(
      cls,
      responses: (
          list[types.Part]
          | list[LlmResponse]
          | list[str]
          | list[list[types.Part]]
      ),
      error: Exception | None = None,
      wait_for_realtime_input: bool = False,
  )

  @classmethod
  def create(
      cls,
      responses: (
          list[types.Part]
          | list[LlmResponse]
          | list[str]
          | list[list[types.Part]]
      ),
      error: Exception | None = None,
      wait_for_realtime_input: bool = False,
  ):
    if error and not responses:
      return cls(
          responses=[],
          error=error,
          wait_for_realtime_input=wait_for_realtime_input,
      )
    if not responses:
      return cls(responses=[], wait_for_realtime_input=wait_for_realtime_input)
    elif isinstance(responses[0], LlmResponse):
      # responses is list[LlmResponse]
      return cls(
          responses=responses, wait_for_realtime_input=wait_for_realtime_input
      )
    else:
      responses = [
          LlmResponse(content=ModelContent(item))
          if isinstance(item, list) and isinstance(item[0], types.Part)
          # responses is list[list[Part]]
          else LlmResponse(
              content=ModelContent(
                  # responses is list[str] or list[Part]
                  [Part(text=item) if isinstance(item, str) else item]
              )
          )
          for item in responses
          if item
      ]

      return cls(
          responses=responses, wait_for_realtime_input=wait_for_realtime_input
      )

  @classmethod
  @override