.

routers/llm_router.py

@@ -11,7 +11,7 @@ router = APIRouter(
 )
 
 @router.post("/askllm")
-def test_home():
+def test_home(userquery: str = Form("")):
     try:
         with get_connection() as conn:
             with conn.cursor() as cur:
@@ -24,7 +24,69 @@ def test_home():
                 """)
                 row = cur.fetchone()
         pollination_api_key=row[0] if row else None
+
+        # pollination key 없으면 그냥 클라한테 오류라고 퉤 뱉기
+        if not pollination_api_key:
+            return CommonResponse(
+                success=False,
+                msg="Pollinations API Key가 없습니다."
+            )
+        
+         # 2. Pollinations Text 모델에 보낼 데이터
+        url = "https://gen.pollinations.ai/v1/chat/completions"
+
+        """
+        택배상자에 뭐 주소나 내 정보 쓰잔아요?
+        컴퓨터의 api 통신에 Content-Type, Authorization 이거 쓰래요
+        Authorization 부분에 pollination_api_key 넣어서 보내래요
+        """
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {pollination_api_key}"
+        }
+
+        """
+        내용물. body 라고도 부름.
+        """
+        payload = {
+            "model": "openai-fast",
+            "messages": [
+                {
+                    "role": "system",
+                    "content": "당신은 친절한 타로 상담사입니다. 초보자도 이해하기 쉽게 한국어로 답변하세요."
+                },
+                {
+                    "role": "user",
+                    "content": userquery
+                }
+            ],
+            "temperature": 0.8,
+            "max_tokens": 800,
+            "stream": False
+        }
         
-        return CommonResponse(success=True)
+        # 3. Pollinations API 호출
+        response = requests.post(url, headers=headers, json=payload)
+
+        # 4. 실패 체크
+        if response.status_code != 200:
+            return CommonResponse(
+                success=False,
+                msg=f"Pollinations API 오류: {response.text}"
+            )
+
+        result = response.json()
+
+        # 5. AI 응답 텍스트 꺼내기
+        answer = result["choices"][0]["message"]["content"]
+
+        # 6. 프론트로 반환
+        return CommonResponse(
+            success=True,
+            data={
+                "question": userquery,
+                "answer": answer
+            }
+        )
     except Exception as e:
         return CommonResponse(success=False, msg=str(e))