From 11c740732288c6801d9eeb60314c67ad333599e6 Mon Sep 17 00:00:00 2001 From: "Julien B." <xm9q8f80@jlnbrtn.me> Date: Sun, 25 Aug 2024 19:47:36 +0200 Subject: [PATCH] fix(endpoints): add inferer config endpoint --- api/models/InfererConfig.py | 6 ++++++ api/routers/endpoints.py | 13 ++++++++++++- 2 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 api/models/InfererConfig.py diff --git a/api/models/InfererConfig.py b/api/models/InfererConfig.py new file mode 100644 index 0000000..88b8d58 --- /dev/null +++ b/api/models/InfererConfig.py @@ -0,0 +1,6 @@ +from pydantic import BaseModel + +class InfererConfig(BaseModel): + provider: str + model_id: str + server_url: str \ No newline at end of file diff --git a/api/routers/endpoints.py b/api/routers/endpoints.py index f4007ea..b1659d4 100644 --- a/api/routers/endpoints.py +++ b/api/routers/endpoints.py @@ -1,7 +1,9 @@ from fastapi import APIRouter +from api.internal_services import database from api.internal_services.background_worker import add_job_to_queue from api.internal_services.logger import logger import uuid +from api.models.InfererConfig import InfererConfig from api.models.Job import Job, JobType from api.models.Sentence import Sentence from api.models.TrainingBody import TrainingBody @@ -34,4 +36,13 @@ def add_sentence_to_process(training_body: TrainingBody): "huggingface_token": training_body.huggingface_token, } add_job_to_queue(new_job) - return {"message": "Job added to the queue for processing.", "job_id": new_job.job_id} \ No newline at end of file + return {"message": "Job added to the queue for processing.", "job_id": new_job.job_id} + +@router.post("/actions/inferer/config") +def add_sentence_to_process(infererConfig: InfererConfig): + new_config = {'model_id': infererConfig.provider} + if infererConfig.provider == 'BERT': + new_config['model_id'] = infererConfig.model_id + new_config['server_url'] = infererConfig.server_url + + database.update_annotator_config({'model_id': 'GPT'}) \ No newline at end of file -- GitLab