Select Git revision
header-snippet.html
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)),
],
),
]