Perplexity Search

api_key = "pplx-abc123..."  # Your real key here

payload = {
    "model": "sonar-pro",  # sonar or sonar-pro both work
    "temperature": 0.2,
    "disable_runs": False,
    "top_p": 0.9,
    "max_tokens": 150,
    "messages": [
        {
            "role": "system",
            "content": (
                """
                Give a short, clear answer in simple spoken language.
                Do not use symbols, citations, or abbreviations.
                """
            )
        },
        {
            "role": "user",
            "content": f"{msg}"
        }
    ]
}

start_time = time.time()

# Send the request to Perplexity
response = requests.post(
    "https://api.perplexity.ai/chat/completions",
    headers={
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json"
    },
    json=payload
)

end_time = time.time()
response_time = round(end_time - start_time, 3)

self.worker.editor_logging_handler.info(f"⏱️ API Response Time: {response_time} seconds")

try:
    result = response.json()
    self.worker.editor_logging_handler.info(f"✅ Parsed JSON Response:\n{json.dumps(result, indent=2)}")
except Exception as e:
    self.worker.editor_logging_handler.info(f"❌ Failed to parse JSON: {e}")
    result = {}

# Extract the assistant's message (final summary)
search_result = result.get("choices", [{}])[0].get("message", {}).get("content", "Sorry, I couldn't find anything.")

# Remove citation markers like [1], [2], etc.
search_result = re.sub(r"\[\d+\]", "", search_result)
search_result = search_result.replace("  ", " ").strip()

# Speak the final summarized result
await self.capability_worker.speak("Here's what I found:")
await self.capability_worker.speak(search_result)

# Resume the normal workflow
self.capability_worker.resume_normal_flow()

"max_tokens": 150,  # Shorter, punchier answers
"max_tokens": 300,  # More detailed explanations

"temperature": 0.2,  # More factual, deterministic
"temperature": 0.7,  # More creative, varied responses