Token (#12)
Browse files- Update to use secret Token (2a2e8d24468d99c2405a70c05153dec868a7011e)
ComputeAgent/graph/graph.py
CHANGED
|
@@ -57,7 +57,15 @@ mcp_client = MultiServerMCPClient(
|
|
| 57 |
"hivecompute": {
|
| 58 |
"command": python_executable,
|
| 59 |
"args": [mcp_server_path],
|
| 60 |
-
"transport": "stdio"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
}
|
| 62 |
}
|
| 63 |
)
|
|
|
|
| 57 |
"hivecompute": {
|
| 58 |
"command": python_executable,
|
| 59 |
"args": [mcp_server_path],
|
| 60 |
+
"transport": "stdio",
|
| 61 |
+
"env": {
|
| 62 |
+
# Pass HF Spaces secrets to the MCP subprocess
|
| 63 |
+
"HIVE_COMPUTE_DEFAULT_API_TOKEN": os.getenv("HIVE_COMPUTE_DEFAULT_API_TOKEN", ""),
|
| 64 |
+
"HIVE_COMPUTE_BASE_API_URL": os.getenv("HIVE_COMPUTE_BASE_API_URL", "https://api.hivecompute.ai"),
|
| 65 |
+
# Also pass these to ensure Python works correctly
|
| 66 |
+
"PATH": os.getenv("PATH", ""),
|
| 67 |
+
"PYTHONPATH": os.getenv("PYTHONPATH", ""),
|
| 68 |
+
}
|
| 69 |
}
|
| 70 |
}
|
| 71 |
)
|
ComputeAgent/graph/graph_deploy.py
CHANGED
|
@@ -49,7 +49,15 @@ mcp_client = MultiServerMCPClient(
|
|
| 49 |
"hivecompute": {
|
| 50 |
"command": python_executable,
|
| 51 |
"args": [mcp_server_path],
|
| 52 |
-
"transport": "stdio"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
}
|
| 54 |
}
|
| 55 |
)
|
|
|
|
| 49 |
"hivecompute": {
|
| 50 |
"command": python_executable,
|
| 51 |
"args": [mcp_server_path],
|
| 52 |
+
"transport": "stdio",
|
| 53 |
+
"env": {
|
| 54 |
+
# Pass HF Spaces secrets to the MCP subprocess
|
| 55 |
+
"HIVE_COMPUTE_DEFAULT_API_TOKEN": os.getenv("HIVE_COMPUTE_DEFAULT_API_TOKEN", ""),
|
| 56 |
+
"HIVE_COMPUTE_BASE_API_URL": os.getenv("HIVE_COMPUTE_BASE_API_URL", "https://api.hivecompute.ai"),
|
| 57 |
+
# Also pass these to ensure Python works correctly
|
| 58 |
+
"PATH": os.getenv("PATH", ""),
|
| 59 |
+
"PYTHONPATH": os.getenv("PYTHONPATH", ""),
|
| 60 |
+
}
|
| 61 |
}
|
| 62 |
}
|
| 63 |
)
|
Compute_MCP/api_data_structure.py
CHANGED
|
@@ -1,5 +1,3 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
import logging
|
| 4 |
from typing import List, Optional, Dict, Any, Union
|
| 5 |
from pydantic import BaseModel, Field
|
|
|
|
|
|
|
|
|
|
| 1 |
import logging
|
| 2 |
from typing import List, Optional, Dict, Any, Union
|
| 3 |
from pydantic import BaseModel, Field
|