-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* migrations for file, knowledge_category and mapping embeddings to a file * document upload now maps embeddings to its file model * resetting the correct migrations * added organization to knowledge category * api to add a knowledge category for an org * validation; knowledge category names to be unique * api to delete the knowledge category; will wipe out all files and embeddings * added knowledge category filter in create chat api and also added a get all categories api * api to get all files of an organization * api to delete a file; category compulsory for an org * use category_id in document upload and chat api
- Loading branch information
1 parent
7896250
commit 7ab0c1a
Showing
5 changed files
with
362 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
44 changes: 44 additions & 0 deletions
44
llm/migrations/0013_knowledgecategory_file_embedding_file.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,44 @@ | ||
# Generated by Django 4.2.6 on 2024-04-17 03:00 | ||
|
||
from django.db import migrations, models | ||
import django.db.models.deletion | ||
import uuid | ||
|
||
|
||
class Migration(migrations.Migration): | ||
|
||
dependencies = [ | ||
('llm', '0012_embedding_num_tokens'), | ||
] | ||
|
||
operations = [ | ||
migrations.CreateModel( | ||
name='KnowledgeCategory', | ||
fields=[ | ||
('id', models.AutoField(primary_key=True, serialize=False)), | ||
('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)), | ||
('name', models.CharField(default='default', max_length=255, unique=True)), | ||
('org', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='llm.organization')), | ||
], | ||
options={ | ||
'db_table': 'knowledge_category', | ||
}, | ||
), | ||
migrations.CreateModel( | ||
name='File', | ||
fields=[ | ||
('id', models.AutoField(primary_key=True, serialize=False)), | ||
('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)), | ||
('name', models.CharField(max_length=255)), | ||
('knowledge_category', models.ForeignKey(null=True, on_delete=django.db.models.deletion.SET_NULL, to='llm.knowledgecategory')), | ||
], | ||
options={ | ||
'db_table': 'files', | ||
}, | ||
), | ||
migrations.AddField( | ||
model_name='embedding', | ||
name='file', | ||
field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='llm.file'), | ||
), | ||
] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
# Generated by Django 4.2.6 on 2024-04-17 03:38 | ||
|
||
from django.db import migrations, models | ||
import django.db.models.deletion | ||
|
||
|
||
class Migration(migrations.Migration): | ||
|
||
dependencies = [ | ||
('llm', '0013_knowledgecategory_file_embedding_file'), | ||
] | ||
|
||
operations = [ | ||
migrations.AlterField( | ||
model_name='file', | ||
name='knowledge_category', | ||
field=models.ForeignKey(null=True, on_delete=django.db.models.deletion.CASCADE, to='llm.knowledgecategory'), | ||
), | ||
] |
Oops, something went wrong.