phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-27 21:31:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: accept MCP authorization headers for MCP toolgroups

Browse source
This commit is contained in:
Ashwin Bharambe 2025-05-22 14:39:59 -07:00
parent 2708312168
commit 0d67e17a91
7 changed files with 294 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/distribution/library_client.py
View file

@ -261,9 +261,11 @@ class AsyncLlamaStackAsLibraryClient(AsyncLlamaStackClient):
raise ValueError("Client not initialized")
# Create headers with provider data if available
headers = {}
headers = options.headers or {}
if self.provider_data:
headers["X-LlamaStack-Provider-Data"] = json.dumps(self.provider_data)
keys = ["X-LlamaStack-Provider-Data", "x-llamastack-provider-data"]
if all(key not in headers for key in keys):
headers["X-LlamaStack-Provider-Data"] = json.dumps(self.provider_data)
# Use context manager for provider data
with request_provider_data_context(headers):