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