Skip to content
Snippets Groups Projects
Select Git revision
  • 0ba6aada835fd97eed6dc659a7fc830c6f806ea2
  • main default protected
2 results

header-snippet.html

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    0001_initial.py 3.23 KiB
    # Generated by Django 4.2.7 on 2023-12-02 15:36
    
    from django.db import migrations, models
    
    
    class Migration(migrations.Migration):
        initial = True
    
        dependencies = []
    
        operations = [
            migrations.CreateModel(
                name="ModelConfig",
                fields=[
                    (
                        "id",
                        models.BigAutoField(
                            auto_created=True,
                            primary_key=True,
                            serialize=False,
                            verbose_name="ID",
                        ),
                    ),
                    (
                        "model_type",
                        models.CharField(
                            choices=[
                                ("MOCK", "Mock Model"),
                                ("TGI", "Text Generation Inference Model"),
                            ],
                            default="TGI",
                            max_length=4,
                        ),
                    ),
                    ("system_prompt", models.TextField(default="This is a test prompt.")),
                ],
            ),
            migrations.CreateModel(
                name="OpenSearchRetrievalConfig",
                fields=[
                    (
                        "id",
                        models.BigAutoField(
                            auto_created=True,
                            primary_key=True,
                            serialize=False,
                            verbose_name="ID",
                        ),
                    ),
                    ("operator", models.CharField(default="or", max_length=10)),
                    ("indices", models.TextField(default="development")),
                    ("analyzer", models.CharField(default="english", max_length=50)),
                    ("fuzziness", models.CharField(default="AUTO", max_length=10)),
                    ("prefix_length", models.IntegerField(default=2)),
                    ("minimum_should_match", models.IntegerField(default=1)),
                    ("max_length", models.IntegerField(default=5000)),
                ],
            ),
            migrations.CreateModel(
                name="RetrievalConfig",
                fields=[
                    (
                        "id",
                        models.BigAutoField(
                            auto_created=True,
                            primary_key=True,
                            serialize=False,
                            verbose_name="ID",
                        ),
                    ),
                    (
                        "retrieval_type",
                        models.CharField(
                            choices=[
                                ("MOCK", "Mock Retrieval"),
                                ("OPNS", "OpenSearch Retrieval"),
                            ],
                            default="MOCK",
                            max_length=4,
                        ),
                    ),
                ],
            ),
            migrations.CreateModel(
                name="TgiConfig",
                fields=[
                    (
                        "id",
                        models.BigAutoField(
                            auto_created=True,
                            primary_key=True,
                            serialize=False,
                            verbose_name="ID",
                        ),
                    ),
                    ("max_new_tokens", models.IntegerField(default=512)),
                ],
            ),
        ]