1 files changed, 17 insertions, 0 deletions
diff --git a/py/types.py b/py/types.py
new file mode 100644
index 0000000..819afe5
--- /dev/null
+++ b/py/types.py
@@ -0,0 +1,17 @@
+from collections.abc import Sequence, Mapping
+from typing import TypedDict, Protocol
+
+
+class AIProvider(Protocol):
+
+  def __init__(self, config: Mapping[str, str]) -> None:
+      pass
+
+  def request(self, messages: Sequence[Message]) -> Generator[str]:
+      pass
+
+class Message(TypedDict):
+    role: str
+    content: str
+    type: str
+