Update agent_with_longterm_memory.py

Author: kyegomez

examples/agents/memory/agents_and_memory/agent_with_longterm_memory.py

@@ -1,15 +1,7 @@
-import os
-
-from dotenv import load_dotenv
-
 # Import the OpenAIChat model and the Agent struct
 from swarms import Agent
-from swarm_models import OpenAIChat
 from swarms_memory import ChromaDB
 
-# Load the environment variables
-load_dotenv()
-
 # Get the API key from the environment
 api_key = os.environ.get("OPENAI_API_KEY")
 
@@ -21,23 +13,16 @@
     docs_folder="artifacts",
 )
 
-# Initialize the language model
-llm = OpenAIChat(
-    temperature=0.5,
-    openai_api_key=api_key,
-    max_tokens=1000,
-)
-
 ## Initialize the workflow
 agent = Agent(
+    agent_name = "Long-Term-Memory-Agent",
     model_name="gpt-4o-mini",
     name="Health and Wellness Blog",
     system_prompt="Generate a 10,000 word blog on health and wellness.",
-    max_loops=4,
+    max_loops=1,
     autosave=True,
     dashboard=True,
     long_term_memory=[chromadb],
-    memory_chunk_size=300,
 )
 
 # Run the workflow on a task