o h@sddlmZmZmZmZmZddlmZddlZddl m Z m Z ee Z de jd<de jd<ee e d d d Ze d d dZe jddgdddZddlZed ZeeZWdn1shwYe dkre jeddeddeddddSdS))Flaskrender_templatesessionjsonifyrequest)SessionN) list_models pull_modelzyour-secret-key SECRET_KEYZ filesystemZ SESSION_TYPE/cCstdS)Nz index.html)rr r -C:\Users\shubh\Videos\agent\agent_prod\app.pyindex srz /api/modelscCst}td|iS)Nmodels)rr)rr r r get_modelss rz/api/pull-modelPOST)methodsc Csxtjd}|stddidfSzt|tdd|diWSty;}ztdt|idfWYd}~Sd}~ww) NZmodelerrorzModel name requiredimessagezModel z pulled successfullyi)rjsongetrr Exceptionstr)Z model_nameer r r add_models  rz config.yaml__main__serverhostportdebug)rrr)flaskrrrrrZ flask_sessionrosZmodules.llm.ollama_clientrr __name__appconfigrouterrrZyamlopenfZ safe_loadrunr r r r s2