graphiti-core 0.3.6__tar.gz → 0.3.8__tar.gz

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: graphiti-core
-Version: 0.3.6
+Version: 0.3.8
 Summary: A temporal graph building library
 License: Apache-2.0
 Author: Paul Paliychuk
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: diskcache (>=5.6.3,<6.0.0)
 Requires-Dist: neo4j (>=5.23.0,<6.0.0)
 Requires-Dist: numpy (>=1.0.0)
-Requires-Dist: openai (>=1.38.0,<2.0.0)
+Requires-Dist: openai (>=1.50.2,<2.0.0)
 Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: tenacity (<9.0.0)
 Description-Content-Type: text/markdown

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/edges.py

@@ -24,9 +24,9 @@ from uuid import uuid4
 from neo4j import AsyncDriver
 from pydantic import BaseModel, Field
 
+from graphiti_core.embedder import EmbedderClient
 from graphiti_core.errors import EdgeNotFoundError, GroupsEdgesNotFoundError
 from graphiti_core.helpers import parse_db_date
-from graphiti_core.llm_client.config import EMBEDDING_DIM
 from graphiti_core.nodes import Node
 
 logger = logging.getLogger(__name__)
@@ -171,17 +171,16 @@ class EntityEdge(Edge):
         default=None, description='datetime of when the fact stopped being true'
     )
 
-    async def generate_embedding(self, embedder, model='text-embedding-3-small'):
+    async def generate_embedding(self, embedder: EmbedderClient):
         start = time()
 
         text = self.fact.replace('\n', ' ')
-        embedding = (await embedder.create(input=[text], model=model)).data[0].embedding
-        self.fact_embedding = embedding[:EMBEDDING_DIM]
+        self.fact_embedding = await embedder.create(input=[text])
 
         end = time()
         logger.info(f'embedded {text} in {end - start} ms')
 
-        return embedding
+        return self.fact_embedding
 
     async def save(self, driver: AsyncDriver):
         result = await driver.execute_query(

graphiti_core-0.3.8/graphiti_core/embedder/__init__.py

@@ -0,0 +1,4 @@
+from .client import EmbedderClient
+from .openai import OpenAIEmbedder, OpenAIEmbedderConfig
+
+__all__ = ['EmbedderClient', 'OpenAIEmbedder', 'OpenAIEmbedderConfig']

graphiti_core-0.3.6/graphiti_core/helpers.py → graphiti_core-0.3.8/graphiti_core/embedder/client.py

@@ -14,10 +14,21 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-from datetime import datetime
+from abc import ABC, abstractmethod
+from typing import Iterable, List, Literal
 
-from neo4j import time as neo4j_time
+from pydantic import BaseModel, Field
 
+EMBEDDING_DIM = 1024
 
-def parse_db_date(neo_date: neo4j_time.DateTime | None) -> datetime | None:
-    return neo_date.to_native() if neo_date else None
+
+class EmbedderConfig(BaseModel):
+    embedding_dim: Literal[1024] = Field(default=EMBEDDING_DIM, frozen=True)
+
+
+class EmbedderClient(ABC):
+    @abstractmethod
+    async def create(
+        self, input: str | List[str] | Iterable[int] | Iterable[Iterable[int]]
+    ) -> list[float]:
+        pass

graphiti_core-0.3.8/graphiti_core/embedder/openai.py

@@ -0,0 +1,48 @@
+"""
+Copyright 2024, Zep Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from typing import Iterable, List
+
+from openai import AsyncOpenAI
+from openai.types import EmbeddingModel
+
+from .client import EmbedderClient, EmbedderConfig
+
+DEFAULT_EMBEDDING_MODEL = 'text-embedding-3-small'
+
+
+class OpenAIEmbedderConfig(EmbedderConfig):
+    embedding_model: EmbeddingModel | str = DEFAULT_EMBEDDING_MODEL
+    api_key: str | None = None
+    base_url: str | None = None
+
+
+class OpenAIEmbedder(EmbedderClient):
+    """
+    OpenAI Embedder Client
+    """
+
+    def __init__(self, config: OpenAIEmbedderConfig | None = None):
+        if config is None:
+            config = OpenAIEmbedderConfig()
+        self.config = config
+        self.client = AsyncOpenAI(api_key=config.api_key, base_url=config.base_url)
+
+    async def create(
+        self, input: str | List[str] | Iterable[int] | Iterable[Iterable[int]]
+    ) -> list[float]:
+        result = await self.client.embeddings.create(input=input, model=self.config.embedding_model)
+        return result.data[0].embedding[: self.config.embedding_dim]

graphiti_core-0.3.8/graphiti_core/embedder/voyage.py

@@ -0,0 +1,47 @@
+"""
+Copyright 2024, Zep Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from typing import Iterable, List
+
+import voyageai  # type: ignore
+from pydantic import Field
+
+from .client import EmbedderClient, EmbedderConfig
+
+DEFAULT_EMBEDDING_MODEL = 'voyage-3'
+
+
+class VoyageAIEmbedderConfig(EmbedderConfig):
+    embedding_model: str = Field(default=DEFAULT_EMBEDDING_MODEL)
+    api_key: str | None = None
+
+
+class VoyageAIEmbedder(EmbedderClient):
+    """
+    VoyageAI Embedder Client
+    """
+
+    def __init__(self, config: VoyageAIEmbedderConfig | None = None):
+        if config is None:
+            config = VoyageAIEmbedderConfig()
+        self.config = config
+        self.client = voyageai.AsyncClient(api_key=config.api_key)
+
+    async def create(
+        self, input: str | List[str] | Iterable[int] | Iterable[Iterable[int]]
+    ) -> list[float]:
+        result = await self.client.embed(input, model=self.config.embedding_model)
+        return result.embeddings[0][: self.config.embedding_dim]

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/graphiti.py

@@ -23,6 +23,7 @@ from dotenv import load_dotenv
 from neo4j import AsyncGraphDatabase
 
 from graphiti_core.edges import EntityEdge, EpisodicEdge
+from graphiti_core.embedder import EmbedderClient, OpenAIEmbedder
 from graphiti_core.llm_client import LLMClient, OpenAIClient
 from graphiti_core.nodes import EntityNode, EpisodeType, EpisodicNode
 from graphiti_core.search.search import SearchConfig, search
@@ -83,6 +84,7 @@ class Graphiti:
         user: str,
         password: str,
         llm_client: LLMClient | None = None,
+        embedder: EmbedderClient | None = None,
         store_raw_episode_content: bool = True,
     ):
         """
@@ -128,6 +130,10 @@ class Graphiti:
             self.llm_client = llm_client
         else:
             self.llm_client = OpenAIClient()
+        if embedder:
+            self.embedder = embedder
+        else:
+            self.embedder = OpenAIEmbedder()
 
     async def close(self):
         """
