chore: 将后端移动到 api/ 目录，准备合并为单仓库

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

api/prompt/consumers.py

@@ -0,0 +1,219 @@
+import json
+from channels.generic.websocket import AsyncWebsocketConsumer
+from channels.db import database_sync_to_async
+from .models import Message
+from .utils import can_access_task, get_or_create_active_conversation
+from .llm import stream_chat, extract_code, stream_guidance, parse_guidance_response
+
+
+class PromptConsumer(AsyncWebsocketConsumer):
+    async def connect(self):
+        self.user = self.scope["user"]
+        if self.user.is_anonymous:
+            await self.close()
+            return
+
+        self.task_id = int(self.scope["url_route"]["kwargs"]["task_id"])
+        if not await self.check_access():
+            await self.close()
+            return
+
+        self.current_user_message = None
+        await self.accept()
+
+        self.conversation = await self.get_or_create_conversation()
+        history = await self.get_history()
+        await self.send(text_data=json.dumps({
+            "type": "init",
+            "conversation_id": str(self.conversation.id),
+            "messages": history,
+        }))
+
+    async def disconnect(self, close_code):
+        if self.current_user_message:
+            await self.delete_message(self.current_user_message)
+            self.current_user_message = None
+
+    async def receive(self, text_data):
+        data = json.loads(text_data)
+        msg_type = data.get("type", "message")
+
+        if msg_type == "new_conversation":
+            return
+
+        prompt = data.get("content", "").strip()
+        model = data.get("model", "")
+        if not prompt:
+            return
+
+        self.current_user_message = await self.save_message("user", prompt)
+
+        try:
+            history = await self.get_history_for_llm()
+
+            full_response = ""
+            try:
+                async for chunk in stream_chat(history, model=model):
+                    full_response += chunk
+                    await self.send(text_data=json.dumps({
+                        "type": "stream",
+                        "content": chunk,
+                    }))
+            except Exception as e:
+                await self.send(text_data=json.dumps({
+                    "type": "error",
+                    "content": f"AI 服务出错：{str(e)}",
+                }))
+                return
+
+            code = extract_code(full_response)
+            assistant_msg = await self.save_message("assistant", full_response, code)
+            self.current_user_message = None
+
+            await self.send(text_data=json.dumps({
+                "type": "complete",
+                "code": code,
+                "message_id": assistant_msg.id,
+            }))
+
+        finally:
+            if self.current_user_message:
+                await self.delete_message(self.current_user_message)
+                self.current_user_message = None
+
+    @database_sync_to_async
+    def check_access(self):
+        return can_access_task(self.user, self.task_id)
+
+    @database_sync_to_async
+    def get_or_create_conversation(self):
+        return get_or_create_active_conversation(self.user, self.task_id)
+
+    @database_sync_to_async
+    def delete_message(self, message):
+        message.delete()
+
+    @database_sync_to_async
+    def save_message(self, role, content, code=None):
+        return Message.objects.create(
+            conversation=self.conversation,
+            role=role,
+            content=content,
+            code_html=code.get("html") if code else None,
+            code_css=code.get("css") if code else None,
+            code_js=code.get("js") if code else None,
+        )
+
+    @database_sync_to_async
+    def get_history(self):
+        messages = self.conversation.messages.filter(source="conversation")
+        return [
+            {
+                "id": m.id,
+                "role": m.role,
+                "content": m.content,
+                "code": {
+                    "html": m.code_html,
+                    "css": m.code_css,
+                    "js": m.code_js,
+                } if m.role == "assistant" else None,
+                "submitted": m.submission_id is not None if m.role == "assistant" else None,
+                "created": m.created.isoformat(),
+            }
+            for m in messages
+        ]
+
+    @database_sync_to_async
+    def get_history_for_llm(self):
+        messages = self.conversation.messages.filter(source="conversation")
+        return [{"role": m.role, "content": m.content} for m in messages]
+
+
+class GuidanceConsumer(AsyncWebsocketConsumer):
+    async def connect(self):
+        self.user = self.scope["user"]
+        if self.user.is_anonymous:
+            await self.close()
+            return
+
+        self.task_id = int(self.scope["url_route"]["kwargs"]["task_id"])
+        if not await self.check_access():
+            await self.close()
+            return
+
+        self.current_user_message = None
+        self.session_messages = []
+        await self.accept()
+
+        self.conversation = await self.get_or_create_conversation()
+        await self.send(text_data=json.dumps({"type": "init", "messages": []}))
+
+    async def disconnect(self, close_code):
+        if self.current_user_message:
+            await self.delete_message(self.current_user_message)
+            self.current_user_message = None
+
+    async def receive(self, text_data):
+        data = json.loads(text_data)
+        prompt = data.get("content", "").strip()
+        if not prompt:
+            return
+
+        self.current_user_message = await self.save_message("user", prompt)
+        self.session_messages.append({"role": "user", "content": prompt})
+
+        try:
+            full_response = ""
+            try:
+                async for chunk in stream_guidance(self.session_messages):
+                    full_response += chunk
+                    await self.send(text_data=json.dumps({
+                        "type": "stream",
+                        "content": chunk,
+                    }))
+            except Exception as e:
+                await self.send(text_data=json.dumps({
+                    "type": "error",
+                    "content": f"AI 服务出错：{str(e)}",
+                }))
+                return
+
+            clean_response, is_ready = parse_guidance_response(full_response)
+            self.session_messages.append({
+                "role": "assistant",
+                "content": clean_response,
+            })
+            assistant_msg = await self.save_message("assistant", clean_response)
+            self.current_user_message = None
+
+            await self.send(text_data=json.dumps({
+                "type": "complete",
+                "message_id": assistant_msg.id,
+                "is_ready": is_ready,
+            }))
+
+        finally:
+            if self.current_user_message:
+                await self.delete_message(self.current_user_message)
+                self.current_user_message = None
+
+    @database_sync_to_async
+    def check_access(self):
+        return can_access_task(self.user, self.task_id)
+
+    @database_sync_to_async
+    def get_or_create_conversation(self):
+        return get_or_create_active_conversation(self.user, self.task_id)
+
+    @database_sync_to_async
+    def delete_message(self, message):
+        message.delete()
+
+    @database_sync_to_async
+    def save_message(self, role, content):
+        return Message.objects.create(
+            conversation=self.conversation,
+            role=role,
+            source="guidance",
+            content=content,
+        )