mirror of
https://github.com/getzep/graphiti.git
synced 2025-07-03 15:11:26 +00:00
16 lines
293 B
Python
16 lines
293 B
Python
|
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]]
|