Updating supported models

Author: dawid-szewc

README.md

@@ -30,7 +30,7 @@ perplexity "What is the meaning of life?"
 
 ## With Additional Options
 ```bash
-perplexity -uc -m llama-3.1-sonar-large-128k-online "Explain Einstein's theory of relativity"
+perplexity -uc -m sonar-pro "Explain Einstein's theory of relativity"
 ```
 
 ## Options
@@ -39,12 +39,13 @@ perplexity -uc -m llama-3.1-sonar-large-128k-online "Explain Einstein's theory o
 -   `-c`, `--citations`: Show citations
 -   `-g`, `--glow`: Use Glow-compatible formatting
 -   `-a API_KEY`, `--api-key API_KEY`: Set the API key (optional, defaults to `PERPLEXITY_API_KEY` environment variable)
--   `-m MODEL`, `--model MODEL`: Choose the language model (default: llama-3.1-sonar-small-128k-online)
+-   `-m MODEL`, `--model MODEL`: Choose the language model (default: sonar-pro)
 
 ## Available Models
--   llama-3.1-sonar-small-128k-online
--   llama-3.1-sonar-large-128k-online
--   llama-3.1-sonar-huge-128k-online
+-   sonar-reasoning-pro
+-   sonar-reasoning
+-   sonar-pro
+-   sonar
 
 ## Configuration
 Set the `PERPLEXITY_API_KEY` environment variable with your Perplexity API key:

perplexity.py

@@ -8,9 +8,10 @@
 import json
 
 AVAILABLE_MODELS = [
-    "llama-3.1-sonar-small-128k-online",
-    "llama-3.1-sonar-large-128k-online",
-    "llama-3.1-sonar-huge-128k-online",
+    "sonar-reasoning-pro",
+    "sonar-reasoning",
+    "sonar-pro",
+    "sonar"
 ]
 
 logger = logging.getLogger(__name__)
@@ -178,10 +179,10 @@ def main() -> None:
         "-m",
         "--model",
         type=str,
-        help="Description for model argument (default: llama-3.1-sonar-small-128k-online) "
+        help="Description for model argument (default: sonar-pro) "
         f"Available models: {AVAILABLE_MODELS}",
         required=False,
-        default="llama-3.1-sonar-small-128k-online",
+        default="sonar-pro",
     )
     args = parser.parse_args()
     log_level = logging.DEBUG if args.verbose else logging.WARNING