Spaces:
Paused
Paused
change host to work with hf
Browse files- Dockerfile +3 -1
- app.py +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
|
| 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
|
| 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', '
|
| 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
|