Add both the /summary and /tags functions

- Add basic incomplete post body testing.
  Running them for real would require OpenAI calls

Author: chiroptical

llm/.gitignore

@@ -1,2 +1,3 @@
 llm-api/
 __pycache__/
+databases/

llm/api.py

@@ -1,23 +1,42 @@
-from flask import Flask, jsonify, abort
-from llm_functions import get_summaries_and_tags_api_function
+from flask import Flask, jsonify, abort, request
+from llm_functions import get_summary_api_function, get_tags_api_function
+import json
 
 app = Flask(__name__)
 
-@app.route('/summary', methods=['POST'])
+
+def is_intersection(keys, required_keys):
+    return (keys & required_keys) == required_keys
+
+
+@app.route("/summary", methods=["POST"])
 def summary():
     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
-    if not (body.keys() & {"bill_id", "bill_title", "bill_text"}):
+    if not is_intersection(body.keys(), {"bill_id", "bill_title", "bill_text"}):
         abort(404, description="requires bill_id, bill_title, and bill_text")
 
-    summary = get_summaries_and_tags_api_function(
-        body["bill_id"],
-        body["bill_title"],
-        body["bill_text"]
+    summary = get_summary_api_function(
+        body["bill_id"], body["bill_title"], body["bill_text"]
     )
 
     if summary["status"] in [-1, -2]:
-        abort(404, description="unable to generate summary or tags")
-    
-    return jsonify(summary)
+        abort(500, description="Unable to generate summary")
+
+    return jsonify(summary["summary"])
+
+
+@app.route("/tags", methods=["POST"])
+def tags():
+    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
+    if not is_intersection(body.keys(), {"bill_id", "bill_title", "bill_text"}):
+        abort(404, description="requires bill_id, bill_title, and bill_text")
+
+    tags = get_tags_api_function(body["bill_id"], body["bill_title"], body["bill_text"])
+
+    if tags["status"] in [-1, -2]:
+        abort(500, description="Unable to generate tags")
+
+    return jsonify(tags["tags"])

llm/requirements.txt

@@ -184,3 +184,4 @@ zipp==3.19.1
 Flask==3.1.0
 itsdangerous==2.2.0
 Werkzeug==3.1.3
+pytest==8.3.5

llm/test_api.py

@@ -0,0 +1,27 @@
+import pytest
+from api import app
+
+
+@pytest.fixture
+def client():
+    return app.test_client()
+
+
+def test_summary_returns_404_with_empty_body(client):
+    response = client.post("/summary", json={})
+    assert response.status_code == 404
+
+
+def test_summary_returns_404_with_partial_body(client):
+    response = client.post("/summary", json={"bill_id": "bill"})
+    assert response.status_code == 404
+
+
+def test_tags_returns_404_with_empty_body(client):
+    response = client.post("/tags", json={})
+    assert response.status_code == 404
+
+
+def test_tags_returns_404_with_partial_body(client):
+    response = client.post("/tags", json={"bill_id": "bill"})
+    assert response.status_code == 404