antonlytics 2.0.0__tar.gz → 2.2.0__tar.gz

antonlytics-2.2.0/.gitignore

@@ -0,0 +1,19 @@
+__pycache__/
+*.py[cod]
+*$py.class
+*.egg-info/
+*.egg
+build/
+dist/
+.eggs/
+.pytest_cache/
+.mypy_cache/
+.coverage
+.coverage.*
+htmlcov/
+.tox/
+.venv/
+venv/
+env/
+.env
+.DS_Store

antonlytics-2.2.0/Makefile

@@ -0,0 +1,37 @@
+# Antonlytics Python SDK — release helpers.
+#
+# Quick reference:
+#   make build              build wheel + sdist into dist/
+#   make publish            upload dist/ to PyPI (uses uv publish)
+#   make release            build + publish in one shot
+#   make clean              wipe dist/ build/ egg-info/
+#
+# Auth: set UV_PUBLISH_TOKEN to your PyPI API token (starts with `pypi-`).
+#   export UV_PUBLISH_TOKEN=pypi-AgEIcHlw...
+# Or pass it inline:
+#   make publish UV_PUBLISH_TOKEN=pypi-...
+# Get a token at: https://pypi.org/manage/account/token/
+
+.PHONY: build publish release clean version
+
+clean:
+	rm -rf dist build *.egg-info
+
+build: clean
+	uv build
+
+publish:
+	@if [ -z "$$UV_PUBLISH_TOKEN" ]; then \
+		echo "error: UV_PUBLISH_TOKEN is not set."; \
+		echo "  get a token at https://pypi.org/manage/account/token/"; \
+		echo "  then: export UV_PUBLISH_TOKEN=pypi-..."; \
+		exit 1; \
+	fi
+	uv publish
+
+release: build publish
+	@VERSION=$$(grep -E '^version' pyproject.toml | head -1 | cut -d'"' -f2); \
+	echo "✓ released antonlytics $$VERSION to PyPI"
+
+version:
+	@grep -E '^version' pyproject.toml | head -1 | cut -d'"' -f2

