NiWaRe commited on
Commit
2325f96
·
1 Parent(s): c2fd356

change host to work with hf

Browse files
Files changed (3) hide show
  1. Dockerfile +3 -1
  2. app.py +3 -3
  3. src/wandb_mcp_server/server.py +1 -1
Dockerfile CHANGED
@@ -29,9 +29,11 @@ COPY app.py .
29
  ENV PYTHONPATH=/app/src
30
  ENV WANDB_SILENT=True
31
  ENV WEAVE_SILENT=True
 
 
32
 
33
  # Expose port for HTTP transport
34
- EXPOSE 8080
35
 
36
  # Run the application
37
  CMD ["python", "app.py"]
 
29
  ENV PYTHONPATH=/app/src
30
  ENV WANDB_SILENT=True
31
  ENV WEAVE_SILENT=True
32
+ ENV MCP_TRANSPORT=http
33
+ ENV HOST=0.0.0.0
34
 
35
  # Expose port for HTTP transport
36
+ EXPOSE 7860
37
 
38
  # Run the application
39
  CMD ["python", "app.py"]
app.py CHANGED
@@ -35,8 +35,8 @@ def main():
35
  sys.argv = [
36
  sys.argv[0], # Keep the script name
37
  "--transport", "http",
38
- "--host", "0.0.0.0", # Listen on all interfaces for HuggingFace Spaces
39
- "--port", str(os.environ.get("PORT", "7860")) # Use PORT env var or default to 8080
40
  ]
41
 
42
  # Check for required environment variables
@@ -48,7 +48,7 @@ def main():
48
  sys.exit(1)
49
 
50
  logger.info(f"WANDB_API_KEY configured: {'Yes' if wandb_api_key else 'No'}")
51
- logger.info(f"Starting HTTP server on port {os.environ.get('PORT', '8080')}")
52
  logger.info("MCP endpoint will be available at: /mcp")
53
 
54
  try:
 
35
  sys.argv = [
36
  sys.argv[0], # Keep the script name
37
  "--transport", "http",
38
+ "--host", os.environ.get("HOST", "0.0.0.0"), # Listen on all interfaces for HuggingFace Spaces
39
+ "--port", str(os.environ.get("PORT", "7860")) # Use PORT env var or default to 7860
40
  ]
41
 
42
  # Check for required environment variables
 
48
  sys.exit(1)
49
 
50
  logger.info(f"WANDB_API_KEY configured: {'Yes' if wandb_api_key else 'No'}")
51
+ logger.info(f"Starting HTTP server on port {os.environ.get('PORT', '7860')}")
52
  logger.info("MCP endpoint will be available at: /mcp")
53
 
54
  try:
src/wandb_mcp_server/server.py CHANGED
@@ -258,7 +258,7 @@ def cli():
258
  logger.info(f"Starting HTTP server on {args.host}:{port}")
259
 
260
  # Create new FastMCP instance with HTTP configuration
261
- http_mcp = FastMCP("weave-mcp-server", port=port, stateless_http=True)
262
 
263
  # Copy all tools from the original mcp instance
264
  # We need to re-register the tools on the new instance
 
258
  logger.info(f"Starting HTTP server on {args.host}:{port}")
259
 
260
  # Create new FastMCP instance with HTTP configuration
261
+ http_mcp = FastMCP("weave-mcp-server", host=args.host, port=port, stateless_http=True)
262
 
263
  # Copy all tools from the original mcp instance
264
  # We need to re-register the tools on the new instance