zep-crewai 1.0.0__tar.gz → 1.1.1__tar.gz

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zep-crewai
-Version: 1.0.0
+Version: 1.1.1
 Summary: CrewAI integration for Zep
 Project-URL: Homepage, https://github.com/getzep/zep
 Project-URL: Documentation, https://help.getzep.com

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/examples/crewai_graph.py

@@ -214,7 +214,7 @@ def main():
         search_results = graph_storage.search("Python", limit=5)
         print(f"Found {len(search_results)} results about Python:")
         for idx, result in enumerate(search_results[:3], 1):
-            print(f"  {idx}. {result.get('memory', '')[:100]}...")
+            print(f"  {idx}. {result.get('context', '')[:100]}...")
 
     except Exception as e:
         print(f"\n❌ Execution failed: {e}")

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/examples/crewai_user.py

@@ -72,7 +72,7 @@ def main():
         client=zep_client,
         user_id=user_id,
         thread_id=thread_id,
-        mode="summary",  # Use summary mode for concise context (or "raw_messages" for full history)
+        mode="summary",  # Use summary mode for concise context (or "basic" for raw context)
     )
     external_memory = ExternalMemory(storage=user_storage)
 
@@ -223,7 +223,7 @@ def main():
         print(f"Found {len(search_results)} relevant memories about mobile app:")
         for idx, result in enumerate(search_results[:3], 1):
             memory_type = result.get("type", "unknown")
-            content = result.get("memory", "")[:100]
+            content = result.get("context", "")[:100]
             print(f"  {idx}. [{memory_type}] {content}...")
 
     except Exception as e:

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/examples/test_context_mock.py

@@ -62,8 +62,8 @@ Key Facts:
     logger.info("Context that CrewAI agent receives:")
     logger.info(context)
 
-    # Example 2: Raw messages mode
-    logger.info("\n2. RAW MESSAGES MODE - thread.get_user_context(mode='raw_messages')")
+    # Example 2: basic mode
+    logger.info("\n2. basic MODE - thread.get_user_context(mode='basic')")
     logger.info("-" * 60)
 
     mock_context_raw = MagicMock()
@@ -82,7 +82,7 @@ Key Facts:
     mock_zep.thread.get_user_context = MagicMock(return_value=mock_context_raw)
 
     storage_raw = ZepUserStorage(
-        client=mock_zep, user_id="user_123", thread_id="thread_456", mode="raw_messages"
+        client=mock_zep, user_id="user_123", thread_id="thread_456", mode="basic"
     )
 
     context = storage_raw.get_context()
@@ -227,7 +227,7 @@ CrewAI agents receive context from Zep in these ways:
 
 1. User Storage (ZepUserStorage):
    - Retrieves user-specific context from conversation threads
-   - Two modes: 'summary' (default) or 'raw_messages'
+   - Two modes: 'summary' (default) or 'basic'
    - Context includes user profile, preferences, conversation history
    - Directly uses thread.get_user_context() from Zep SDK
 

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zep-crewai"
-version = "1.0.0"
+version = "1.1.1"
 description = "CrewAI integration for Zep"
 readme = "README.md"
 requires-python = ">=3.10"

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/src/zep_crewai/graph_storage.py

@@ -127,7 +127,7 @@ class ZepGraphStorage(Storage):
             if context:
                 self._logger.info(f"Composed context for query: {query}")
                 return [
-                    {"memory": context, "type": "graph_context", "source": "graph", "query": query}
+                    {"context": context, "type": "graph_context", "source": "graph", "query": query}
                 ]
 
             self._logger.info(f"No results found for query: {query}")

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/src/zep_crewai/memory.py

@@ -156,10 +156,10 @@ class ZepStorage(Storage):
             self._logger.debug(f"Failed to search user memories: {e}")
 
         if thread_context and hasattr(thread_context, "context") and thread_context.context:
-            results.append({"memory": thread_context.context})
+            results.append({"context": thread_context.context})
 
         for result in edges_search_results:
-            results.append({"memory": result})
+            results.append({"context": result})
 
         return results
 

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/src/zep_crewai/user_storage.py

