hermes-admin/hermes-agent
1
0
Fork 0
Code Issues Pull Requests Actions 29 Packages Projects Releases Wiki Activity

fix(aux-hooks): tolerate test seams that stub relay metadata without task keys

Browse Source 
test_chat_sdk_transform_bypass stubs _relay_auxiliary_metadata with an empty
metadata dict; read aux_task/api_mode with .get so the seam keeps working.
This commit is contained in:
teknium1
2026-09-22 01:02:01 -07:00
committed by Teknium
parent 0e5809566f
commit 966d091d6a
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
agent/auxiliary_client.py
View File

@@ -2614,8 +2614,8 @@ def _relay_sync_completion(
name=provider_name, model_name=model_name, metadata=metadata,
defer_logical_completion=True,
),
aux_task=metadata["auxiliary_task"], metadata=metadata, client=client, kwargs=kwargs,
provider=provider_name, model=model_name, api_mode=metadata["api_mode"],
aux_task=str(metadata.get("auxiliary_task") or ""), metadata=metadata, client=client, kwargs=kwargs,
provider=provider_name, model=model_name, api_mode=str(metadata.get("api_mode") or ""),
)
@@ -2640,8 +2640,8 @@ async def _relay_async_completion(
kwargs, callback, name=provider_name, model_name=model_name,
metadata=metadata, defer_logical_completion=True,
),
aux_task=metadata["auxiliary_task"], metadata=metadata, client=client, kwargs=kwargs,
provider=provider_name, model=model_name, api_mode=metadata["api_mode"],
aux_task=str(metadata.get("auxiliary_task") or ""), metadata=metadata, client=client, kwargs=kwargs,
provider=provider_name, model=model_name, api_mode=str(metadata.get("api_mode") or ""),
)
@@ -2666,8 +2666,8 @@ def _relay_sync_stream(
kwargs, create, name=provider_name, model_name=model_name, finalizer=dict,
metadata=metadata, completed_response_predicate=lambda value: hasattr(value, "choices"),
),
aux_task=metadata["auxiliary_task"], metadata=metadata, client=client, kwargs=kwargs,
provider=provider_name, model=model_name, api_mode=metadata["api_mode"], streaming=True,
aux_task=str(metadata.get("auxiliary_task") or ""), metadata=metadata, client=client, kwargs=kwargs,
provider=provider_name, model=model_name, api_mode=str(metadata.get("api_mode") or ""), streaming=True,
)