Initial working commit

Author: chiroptical

infra/docker-compose.yml

@@ -29,7 +29,7 @@ services:
     build:
       context: ..
       dockerfile: infra/Dockerfile.firebase
-    command: yarn emulators:start
+    command: yarn emulators:start --debug # TODO: REMOVE THIS
     working_dir: /app
     depends_on:
       init:

infra/firebase.compose.json

@@ -12,7 +12,7 @@
       "runtime": "nodejs18"
     },
     {
-      "predeploy": ["source /app/llm/venv/bin/activate && python3.10 -m pip install -r /app/llm/requirements.txt"],
+      "predeploy": [". llm/venv/bin/activate && python3.10 -m pip install -r llm/requirements.txt"],
       "source": "llm",
       "codebase": "maple-llm",
       "runtime": "python310"

llm/main.py

@@ -3,6 +3,7 @@
 import json
 from firebase_admin import initialize_app
 from firebase_functions import https_fn, options
+import os
 
 initialize_app()
 app = Flask(__name__)
@@ -12,8 +13,14 @@ def is_intersection(keys, required_keys):
     return (keys & required_keys) == required_keys
 
 
+def set_openai_api_key():
+    if os.environ.get("OPENAI_DEV") != None:
+        os.environ["OPENAI_API_KEY"] = os.environ["OPENAI_DEV"]
+
+
 @app.route("/summary", methods=["POST"])
 def summary():
+    set_openai_api_key()
     body = json.loads(request.data)
     # We require bill_id, bill_title, bill_text to exist as keys in the POST
     if not is_intersection(body.keys(), {"bill_id", "bill_title", "bill_text"}):
@@ -28,8 +35,10 @@ def summary():
 
     return jsonify(summary["summary"])
 
+
 @app.route("/tags", methods=["POST"])
 def tags():
+    set_openai_api_key()
     body = json.loads(request.data)
     # We require bill_id, bill_title, bill_text to exist as keys in the POST
     # Note: & is essentially set intersection
@@ -43,11 +52,15 @@ def tags():
 
     return jsonify(tags["tags"])
 
+
 @app.route("/ready", methods=["GET"])
 def ready():
     return ""
 
-@https_fn.on_request(secrets=["OPENAI_DEV"], timeout_sec=300, memory=options.MemoryOption.GB_1)
+
+@https_fn.on_request(
+    secrets=["OPENAI_DEV"], timeout_sec=300, memory=options.MemoryOption.GB_1
+)
 def httpsflaskexample(req: https_fn.Request) -> https_fn.Response:
     with app.request_context(req.environ):
         return app.full_dispatch_request()