Django + ChatGPT интеграция

pip install openai django

# settings.py
AITUNNEL_API_KEY = "sk-aitunnel-xxx"
AITUNNEL_BASE_URL = "https://api.aitunnel.ru/v1/"

# services/ai_service.py
from django.conf import settings
from openai import OpenAI

class AIService:
    def __init__(self):
        self.client = OpenAI(
            api_key=settings.AITUNNEL_API_KEY,
            base_url=settings.AITUNNEL_BASE_URL
        )
    
    def chat(self, message: str, model: str = "gpt-5.2") -> str:
        response = self.client.chat.completions.create(
            model=model,
            messages=[{"role": "user", "content": message}]
        )
        return response.choices[0].message.content
    
    def summarize(self, text: str) -> str:
        return self.chat(f"Кратко изложи: {text}")
    
    def translate(self, text: str, target_lang: str = "ru") -> str:
        return self.chat(f"Переведи на {target_lang}: {text}")

ai_service = AIService()

# views.py
from django.http import JsonResponse
from django.views.decorators.http import require_POST
from .services.ai_service import ai_service
import json

@require_POST
def chat_view(request):
    data = json.loads(request.body)
    message = data.get("message")
    
    response = ai_service.chat(message)
    return JsonResponse({"response": response})

@require_POST
def summarize_view(request):
    data = json.loads(request.body)
    text = data.get("text")
    
    summary = ai_service.summarize(text)
    return JsonResponse({"summary": summary})

# serializers.py
from rest_framework import serializers

class ChatSerializer(serializers.Serializer):
    message = serializers.CharField()
    model = serializers.CharField(default="gpt-5.2")

# views.py
from rest_framework.views import APIView
from rest_framework.response import Response

class ChatAPIView(APIView):
    def post(self, request):
        serializer = ChatSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        
        response = ai_service.chat(
            serializer.validated_data["message"],
            serializer.validated_data["model"]
        )
        return Response({"response": response})