+
/
diff --git a/llm-server/app.py b/llm-server/app.py
index 83af2b667..445291429 100644
--- a/llm-server/app.py
+++ b/llm-server/app.py
@@ -11,6 +11,7 @@
from utils.base import try_to_match_and_call_api_endpoint
from models.models import AiResponseFormat
from routes.workflow.workflow_controller import workflow
+from routes.swagger_controller.swagger_api import swagger_workflow
import json
from typing import Any, Tuple
from prompts.base import api_base_prompt, non_api_base_prompt
@@ -29,6 +30,8 @@
app = Flask(__name__)
app.register_blueprint(workflow, url_prefix="/workflow")
+app.register_blueprint(swagger_workflow, url_prefix="/swagger_api")
+
## TODO: Implement caching for the swagger file content (no need to load it everytime)
@app.route("/handle", methods=["POST", "OPTIONS"])
diff --git a/llm-server/routes/swagger_controller/swagger_api.py b/llm-server/routes/swagger_controller/swagger_api.py
new file mode 100644
index 000000000..2426d4ee8
--- /dev/null
+++ b/llm-server/routes/swagger_controller/swagger_api.py
@@ -0,0 +1,155 @@
+from flask import Flask, request, jsonify, Blueprint, request, Response
+from flask_pymongo import PyMongo
+
+import json, yaml
+from bson import ObjectId
+
+from utils.db import Database
+from typing import Any
+import requests
+
+db_instance = Database()
+mongo = db_instance.get_db()
+swagger_workflow = Blueprint("swagger_workflow", __name__)
+
+
+@swagger_workflow.route("/", methods=["GET"])
+def get_swagger_files() -> Response:
+ # Get page and page_size query params
+ page = int(request.args.get("page", 1))
+ page_size = int(request.args.get("page_size", 10))
+
+ # Calculate document skip/limit
+ skip = (page - 1) * page_size
+ limit = page_size
+
+ # Query for paginated docs
+ files = [
+ doc.update({"_id": str(doc["_id"])}) or doc
+ for doc in mongo.swagger_files.find({}, {}).skip(skip).limit(limit)
+ ]
+
+ # Get total docs count
+ total = mongo.swagger_files.count_documents({})
+
+ # Prepare response data
+ data = {"total": total, "page": page, "page_size": page_size, "files": files}
+
+ return jsonify(data)
+
+
+@swagger_workflow.route("/", methods=["POST"])
+def add_swagger_file():
+ if request.content_type == "application/json":
+ # JSON file
+ file_content = request.get_json()
+
+ elif "multipart/form-data" in request.content_type:
+ # Uploaded file
+ file = request.files.get("file")
+ if file is None:
+ return jsonify({"error": "File upload is required"}), 400
+
+ if file.filename.endswith(".json"):
+ try:
+ file_content = json.load(file)
+ except json.JSONDecodeError as e:
+ return (
+ jsonify({"error": "Invalid JSON format in the uploaded file"}),
+ 400,
+ )
+
+ elif file.filename.endswith(".yaml") or file.filename.endswith(".yml"):
+ try:
+ file_content = yaml.safe_load(file)
+ except yaml.YAMLError as e:
+ return (
+ jsonify({"error": "Invalid YAML format in the uploaded file"}),
+ 400,
+ )
+
+ else:
+ return jsonify({"error": "Unsupported content type"}), 400
+
+ # Insert into MongoDB
+ mongo.swagger_files.insert_one(file_content)
+
+ return jsonify({"message": "File added successfully"})
+
+
+@swagger_workflow.route("/
", methods=["GET"])
+def get_swagger_file(id: str) -> Response:
+ file = mongo.swagger_files.find_one({"_id": ObjectId(id)})
+ if not file:
+ return jsonify({"message": "Swagger file not found"})
+
+ file["_id"] = str(file["_id"])
+ return jsonify(file)
+
+
+@swagger_workflow.route("/transform/<_id>", methods=["GET"])
+def get_transformed_swagger_file(_id: str) -> Response:
+ swagger_json = mongo.swagger_files.aggregate(
+ [
+ {"$match": {"_id": ObjectId(_id)}},
+ {"$project": {"paths": 1}},
+ {
+ "$project": {
+ "methods": {
+ "$reduce": {
+ "input": {"$objectToArray": "$paths"},
+ "initialValue": [],
+ "in": {
+ "$concatArrays": [
+ "$$value",
+ {
+ "$map": {
+ "input": {"$objectToArray": "$$this.v"},
+ "as": "path",
+ "in": {
+ "$mergeObjects": [
+ "$$path.v",
+ {
+ "method": "$$path.k",
+ "path": "$$this.k",
+ },
+ ]
+ },
+ }
+ },
+ ]
+ },
+ }
+ }
+ }
+ },
+ {
+ "$project": {
+ "methods.requestBody": 0,
+ "methods.responses": 0,
+ "methods.security": 0,
+ }
+ },
+ ]
+ )
+
+ swagger_json = [doc.update({"_id": str(doc["_id"])}) or doc for doc in swagger_json]
+
+ return jsonify(list(swagger_json))
+
+
+@swagger_workflow.route("/", methods=["PUT"])
+def update_swagger_file(id: str) -> Response:
+ data = request.get_json()
+ result = mongo.swagger_files.update_one({"_id": ObjectId(id)}, {"$set": data})
+ if result.modified_count == 1:
+ return jsonify({"message": "Swagger file updated successfully"})
+ return jsonify({"message": "Swagger file not found"})
+
+
+@swagger_workflow.route("/", methods=["DELETE"])
+def delete_swagger_file(id: str) -> Response:
+ result = mongo.swagger_files.delete_one({"_id": ObjectId(id)})
+ if result.deleted_count == 1:
+ return jsonify({"message": "Swagger file deleted successfully"})
+ return jsonify({"message": "Swagger file not found"})
diff --git a/nginx/nginx.conf b/nginx/nginx.conf
index 0a7e95ddb..ccd4550d6 100644
--- a/nginx/nginx.conf
+++ b/nginx/nginx.conf
@@ -12,6 +12,11 @@ http {
proxy_pass http://llm-server:8002/;
}
+ location /backend/swagger_api {
+ rewrite /backend/swagger_api/(.*) /swagger_api/$1 break;
+ proxy_pass http://llm-server:8002/;
+ }
+
location /backend/ {
proxy_pass http://backend:5000/;
}