phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix clients

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-22 17:25:01 -07:00
parent ef042e1605
commit 622f143d6b
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/memory_banks/client.py
View file

@ -40,8 +40,8 @@ class MemoryBanksClient(MemoryBanks):
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
response = await client.get( response = await client.get(
f"{self.base_url}/memory_banks/get", f"{self.base_url}/memory_banks/get",
json={ params={
"bank_type": bank_type, "bank_type": bank_type.value,
}, },
headers={"Content-Type": "application/json"}, headers={"Content-Type": "application/json"},
) )

2
llama_stack/apis/models/client.py
View file

@ -38,7 +38,7 @@ class ModelsClient(Models):
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
response = await client.get( response = await client.get(
f"{self.base_url}/models/get", f"{self.base_url}/models/get",
json={ params={
"core_model_id": core_model_id, "core_model_id": core_model_id,
}, },
headers={"Content-Type": "application/json"}, headers={"Content-Type": "application/json"},

2
llama_stack/apis/shields/client.py
View file

@ -38,7 +38,7 @@ class ShieldsClient(Shields):
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
response = await client.get( response = await client.get(
f"{self.base_url}/shields/get", f"{self.base_url}/shields/get",
json={ params={
"shield_type": shield_type, "shield_type": shield_type,
}, },
headers={"Content-Type": "application/json"}, headers={"Content-Type": "application/json"},