getzep
diff --git a/‎examples/chat_history/memory.py‎
Lines changed: 20 additions & 56 deletions b/‎examples/chat_history/memory.py‎
Lines changed: 20 additions & 56 deletions
diff --git a/‎examples/graph_example/user_graph_example.py‎
Lines changed: 10 additions & 29 deletions b/‎examples/graph_example/user_graph_example.py‎
Lines changed: 10 additions & 29 deletions
diff --git a/‎pyproject.toml‎
Lines changed: 1 addition & 1 deletion b/‎pyproject.toml‎
Lines changed: 1 addition & 1 deletion
@@ -20,7 +20,7 @@
 from chat_history_shoe_purchase import history
 
 from zep_cloud.client import AsyncZep
-from zep_cloud.types import Message, FactRatingInstruction, FactRatingExamples
+from zep_cloud.types import Message
 
 load_dotenv(
     dotenv_path=find_dotenv()
@@ -36,15 +36,7 @@ async def main() -> None:
 
     # Create a user
     user_id = uuid.uuid4().hex  # unique user id. can be any alphanum string
-    fact_rating_instruction = """Rate the facts by poignancy. Highly poignant 
-    facts have a significant emotional impact or relevance to the user. 
-    Facts with low poignancy are minimally relevant or of little emotional
-    significance."""
-    fact_rating_examples = FactRatingExamples(
-        high="The user received news of a family member's serious illness.",
-        medium="The user completed a challenging marathon.",
-        low="The user bought a new brand of toothpaste.",
-    )
+
     await client.user.add(
         user_id=user_id,
         email="user@example.com",
@@ -55,69 +47,41 @@ async def main() -> None:
 
     # await asyncio.sleep(1)
     print(f"User added: {user_id}")
-    session_id = uuid.uuid4().hex  # unique session id. can be any alphanum string
+    thread_id = uuid.uuid4().hex  # unique session id. can be any alphanum string
 
     # Create session associated with the above user
-    print(f"\n---Creating session: {session_id}")
+    print(f"\n---Creating thread: {thread_id}")
 
-    await client.memory.add_session(
-        session_id=session_id,
+    await client.thread.create(
+        thread_id=thread_id,
         user_id=user_id,
-        metadata={"foo": "bar"},
     )
-    # await asyncio.sleep(1)
-    # Update session metadata
-    print(f"\n---Updating session: {session_id}")
-    await client.memory.update_session(session_id=session_id, metadata={"bar": "foo"})
-    # await asyncio.sleep(3)
+
     # Get session
-    print(f"\n---Getting session: {session_id}")
-    session = await client.memory.get_session(session_id)
-    print(f"Session details: {session}")
-    # await asyncio.sleep(3)
+    print(f"\n---Getting thread: {thread_id}")
+    session = await client.thread.get(thread_id)
+    print(f"Thread details: {session}")
 
-    # Add Memory for session
-    print(f"\n---Add Memory for Session: {session_id}")
+    # Add messages to the thread
+    print(f"\n---Add messages to the thread: {thread_id}")
     for m in history:
         print(f"{m['role']}: {m['content']}")
-        await client.memory.add(session_id=session_id, messages=[Message(**m)])
+        await client.thread.add_messages(thread_id=thread_id, messages=[Message(**m)])
         # await asyncio.sleep(0.5)
 
     #  Wait for the messages to be processed
     await asyncio.sleep(50)
 
-    # Synthesize a question from most recent messages.
-    # Useful for RAG apps. This is faster than using an LLM chain.
-    print("\n---Synthesize a question from most recent messages")
-    question = await client.memory.synthesize_question(session_id, last_n_messages=3)
-    print(f"Question: {question}")
-
-    # Classify the session.
-    # Useful for semantic routing, filtering, and many other use cases.
-    print("\n---Classify the session")
-    classes = [
-        "low spender <$50",
-        "medium spender >=$50, <$100",
-        "high spender >=$100",
-        "unknown",
-    ]
-    classification = await client.memory.classify_session(
-        session_id, name="spender_category", classes=classes, persist=True
-    )
-    print(f"Classification: {classification}")
-
-    # Get Memory for session
-    print(f"\n---Get Perpetual Memory for Session: {session_id}")
-    memory = await client.memory.get(session_id)
-    print(f"Memory: {memory}")
-    print("\n---End of Memory")
+    # Get user context for thread
+    print(f"\n---Get user context for thread: {thread_id}")
+    user_context = await client.thread.get_user_context(thread_id)
 
-    print(f"Memory context: {memory.context}")
+    print(f"User context: {user_context.context}")
 
-    # Delete Memory for session
+    # Delete thread and clear context
     # Uncomment to run
-    # print(f"\n6---deleteMemory for Session: {session_id}")
-    # await client.memory.delete(session_id)
+    # print(f"\n6---delete thread memory : {thread_id}")
+    # await client.thread.delete(thread_id)
 
 
 if __name__ == "__main__":
 
@@ -3,15 +3,15 @@
 
 This script demonstrates the following functionality:
 - Creating a user.
-- Creating a session associated with the created user.
-- Adding messages to the session.
+- Creating a thread associated with the created user.
+- Adding messages to the thread.
 - Retrieving episodes, edges, and nodes for a user.
 - Searching the user's graph memory.
 - Adding text and JSON episodes to the graph.
 - Performing a centered search on a specific node.
 
 The script showcases various operations using the Zep Graph API, including
-user and session management, adding different types of episodes, and querying
+user and thread management, adding different types of episodes, and querying
 the graph structure.
 """
 
@@ -37,14 +37,14 @@ async def main() -> None:
         api_key=API_KEY,
     )
     user_id = uuid.uuid4().hex
-    session_id = uuid.uuid4().hex
+    thread_id = uuid.uuid4().hex
     await client.user.add(user_id=user_id, first_name="Paul")
     print(f"User {user_id} created")
-    await client.memory.add_session(session_id=session_id, user_id=user_id)
-    print(f"Session {session_id} created")
+    await client.thread.create(thread_id=thread_id, user_id=user_id)
+    print(f"thread {thread_id} created")
     for message in history[2]:
         await client.memory.add(
-            session_id,
+            thread_id,
             messages=[
                 Message(
                     role_type=message["role_type"],
@@ -55,19 +55,9 @@ async def main() -> None:
 
     print("Waiting for the graph to be updated...")
     await asyncio.sleep(30)
-    print("Getting memory for session")
-    session_memory = await client.memory.get(session_id)
-    print(session_memory)
-    print("Searching user memory...")
-    search_results = await client.memory.search_sessions(
-        text="What is the weather in San Francisco?",
-        user_id=user_id,
-        search_scope="facts",
-    )
-    print(search_results)
-    sessions = await client.user.get_sessions(user_id)
-    print(sessions)
-    print("Getting episodes for user")
+    print("Getting user context for thread")
+    thread_context = await client.thread.get_user_context(thread_id)
+    print(thread_context)
     episode_result = await client.graph.episode.get_by_user_id(user_id, lastn=3)
     episodes = episode_result.episodes
     print(f"Episodes for user {user_id}:")
@@ -154,15 +144,6 @@ async def main() -> None:
         scope="nodes",
     )
     print(search_results.nodes)
-    print("Getting all user facts")
-    result = await client.user.get_facts(user_id)
-    print(result.facts)
-
-    for fact in result.facts:
-        if fact.valid_at or fact.invalid_at:
-            print(
-                f"Fact {fact.fact} is valid at {fact.valid_at} and invalid at {fact.invalid_at}\n "
-            )
 
     # Uncomment to delete the user
     # await client.user.delete(user_id)
 
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zep-cloud"
-version = "2.18.0"
+version = "2.19.0"
 description = ""
 readme = "README.md"
 authors = []