From 9ebd54de574fffe302583432d64cf19cb4b9286f Mon Sep 17 00:00:00 2001
From: CI-DEV <154627941+IlumCI@users.noreply.github.com>
Date: Mon, 29 Sep 2025 18:53:39 +0300
Subject: [PATCH] Update meme_agent_persona_generator.py

---
 swarms/structs/meme_agent_persona_generator.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/swarms/structs/meme_agent_persona_generator.py b/swarms/structs/meme_agent_persona_generator.py
index 1866ea45..f8b61889 100644
--- a/swarms/structs/meme_agent_persona_generator.py
+++ b/swarms/structs/meme_agent_persona_generator.py
@@ -9,7 +9,7 @@ from pydantic import BaseModel, Field
 
 from swarms.structs.agent import Agent
 from swarms.structs.swarm_router import SwarmRouter
-from swarms.utils.function_caller_model import OpenAIFunctionCaller
+from swarms.utils.litellm_wrapper import LiteLLM
 
 load_dotenv()
 
@@ -162,11 +162,11 @@ class MemeAgentGenerator:
             list: List of created agents
         """
         logger.info("Creating agents for task")
-        model = OpenAIFunctionCaller(
+        model = LiteLLM(
+            model_name="gpt-4o",
             system_prompt=BOSS_SYSTEM_PROMPT,
-            api_key=os.getenv("OPENAI_API_KEY"),
             temperature=0.1,
-            base_model=MemeSwarmConfig,
+            response_format=MemeSwarmConfig,
         )
 
         agents_dictionary = model.run(task)