poetry

libs/langchain/langchain/chat_models/fireworks.py

@@ -1,8 +1,9 @@
+from typing import Any, Callable, Dict, Iterator, List, Mapping, Optional, Tuple, Union
+
 import fireworks
 import fireworks.client
-from langchain.utils.env import get_from_dict_or_env
 from pydantic import root_validator
-from typing import Any, Callable, Dict, Iterator, List, Mapping, Optional, Tuple, Union
+
 from langchain.adapters.openai import convert_message_to_dict
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,
@@ -16,18 +17,16 @@ from langchain.schema.messages import (
     BaseMessage,
     BaseMessageChunk,
     ChatMessage,
-    FunctionMessage,
-    HumanMessage,
-    SystemMessage,
-    AIMessageChunk,
-    BaseMessage,
-    BaseMessageChunk,
     ChatMessageChunk,
+    FunctionMessage,
     FunctionMessageChunk,
+    HumanMessage,
     HumanMessageChunk,
+    SystemMessage,
     SystemMessageChunk,
 )
 from langchain.schema.output import ChatGeneration, ChatGenerationChunk, ChatResult
+from langchain.utils.env import get_from_dict_or_env
 
 
 def _convert_delta_to_message_chunk(

libs/langchain/langchain/llms/fireworks.py

@@ -1,6 +1,9 @@
+from typing import Any, Callable, Dict, Iterator, List, Optional, Union
+
 import fireworks
 import fireworks.client
-from typing import Any, Callable, Dict, Iterator, List, Optional, Union
+from pydantic import root_validator
+
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
@@ -10,7 +13,6 @@ from langchain.schema.language_model import LanguageModelInput
 from langchain.schema.output import GenerationChunk
 from langchain.schema.runnable.config import RunnableConfig
 from langchain.utils.env import get_from_dict_or_env
-from pydantic import root_validator
 
 
 def _stream_response_to_generation_chunk(

libs/langchain/tests/integration_tests/llms/test_fireworks.py

@@ -1,14 +1,16 @@
 """Test Fireworks AI API Wrapper."""
 from typing import Generator
+
+import pytest
+
 from langchain.chains import LLMChain
-from langchain.prompts import PromptTemplate
 from langchain.llms.fireworks import Fireworks
+from langchain.prompts import PromptTemplate
 from langchain.prompts.chat import (
     ChatPromptTemplate,
     HumanMessagePromptTemplate,
 )
 from langchain.schema import LLMResult
-import pytest
 
 
 def test_fireworks_call() -> None: