Merge branch 'main' of github.com:getzep/zep-python

Author: paul-paliychuk

examples/chat_history/memory.py

@@ -20,7 +20,7 @@
 from chat_history_shoe_purchase import history
 
 from zep_cloud.client import AsyncZep
-from zep_cloud.types import Message
+from zep_cloud.types import Message, FactRatingInstruction, FactRatingExamples
 
 load_dotenv(
     dotenv_path=find_dotenv()
@@ -36,7 +36,15 @@ 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",
@@ -47,41 +55,69 @@ async def main() -> None:
 
     # await asyncio.sleep(1)
     print(f"User added: {user_id}")
-    thread_id = uuid.uuid4().hex  # unique session id. can be any alphanum string
+    session_id = uuid.uuid4().hex  # unique session id. can be any alphanum string
 
     # Create session associated with the above user
-    print(f"\n---Creating thread: {thread_id}")
+    print(f"\n---Creating session: {session_id}")
 
-    await client.thread.create(
-        thread_id=thread_id,
+    await client.memory.add_session(
+        session_id=session_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 thread: {thread_id}")
-    session = await client.thread.get(thread_id)
-    print(f"Thread details: {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)
 
-    # Add messages to the thread
-    print(f"\n---Add messages to the thread: {thread_id}")
+    # Add Memory for session
+    print(f"\n---Add Memory for Session: {session_id}")
     for m in history:
         print(f"{m['role']}: {m['content']}")
-        await client.thread.add_messages(thread_id=thread_id, messages=[Message(**m)])
+        await client.memory.add(session_id=session_id, messages=[Message(**m)])
         # await asyncio.sleep(0.5)
 
     #  Wait for the messages to be processed
     await asyncio.sleep(50)
 
-    # 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)
+    # 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")
 
-    print(f"User context: {user_context.context}")
+    print(f"Memory context: {memory.context}")
 
-    # Delete thread and clear context
+    # Delete Memory for session
     # Uncomment to run
-    # print(f"\n6---delete thread memory : {thread_id}")
-    # await client.thread.delete(thread_id)
+    # print(f"\n6---deleteMemory for Session: {session_id}")
+    # await client.memory.delete(session_id)
 
 
 if __name__ == "__main__":

examples/graph_example/user_graph_example.py

@@ -3,15 +3,15 @@
 
 This script demonstrates the following functionality:
 - Creating a user.
-- Creating a thread associated with the created user.
-- Adding messages to the thread.
+- Creating a session associated with the created user.
+- Adding messages to the session.
 - 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 thread management, adding different types of episodes, and querying
+user and session 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
-    thread_id = uuid.uuid4().hex
+    session_id = uuid.uuid4().hex
     await client.user.add(user_id=user_id, first_name="Paul")
     print(f"User {user_id} created")
-    await client.thread.create(thread_id=thread_id, user_id=user_id)
-    print(f"thread {thread_id} created")
+    await client.memory.add_session(session_id=session_id, user_id=user_id)
+    print(f"Session {session_id} created")
     for message in history[2]:
         await client.memory.add(
-            thread_id,
+            session_id,
             messages=[
                 Message(
                     role_type=message["role_type"],
@@ -55,9 +55,19 @@ async def main() -> None:
 
     print("Waiting for the graph to be updated...")
     await asyncio.sleep(30)
-    print("Getting user context for thread")
-    thread_context = await client.thread.get_user_context(thread_id)
-    print(thread_context)
+    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")
     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}:")
@@ -144,6 +154,15 @@ 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)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zep-cloud"
-version = "2.19.0"
+version = "2.21.0"
 description = ""
 readme = "README.md"
 authors = []