Migrate langchain deps (#452)

This commit is contained in:
William FH
2024-05-14 16:48:21 -07:00
committed by GitHub
parent 77db7bf825
commit 72cb4bac1b
10 changed files with 86 additions and 53 deletions
@@ -1,6 +1,7 @@
import functools
from typing import Annotated, Any, Callable, Dict, List, Optional, Union
from langchain_community.adapters.openai import convert_message_to_dict
from langchain_core.messages import AIMessage, AnyMessage, BaseMessage, HumanMessage
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_core.runnables import Runnable, RunnableLambda
@@ -21,7 +22,6 @@ def langchain_to_openai_messages(messages: List[BaseMessage]):
Returns:
List[dict]: A list of openai messages.
"""
from langchain.adapters.openai import convert_message_to_dict # noqa: I001
return [
convert_message_to_dict(m) if isinstance(m, BaseMessage) else m