carraraig commited on
Commit
5d5ee81
·
verified ·
1 Parent(s): 93850a2

- 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