diff --git a/requirements.txt b/requirements.txt index e216504..1a9f46b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -20,6 +20,7 @@ librosa>=0.10.2.post1 selenium>=4.27.1 markdownify>=1.1.0 text2emotion>=0.0.5 +adaptive-classifier>=0.0.10 langid>=1.1.6 chromedriver-autoinstaller>=0.6.4 httpx>=0.27,<0.29 diff --git a/setup.py b/setup.py index bde5055..8f22c71 100644 --- a/setup.py +++ b/setup.py @@ -36,6 +36,7 @@ setup( "text2emotion>=0.0.5", "python-dotenv>=1.0.0", "langid>=1.1.6", + "adaptive-classifier>=0.0.10", "httpx>=0.27,<0.29", "anyio>=3.5.0,<5", "distro>=1.7.0,<2", diff --git a/sources/router.py b/sources/router.py index 3152915..8a4b4dd 100644 --- a/sources/router.py +++ b/sources/router.py @@ -118,6 +118,9 @@ class AgentRouter: ("Write a Python script to count words in a text file", "LOW"), ("Find a public API for sports scores and build a web app to show live updates", "HIGH"), ("Create a simple HTML page with CSS styling", "LOW"), + ("hi", "LOW"), + ("Bonjour", "LOW"), + ("What's up ?", "LOW"), ] texts = [text for text, _ in few_shots] labels = [label for _, label in few_shots]