graphiti/core/prompts/models.py

from typing import Callable, Protocol

from pydantic import BaseModel


class Message(BaseModel):
	role: str
	content: str


class PromptVersion(Protocol):
	def __call__(self, context: dict[str, any]) -> list[Message]: ...


PromptFunction = Callable[[dict[str, any]], list[Message]]