Merge pull request #136 from langchain-ai/nc/format

Format
langchain-ai · Jan 26, 2024 · 7dd02a6 · 7dd02a6
2 parents ace77a9 + fff4982
commit 7dd02a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/backend/app/storage.py b/backend/app/storage.py
@@ -1,7 +1,6 @@
 import os
 from datetime import datetime
 from typing import List, Sequence
-from app.stream import map_chunk_to_msg
 
 import orjson
 from langchain.schema.messages import AnyMessage
@@ -12,6 +11,7 @@
 
 from app.agent import AgentType, get_agent_executor
 from app.schema import Assistant, AssistantWithoutUserId, Thread, ThreadWithoutUserId
+from app.stream import map_chunk_to_msg
 
 
 def assistants_list_key(user_id: str) -> str:

diff --git a/backend/app/stream.py b/backend/app/stream.py
@@ -1,8 +1,10 @@
 from typing import AsyncIterator, Optional, Sequence, Union
 
+import orjson
 from langchain_core.messages import (
     AIMessage,
     AIMessageChunk,
+    AnyMessage,
     BaseMessage,
     BaseMessageChunk,
     ChatMessage,
@@ -11,11 +13,9 @@
     FunctionMessageChunk,
     HumanMessage,
     HumanMessageChunk,
-    AnyMessage,
 )
 from langchain_core.runnables import Runnable, RunnableConfig
 from langserve.serialization import WellKnownLCSerializer
-import orjson
 
 MessagesStream = AsyncIterator[Union[list[AnyMessage], str]]