(command, vega_lite_spec)
| 78 | |
| 79 | |
| 80 | def get_changed_vega(command, vega_lite_spec) -> Dict: |
| 81 | messages = make_default_visualization_change_messages() |
| 82 | messages.append({ |
| 83 | "role": "user", |
| 84 | "content": make_visualization_change_message().format( |
| 85 | command=command, |
| 86 | vega_lite_spec=vega_lite_spec |
| 87 | ) |
| 88 | }) |
| 89 | vega = extract_json_str_from_markdown( |
| 90 | get_assistant_message(messages)["message"]["content"] |
| 91 | ) |
| 92 | return json.loads(vega) |
| 93 | |
| 94 | |
| 95 | def extract_json_str_from_markdown(assistant_message_content) -> str: |
no test coverage detected