diff --git a/docs/features/custom-categories.mdx b/docs/features/custom-categories.mdx
index cc60934c..faaca07b 100644
--- a/docs/features/custom-categories.mdx
+++ b/docs/features/custom-categories.mdx
@@ -19,9 +19,12 @@ Here's how to set custom categories:
 
 <CodeGroup>
 ```python Code
+import os
 from mem0 import MemoryClient
 
-client = MemoryClient(api_key="<your_mem0_api_key>")
+os.environ["MEM0_API_KEY"] = "your-api-key"
+
+client = MemoryClient()
 
 # Update custom categories
 new_categories = [
@@ -95,8 +98,11 @@ You can also set custom categories during the `add` API call. This will override
 
 <CodeGroup>
 ```python Code
+import os
 from mem0 import MemoryClient
 
+os.environ["MEM0_API_KEY"] = "your-api-key"
+
 client = MemoryClient(api_key="<your_mem0_api_key>")
 
 custom_categories = [
@@ -148,9 +154,12 @@ Here is the list of **default categories**. If you don't specify any custom cate
 
 <CodeGroup>
 ```python Code
+import os
 from mem0 import MemoryClient
 
-client = MemoryClient(api_key="<your_mem0_api_key>")
+os.environ["MEM0_API_KEY"] = "your-api-key"
+
+client = MemoryClient()
 
 messages = [
     {"role": "user", "content": "Hi, my name is Alice."},
diff --git a/docs/features/selective-memory.mdx b/docs/features/selective-memory.mdx
index 9e2edd2c..919f174a 100644
--- a/docs/features/selective-memory.mdx
+++ b/docs/features/selective-memory.mdx
@@ -29,9 +29,12 @@ Users can define specific kinds of memories to store. This feature enhances memo
 Here’s how you can do it:
 
 ```python
+import os
 from mem0 import MemoryClient
 
-m = MemoryClient(api_key="xxx")
+os.environ["MEM0_API_KEY"] = "your-api-key"
+
+m = MemoryClient()
 
 # Define what to include
 includes = "sports related things"
diff --git a/docs/integrations/autogen.mdx b/docs/integrations/autogen.mdx
index 0412cee9..53178205 100644
--- a/docs/integrations/autogen.mdx
+++ b/docs/integrations/autogen.mdx
@@ -30,9 +30,10 @@ USER_ID = "customer_service_bot"
 
 # Set up OpenAI API key
 os.environ['OPENAI_API_KEY'] = OPENAI_API_KEY
+os.environ['MEM0_API_KEY'] = MEM0_API_KEY
 
 # Initialize Mem0 and AutoGen agents
-memory_client = MemoryClient(api_key=MEM0_API_KEY)
+memory_client = MemoryClient()
 agent = ConversableAgent(
     "chatbot",
     llm_config={"config_list": [{"model": "gpt-4", "api_key": OPENAI_API_KEY}]},
diff --git a/docs/integrations/langchain.mdx b/docs/integrations/langchain.mdx
index 6d6fdb39..1d664c1f 100644
--- a/docs/integrations/langchain.mdx
+++ b/docs/integrations/langchain.mdx
@@ -37,7 +37,7 @@ os.environ["MEM0_API_KEY"] = "your-mem0-api-key"
 
 # Initialize LangChain and Mem0
 llm = ChatOpenAI(model="gpt-4o-mini")
-mem0 = MemoryClient(api_key=os.environ["MEM0_API_KEY"])
+mem0 = MemoryClient()
 ```
 
 ## Create Prompt Template
diff --git a/docs/integrations/multion.mdx b/docs/integrations/multion.mdx
index 620273fc..610a8807 100644
--- a/docs/integrations/multion.mdx
+++ b/docs/integrations/multion.mdx
@@ -34,10 +34,11 @@ USER_ID = "your-user-id"
 
 # Set up OpenAI API key
 os.environ['OPENAI_API_KEY'] = OPENAI_API_KEY
+os.environ['MEM0_API_KEY'] = MEM0_API_KEY
 
 # Initialize Mem0 and MultiOn
 memory = Memory()  # For local usage
-memory_client = MemoryClient(api_key=MEM0_API_KEY)  # For API usage
+memory_client = MemoryClient()  # For API usage
 multion = MultiOn(api_key=MULTION_API_KEY)
 ```