From eeb4e0993b303325273e9709f5623e8ac25fab52 Mon Sep 17 00:00:00 2001
From: HamadaSalhab <hamada.a.salhab@gmail.com>
Date: Tue, 3 Sep 2024 22:57:06 +0300
Subject: [PATCH] Transform the spec type when retreiving agent tools

---
 agents-api/agents_api/models/tools/list_tools.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/agents-api/agents_api/models/tools/list_tools.py b/agents-api/agents_api/models/tools/list_tools.py
index 4b44fc1e0..931ca3ca9 100644
--- a/agents-api/agents_api/models/tools/list_tools.py
+++ b/agents-api/agents_api/models/tools/list_tools.py
@@ -27,7 +27,13 @@
         TypeError: partialclass(HTTPException, status_code=400),
     }
 )
-@wrap_in_class(Tool)
+@wrap_in_class(
+    Tool,
+    transform=lambda d: {
+        d["type"]: {**d.pop("spec"), "name": d["name"]},
+        **d,
+    },
+)
 @cozo_query
 @beartype
 def list_tools(