{antonlytics-2.0.0 → antonlytics-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: antonlytics
-Version: 2.0.0
+Version: 2.2.0
 Summary: Memory for AI Agents - Simple natural language SDK
 Home-page: https://github.com/Voidback-Inc/antonlytics-python-sdk
 Author: Voidback
@@ -31,6 +31,9 @@ Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
 Requires-Dist: black>=23.0.0; extra == "dev"
 Requires-Dist: flake8>=6.0.0; extra == "dev"
 Requires-Dist: mypy>=1.0.0; extra == "dev"
+Provides-Extra: langchain
+Requires-Dist: langchain-core>=0.2.0; extra == "langchain"
+Requires-Dist: pydantic>=2.0.0; extra == "langchain"
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file

{antonlytics-2.0.0 → antonlytics-2.2.0}/antonlytics/__init__.py

@@ -7,5 +7,5 @@ Simple SDK for giving your AI agent persistent memory.
 from .agent import Agent
 from .exceptions import AntonlyticsError, APIError, AuthenticationError
 
-__version__ = "2.0.0"
+__version__ = "2.2.0"
 __all__ = ["Agent", "AntonlyticsError", "APIError", "AuthenticationError"]

{antonlytics-2.0.0 → antonlytics-2.2.0}/antonlytics/agent.py

@@ -2,7 +2,7 @@
 Agent class for interacting with Antonlytics API.
 """
 
-from typing import Dict, Any, Optional, List
+from typing import Dict, Any, Optional, List, Iterator
 from .http_client import HTTPClient
 from .exceptions import AntonlyticsError
 
@@ -121,41 +121,89 @@ class Agent:
             
         return self.client.post('/api/v1/memory/chat/', payload)
     
-    def get_memory(self, query: Optional[str] = None) -> Dict[str, Any]:
+    def get_memory(
+        self,
+        query: Optional[str] = None,
+        *,
+        max_entities: int = 10_000,
+        page_size: int = 500,
+    ) -> Dict[str, Any]:
         """
         Get memory context for your own agent/model.
-        Returns structured knowledge graph data.
-        
+
+        Two modes:
+          - ``query`` provided: semantic top-K retrieval (single shot, server-ranked).
+            Returns the most relevant entities and their relationships.
+          - ``query=None``: enumerates the full project graph via cursor pagination,
+            auto-iterating pages internally. Stops at ``max_entities`` for safety.
+
         Args:
-            query: Optional natural language query to filter context
-            
+            query: Optional natural language query for ranked retrieval.
+            max_entities: Safety ceiling for full-graph mode (default 10000).
+            page_size: Pagination page size for full-graph mode (1-1000, default 500).
+
         Returns:
-            Dict with entities and relationships
-            
+            Dict with entities and relationships.
+
         Example:
-            >>> # Get all memory
-            >>> memory = agent.get_memory()
-            >>> 
-            >>> # Use with your own model
-            >>> your_model.chat(
-            ...     system="You are a sales assistant",
-            ...     context=memory,
-            ...     message="Who to follow up?"
-            ... )
+            >>> # Top-K retrieval for prompt context
+            >>> memory = agent.get_memory(query="What did Sarah say?")
+            >>>
+            >>> # Full project dump (auto-paginated)
+            >>> all_memory = agent.get_memory()
+            >>>
+            >>> # For very large projects, stream pages:
+            >>> for page in agent.iter_memory(page_size=500):
+            ...     process(page)
         """
         if query:
             response = self.client.post('/api/v1/memory/query/', {
                 'question': query,
-                'project_id': self.project_id
-            })
-            return response.get('graph_context', {})
-        else:
-            # Get all memory
-            response = self.client.post('/api/v1/memory/query/', {
-                'question': 'What do you know?',
-                'project_id': self.project_id
+                'project_id': self.project_id,
             })
             return response.get('graph_context', {})
+
+        # Full-graph mode: auto-paginate via /memory/list/.
+        entities: List[Dict[str, Any]] = []
+        relationships: List[Dict[str, Any]] = []
+        for page in self.iter_memory(page_size=page_size):
+            entities.extend(page.get('entities', []))
+            relationships.extend(page.get('relationships', []))
+            if len(entities) >= max_entities:
+                entities = entities[:max_entities]
+                break
+        return {'entities': entities, 'relationships': relationships}
+
+    def iter_memory(self, page_size: int = 500) -> Iterator[Dict[str, Any]]:
+        """
+        Stream the full project graph one page at a time.
+
+        Yields a dict ``{entities, relationships, next_cursor, has_more}`` per page.
+        Iteration stops automatically when the server reports no more pages.
+
+        Args:
+            page_size: Rows per page (1-1000, default 500).
+
+        Example:
+            >>> for page in agent.iter_memory(page_size=200):
+            ...     for entity in page["entities"]:
+            ...         print(entity["name"])
+        """
+        cursor: Optional[str] = None
+        while True:
+            payload: Dict[str, Any] = {
+                'project_id': self.project_id,
+                'limit': page_size,
+            }
+            if cursor:
+                payload['cursor'] = cursor
+            page = self.client.post('/api/v1/memory/list/', payload)
+            yield page
+            if not page.get('has_more'):
+                break
+            cursor = page.get('next_cursor')
+            if not cursor:
+                break
     
     def set_system_prompt(self, prompt: str) -> Dict[str, Any]:
         """

antonlytics-2.2.0/antonlytics/integrations/__init__.py

@@ -0,0 +1 @@
+"""Optional framework adapters. Import directly: `from antonlytics.integrations.langchain import AntonlyticsMemory`."""

antonlytics-2.2.0/antonlytics/integrations/langchain.py

@@ -0,0 +1,124 @@
+"""
+LangChain memory adapter.
+
+Usage:
+    from antonlytics import Agent
+    from antonlytics.integrations.langchain import AntonlyticsMemory
+
+    agent  = Agent(api_key="...", project_id="...")
+    memory = AntonlyticsMemory(agent=agent)
+
+    # Drop into any LangChain chain that accepts a memory.
+
+Install the LangChain extras first:
+    pip install antonlytics[langchain]
+"""
+from typing import Any, Dict, List, Optional
+
+try:
+    from langchain_core.memory import BaseMemory
+    from pydantic import Field
+except ImportError as exc:  # pragma: no cover
+    raise ImportError(
+        "AntonlyticsMemory requires langchain-core. Install with: "
+        "pip install antonlytics[langchain]"
+    ) from exc
+
+from ..agent import Agent
+
+
+def _format_graph(graph: Dict[str, Any]) -> str:
+    entities = graph.get("entities") or []
+    rels = graph.get("relationships") or []
+    if not entities and not rels:
+        return ""
+
+    lines: List[str] = []
+    if entities:
+        lines.append("Known facts:")
+        for e in entities[:50]:
+            props = e.get("properties") or {}
+            name = props.get("name") or e.get("name") or e.get("external_id") or ""
+            extras = ", ".join(f"{k}: {v}" for k, v in props.items() if k != "name")
+            lines.append(f"- [{e.get('type', '?')}] {name}" + (f" — {extras}" if extras else ""))
+
+    if rels:
+        # Build id → name map from entities for readable relationships.
+        id_to_name = {e.get("id"): (e.get("properties") or {}).get("name", "") for e in entities}
+        lines.append("Relationships:")
+        for r in rels[:50]:
+            src = id_to_name.get(r.get("source_id"), r.get("source_id", "?"))
+            tgt = id_to_name.get(r.get("target_id"), r.get("target_id", "?"))
+            lines.append(f"- {src} --[{r.get('type', '?')}]--> {tgt}")
+
+    return "\n".join(lines)
+
+
+class AntonlyticsMemory(BaseMemory):
+    """LangChain BaseMemory backed by an Antonlytics project.
+
+    `load_memory_variables` fetches a question-scoped slice of the knowledge
+    graph and formats it into a string suitable for a prompt.
+    `save_context` ingests the new turn so the next call sees it.
+    """
+
+    agent: Agent
+    memory_key: str = Field(default="history")
+    input_key: Optional[str] = Field(default="input")
+    output_key: Optional[str] = Field(default=None)
+    return_messages: bool = Field(default=False)
+
+    class Config:
+        arbitrary_types_allowed = True
+
+    @property
+    def memory_variables(self) -> List[str]:
+        return [self.memory_key]
+
+    def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, Any]:
+        query = ""
+        if self.input_key and self.input_key in inputs:
+            query = str(inputs[self.input_key])
+        elif inputs:
+            # Fall back to the first stringy value.
+            for v in inputs.values():
+                if isinstance(v, str):
+                    query = v
+                    break
+        try:
+            graph = self.agent.get_memory(query=query or None)
+        except Exception:
+            graph = {}
+        return {self.memory_key: _format_graph(graph)}
+
+    def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, Any]) -> None:
+        user_msg = ""
+        if self.input_key and self.input_key in inputs:
+            user_msg = str(inputs[self.input_key])
+        ai_msg = ""
+        if self.output_key and self.output_key in outputs:
+            ai_msg = str(outputs[self.output_key])
+        elif outputs:
+            for v in outputs.values():
+                if isinstance(v, str):
+                    ai_msg = v
+                    break
+
+        turn = ""
+        if user_msg:
+            turn += f"User: {user_msg}\n"
+        if ai_msg:
+            turn += f"Assistant: {ai_msg}"
+        turn = turn.strip()
+        if not turn:
+            return
+
+        try:
+            self.agent.ingest(turn)
+        except Exception:
+            # Never break the chain just because ingest hiccuped.
+            pass
+
+    def clear(self) -> None:
+        # Project-level wipe is not exposed by the API yet — intentional no-op.
+        return None

