From 008410f8829d166beb978afa44f1b3f07b2f1aba Mon Sep 17 00:00:00 2001
From: Vignesh Aigal <vignesh.aigal@gmail.com>
Date: Wed, 20 Mar 2024 23:11:06 -0700
Subject: [PATCH] Remove old unwanted tables

---
 .../datasources/migrations/0003_userfiles.py  | 29 -------------------
 .../migrations/0004_alter_userfiles_user.py   | 21 --------------
 .../migrations/0005_delete_userfiles.py       | 16 ----------
 llmstack/datasources/models.py                | 16 ----------
 4 files changed, 82 deletions(-)
 delete mode 100644 llmstack/datasources/migrations/0003_userfiles.py
 delete mode 100644 llmstack/datasources/migrations/0004_alter_userfiles_user.py
 delete mode 100644 llmstack/datasources/migrations/0005_delete_userfiles.py

diff --git a/llmstack/datasources/migrations/0003_userfiles.py b/llmstack/datasources/migrations/0003_userfiles.py
deleted file mode 100644
index 1bda2cc091d..00000000000
--- a/llmstack/datasources/migrations/0003_userfiles.py
+++ /dev/null
@@ -1,29 +0,0 @@
-# Generated by Django 4.2.10 on 2024-03-16 10:53
-
-from django.conf import settings
-from django.db import migrations, models
-import django.db.models.deletion
-import llmstack.datasources.models
-import uuid
-
-
-class Migration(migrations.Migration):
-
-    dependencies = [
-        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
-        ('datasources', '0002_datasource_config_and_more'),
-    ]
-
-    operations = [
-        migrations.CreateModel(
-            name='UserFiles',
-            fields=[
-                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
-                ('uuid', models.UUIDField(default=uuid.uuid4, editable=False, help_text='UUID of the asset')),
-                ('file', models.FileField(blank=True, null=True, storage=llmstack.datasources.models.select_storage, upload_to=llmstack.datasources.models.upload_to)),
-                ('metadata', models.JSONField(blank=True, default=dict, help_text='Metadata for the asset', null=True)),
-                ('created_at', models.DateTimeField(auto_now_add=True)),
-                ('user', models.OneToOneField(help_text='User this asset belongs to', on_delete=django.db.models.deletion.DO_NOTHING, to=settings.AUTH_USER_MODEL)),
-            ],
-        ),
-    ]
diff --git a/llmstack/datasources/migrations/0004_alter_userfiles_user.py b/llmstack/datasources/migrations/0004_alter_userfiles_user.py
deleted file mode 100644
index 0ebfea1f681..00000000000
--- a/llmstack/datasources/migrations/0004_alter_userfiles_user.py
+++ /dev/null
@@ -1,21 +0,0 @@
-# Generated by Django 4.2.10 on 2024-03-17 16:02
-
-from django.conf import settings
-from django.db import migrations, models
-import django.db.models.deletion
-
-
-class Migration(migrations.Migration):
-
-    dependencies = [
-        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
-        ('datasources', '0003_userfiles'),
-    ]
-
-    operations = [
-        migrations.AlterField(
-            model_name='userfiles',
-            name='user',
-            field=models.ForeignKey(help_text='User this asset belongs to', on_delete=django.db.models.deletion.DO_NOTHING, to=settings.AUTH_USER_MODEL),
-        ),
-    ]
diff --git a/llmstack/datasources/migrations/0005_delete_userfiles.py b/llmstack/datasources/migrations/0005_delete_userfiles.py
deleted file mode 100644
index 93c4311de4f..00000000000
--- a/llmstack/datasources/migrations/0005_delete_userfiles.py
+++ /dev/null
@@ -1,16 +0,0 @@
-# Generated by Django 4.2.10 on 2024-03-21 00:01
-
-from django.db import migrations
-
-
-class Migration(migrations.Migration):
-
-    dependencies = [
-        ('datasources', '0004_alter_userfiles_user'),
-    ]
-
-    operations = [
-        migrations.DeleteModel(
-            name='UserFiles',
-        ),
-    ]
diff --git a/llmstack/datasources/models.py b/llmstack/datasources/models.py
index 214a54600bd..8e27f6e2476 100644
--- a/llmstack/datasources/models.py
+++ b/llmstack/datasources/models.py
@@ -176,19 +176,3 @@ def user_can_read(self, user) -> bool:
                 ).organization
             )
         return False
-
-
-def select_storage():
-    from django.core.files.storage import storages
-
-    return storages["useruploads"]
-
-
-def upload_to(instance, filename):
-    return "/".join(
-        [
-            str(instance.profile_uuid),
-            instance.path,
-            filename,
-        ]
-    )