Add quick start and utils for extracting code blocks (#404)

* Add quick start and utils for extracting code blocks

* format

* Spelling
This commit is contained in:
Eric Zhu
2024-08-26 17:23:27 -07:00
committed by GitHub
parent c779d4177b
commit c8084ada4e
5 changed files with 304 additions and 2 deletions

View File

@@ -0,0 +1,37 @@
from agnext.components.code_executor import extract_markdown_code_blocks
def test_extract_markdown_code_blocks() -> None:
text = """# This is a markdown text
```python
print("Hello World")
```
"""
code_blocks = extract_markdown_code_blocks(text)
assert len(code_blocks) == 1
assert code_blocks[0].language == "python"
assert code_blocks[0].code == 'print("Hello World")\n'
text = """More markdown text
```python
print("Hello World")
```
Another code block.
```python
print("Hello World 2")
```
"""
code_blocks = extract_markdown_code_blocks(text)
assert len(code_blocks) == 2
assert code_blocks[0].language == "python"
assert code_blocks[0].code == 'print("Hello World")\n'
assert code_blocks[1].language == "python"
assert code_blocks[1].code == 'print("Hello World 2")\n'