def edit()

in databao/visualizers/vega_chat.py [0:0]

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

    def edit(self, request: str, visualization: VisualisationResult, *, stream: bool = False) -> VegaChatResult:
        if not isinstance(visualization, VegaChatResult):
            raise ValueError(f"{self.__class__.__name__} can only edit {VegaChatResult.__name__} objects")
        if visualization.spec_df is None:
            raise ValueError("No dataframe found in the provided visualization")
        messages = visualization.meta.get("messages", None)
        if messages is None:
            raise ValueError("No message history found in the provided visualization")
        return self._run_vega_chat(request, visualization.spec_df, messages=messages, stream=stream)