sajad-dev
2c42ebe01c
- Removed deprecated user_info files and paths from configuration. - Added user soil data integration in chat context to improve response accuracy. - Updated build_rag_context and chat_rag_stream functions to include sensor_uuid for user-specific data retrieval. - Enhanced load_sources function to load user data from the database. - Implemented filtering in search_with_query and QdrantVectorStore to isolate user data based on sensor_uuid. - Introduced Celery Beat schedule for periodic user data ingestion.
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
"""
|
|
ویوهای RAG — چت با استریم
|
|
"""
|
|
from django.http import StreamingHttpResponse
|
|
from rest_framework import status
|
|
from rest_framework.request import Request
|
|
from rest_framework.response import Response
|
|
from rest_framework.views import APIView
|
|
|
|
from .chat import chat_rag_stream
|
|
|
|
|
|
class ChatView(APIView):
|
|
"""
|
|
چت RAG با استریم.
|
|
POST با {"message": "متن سوال", "sensor_uuid": "uuid-سنسور"}
|
|
sensor_uuid اجباری — هر کاربر فقط به دیتای خودش دسترسی دارد.
|
|
"""
|
|
|
|
def post(self, request: Request):
|
|
data = request.data if request.method == "POST" else request.query_params
|
|
message = data.get("message")
|
|
sensor_uuid = data.get("sensor_uuid")
|
|
|
|
if not message or not isinstance(message, str):
|
|
return Response(
|
|
{"code": 400, "msg": "پارامتر message الزامی است."},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
message = str(message).strip()
|
|
if not message:
|
|
return Response(
|
|
{"code": 400, "msg": "پیام نباید خالی باشد."},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
if not sensor_uuid or not isinstance(sensor_uuid, str):
|
|
return Response(
|
|
{"code": 400, "msg": "پارامتر sensor_uuid الزامی است."},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
sensor_uuid = str(sensor_uuid).strip()
|
|
if not sensor_uuid:
|
|
return Response(
|
|
{"code": 400, "msg": "sensor_uuid نباید خالی باشد."},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def generate():
|
|
try:
|
|
for chunk in chat_rag_stream(message, sensor_uuid=sensor_uuid):
|
|
yield chunk
|
|
except Exception as e:
|
|
yield f"\n[خطا: {e}]"
|
|
|
|
return StreamingHttpResponse(
|
|
generate(),
|
|
content_type="text/plain; charset=utf-8",
|
|
)
|