import from monorepo

worker/orca-agent/src/server/routes/task.py

@@ -0,0 +1,48 @@
+from flask import Blueprint, jsonify, request
+from src.server.services import repo_summary_service
+
+bp = Blueprint("task", __name__)
+
+
+@bp.post("/worker-task/<round_number>")
+def start_task(round_number):
+    logger = repo_summary_service.logger
+    logger.info(f"Task started for round: {round_number}")
+
+    data = request.get_json()
+    logger.info(f"Task data: {data}")
+    required_fields = ["taskId", "round_number", "repo_url"]
+    if any(data.get(field) is None for field in required_fields):
+        return jsonify({"error": "Missing data"}), 401
+
+    result = repo_summary_service.handle_task_creation(
+        task_id=data["taskId"],
+        round_number=int(round_number),
+        repo_url=data["repo_url"],
+    )
+
+    return result
+
+
+if __name__ == "__main__":
+    from flask import Flask
+
+    # Create a Flask app instance
+    app = Flask(__name__)
+    app.register_blueprint(bp)
+
+    # Test data
+    test_data = {
+        "taskId": "fake",
+        "round_number": "1",
+        "repo_url": "https://github.com/koii-network/docs",
+    }
+
+    # Set up test context
+    with app.test_client() as client:
+        # Make a POST request to the endpoint
+        response = client.post("/repo_summary/1", json=test_data)
+
+        # Print the response
+        print(f"Status Code: {response.status_code}")
+        print(f"Response: {response.get_json()}")