Add Intelligent Cognitive Agent Python Script

examples/python/tools/exa-tool/chile_government_services_assistant_.py

@@ -0,0 +1,141 @@
+# # -*- coding: utf-8 -*-
+# """Chile_Government_Services_Assistant .ipynb
+
+# Automatically generated by Colab.
+
+# Original file is located at
+#     https://colab.research.google.com/drive/13IOn2Vhg0EUwuIGpVilImW0rEpYYMoFQ
+
+# # Chile Government Services Assistant - AI Chatbot
+
+# This notebook demonstrates how to use an AI-powered assistant to answer questions about Chilean government services and procedures, using the Firecrawl API and a friendly, step-by-step conversational approach.
+
+# [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DhivyaBharathy-web/PraisonAI/blob/main/examples/cookbooks/Chile_Government_Services_Assistant.ipynb)
+
+# # Install dependencies
+# """
+
+# !pip install flask firecrawl praisonaiagents google-genai python-dotenv deep-translator
+
+# """# Set API Keys"""
+
+import os
+
+os.environ['FIRECRAWL_API_KEY'] = "your api key here"
+os.environ['OPENAI_API_KEY'] = "your api key here"
+
+# """# Import Libraries & Translator"""
+
+from firecrawl import FirecrawlApp, ScrapeOptions
+from deep_translator import GoogleTranslator
+import re
+
+def translate_to_spanish(text):
+    try:
+        return GoogleTranslator(source='auto', target='es').translate(text)
+    except Exception as e:
+        print("Translation to Spanish failed:", e)
+        return text
+
+def translate_to_english(text):
+    try:
+        # Remove Markdown images and None values before translation
+        text = str(text).replace("None", "")
+        text = re.sub(r'!\[.*?\]\(.*?\)', '', text)
+        return GoogleTranslator(source='auto', target='en').translate(text)
+    except Exception as e:
+        print("Translation to English failed:", e)
+        return text
+
+# """# Firecrawl Tool Class"""
+
+class FirecrawlTool:
+    def __init__(self, api_key, instruction: str, template: str):
+        if not api_key:
+            raise ValueError("Firecrawl API key not provided.")
+        self.app = FirecrawlApp(api_key=api_key)
+        self.instruction = instruction
+        self.template = template
+
+    def search(self, search: str) -> str:
+        if not search or len(search) < 5:
+            return "Error: Please provide a valid search query (at least 5 characters)."
+        response_md = ""
+        try:
+            search_result = self.app.search(
+                query=self.instruction + search,
+                limit=2,
+                country="cl",
+                lang="es",  # Always search in Spanish for best results
+                scrape_options=ScrapeOptions(formats=["markdown", "links"])
+            )
+            if search_result and hasattr(search_result, 'data') and search_result.data:
+                filtered_results = [
+                    result for result in search_result.data
+                    if str(result.get("url", "")).startswith("https://www.chileatiende.gob.cl/fichas") and not str(result.get("url", "")).endswith("pdf")
+                ]
+                if filtered_results:
+                    for num, result in enumerate(filtered_results, start=1):
+                        response_md += self.template.format(
+                            result_number=num,
+                            page_title=str(result.get("title", "")),
+                            page_url=str(result.get("url", "")),
+                            page_content=str(result.get("markdown", ""))
+                        )
+                    return response_md
+                else:
+                    return None
+            else:
+                return None
+        except Exception as e:
+            return f"Error during search: {e}"
+
+# """# Firecrawl Prompt Template"""
+
+# FIRECRAWL_INSTRUCTION = "ChileAtiende: "
+# FIRECRAWL_TEMPLATE = """
+# # Result {result_number}
+
+# ## Page Name:
+# "{page_title}"
+
+# ## URL:
+# {page_url}
+
+# ## Content:
+# {page_content}
+
+# """
+
+# """# Initialize Firecrawl Tool"""
+
+firecrawl_tool = FirecrawlTool(
+    api_key=os.environ['FIRECRAWL_API_KEY'],
+    instruction=FIRECRAWL_INSTRUCTION,
+    template=FIRECRAWL_TEMPLATE
+)
+
+# """# Main Chat Loop"""
+
+print("Hello! I am your ChileAtiende assistant, Tomás. How can I help you today?")
+print("You can ask me, for example: How to renew your ID card, How to apply for the Winter Bonus, etc.")
+
+while True:
+    user_input = input("\nYou: ")
+    if user_input.lower() in ["exit", "quit"]:
+        print("Tomás: It was a pleasure to help you. Goodbye!")
+        break
+
+    # Translate English input to Spanish for Firecrawl
+    spanish_query = translate_to_spanish(user_input)
+    spanish_answer = firecrawl_tool.search(spanish_query)
+
+    # Only translate if we got a real answer
+    if spanish_answer and isinstance(spanish_answer, str) and spanish_answer.strip() and "Error" not in spanish_answer:
+        try:
+            english_answer = translate_to_english(spanish_answer)
+            print("\nTomás (in English):\n", english_answer)
+        except Exception as e:
+            print(f"\nTomás: I found information, but couldn't translate it. Here it is in Spanish:\n{spanish_answer}\n(Translation error: {e})")
+    else:
+        print("\nTomás: Sorry, I couldn't find relevant information. Try rephrasing your question or ask about another service.")

