From 655687671a45f0dd6d12baffc9f485d299224301 Mon Sep 17 00:00:00 2001
From: CI-DEV <154627941+IlumCI@users.noreply.github.com>
Date: Mon, 29 Sep 2025 18:56:38 +0300
Subject: [PATCH] Update hierarchical_swarm_example.py

---
 .../hiearchical_swarm/hierarchical_swarm_example.py       | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/multi_agent/hiearchical_swarm/hierarchical_swarm_example.py b/examples/multi_agent/hiearchical_swarm/hierarchical_swarm_example.py
index 5be9008f..a73e8bc1 100644
--- a/examples/multi_agent/hiearchical_swarm/hierarchical_swarm_example.py
+++ b/examples/multi_agent/hiearchical_swarm/hierarchical_swarm_example.py
@@ -7,7 +7,7 @@ from swarms.structs.hiearchical_swarm import (
     HierarchicalSwarm,
     SwarmSpec,
 )
-from swarms.utils.function_caller_model import OpenAIFunctionCaller
+from swarms.utils.litellm_wrapper import LiteLLM
 
 load_dotenv()
 
@@ -15,9 +15,9 @@ load_dotenv()
 # ------------------------------------------------------------------------------
 # Director LLM: Responsible for orchestrating tasks among the agents
 # ------------------------------------------------------------------------------
-llm = OpenAIFunctionCaller(
-    base_model=SwarmSpec,
-    api_key=os.getenv("OPENAI_API_KEY"),
+llm = LiteLLM(
+    model_name="gpt-4o",
+    response_format=SwarmSpec,
     system_prompt=(
         "As the Director of this Hierarchical Agent Swarm, you are in charge of "
         "coordinating and overseeing all tasks, ensuring that each is executed "