Skip to content

Commit

Permalink
✨ Add types for funcchain
Browse files Browse the repository at this point in the history
  • Loading branch information
shroominic committed Feb 11, 2024
1 parent a887bde commit c26e238
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions src/funcchain/schema/types.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
from typing import Callable, Optional, Union

from langchain_core.chat_history import BaseChatMessageHistory
from langchain_core.language_models import BaseChatModel
from langchain_core.messages import AIMessage, HumanMessage
from langchain_core.runnables import Runnable

UniversalChatModel = Union[BaseChatModel, str, None]

ChatHandler = Runnable[HumanMessage, AIMessage]

ChatHistoryFactory = Callable[..., BaseChatMessageHistory]

OptionalChatHistoryFactory = Optional[ChatHistoryFactory]

0 comments on commit c26e238

Please sign in to comment.