@@ -161,7 +167,7 @@ class Graphiti:
         """
         await self.driver.close()
 
-    async def build_indices_and_constraints(self):
+    async def build_indices_and_constraints(self, delete_existing: bool = False):
         """
         Build indices and constraints in the Neo4j database.
 
@@ -171,6 +177,9 @@ class Graphiti:
         Parameters
         ----------
         self
+        delete_existing : bool, optional
+            Whether to clear existing indices before creating new ones.
+
 
         Returns
         -------
@@ -191,7 +200,7 @@ class Graphiti:
         Caution: Running this method on a large existing database may take some time
         and could impact database performance during execution.
         """
-        await build_indices_and_constraints(self.driver)
+        await build_indices_and_constraints(self.driver, delete_existing)
 
     async def retrieve_episodes(
         self,
@@ -287,7 +296,6 @@ class Graphiti:
             start = time()
 
             entity_edges: list[EntityEdge] = []
-            embedder = self.llm_client.get_embedder()
             now = datetime.now()
 
             previous_episodes = await self.retrieve_episodes(
@@ -315,7 +323,7 @@ class Graphiti:
             # Calculate Embeddings
 
             await asyncio.gather(
-                *[node.generate_name_embedding(embedder) for node in extracted_nodes]
+                *[node.generate_name_embedding(self.embedder) for node in extracted_nodes]
             )
 
             # Resolve extracted nodes with nodes already in the graph and extract facts
@@ -343,7 +351,7 @@ class Graphiti:
             # calculate embeddings
             await asyncio.gather(
                 *[
-                    edge.generate_embedding(embedder)
+                    edge.generate_embedding(self.embedder)
                     for edge in extracted_edges_with_resolved_pointers
                 ]
             )
@@ -436,7 +444,7 @@ class Graphiti:
             if update_communities:
                 await asyncio.gather(
                     *[
-                        update_community(self.driver, self.llm_client, embedder, node)
+                        update_community(self.driver, self.llm_client, self.embedder, node)
                         for node in nodes
                     ]
                 )
@@ -485,7 +493,6 @@ class Graphiti:
         """
         try:
             start = time()