@@ -31,7 +31,7 @@ class ZepUserStorage(Storage):
         search_filters: SearchFilters | None = None,
         facts_limit: int = 20,
         entity_limit: int = 5,
-        mode: Literal["summary", "raw_messages"] = "summary",
+        mode: Literal["summary", "basic"] = "summary",
         **kwargs: Any,
     ) -> None:
         """
@@ -44,7 +44,7 @@ class ZepUserStorage(Storage):
             search_filters: Optional filters for search operations
             facts_limit: Maximum number of facts (edges) to retrieve for context
             entity_limit: Maximum number of entities (nodes) to retrieve for context
-            mode: Mode for thread context retrieval ("summary" or "raw_messages")
+            mode: Mode for thread context retrieval ("summary" or "basic")
             **kwargs: Additional configuration options
         """
         if not isinstance(client, Zep):
@@ -155,7 +155,7 @@ class ZepUserStorage(Storage):
                 self._logger.info(f"Composed context for query: {query}")
                 return [
                     {
-                        "memory": context,
+                        "context": context,
                         "type": "user_graph_context",
                         "source": "user_graph",
                         "query": query,

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/tests/test_graph_storage.py

@@ -125,7 +125,7 @@ class TestZepGraphStorage:
         assert len(results) == 1  # Single composed result
 
         # Check the composed result
-        assert results[0]["memory"] == (
+        assert results[0]["context"] == (
             "Facts:\n- Python is used for AI\n\n"
             "Entities:\n- Python: A programming language\n\n"
             "Episodes:\n- Discussion about Python"

{zep_crewai-1.0.0 → zep_crewai-1.1.1}/tests/test_user_storage.py

@@ -180,7 +180,7 @@ class TestZepUserStorage:
         assert isinstance(results, list)
         assert len(results) == 1
         assert results[0]["type"] == "user_graph_context"
-        assert results[0]["memory"] == "Context: User likes Python"
+        assert results[0]["context"] == "Context: User likes Python"
 
     @patch("zep_crewai.utils.compose_context_string")
     @patch("zep_crewai.utils.ThreadPoolExecutor")
@@ -256,8 +256,8 @@ class TestZepUserStorage:
         # Verify results
         assert len(results) == 1
         assert results[0]["type"] == "user_graph_context"
-        assert "User prefers Python" in results[0]["memory"]
-        assert "UserPreference" in results[0]["memory"]
+        assert "User prefers Python" in results[0]["context"]
+        assert "UserPreference" in results[0]["context"]
 
     def test_get_context_with_thread(self):
         """Test get_context retrieves context using thread.get_user_context."""
@@ -285,28 +285,28 @@ class TestZepUserStorage:
 
         assert context == "User's conversation context with summary"
 
-    def test_get_context_with_raw_messages_mode(self):
-        """Test get_context with raw_messages mode."""
+    def test_get_context_with_basic_mode(self):
+        """Test get_context with basic mode."""
         from zep_cloud.client import Zep
 
         mock_client = MagicMock(spec=Zep)
         mock_client.thread = MagicMock()
 
-        # Mock thread context response with raw messages
+        # Mock thread context response with basic
         mock_context = MagicMock()
         mock_context.context = "User: Hello\nAssistant: Hi there!\nUser: How are you?"
         mock_client.thread.get_user_context.return_value = mock_context
 
         storage = ZepUserStorage(
-            client=mock_client, user_id="test-user", thread_id="test-thread", mode="raw_messages"
+            client=mock_client, user_id="test-user", thread_id="test-thread", mode="basic"
         )
 
         # Get context
         context = storage.get_context()
 
-        # Verify thread.get_user_context was called with raw_messages mode
+        # Verify thread.get_user_context was called with basic mode
         mock_client.thread.get_user_context.assert_called_once_with(
-            thread_id="test-thread", mode="raw_messages"
+            thread_id="test-thread", mode="basic"
         )
 
         assert context == "User: Hello\nAssistant: Hi there!\nUser: How are you?"