mishrabp commited on
Commit
c9696f3
·
verified ·
1 Parent(s): 7ae6787

Upload folder using huggingface_hub

Browse files
Dockerfile CHANGED
@@ -17,4 +17,6 @@ ENV PYTHONPATH=/app/src
17
 
18
  EXPOSE 7860
19
 
20
- CMD ["python", "src/mcp-github/server.py", "--transport", "sse", "--port", "7860", "--host", "0.0.0.0"]
 
 
 
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
- CMD ["python", "src/mcp-github/server.py", "--transport", "sse", "--port", "7860", "--host", "0.0.0.0"]
 
 
 
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
- mcp.run()
 
 
 
 
 
 
 
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()