Merge pull request #578 from jupyter-naas/510-use-langchain---chatperplexity-in-perplexity-agent

feat: use langchain chatperplexity in perplexity agent

Author: Dr0p42

lib/abi/services/agent/Agent.py

@@ -286,7 +286,9 @@ def __init__(
 
         # TODO: Make sure the Agent does not call the version without tools.
         self._chat_model = chat_model
-        self._chat_model_with_tools = chat_model.bind_tools(self._structured_tools)
+        self._chat_model_with_tools = chat_model
+        if self._tools:
+            self._chat_model_with_tools = chat_model.bind_tools(self._structured_tools)
 
         # Use provided memory or create based on environment
         if memory is None:
@@ -376,7 +378,7 @@ def build_graph(self, patcher: Optional[Callable] = None):
         graph = StateGraph(MessagesState)
         graph.add_node(self.call_model)
         graph.add_edge(START, "call_model")
-
+        
         graph.add_node(self.call_tools)
         # This is not needed because the call_tools is generating the proper Command to go to the right node after each tool call.
         # graph.add_edge("call_tools", "call_model")

pyproject.toml

@@ -63,6 +63,7 @@ dependencies = [
     "psycopg[binary,pool]>=3.2.9",
     "langgraph-checkpoint-postgres>=2.0.23",
     "mypy>=1.17.1",
+    "langchain-perplexity>=0.1.2",
 ]
 
 [project.scripts]

src/__modules__.py

@@ -40,19 +40,17 @@ def analyze_module_dependencies(module_paths: List[str]) -> Dict[str, Set[str]]:
                     if isinstance(node, ast.Call):
                         if isinstance(node.func, ast.Name) and node.func.id == 'get_modules':
                             get_modules_found = True
-                            logger.debug(f"Found get_modules() call in {py_file}")
                             break
                         elif isinstance(node.func, ast.Attribute) and node.func.attr == 'get_modules':
                             get_modules_found = True
-                            logger.debug(f"Found get_modules() call in {py_file}")
                             break
 
                 # If get_modules() is found, this module depends on ALL other modules
                 if get_modules_found:
                     for other_module in module_paths:
                         if other_module != module_path:
                             dependencies[module_path].add(other_module)
-                    logger.info(f"Module {module_path} calls get_modules() - depends on ALL other modules")
+                    logger.debug(f"Module {module_path} depends on ALL other modules")
                 else:
                     # Regular dependency analysis for import statements
                     for node in ast.walk(tree):

src/core/modules/perplexity/agents/PerplexityAgent.py

@@ -1,171 +1,66 @@
-from abi.services.agent.IntentAgent import (
-    IntentAgent,
-    Intent,
-    IntentType,
+from abi.services.agent.Agent import (
+    Agent,
     AgentConfiguration,
     AgentSharedState,
+    MemorySaver,
 )
-from abi.services.agent.Agent import Agent
-from fastapi import APIRouter
-from src.core.modules.perplexity.models.perplexity_gpt_4o import model
 from typing import Optional
-from enum import Enum
-from abi import logger
+from src.core.modules.perplexity.models.sonar_pro import model
 
-AVATAR_URL = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi/assets/perplexity.png"
 NAME = "Perplexity"
-TYPE = "core"
-SLUG = "perplexity"
 DESCRIPTION = "Perplexity Agent that provides real-time answers to any question on the web using Perplexity AI."
-MODEL = "perplexity-gpt-4o"
-SYSTEM_PROMPT = """
-Role:
-You are Perplexity, a researcher agent with access to Perplexity AI search engine.
-
-Objective: 
-Provide accurate and comprehensive information to user inquiries using your web search capabilities.
-
-Context:
-You will receive prompts from users but also from a supervisor agent named 'abi' that already handle the conversation with the user.
-When users say things like "ask perplexity", "parler à perplexity", "I want to talk to perplexity", or similar phrases referring to YOU:
-- Recognize that YOU ARE Perplexity - don't try to "connect" them to Perplexity
-- Respond directly as Perplexity without any delegation confusion
-- Simply acknowledge and proceed to help them directly
-- Never say "I cannot connect you to Perplexity" - you ARE Perplexity!
-
-Tasks:
-- Answer the user's question
-- If you don't know the answer, you must ask more information to the user or propose to use differents agents to answer the question.
-
-Tools:
-- current_datetime: Get the current datetime in Paris timezone.
-- ask_question: Ask a question to Perplexity AI
-
-Operating Guidelines:
-1. Get the date and time using current_datetime tool and add it to user's question
-For example: 
-"le gagnant de la dernière ligue des champions masculin" -> "Current datetime: 2025-06-25 10:00:00 : le gagnant de la dernière ligue des champions masculin"
-2. Perform the search using Perplexity AI tool: ask_question
-3. Present information as found in search results
-
-Constraints:
-- Must follow the operating guidelines from any prompts you receive
-- NEVER use your internal knowledge base to answer questions
-- Try to find at least 1 source in the response which is not a link to the Perplexity page.
-- Must display sources full URL
-- Must format sources as at the end of the response after 2 blank lines: 
-```
-**Sources:**
-- [Source Name](source_url)
-```
-
-Examples:
-```
-**Sources:**
-- [Les echos](https://www.lesechos.fr/entreprises-et-marches/actualites/2025-06-25/)
-- [Le Monde](https://www.lemonde.fr/economie/article
-- [Le Parisien](https://www.leparisien.fr/economie/article
-```
-"""
-TEMPERATURE = 0
-DATE = True
-INSTRUCTIONS_TYPE = "system"
-ONTOLOGY = True
+AVATAR_URL = "https://naasai-public.s3.eu-west-3.amazonaws.com/abi/assets/perplexity.png"
 SUGGESTIONS: list = []
 
 def create_agent(
     agent_shared_state: Optional[AgentSharedState] = None, 
     agent_configuration: Optional[AgentConfiguration] = None
-) -> Optional[IntentAgent]:
-    # Check if model is available
-    if model is None:
-        logger.error("Perplexity model not available - missing OpenAI API key and Perplexity API key")
-        return None
-    
+) -> Agent:  
+    # System prompt
+    SYSTEM_PROMPT = """
+    Role:
+    You are Perplexity, an advanced AI research agent powered by the Perplexity AI search engine. 
+    You excel at real-time information gathering, fact-checking, and providing up-to-date insights across all fields of knowledge. 
+
+    Objective:
+    - Deliver comprehensive, well-researched answers by leveraging real-time web search capabilities
+    - Synthesize information from multiple reliable sources to provide balanced perspectives
+    - Present complex topics in a clear, accessible manner while maintaining accuracy
+    - Proactively fact-check information and acknowledge any limitations in available data
+    - Include relevant context and background information when beneficial to understanding
+
+    Constraints:
+    - You maintain a professional yet approachable tone, always striving for accuracy and clarity in your responses.
+    - Only include source references when you have actual URLs to cite
+    - When citing sources, place them at the end of your response after two blank lines
+    - Format sources exactly as shown in the example below:
+
+    Examples:
+    ```
+    **Sources:**
+    - [1](https://www.lesechos.fr/entreprises-et-marches/actualites/2025-06-25/)
+    - [2](https://www.lemonde.fr/economie/article)
+    - [3](https://www.leparisien.fr/economie/article)
+    ```
+    """
     # Set configuration
     if agent_configuration is None:
         agent_configuration = AgentConfiguration(
             system_prompt=SYSTEM_PROMPT,
         )
     if agent_shared_state is None:
         agent_shared_state = AgentSharedState(thread_id="0")
-    
-    
-    # Init
-    from langchain_core.tools import Tool
-    from typing import List, Union
-    
-    tools: List[Union[Tool, Agent]] = []
-
-    from src import secret
-    from src.core.modules.perplexity.integrations import PerplexityIntegration
-    from src.core.modules.perplexity.integrations.PerplexityIntegration import PerplexityIntegrationConfiguration
-
-    perplexity_api_key = secret.get('PERPLEXITY_API_KEY')
-    if perplexity_api_key is not None:
-        perplexity_integration_configuration = PerplexityIntegrationConfiguration(  
-            api_key=perplexity_api_key
-        )
-        tools += PerplexityIntegration.as_tools(perplexity_integration_configuration)
-    else:
-        logger.error("Perplexity model not available - missing Perplexity API key")
-        return None
-
-    from datetime import datetime
-    from zoneinfo import ZoneInfo
-    from langchain_core.tools import StructuredTool
-    from pydantic import BaseModel
-    
-    class EmptySchema(BaseModel):
-        pass
-        
-    current_datetime_tool = StructuredTool(
-        name="current_datetime", 
-        description="Get the current datetime in Paris timezone.",
-        func=lambda : datetime.now(tz=ZoneInfo('Europe/Paris')),
-        args_schema=EmptySchema
-    )
-    
-    from typing import cast
-    tools += [cast(Tool, current_datetime_tool)]
-
-    intents: list = [
-        Intent(
-            intent_value="what is your name",
-            intent_type=IntentType.RAW,
-            intent_target="I am Perplexity, an AI research assistant that provides real-time answers using web search capabilities.",
-        ),
-        Intent(
-            intent_value="what can you do",
-            intent_type=IntentType.RAW,
-            intent_target="I can search the web in real-time, provide up-to-date information, research any topic, and answer questions using the latest information from the internet.",
-        ),
-    ]
 
     return PerplexityAgent(
         name=NAME,
         description=DESCRIPTION,
-        chat_model=model.model,
-        tools=tools, 
+        chat_model=model,
+        tools=[],
         agents=[],
-        intents=intents,
         state=agent_shared_state, 
         configuration=agent_configuration, 
-        memory=None
+        memory=MemorySaver()
     ) 
 
-class PerplexityAgent(IntentAgent):
-    def as_api(
-        self, 
-        router: APIRouter, 
-        route_name: str = NAME, 
-        name: str = NAME.replace("_", " "), 
-        description: str = "API endpoints to call the Perplexity agent completion.", 
-        description_stream: str = "API endpoints to call the Perplexity agent stream completion.",
-        tags: Optional[list[str | Enum]] = None,
-    ) -> None:
-        if tags is None:
-            tags = []
-        return super().as_api(
-            router, route_name, name, description, description_stream, tags
-        )
+class PerplexityAgent(Agent):
+    pass

src/core/modules/perplexity/agents/PerplexityAgent_test.py

@@ -1,27 +1,21 @@
 import pytest
 
-from src.core.modules.perplexity.agents.PerplexityAgent import create_agent
-
 @pytest.fixture
 def agent():
+    from src.core.modules.perplexity.agents.PerplexityAgent import create_agent
     return create_agent()
 
 def test_agent_name(agent):
-    result = agent.invoke("what is your name")
+    result = agent.invoke("What is your name?")
 
     assert result is not None, result
     assert "Perplexity" in result, result
 
-def test_search_news(agent):
-    result = agent.invoke("Which team won the last men's Champions League?")
-
-    assert result is not None, result
-    assert "Paris Saint-Germain" in result, result
-
 def test_search_news_with_datetime(agent):
     from datetime import datetime
 
     result = agent.invoke("What are the news of the day? Start by: 'As of today the date is YYYY-MM-DD.'")
 
     assert result is not None, result
-    assert datetime.now().strftime("%Y-%m-%d") in result, result
+    assert datetime.now().strftime("%Y-%m-%d") in result, result
+    assert "sources" in result.lower(), result

src/core/modules/perplexity/models/perplexity_gpt_4o.py

@@ -0,0 +1,13 @@
+from langchain_perplexity import ChatPerplexity
+from src import secret
+from pydantic import SecretStr
+from abi import logger
+
+perplexity_api_key = secret.get("PERPLEXITY_API_KEY")
+model = ChatPerplexity(
+    model="sonar",
+    temperature=0,
+    api_key=SecretStr(perplexity_api_key),
+    timeout=120
+)
+logger.debug("✅ Perplexity GPT-4o model loaded successfully via OpenAI")

src/core/modules/perplexity/models/sonar.py

@@ -0,0 +1,13 @@
+from langchain_perplexity import ChatPerplexity
+from src import secret
+from pydantic import SecretStr
+from abi import logger
+
+perplexity_api_key = secret.get("PERPLEXITY_API_KEY")
+model = ChatPerplexity(
+    model="sonar-pro",
+    temperature=0,
+    api_key=SecretStr(perplexity_api_key),
+    timeout=120
+)
+logger.debug("✅ Perplexity GPT-4o model loaded successfully via OpenAI")