← [BACK TO REPOSITORY]    Media: app.cpython-310.pyc (1.4 KB)
Format: application/x-python-code
⬇ Download File (1.4 KB) 
   1  o
   2  
   3  ��h��@sddlmZmZmZmZmZddlmZddlZddl	m
   4  Z
   5  mZee�Z
   6  de
   7  jd<de
   8  jd<ee
   9  �e
  10  �d	�d
  11  d��Ze
  12  �d�d
  13  d��Ze
  14  jddgd�dd��ZddlZed��
  15  Ze�e�ZWd�n1shwYedkr�e
  16  jeddeddeddd�dSdS)�)�Flask�render_template�session�jsonify�request)�SessionN)�list_models�
  17  pull_modelzyour-secret-key�
  18  SECRET_KEYZ
  19  filesystemZSESSION_TYPE�/cCstd�S)Nz
  20  index.html)r�rr�-C:\Users\shubh\Videos\agent\agent_prod\app.py�index
  21  srz/api/modelscCst�}td|i�S)N�models)rr)rrrr
  22  �
  23  get_modelssrz/api/pull-model�POST)�methodsc
  24  Csxtj�d�}|stddi�dfSzt|�tdd|�d�i�WSty;}ztdt|�i�dfWYd}~Sd}~ww)	NZmodel�errorzModel name requiredi��messagezModel z pulled successfullyi�)r�json�getrr	�	Exception�str)Z
  25  model_name�errr
  26  �	add_models ��rzconfig.yaml�__main__�server�host�port�debug)rrr)�flaskrrrrrZ
  27  flask_sessionr�osZmodules.llm.ollama_clientrr	�__name__�app�config�routerrrZyaml�open�fZ	safe_load�runrrrr
  28  �<module>s2
  29  
  30  
  31  
  32  
  33  
  34  �
  35  
  36  
  37  
  38  ��