examples/python/tools/exa-tool/intelligent_cognitive_agent.py

@@ -0,0 +1,81 @@
+# # -*- coding: utf-8 -*-
+# """Intelligent_Cognitive_Agent.ipynb
+
+# Automatically generated by Colab.
+
+# Original file is located at
+#     https://colab.research.google.com/drive/1UssW8MZyZpMX6oVjhqn4O6euxLLIVgGM
+
+# # 🧠 Cognitive Assistant Agent Team (PraisonAI)
+
+# This notebook demonstrates a multi-specialist cognitive assistant using PraisonAI, with each specialist as a tool.
+
+# [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DhivyaBharathy-web/PraisonAI/blob/main/examples/cookbooks/Intelligent_Cognitive_Agent.ipynb)
+
+# # Install Dependencies
+# """
+
+# !pip install praisonaiagents
+
+# """# Set API Key"""
+
+import os
+
+# Set your OpenAI or OpenRouter API key for PraisonAI
+os.environ['OPENAI_API_KEY'] = 'Enter your api key'  # <-- Replace with your actual OpenAI or OpenRouter API key
+
+"""# Imports and Specialist Agent Definitions"""
+
+import re
+from praisonaiagents import Agent
+
+# Dummy specialist agent classes for demonstration.
+# Replace these with your actual specialist agent logic if needed.
+class SpecialistAgent:
+    def __init__(self, name):
+        self.name = name
+    def run(self, query):
+        return f"{self.name} specialist response to: {query}"
+
+# Instantiate all specialist agents (replace with your real classes if available)
+specialists = [
+    SpecialistAgent("decision_risk_agent"),
+    SpecialistAgent("problem_solving_innovation_agent"),
+    SpecialistAgent("Systems___Complexity_Agent"),
+    SpecialistAgent("bias_psychology_agent"),
+    SpecialistAgent("Strategy___Competition_Agent"),
+    SpecialistAgent("Learning___Communication_Agent"),
+    SpecialistAgent("Efficiency___Process_Agent"),
+    SpecialistAgent("motivation_human_factors_agent"),
+]
+
+"""# Tool Wrapping and Agent Setup"""
+
+def make_tool(agent):
+    def tool(query: str):
+        return agent.run(query)
+    # Sanitize the tool name to match PraisonAI's requirements
+    safe_name = re.sub(r'[^a-zA-Z0-9_-]', '_', agent.name or "tool")
+    tool.__name__ = safe_name
+    return tool
+
+tools = [make_tool(agent) for agent in specialists]
+
+agent = Agent(
+    instructions="""
+        You are a cognitive assistant with access to a set of specialist tools (decision making, problem solving, systems thinking, psychology, strategy, learning, efficiency, motivation).
+        Use the most relevant tool(s) to answer the user's query.
+    """,
+    tools=tools,
+)
+
+# """# User Query and Agent Response"""
+
+user_query = input("Ask your question (or type 'quit' to exit): ")
+while user_query.lower() != 'quit':
+    print("\n--- Thinking ---")
+    response = agent.start(user_query)
+    print(response)
+    print("\n---------------")
+    user_query = input("Ask your question (or type 'quit' to exit): ")
+print("Cognitive Assistant Agent Team - Finished.")