{antonlytics-2.0.0 → antonlytics-2.2.0}/antonlytics.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: antonlytics
-Version: 2.0.0
+Version: 2.2.0
 Summary: Memory for AI Agents - Simple natural language SDK
 Home-page: https://github.com/Voidback-Inc/antonlytics-python-sdk
 Author: Voidback
@@ -31,6 +31,9 @@ Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
 Requires-Dist: black>=23.0.0; extra == "dev"
 Requires-Dist: flake8>=6.0.0; extra == "dev"
 Requires-Dist: mypy>=1.0.0; extra == "dev"
+Provides-Extra: langchain
+Requires-Dist: langchain-core>=0.2.0; extra == "langchain"
+Requires-Dist: pydantic>=2.0.0; extra == "langchain"
 Dynamic: author
 Dynamic: home-page
 Dynamic: license-file

{antonlytics-2.0.0 → antonlytics-2.2.0}/antonlytics.egg-info/SOURCES.txt

@@ -1,6 +1,7 @@
-.DS_Store
+.gitignore
 LICENSE
 MANIFEST.in
+Makefile
 README.md
 pyproject.toml
 setup.py
@@ -13,4 +14,7 @@ antonlytics.egg-info/PKG-INFO
 antonlytics.egg-info/SOURCES.txt
 antonlytics.egg-info/dependency_links.txt
 antonlytics.egg-info/requires.txt
