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

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

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"