-            embedder = self.llm_client.get_embedder()
             now = datetime.now()
 
             episodes = [
@@ -517,8 +524,8 @@ class Graphiti:
 
             # Generate embeddings
             await asyncio.gather(
-                *[node.generate_name_embedding(embedder) for node in extracted_nodes],
-                *[edge.generate_embedding(embedder) for edge in extracted_edges],
+                *[node.generate_name_embedding(self.embedder) for node in extracted_nodes],
+                *[edge.generate_embedding(self.embedder) for edge in extracted_edges],
             )
 
             # Dedupe extracted nodes, compress extracted edges
@@ -561,14 +568,14 @@ class Graphiti:
             raise e
 
     async def build_communities(self):
-        embedder = self.llm_client.get_embedder()
-
         # Clear existing communities
         await remove_communities(self.driver)
 
         community_nodes, community_edges = await build_communities(self.driver, self.llm_client)
 
-        await asyncio.gather(*[node.generate_name_embedding(embedder) for node in community_nodes])
+        await asyncio.gather(
+            *[node.generate_name_embedding(self.embedder) for node in community_nodes]
+        )
 
         await asyncio.gather(*[node.save(self.driver) for node in community_nodes])
         await asyncio.gather(*[edge.save(self.driver) for edge in community_edges])
@@ -619,7 +626,7 @@ class Graphiti:
         edges = (
             await search(
                 self.driver,
-                self.llm_client.get_embedder(),
+                self.embedder,
                 query,
                 group_ids,
                 search_config,
@@ -636,9 +643,7 @@ class Graphiti:
         group_ids: list[str] | None = None,
         center_node_uuid: str | None = None,
     ) -> SearchResults:
-        return await search(
-            self.driver, self.llm_client.get_embedder(), query, group_ids, config, center_node_uuid
-        )
+        return await search(self.driver, self.embedder, query, group_ids, config, center_node_uuid)
 
     async def get_nodes_by_query(
         self,
@@ -683,14 +688,15 @@ class Graphiti:
         to each individual search method before results are combined and deduplicated.
         If not specified, a default limit (defined in the search functions) will be used.
         """
-        embedder = self.llm_client.get_embedder()
         search_config = (
             NODE_HYBRID_SEARCH_RRF if center_node_uuid is None else NODE_HYBRID_SEARCH_NODE_DISTANCE
         )
         search_config.limit = limit
 
         nodes = (
-            await search(self.driver, embedder, query, group_ids, search_config, center_node_uuid)
+            await search(
+                self.driver, self.embedder, query, group_ids, search_config, center_node_uuid
+            )
         ).nodes
         return nodes
 

graphiti_core-0.3.8/graphiti_core/helpers.py

@@ -0,0 +1,53 @@
+"""
+Copyright 2024, Zep Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from datetime import datetime
+
+from neo4j import time as neo4j_time
+
+
+def parse_db_date(neo_date: neo4j_time.DateTime | None) -> datetime | None:
+    return neo_date.to_native() if neo_date else None
+
+
+def lucene_sanitize(query: str) -> str:
+    # Escape special characters from a query before passing into Lucene
+    # + - && || ! ( ) { } [ ] ^ " ~ * ? : \
+    escape_map = str.maketrans(
+        {
+            '+': r'\+',
+            '-': r'\-',
+            '&': r'\&',
+            '|': r'\|',
+            '!': r'\!',
+            '(': r'\(',
+            ')': r'\)',
+            '{': r'\{',
+            '}': r'\}',
+            '[': r'\[',
+            ']': r'\]',
+            '^': r'\^',
+            '"': r'\"',
+            '~': r'\~',
+            '*': r'\*',
+            '?': r'\?',
+            ':': r'\:',
+            '\\': r'\\',
+        }
+    )
+
+    sanitized = query.translate(escape_map)
+    return sanitized

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/llm_client/anthropic_client.py

@@ -20,7 +20,6 @@ import typing
 
 import anthropic
 from anthropic import AsyncAnthropic
-from openai import AsyncOpenAI
 
 from ..prompts.models import Message
 from .client import LLMClient
@@ -47,10 +46,6 @@ class AnthropicClient(LLMClient):
             max_retries=1,
         )
 
-    def get_embedder(self) -> typing.Any:
-        openai_client = AsyncOpenAI()
-        return openai_client.embeddings
-
     async def _generate_response(self, messages: list[Message]) -> dict[str, typing.Any]:
         system_message = messages[0]
         user_messages = [{'role': m.role, 'content': m.content} for m in messages[1:]] + [

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/llm_client/client.py

@@ -55,10 +55,6 @@ class LLMClient(ABC):
         self.cache_enabled = cache
         self.cache_dir = Cache(DEFAULT_CACHE_DIR)  # Create a cache directory
 
-    @abstractmethod
-    def get_embedder(self) -> typing.Any:
-        pass
-
     @retry(
         stop=stop_after_attempt(4),
         wait=wait_random_exponential(multiplier=10, min=5, max=120),

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/llm_client/config.py

@@ -14,7 +14,6 @@ See the License for the specific language governing permissions and
 limitations under the License.
 """
 
-EMBEDDING_DIM = 1024
 DEFAULT_MAX_TOKENS = 16384
 DEFAULT_TEMPERATURE = 0
 

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/llm_client/groq_client.py

@@ -21,7 +21,6 @@ import typing
 import groq
 from groq import AsyncGroq
 from groq.types.chat import ChatCompletionMessageParam
-from openai import AsyncOpenAI
 
 from ..prompts.models import Message
 from .client import LLMClient
@@ -44,10 +43,6 @@ class GroqClient(LLMClient):
 
         self.client = AsyncGroq(api_key=config.api_key)
 
-    def get_embedder(self) -> typing.Any:
-        openai_client = AsyncOpenAI()
-        return openai_client.embeddings
-
     async def _generate_response(self, messages: list[Message]) -> dict[str, typing.Any]:
         msgs: list[ChatCompletionMessageParam] = []
         for m in messages:

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/llm_client/openai_client.py

@@ -49,9 +49,6 @@ class OpenAIClient(LLMClient):
         __init__(config: LLMConfig | None = None, cache: bool = False, client: typing.Any = None):
             Initializes the OpenAIClient with the provided configuration, cache setting, and client.
 
-        get_embedder() -> typing.Any:
-            Returns the embedder from the OpenAI client.
-
         _generate_response(messages: list[Message]) -> dict[str, typing.Any]:
             Generates a response from the language model based on the provided messages.
     """
@@ -78,9 +75,6 @@ class OpenAIClient(LLMClient):
         else:
             self.client = client
 
-    def get_embedder(self) -> typing.Any:
-        return self.client.embeddings
-
     async def _generate_response(self, messages: list[Message]) -> dict[str, typing.Any]:
         openai_messages: list[ChatCompletionMessageParam] = []
         for m in messages:

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/llm_client/utils.py

@@ -15,22 +15,18 @@ limitations under the License.
 """
 
 import logging
-import typing
 from time import time
 
-from graphiti_core.llm_client.config import EMBEDDING_DIM
+from graphiti_core.embedder.client import EmbedderClient
 
 logger = logging.getLogger(__name__)
 
 
-async def generate_embedding(
-    embedder: typing.Any, text: str, model: str = 'text-embedding-3-small'
-):
+async def generate_embedding(embedder: EmbedderClient, text: str):
     start = time()
 
     text = text.replace('\n', ' ')
-    embedding = (await embedder.create(input=[text], model=model)).data[0].embedding
-    embedding = embedding[:EMBEDDING_DIM]
+    embedding = await embedder.create(input=[text])
 
     end = time()
     logger.debug(f'embedded text of length {len(text)} in {end - start} ms')

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/nodes.py

@@ -25,8 +25,8 @@ from uuid import uuid4
 from neo4j import AsyncDriver
 from pydantic import BaseModel, Field
 
+from graphiti_core.embedder import EmbedderClient
 from graphiti_core.errors import NodeNotFoundError
-from graphiti_core.llm_client.config import EMBEDDING_DIM
 
 logger = logging.getLogger(__name__)
 
@@ -212,15 +212,14 @@ class EntityNode(Node):
     name_embedding: list[float] | None = Field(default=None, description='embedding of the name')
     summary: str = Field(description='regional summary of surrounding edges', default_factory=str)
 
-    async def generate_name_embedding(self, embedder, model='text-embedding-3-small'):
+    async def generate_name_embedding(self, embedder: EmbedderClient):
         start = time()
         text = self.name.replace('\n', ' ')
-        embedding = (await embedder.create(input=[text], model=model)).data[0].embedding
-        self.name_embedding = embedding[:EMBEDDING_DIM]
+        self.name_embedding = await embedder.create(input=[text])
         end = time()
         logger.info(f'embedded {text} in {end - start} ms')
 
-        return embedding
+        return self.name_embedding
 
     async def save(self, driver: AsyncDriver):
         result = await driver.execute_query(
@@ -323,15 +322,14 @@ class CommunityNode(Node):
 
         return result
 
-    async def generate_name_embedding(self, embedder, model='text-embedding-3-small'):
+    async def generate_name_embedding(self, embedder: EmbedderClient):
         start = time()
         text = self.name.replace('\n', ' ')
-        embedding = (await embedder.create(input=[text], model=model)).data[0].embedding
-        self.name_embedding = embedding[:EMBEDDING_DIM]
+        self.name_embedding = await embedder.create(input=[text])
         end = time()
         logger.info(f'embedded {text} in {end - start} ms')
 
-        return embedding
+        return self.name_embedding
 
     @classmethod
     async def get_by_uuid(cls, driver: AsyncDriver, uuid: str):

graphiti_core-0.3.8/graphiti_core/prompts/eval.py

@@ -0,0 +1,90 @@
+"""
+Copyright 2024, Zep Software, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import json
+from typing import Any, Protocol, TypedDict
+
+from .models import Message, PromptFunction, PromptVersion
+
+
+class Prompt(Protocol):
+    qa_prompt: PromptVersion
+    eval_prompt: PromptVersion
+
+
+class Versions(TypedDict):
+    qa_prompt: PromptFunction
+    eval_prompt: PromptFunction
+
+
+def qa_prompt(context: dict[str, Any]) -> list[Message]:
+    sys_prompt = """You are Alice and should respond to all questions from the first person perspective of Alice"""
+
+    user_prompt = f"""
+    Your task is to briefly answer the question in the way that you think Alice would answer the question.
+    You are given the following entity summaries and facts to help you determine the answer to your question.
+    <ENTITY_SUMMARIES>
+    {json.dumps(context['entity_summaries'])}
+    </ENTITY_SUMMARIES
+    <FACTS>
+    {json.dumps(context['facts'])}
+    </FACTS>
+    <QUESTION>
+    {context['query']}
+    </QUESTION>
+    respond with a JSON object in the following format:
+    {{
+        "ANSWER": "how Alice would answer the question"
+    }}
+    """
+    return [
+        Message(role='system', content=sys_prompt),
+        Message(role='user', content=user_prompt),
+    ]
+
+
+def eval_prompt(context: dict[str, Any]) -> list[Message]:
+    sys_prompt = (
+        """You are a judge that determines if answers to questions match a gold standard answer"""
+    )
+
+    user_prompt = f"""
+    Given the QUESTION and the gold standard ANSWER determine if the RESPONSE to the question is correct or incorrect.
+    Although the RESPONSE may be more verbose, mark it as correct as long as it references the same topic 
+    as the gold standard ANSWER. Also include your reasoning for the grade.
+    <QUESTION>
+    {context['query']}
+    </QUESTION>
+    <ANSWER>
+    {context['answer']}
+    </ANSWER>
+    <RESPONSE>
+    {context['response']}
+    </RESPONSE>
+    
+    respond with a JSON object in the following format:
+    {{
+        "is_correct": "boolean if the answer is correct or incorrect"
+        "reasoning": "why you determined the response was correct or incorrect"
+    }}
+    """
+    return [
+        Message(role='system', content=sys_prompt),
+        Message(role='user', content=user_prompt),
+    ]
+
+
+versions: Versions = {'qa_prompt': qa_prompt, 'eval_prompt': eval_prompt}

{graphiti_core-0.3.6 → graphiti_core-0.3.8}/graphiti_core/prompts/lib.py

@@ -34,6 +34,9 @@ from .dedupe_nodes import (
 from .dedupe_nodes import (
     versions as dedupe_nodes_versions,
 )
+from .eval import Prompt as EvalPrompt
+from .eval import Versions as EvalVersions
+from .eval import versions as eval_versions
 from .extract_edge_dates import (
     Prompt as ExtractEdgeDatesPrompt,
 )
@@ -84,6 +87,7 @@ class PromptLibrary(Protocol):
     invalidate_edges: InvalidateEdgesPrompt
     extract_edge_dates: ExtractEdgeDatesPrompt
     summarize_nodes: SummarizeNodesPrompt
+    eval: EvalPrompt
 
 
 class PromptLibraryImpl(TypedDict):
@@ -94,6 +98,7 @@ class PromptLibraryImpl(TypedDict):
     invalidate_edges: InvalidateEdgesVersions
     extract_edge_dates: ExtractEdgeDatesVersions
     summarize_nodes: SummarizeNodesVersions
+    eval: EvalVersions
 
 
 class VersionWrapper:
@@ -124,5 +129,6 @@ PROMPT_LIBRARY_IMPL: PromptLibraryImpl = {
     'invalidate_edges': invalidate_edges_versions,
     'extract_edge_dates': extract_edge_dates_versions,
     'summarize_nodes': summarize_nodes_versions,
+    'eval': eval_versions,
 }
 prompt_library: PromptLibrary = PromptLibraryWrapper(PROMPT_LIBRARY_IMPL)  # type: ignore[assignment]