mirror of
https://github.com/getzep/graphiti.git
synced 2025-06-27 02:00:02 +00:00
f1c2224c0e
* chore: Initial draft of stubs * chore: Add comments and mock implementation of the add_episode method * chore: Add success and error callbacks * chore: Add success and error callbacks * refactor: Fix conflicts with the latest merge
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]]
|