Tengo un fastapi para hacer predicciones y devolver la salida como respuesta, pero he implementado la verificación de entrada donde, si el usuario da una entrada no admitida, devuelve una Invalid smile , pero el problema aquí es que el diccionario de respuestas no se reemplaza.
cuando hago la predicción obtuve esta respuesta
{"result":{"interaction_map":[[15.0,5.0,14.0,15.0,15.0],[19.0,7.0,20.0,19.0,19.0],[13.0,6.0,18.0,13.0,13.0],[15.0,5.0,14.0,15.0,15.0],[15.0,5.0,14.0,15.0,15.0]],"predictions":-3.405024290084839}}Pero cuando doy una entrada incorrecta, obtuve esta respuesta
{"result":{"interaction_map":[[15.0,5.0,14.0,15.0,15.0],[19.0,7.0,20.0,19.0,19.0],[13.0,6.0,18.0,13.0,13.0],[15.0,5.0,14.0,15.0,15.0],[15.0,5.0,14.0,15.0,15.0]],"predictions":"invalid SMILES"}}pero espero esta respuesta
{"predictions":"invalid SMILES"}este es el codigo que estoy usando
response = {} async def predictions(solute, solvent): m = Chem.MolFromSmiles(solute,sanitize=False) n = Chem.MolFromSmiles(solvent,sanitize=False) if (m == None or n == None): response['predictions']= 'invalid SMILES' print('invalid SMILES') else: mol = Chem.MolFromSmiles(solute) mol = Chem.AddHs(mol) solute = Chem.MolToSmiles(mol) solute_graph = get_graph_from_smile(solute) mol = Chem.MolFromSmiles(solvent) mol = Chem.AddHs(mol) solvent = Chem.MolToSmiles(mol) solvent_graph = get_graph_from_smile(solvent) delta_g, interaction_map = model([solute_graph.to(device), solvent_graph.to(device)]) interaction_map_one = torch.trunc(interaction_map) response["interaction_map"] = (interaction_map_one.detach().numpy()).tolist() response["predictions"] = delta_g.item() @app.get('/predict_solubility') async def post(): return {'result': response} @app.get('/predict') async def predict(background_tasks: BackgroundTasks,solute,solvent): background_tasks.add_task(predictions,solute,solvent) return {'success'}El problema es que la response es una variable global, por lo que los elementos que escribe en su primera solicitud aún permanecen allí en la segunda solicitud.
Una solución rápida sería borrar el dictado de response al comienzo de la solicitud a /predict_solubility :
async def predictions(solute, solvent): response.clear() Pero, en general, me parece problemático que una sola solicitud para /predict establezca algún tipo de estado global que podría sobrescribirse, en lugar de devolver un jobid de trabajo donde puede verificar un trabajo/tarea específico.