Update meme_agent_persona_generator.py

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)