Delete unused models (#267)

Author: vegito22

llmstack/processors/admin.py

@@ -5,8 +5,6 @@
     ApiProvider,
     Endpoint,
     Feedback,
-    Request,
-    Response,
     RunEntry,
     ShareTag,
     VersionedEndpoint,
@@ -17,7 +15,5 @@
 admin.site.register(Endpoint)
 admin.site.register(VersionedEndpoint)
 admin.site.register(Feedback)
-admin.site.register(Request)
-admin.site.register(Response)
 admin.site.register(RunEntry)
 admin.site.register(ShareTag)

llmstack/processors/migrations/0010_remove_response_request_remove_feedback_request_and_more.py

@@ -0,0 +1,35 @@
+# Generated by Django 4.2.14 on 2024-08-09 18:17
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apiabstractor', '0009_remove_runentry_processor_runs'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='response',
+            name='request',
+        ),
+        migrations.RemoveField(
+            model_name='feedback',
+            name='request',
+        ),
+        migrations.RemoveField(
+            model_name='testresult',
+            name='request',
+        ),
+        migrations.RemoveField(
+            model_name='testresult',
+            name='response',
+        ),
+        migrations.DeleteModel(
+            name='Request',
+        ),
+        migrations.DeleteModel(
+            name='Response',
+        ),
+    ]

llmstack/processors/models.py

@@ -219,92 +219,6 @@ def __str__(self):
         return self.version.__str__()
 
 
-class Request(models.Model):
-    """
-    Model representing the request made against a versioned endpoint
-    """
-
-    from llmstack.apps.models import AppSession
-
-    endpoint = models.ForeignKey(
-        Endpoint,
-        on_delete=models.DO_NOTHING,
-        help_text="Version of endpoint this request made for",
-    )
-    param_values = models.JSONField(
-        default=dict,
-        help_text="Override param values with these",
-    )
-    prompt_values = models.JSONField(
-        default=dict,
-        help_text="Values for placeholders in the prompt",
-    )
-    input = models.JSONField(
-        default=dict,
-        help_text="Input to the API",
-    )
-    config = models.JSONField(
-        default=dict,
-        help_text="Configuration for this endpoint. Values for this will be set in endpoint when it is created",
-        blank=True,
-        null=True,
-    )
-    template_values = models.JSONField(
-        default=dict,
-        help_text="Values for placeholders in the input",
-        blank=True,
-        null=True,
-    )
-    created_on = models.DateTimeField(auto_now_add=True)
-    app_session = models.ForeignKey(
-        AppSession,
-        on_delete=models.DO_NOTHING,
-        help_text="App session this request was made for",
-        null=True,
-        blank=True,
-        default=None,
-    )
-    app_session_key = models.CharField(
-        max_length=100,
-        help_text="App session this request was made for",
-        null=True,
-        blank=True,
-        default=None,
-    )
-    app_id = models.IntegerField(
-        default=None,
-        null=True,
-        blank=True,
-        help_text="App this request was made for",
-    )
-
-    def __str__(self):
-        return self.endpoint.name + ":" + str(self.endpoint.version)
-
-
-class Response(models.Model):
-    """
-    Model that captures the repsonse from API backend as well as the response sent to the user
-    """
-
-    request = models.ForeignKey(
-        Request,
-        on_delete=models.DO_NOTHING,
-        help_text="Request this response corresponds to",
-    )
-    raw_response = models.TextField(help_text="Raw response JSON from backend")
-    processed_response = models.TextField(
-        help_text="Output returned to the user after running the post processor",
-    )
-    response_code = models.IntegerField(
-        help_text="Response code from the API backend",
-    )
-    created_on = models.DateTimeField(auto_now_add=True)
-
-    def __str__(self):
-        return self.request.__str__()
-
-
 class RunEntry(models.Model):
     """
     Represents a run of an app or an endpoint
@@ -478,11 +392,6 @@ class Feedback(models.Model):
     This is used to collect feedback about the response generated by backend. User can later collect these feedbacks and use them for finetuning
     """
 
-    request = models.ForeignKey(
-        Request,
-        on_delete=models.DO_NOTHING,
-        help_text="Request object this feedback is collected against",
-    )
     response_quality = models.CharField(
         max_length=100,
         help_text="Quality rating for the response",
@@ -584,16 +493,6 @@ class TestResult(models.Model):
         default=uuid.uuid4,
         help_text="Test Case Identifier",
     )
-    request = models.ForeignKey(
-        Request,
-        on_delete=models.DO_NOTHING,
-        help_text="Request made for the test case",
-    )
-    response = models.ForeignKey(
-        Response,
-        on_delete=models.DO_NOTHING,
-        help_text="Response for the test run",
-    )
     testrun = models.ForeignKey(
         TestRun,
         on_delete=models.DO_NOTHING,

llmstack/processors/serializers.py

@@ -5,7 +5,7 @@
 
 from llmstack.processors.providers.api_processors import ApiProcessorFactory
 
-from .models import ApiBackend, ApiProvider, Endpoint, Request, Response, RunEntry
+from .models import ApiBackend, ApiProvider, Endpoint, RunEntry
 
 
 class ApiProviderSerializer(serializers.ModelSerializer):
@@ -131,34 +131,6 @@ class Meta:
         ]
 
 
-class RequestSerializer(serializers.ModelSerializer):
-    endpoint = EndpointSerializer()
-
-    class Meta:
-        model = Request
-        fields = [
-            "endpoint",
-            "input",
-            "param_values",
-            "prompt_values",
-            "created_on",
-        ]
-
-
-class ResponseSerializer(serializers.ModelSerializer):
-    request = RequestSerializer()
-
-    class Meta:
-        model = Response
-        fields = [
-            "request",
-            "raw_response",
-            "processed_response",
-            "response_code",
-            "created_on",
-        ]
-
-
 class HistorySerializer(serializers.ModelSerializer):
     app_detail = serializers.SerializerMethodField()
     processor_runs = serializers.SerializerMethodField()