Add: Perplexity AI support with multiple models

app.py

@@ -81,7 +81,22 @@ def create_gradio_interface():
                 
                 with gr.Row():
                     model_dropdown = gr.Dropdown(
-                        choices=["gpt-3.5-turbo", "gpt-4", "claude-3-opus"],
+                        choices=[
+                            # OpenAI Models
+                            "gpt-3.5-turbo",
+                            "gpt-4",
+                            # Anthropic Models
+                            "claude-3-opus",
+                            # Perplexity Current Models
+                            "sonar-reasoning-pro",
+                            "sonar-reasoning",
+                            "sonar-pro",
+                            "sonar",
+                            # Perplexity Legacy Models
+                            "llama-3.1-sonar-small-128k-online",
+                            "llama-3.1-sonar-large-128k-online",
+                            "llama-3.1-sonar-huge-128k-online"
+                        ],
                         value="gpt-3.5-turbo",
                         label="Model"
                     )

requirements.txt

@@ -22,4 +22,5 @@ openai>=1.12.0
 anthropic>=0.18.1
 duckduckgo-search>=4.4.3
 orjson>=3.9.15
-python-dotenv>=1.0.0 
+python-dotenv>=1.0.0
+perplexity-python>=0.1.1 

src/agents.py

@@ -3,6 +3,7 @@ from typing import Optional
 import openai
 import anthropic
 from duckduckgo_search import DDGS
+from perplexity import Perplexity
 
 class Agent:
     def __init__(self, base_model: str = "gpt-3.5-turbo", search_engine: str = "duckduckgo"):
@@ -14,6 +15,8 @@ class Agent:
             openai.api_key = os.getenv("OPENAI_API_KEY")
         elif "claude" in base_model:
             self.claude = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
+        elif "sonar" in base_model:
+            self.perplexity = Perplexity(api_key=os.getenv("PERPLEXITY_API_KEY"))
 
     def execute(self, message: str, project_name: str) -> Optional[str]:
         try:
@@ -22,6 +25,8 @@ class Agent:
                 response = self._process_with_gpt(message)
             elif "claude" in self.base_model:
                 response = self._process_with_claude(message)
+            elif "sonar" in self.base_model:
+                response = self._process_with_perplexity(message)
             else:
                 response = "Unsupported model selected"
                 
@@ -50,6 +55,29 @@ class Agent:
         )
         return response.content[0].text
 
+    def _process_with_perplexity(self, message: str) -> str:
+        # Map model names to Perplexity models
+        model_mapping = {
+            # Current models
+            "sonar-reasoning-pro": "sonar-reasoning-pro",  # 127k context, 8k output
+            "sonar-reasoning": "sonar-reasoning",          # 127k context
+            "sonar-pro": "sonar-pro",                     # 200k context, 8k output
+            "sonar": "sonar",                             # 127k context
+            # Legacy models (will be deprecated after 2/22/2025)
+            "llama-3.1-sonar-small-128k-online": "llama-3.1-sonar-small-128k-online",
+            "llama-3.1-sonar-large-128k-online": "llama-3.1-sonar-large-128k-online",
+            "llama-3.1-sonar-huge-128k-online": "llama-3.1-sonar-huge-128k-online"
+        }
+        
+        # Use the mapped model or default to sonar
+        model = model_mapping.get(self.base_model, "sonar")
+        
+        response = self.perplexity.chat.create(
+            model=model,
+            messages=[{"role": "user", "content": message}]
+        )
+        return response.choices[0].message.content
+
     def _search_web(self, query: str, num_results: int = 5) -> list:
         if self.search_engine == "duckduckgo":
             with DDGS() as ddgs:

src/config.py

@@ -19,10 +19,12 @@ class Config:
                 "BING": os.getenv("BING_API_KEY", ""),
                 "GOOGLE_SEARCH": os.getenv("GOOGLE_API_KEY", ""),
                 "GOOGLE_SEARCH_ENGINE_ID": os.getenv("GOOGLE_SEARCH_ENGINE_ID", ""),
+                "PERPLEXITY": os.getenv("PERPLEXITY_API_KEY", ""),
             },
             "API_ENDPOINTS": {
                 "BING": "https://api.bing.microsoft.com/v7.0/search",
                 "GOOGLE": "https://www.googleapis.com/customsearch/v1",
+                "PERPLEXITY": "https://api.perplexity.ai",
             }
         }
         
@@ -41,4 +43,7 @@ class Config:
         return self.config["API_KEYS"]["GOOGLE_SEARCH"]
 
     def get_google_search_engine_id(self):
-        return self.config["API_KEYS"]["GOOGLE_SEARCH_ENGINE_ID"] 
+        return self.config["API_KEYS"]["GOOGLE_SEARCH_ENGINE_ID"]
+        
+    def get_perplexity_api_key(self):
+        return self.config["API_KEYS"]["PERPLEXITY"]