Merge pull request #48 from vgulerianb/vikrant/add-templates-on-start

chore(Templates): Add templates on boot up

server/app.py

@@ -7,7 +7,7 @@
 from fastapi.templating import Jinja2Templates
 from fastapi.middleware.cors import CORSMiddleware
 from commons import config as c
-from commons.utils import add_default_user
+from commons.utils import add_default_user, add_default_templates
 
 # Routers
 from api.auth import auth_router
@@ -37,6 +37,7 @@
 )
 
 add_default_user()
+add_default_templates()
 
 ####################################################
 ################ INITIALIZE ########################

server/commons/utils.py

@@ -1,11 +1,12 @@
 import jwt
 from passlib.hash import sha256_crypt
 from sqlalchemy.exc import IntegrityError
-from database import db_session, User, Prompt, IntermediateStep
+from database import db_session, User, Prompt, IntermediateStep, Template
 from commons import config as c
 import database_constants as constants
 from fastapi import HTTPException
 from sqlalchemy import func
+import json
 
 logger = c.get_logger(__name__)
 
@@ -20,6 +21,29 @@ def add_default_user():
     except IntegrityError as e:
         logger.info("Not adding default user")
 
+def add_default_templates():
+    db = db_session()
+    try:
+        with open("./examples/index.json") as f:
+            data = json.load(f)
+        for template_data in data:
+            template = db.query(Template).filter_by(id=template_data["id"]).first()
+            if template:
+                template.name = template_data["name"]
+                template.description = template_data["description"]
+                with open("./examples/"+template_data['dag']) as f:
+                    dag = json.load(f)
+                template.dag = dag
+            else:
+                with open("./examples/"+template_data["dag"]) as f:
+                    dag = json.load(f)
+                template = Template(name=template_data["name"], description=template_data["description"], dag=dag)
+                db.add(template)
+        db.commit()
+    except IntegrityError as e:
+        logger.info("Not adding default templates")
+
+
 
 def get_user_from_jwt(token):
     try:

server/examples/index.json

@@ -0,0 +1,8 @@
+[
+    {
+        "id": 1,
+        "name": "Base Template",
+        "description": "This is a basic prompt to start from. It contains just a simple conversational chain.",
+        "dag": "basic_template.json"
+    }
+]