Skip to content
Snippets Groups Projects
Commit 11c74073 authored by Julien B.'s avatar Julien B.
Browse files

fix(endpoints): add inferer config endpoint

parent 59374ec1
No related branches found
No related tags found
No related merge requests found
from pydantic import BaseModel
class InfererConfig(BaseModel):
provider: str
model_id: str
server_url: str
\ No newline at end of file
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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment