카카오톡 챗봇을 만들고 싶은 거북이의 사례 2탄! Fast API와 퍼플렉시티를 활용한 카톡봇만들기

"""
FastAPI 기반 Perplexity 검색 서버
"""

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import httpx
import os
from dotenv import load_dotenv

# 환경변수 로딩
load_dotenv()
PERPLEXITY_API_KEY = os.getenv("PERPLEXITY_API_KEY")

app = FastAPI(title="Perplexity 검색 서버")

API_TIMEOUT = 30  # 초

class SearchRequest(BaseModel):
    query: str

class SearchResponse(BaseModel):
    answer: str

# Perplexity API 호출 함수
async def call_perplexity(query: str) -> str:
    headers = {
        "Authorization": f"Bearer {PERPLEXITY_API_KEY}",
        "Content-Type": "application/json"
    }
    data = {
        "model": "sonar-pro",
        "messages": [{"role": "user", "content": query}]
    }
    try:
        async with httpx.AsyncClient() as client:
            response = await client.post(
                "https://api.perplexity.ai/chat/completions",
                headers=headers,
                json=data,
                timeout=API_TIMEOUT
            )
            if response.status_code == 200:
                result = response.json()
                return result.get("choices", [{}])[0].get("message", {}).get("content", "응답 없음")
            else:
                return f"❌ API 오류: HTTP {response.status_code}"
    except httpx.TimeoutException:
        return "⏱️ 요청 시간이 초과되었습니다."
    except Exception as e:
        return f"❌ 예외 발생: {str(e)}"

@app.post("/search", response_model=SearchResponse)
async def search(request: SearchRequest):
    if not PERPLEXITY_API_KEY:
        raise HTTPException(status_code=500, detail="API 키가 설정되지 않았습니다.")
    answer = await call_perplexity(request.query)
    return SearchResponse(answer=answer)

@app.get("/")
async def health_check():
    return {"status": "ok", "service": "Perplexity 검색 서버"}