From 622f143d6bb1b2521aaf031e5fe5933ab9a580dd Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Sun, 22 Sep 2024 17:25:01 -0700
Subject: [PATCH] fix clients

---
 llama_stack/apis/memory_banks/client.py | 4 ++--
 llama_stack/apis/models/client.py       | 2 +-
 llama_stack/apis/shields/client.py      | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/llama_stack/apis/memory_banks/client.py b/llama_stack/apis/memory_banks/client.py
index 1a648927e..d43b96be3 100644
--- a/llama_stack/apis/memory_banks/client.py
+++ b/llama_stack/apis/memory_banks/client.py
@@ -40,8 +40,8 @@ class MemoryBanksClient(MemoryBanks):
         async with httpx.AsyncClient() as client:
             response = await client.get(
                 f"{self.base_url}/memory_banks/get",
-                json={
-                    "bank_type": bank_type,
+                params={
+                    "bank_type": bank_type.value,
                 },
                 headers={"Content-Type": "application/json"},
             )
diff --git a/llama_stack/apis/models/client.py b/llama_stack/apis/models/client.py
index d97184997..dbd26146d 100644
--- a/llama_stack/apis/models/client.py
+++ b/llama_stack/apis/models/client.py
@@ -38,7 +38,7 @@ class ModelsClient(Models):
         async with httpx.AsyncClient() as client:
             response = await client.get(
                 f"{self.base_url}/models/get",
-                json={
+                params={
                     "core_model_id": core_model_id,
                 },
                 headers={"Content-Type": "application/json"},
diff --git a/llama_stack/apis/shields/client.py b/llama_stack/apis/shields/client.py
index 5e5001d90..60ea56fae 100644
--- a/llama_stack/apis/shields/client.py
+++ b/llama_stack/apis/shields/client.py
@@ -38,7 +38,7 @@ class ShieldsClient(Shields):
         async with httpx.AsyncClient() as client:
             response = await client.get(
                 f"{self.base_url}/shields/get",
-                json={
+                params={
                     "shield_type": shield_type,
                 },
                 headers={"Content-Type": "application/json"},