fix: advanced search not existing

Author: FlorentRvl

src/core/perplexity/integrations/PerplexityIntegration.py

@@ -124,31 +124,35 @@ def as_tools(configuration: PerplexityIntegrationConfiguration):
 
     integration = PerplexityIntegration(configuration)
 
-    class AskQuestionSchema(BaseModel):
+    class SearchWebSchema(BaseModel):
         question: str = Field(..., description="The question to ask Perplexity AI")
         user_location: Optional[Annotated[str, Field(..., description="The user location to use for the search")]] = "FR"
         search_context_size: Optional[Annotated[str, Field(..., description="The search context size to use for the search", pattern="^(low|medium|high)$")]] = "medium"
 
+    class AdvancedSearchSchema(BaseModel):
+        question: str = Field(..., description="The question to ask Perplexity AI")
+        user_location: Optional[Annotated[str, Field(..., description="The user location to use for the search")]] = "FR"
+
     return [
         StructuredTool(
             name="perplexity_quick_search",
             description="A lightweight, cost-effective search model optimized for quick, grounded answers with real-time web search.",
             func=lambda question, user_location, search_context_size: integration.search_web(question=question, user_location=user_location, search_context_size=search_context_size, model="sonar"),
-            args_schema=AskQuestionSchema,
+            args_schema=SearchWebSchema,
             return_direct=True
         ),
         StructuredTool(
             name="perplexity_search",
             description="Advanced search model designed for complex queries, delivering deeper content understanding with enhanced search result accuracy and 2x more search results than standard Sonar.",
             func=lambda question, user_location, search_context_size: integration.search_web(question=question, user_location=user_location, search_context_size=search_context_size, model="sonar-pro"),
-            args_schema=AskQuestionSchema,
+            args_schema=SearchWebSchema,
             return_direct=True
         ),
         StructuredTool(
             name="perplexity_advanced_search",
             description="Advanced search model designed for complex queries, delivering deeper content understanding with enhanced search result accuracy and 2x more search results than standard Sonar with high context size",
-            func=lambda question, user_location, search_context_size: integration.search_web(question=question, user_location=user_location, search_context_size=search_context_size, model="sonar-pro-search"),
-            args_schema=AskQuestionSchema,
+            func=lambda question, user_location: integration.search_web(question=question, user_location=user_location, search_context_size="high", model="sonar-pro"),
+            args_schema=AdvancedSearchSchema,
             return_direct=True
         ),
     ]

src/core/perplexity/integrations/PerplexityIntegration_test.py

@@ -10,9 +10,33 @@ def perplexity_integration() -> PerplexityIntegration:
     )
     return PerplexityIntegration(configuration)
 
-def test_ask_question(perplexity_integration: PerplexityIntegration):
+def test_perplexity_quick_search(perplexity_integration: PerplexityIntegration):
     response = perplexity_integration.search_web(
         question="What is the capital of France?",
+        user_location="FR",
+        search_context_size="medium",
+        model="sonar",
+    )
+    assert response is not None
+    assert "France" in response
+
+def test_perplexity_search(perplexity_integration: PerplexityIntegration):
+    response = perplexity_integration.search_web(
+        question="What is the capital of France?",
+        user_location="FR",
+        search_context_size="medium",
+        model="sonar-pro",
+    )
+    assert response is not None
+    assert "France" in response
+
+
+def test_perplexity_advanced_search(perplexity_integration: PerplexityIntegration):
+    response = perplexity_integration.search_web(
+        question="What is the capital of France?",
+        user_location="FR",
+        search_context_size="high",
+        model="sonar-pro",
     )
     assert response is not None
     assert "France" in response