You've already forked RekomenciBackend
fix add resume
This commit is contained in:
@@ -29,7 +29,7 @@ class MlApiGateway:
|
||||
self._client = client
|
||||
|
||||
async def generate_embedding(self, text: str) -> list[float]:
|
||||
response = await self._client.post("/get_embedding", json={"text": text})
|
||||
response = await self._client.post("/get_embedding", json={"text": text}, timeout=100)
|
||||
return cast(list[float], response.json()["embedding"])
|
||||
|
||||
async def generate_resume_prediction(
|
||||
@@ -39,9 +39,9 @@ class MlApiGateway:
|
||||
suituble_vacancies: Sequence[SuitableVacancyDs],
|
||||
) -> GenerateResumePredictionResponse:
|
||||
response = await self._client.post(
|
||||
"/predict_salary",
|
||||
"/predict",
|
||||
json={
|
||||
"resume_id": resume_id,
|
||||
"resume_id": str(resume_id),
|
||||
"key_skills": key_skills,
|
||||
"vacancies": [
|
||||
{
|
||||
@@ -53,6 +53,7 @@ class MlApiGateway:
|
||||
} for suituble_vacancy in suituble_vacancies
|
||||
],
|
||||
},
|
||||
timeout=100,
|
||||
)
|
||||
|
||||
response_json = response.json()
|
||||
|
||||
Reference in New Issue
Block a user