-antonlytics.egg-info/top_level.txt
+antonlytics.egg-info/top_level.txt
+antonlytics/integrations/__init__.py
+antonlytics/integrations/langchain.py
+examples/langchain_quickstart.py

{antonlytics-2.0.0 → antonlytics-2.2.0}/antonlytics.egg-info/requires.txt

@@ -6,3 +6,7 @@ pytest-cov>=4.0.0
 black>=23.0.0
 flake8>=6.0.0
 mypy>=1.0.0
+
+[langchain]
+langchain-core>=0.2.0
+pydantic>=2.0.0

antonlytics-2.2.0/examples/langchain_quickstart.py

@@ -0,0 +1,51 @@
+"""
+LangChain + Antonlytics quickstart.
+
+Install:
+    pip install antonlytics[langchain] langchain langchain-anthropic
+
+Run:
+    export ANTONLYTICS_API_KEY=...
+    export ANTONLYTICS_PROJECT_ID=...
+    export ANTHROPIC_API_KEY=...
+    python examples/langchain_quickstart.py
+"""
+import os
+
+from antonlytics import Agent
+from antonlytics.integrations.langchain import AntonlyticsMemory
+
+from langchain_anthropic import ChatAnthropic
+from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langchain.chains import ConversationChain
+
+
+def main() -> None:
+    agent = Agent(
+        api_key=os.environ["ANTONLYTICS_API_KEY"],
+        project_id=os.environ["ANTONLYTICS_PROJECT_ID"],
+    )
+
+    memory = AntonlyticsMemory(agent=agent, memory_key="history", input_key="input")
+
+    llm = ChatAnthropic(model="claude-sonnet-4-5", temperature=0)
+
+    prompt = ChatPromptTemplate.from_messages([
+        ("system",
+         "You are a helpful assistant. Use the prior knowledge below when relevant.\n\n"
+         "{history}"),
+        ("human", "{input}"),
+    ])
+
+    chain = ConversationChain(llm=llm, prompt=prompt, memory=memory, input_key="input")
+
+    # Teach it something.
+    chain.invoke({"input": "I had a sales call with Sarah from TechCorp about Enterprise pricing."})
+
+    # Ask it back with no keyword overlap. Hybrid retrieval should still pull Sarah/TechCorp.
+    out = chain.invoke({"input": "Anyone I should circle back with on a deal?"})
+    print(out["response"])
+
+
+if __name__ == "__main__":
+    main()

{antonlytics-2.0.0 → antonlytics-2.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "antonlytics"
-version = "2.0.0"
+version = "2.2.0"
 description = "Memory for AI Agents - Simple natural language SDK"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -37,6 +37,10 @@ dev = [
     "flake8>=6.0.0",
     "mypy>=1.0.0",
 ]
+langchain = [
+    "langchain-core>=0.2.0",
+    "pydantic>=2.0.0",
+]
 
 [project.urls]
 Homepage = "https://antonlytics.com"

{antonlytics-2.0.0 → antonlytics-2.2.0}/setup.py

@@ -9,7 +9,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="antonlytics",
-    version="2.0.0",
+    version="2.2.0",
     author="Voidback",
     author_email="hello@voidback.com",
     description="Memory for AI Agents - Simple natural language SDK",
@@ -48,6 +48,10 @@ setup(
             "flake8>=6.0.0",
             "mypy>=1.0.0",
         ],
+        "langchain": [
+            "langchain-core>=0.2.0",
+            "pydantic>=2.0.0",
+        ],
     },
     keywords="ai agent memory llm knowledge-graph natural-language",
 )