diff --git a/examples/multi_agent/hiearchical_swarm/hs_stock_team.py b/examples/multi_agent/hiearchical_swarm/hs_stock_team.py
index d4cbe763..898c8ab9 100644
--- a/examples/multi_agent/hiearchical_swarm/hs_stock_team.py
+++ b/examples/multi_agent/hiearchical_swarm/hs_stock_team.py
@@ -7,16 +7,16 @@ 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()
 
 # ------------------------------------------------------------------------------
 # Trading Director: Responsible for orchestrating tasks among multiple stock analysts
 # ------------------------------------------------------------------------------
-director_llm = OpenAIFunctionCaller(
-    base_model=SwarmSpec,
-    api_key=os.getenv("OPENAI_API_KEY"),
+director_llm = LiteLLM(
+    model_name="gpt-4o",
+    response_format=SwarmSpec,
     system_prompt=(
         "You are the Trading Director in charge of coordinating a team of specialized "
         "Stock Analysts. Your responsibilities include:\n\n"