refactor: removendo códigos de testes

endpoint_teste/models.py

@@ -1,8 +0,0 @@
-from django.db import models
-
-# Create your models here.
-class EndpointTesteModel(models.Model):
-    teste = models.CharField(blank=False, max_length=300)
-    
-    def __str__(self):
-        return self.teste

{endpoint_teste → pdfs}/admin.py

@@ -1,7 +1,7 @@
 from django.contrib import admin
 
-from endpoint_teste.models import EndpointTesteModel
+# from pdfs.models import PDFsModel
 
 # Register your models here.
 
-admin.site.register(EndpointTesteModel)
+# admin.site.register(PDFsModel)

{endpoint_teste → pdfs}/apps.py

@@ -1,6 +1,6 @@
 from django.apps import AppConfig
 
 
-class EndpointTesteConfig(AppConfig):
+class PdfsConfig(AppConfig):
     default_auto_field = "django.db.models.BigAutoField"
-    name = "endpoint_teste"
+    name = "pdfs"

pdfs/migrations/0002_delete_endpointtestemodel.py

@@ -0,0 +1,16 @@
+# Generated by Django 4.1 on 2024-11-16 00:46
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('pdfs', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.DeleteModel(
+            name='EndpointTesteModel',
+        ),
+    ]

pdfs/models.py

@@ -0,0 +1,4 @@
+from django.db import models
+
+# Create your models here.
+# class PDFsModel(models.Model):

{endpoint_teste → pdfs}/serializer.py

@@ -1,20 +1,6 @@
 from rest_framework import serializers
 
-from endpoint_teste.models import EndpointTesteModel
-
-class EndpointTesteSerializer(serializers.ModelSerializer):
-    class Meta:
-        model = EndpointTesteModel
-        fields = "__all__"
-    
-class TesteSerializer(serializers.Serializer):
-    history_chat = serializers.ListField(read_only=True, required=False)
-    system_prompt = serializers.CharField(required=True)
-    user_message = serializers.CharField(required=True)
-    pdf_url = serializers.CharField(required=False)
-
 class PDFUploadSerializer(serializers.Serializer):
-    # file = serializers.FileField()
     files = serializers.ListField(child=serializers.FileField(), required=True)
     system_prompt = serializers.CharField(required=True)
     user_message = serializers.CharField(required=True)

{endpoint_teste → pdfs}/views.py

@@ -1,8 +1,5 @@
 import tempfile, os
-from rest_framework import viewsets, filters
-from django_filters.rest_framework import DjangoFilterBackend
-from endpoint_teste.models import EndpointTesteModel
-from endpoint_teste.serializer import EndpointTesteSerializer, PDFUploadSerializer
+from pdfs.serializer import PDFUploadSerializer
 from setup.environment import default_model
 from drf_spectacular.utils import extend_schema
 
@@ -11,52 +8,6 @@ from rest_framework.parsers import MultiPartParser
 from rest_framework.response import Response
 
 from _utils.main import get_llm_answer
-from .serializer import TesteSerializer
-from langchain_huggingface import HuggingFaceEndpoint
-
-class EndpointTesteViewSet(viewsets.ModelViewSet):
-    """Mostrará todas as tarefas"""
-    queryset = EndpointTesteModel.objects.order_by("id").all()
-    serializer_class = EndpointTesteSerializer
-    filter_backends = [DjangoFilterBackend, filters.SearchFilter]
-    search_fields = ["id"]
-
-@api_view(["GET", "POST"])
-def getTeste(request):
-    if request.method == "POST":
-        serializer = TesteSerializer(data=request.data)
-        if serializer.is_valid(raise_exception=True):
-            data = request.data
-            pdf_url = None
-            if data["pdf_url"]:
-                pdf_url = data["pdf_url"]
-            resposta_llm = get_llm_answer(data["system_prompt"], data["user_message"], pdf_url)
-            return Response({
-                "Resposta": resposta_llm
-            })
-    if request.method == "GET":
-        # hugging_face_token = os.environ.get("hugging_face_token")
-        # API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B"
-        # headers = {"Authorization": "Bearer " + hugging_face_token}
-        # def query(payload):
-        #     response = requests.post(API_URL, headers=headers, json=payload)
-        #     return response.json()
-            
-        # output = query({
-        #     "inputs": "Can you please let us know more details about your something I don't know",
-        # })
-        # print('output: ', output)
-        # print('output: ', dir(output))
-        llm = HuggingFaceEndpoint(
-            repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
-            task="text-generation",
-            max_new_tokens=100,
-            do_sample=False,
-            huggingfacehub_api_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
-        )
-        result = llm.invoke("Hugging Face is")
-        print('result: ', result)
-        return Response(result)
 
 @extend_schema(
     request=PDFUploadSerializer,

setup/settings.py

@@ -47,7 +47,7 @@ INSTALLED_APPS = [
     "rest_framework",
     "django_filters",
     "corsheaders",
-    "endpoint_teste",
+    "pdfs",
     "drf_spectacular",
     "resumos",
     "modelos_usuarios"
@@ -160,9 +160,6 @@ CORS_ORIGIN_WHITELIST = [
     "http://localhost",
     "https://luanpoppe-projeto-y-teste-01.hf.space",
     "https://*"
-    # "http://localhost:5000",
-    # "https://luan-pessoal.vercel.app",
-    # "http://localhost:5173",
 ]
 
 REST_FRAMEWORK = {

setup/urls.py

@@ -4,7 +4,7 @@ from rest_framework import routers
 from drf_spectacular.views import SpectacularSwaggerView, SpectacularAPIView
 
 
-from endpoint_teste.views import EndpointTesteViewSet, getTeste, getPDF
+from pdfs.views import getPDF
 from resumos.views import ResumoView
 from modelos_usuarios.views import ListCreateModeloUsuarioView, CreateUpdateDeleteModeloUsuarioView, ListModelosPorUsuarioView
 
@@ -16,7 +16,7 @@ urlpatterns = [
     path('swagger/', SpectacularSwaggerView.as_view(url_name='schema'), name='swagger-ui'),
     path("admin/", admin.site.urls),
     path('', include(router.urls)),
-    path("teste", getTeste),
+
     path('pdf', getPDF, name='upload-pdf'),
     path('resumo', ResumoView.as_view(), name='summary-pdf'),
     path("modelo", ListCreateModeloUsuarioView.as_view()),