Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- Dockerfile +3 -1
- src/mcp-github/Dockerfile +3 -1
- src/mcp-github/server.py +7 -1
Dockerfile
CHANGED
|
@@ -17,4 +17,6 @@ ENV PYTHONPATH=/app/src
|
|
| 17 |
|
| 18 |
EXPOSE 7860
|
| 19 |
|
| 20 |
-
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
EXPOSE 7860
|
| 19 |
|
| 20 |
+
ENV MCP_TRANSPORT=sse
|
| 21 |
+
|
| 22 |
+
CMD ["python", "src/mcp-github/server.py"]
|
src/mcp-github/Dockerfile
CHANGED
|
@@ -17,4 +17,6 @@ ENV PYTHONPATH=/app/src
|
|
| 17 |
|
| 18 |
EXPOSE 7860
|
| 19 |
|
| 20 |
-
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
EXPOSE 7860
|
| 19 |
|
| 20 |
+
ENV MCP_TRANSPORT=sse
|
| 21 |
+
|
| 22 |
+
CMD ["python", "src/mcp-github/server.py"]
|
src/mcp-github/server.py
CHANGED
|
@@ -174,4 +174,10 @@ def get_pull_request(owner: str, repo_name: str, pr_number: int) -> Dict[str, An
|
|
| 174 |
return {"error": str(e)}
|
| 175 |
|
| 176 |
if __name__ == "__main__":
|
| 177 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 174 |
return {"error": str(e)}
|
| 175 |
|
| 176 |
if __name__ == "__main__":
|
| 177 |
+
import os
|
| 178 |
+
if os.environ.get("MCP_TRANSPORT") == "sse":
|
| 179 |
+
import uvicorn
|
| 180 |
+
port = int(os.environ.get("PORT", 7860))
|
| 181 |
+
uvicorn.run(mcp.sse_app(), host="0.0.0.0", port=port)
|
| 182 |
+
else:
|
| 183 |
+
mcp.run()
|