ts/virgil-ai
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix ai estimation

Browse source
This commit is contained in:
ipu 2025-08-08 18:16:43 +03:00
parent 91abde04a6
commit 29697f6274
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/services/estimation_service_v2.py
View file

@ -1,5 +1,6 @@
from datetime import date from datetime import date
from enum import Enum from enum import Enum
import json
from typing import Optional from typing import Optional
from src.cache.drug_cache import fetch_drug_with_dosage from src.cache.drug_cache import fetch_drug_with_dosage
from src.models import PHQ, Applicant, Plan, EstimationResponse, EstimationDetails, EstimationResult from src.models import PHQ, Applicant, Plan, EstimationResponse, EstimationDetails, EstimationResult
@ -121,11 +122,11 @@ class EstimationService:
"/chat/", "/chat/",
json={"chat_session_id": session_id, "user_message": f"Check if this applicant eligible for insurance:\n {phq.model_dump_json()}"}, json={"chat_session_id": session_id, "user_message": f"Check if this applicant eligible for insurance:\n {phq.model_dump_json()}"},
) )
response_json = response.json() response_json = response.json()['message']
print(f"AI DTQ Response: {response_json}") print(f"AI DTQ Response: {response_json}")
ai_response = json.loads(response_json)
dtq = not response_json["message"]["insurable"] dtq = not ai_response["insurable"]
reason = response_json["message"]["reason"] reason = ai_response["reason"]
return dtq, reason return dtq, reason
except Exception as e: except Exception as e:
print(f"Error in AI DTQ check: {e}") print(f"Error in AI DTQ check: {e}")