@@ -26,10 +26,7 @@ def __init__(self, api_key: str | None = None, base_url: str | None = None):
2626 if not api_key :
2727 raise ValueError ("MemOS API key is required" )
2828
29- self .headers = {
30- "Content-Type" : "application/json" ,
31- "Authorization" : f"Token { api_key } "
32- }
29+ self .headers = {"Content-Type" : "application/json" , "Authorization" : f"Token { api_key } " }
3330
3431 def _validate_required_params (self , ** params ):
3532 """Validate required parameters - if passed, they must not be empty"""
@@ -45,7 +42,7 @@ def get_message(
4542 self ._validate_required_params (user_id = user_id )
4643
4744 url = f"{ self .base_url } /get/message"
48- payload = { "user_id" : user_id , "conversation_id" : conversation_id }
45+ payload = {"user_id" : user_id , "conversation_id" : conversation_id }
4946 for retry in range (MAX_RETRY_COUNT ):
5047 try :
5148 response = requests .post (
@@ -70,7 +67,7 @@ def add_message(
7067 )
7168
7269 url = f"{ self .base_url } /add/message"
73- payload = { "messages" : messages , "user_id" : user_id , "conversation_id" : conversation_id }
70+ payload = {"messages" : messages , "user_id" : user_id , "conversation_id" : conversation_id }
7471 for retry in range (MAX_RETRY_COUNT ):
7572 try :
7673 response = requests .post (
0 commit comments