core-lead/molecule-core
1
0
Fork 0
forked from molecule-ai/molecule-core
Code Pull Requests Activity

Merge pull request #235 from Molecule-AI/fix/issue-220-initial-idle-prompt-auth

Browse Source 
fix(workspace-template): #220 — auth_headers on initial_prompt + idle loop
This commit is contained in:
Hongming Wang 2026-04-15 12:02:06 -07:00 committed by GitHub
commit 7d89fd4ea4
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
workspace-template/main.py
View File

@ -347,6 +347,12 @@ async def main(): # pragma: no cover
},
}).encode()
# #220: include platform bearer token so the request isn't
# silently rejected once any workspace has a live token on
# file. Without this, initial_prompt 401s in multi-tenant
# mode exactly like /registry/register did in #215.
headers = {"Content-Type": "application/json", **auth_headers()}
# Retry with backoff — the platform proxy may not be able to
# reach us yet (container networking takes a moment to settle).
max_retries = 5
@ -355,7 +361,7 @@ async def main(): # pragma: no cover
req = urllib.request.Request(
f"{platform_url}/workspaces/{workspace_id}/a2a",
data=payload,
headers={"Content-Type": "application/json"},
headers=headers,
)
with urllib.request.urlopen(req, timeout=600) as resp:
resp.read()
@ -435,11 +441,14 @@ async def main(): # pragma: no cover
def _post_sync():
# Returns (status_code, error_type) so the caller logs the
# actual outcome instead of a bare "post failed" line.
# #220: include auth_headers() on every idle fire. Without
# this, the idle loop 401s in multi-tenant mode.
headers = {"Content-Type": "application/json", **auth_headers()}
try:
req = _urlreq.Request(
f"{platform_url}/workspaces/{workspace_id}/a2a",
data=payload,
headers={"Content-Type": "application/json"},
headers=headers,
)
with _urlreq.urlopen(req, timeout=IDLE_FIRE_TIMEOUT_SECONDS) as resp:
resp.read()