| import pytest
|
|
|
| from agent.routing.task_classifier import TaskClassifier
|
|
|
|
|
| CODING_PROMPTS = [
|
| 'jalankan hello.py',
|
| 'ubah hello.py agar mencetak Hello ZEVORA',
|
| 'jalankan lagi',
|
| 'buatkan package.json dan jalankan npm test',
|
| 'run npm test',
|
| 'install dependency',
|
| 'jalankan aplikasi',
|
| 'buat project React',
|
| 'ubah CSS',
|
| 'buat API endpoint',
|
| 'buat smart contract',
|
| 'ubah fungsi login',
|
| 'buat file hello.py yang mencetak Hello World',
|
| 'perbaiki error project ini dan jalankan test',
|
| 'create a JavaScript program',
|
| 'build aplikasi HTML dan CSS',
|
| 'compile Java project ini',
|
| 'edit function authentication',
|
| 'refactor kode TypeScript',
|
| 'jalanin script test.js',
|
| 'fix bug pada API endpoint',
|
| ]
|
|
|
|
|
| @pytest.mark.parametrize('prompt', CODING_PROMPTS)
|
| def test_bilingual_coding_prompts_are_classified_for_workspace(prompt):
|
| task = TaskClassifier().classify(prompt)
|
|
|
| assert 'coding' in task.labels
|
|
|
|
|
| @pytest.mark.parametrize('prompt', [
|
| 'apa itu Python?',
|
| 'ceritakan sejarah AI',
|
| 'jelaskan apa itu JavaScript',
|
| 'siapa pencipta bahasa Java?',
|
| 'ringkas artikel ini',
|
| ])
|
| def test_non_coding_prompts_do_not_trigger_coding_workspace(prompt):
|
| task = TaskClassifier().classify(prompt)
|
|
|
| assert 'coding' not in task.labels
|
| assert 'debugging' not in task.labels
|
| assert 'tool_task' not in task.labels
|
|
|
|
|
| def test_file_and_action_prompts_expose_navigation_tool_hints():
|
| classifier = TaskClassifier()
|
|
|
| assert 'filesystem.read' in classifier.classify('ubah app.py').requires_tools
|
| assert 'terminal.execute' in classifier.classify('jalankan npm test').requires_tools
|
| assert 'project.create' in classifier.classify('buat project React').requires_tools
|
|
|
