Skip to content

Commit

Permalink
🔧 fix pre-commit formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
shroominic committed Dec 15, 2023
1 parent e2c60db commit 6276d2f
Show file tree
Hide file tree
Showing 4 changed files with 28 additions and 5 deletions.
9 changes: 8 additions & 1 deletion src/funcchain/_llms.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,14 @@

from langchain_core.callbacks.manager import CallbackManagerForLLMRun
from langchain_core.language_models import BaseChatModel, BaseLanguageModel
from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage, ChatMessage, HumanMessage, SystemMessage
from langchain_core.messages import (
AIMessage,
AIMessageChunk,
BaseMessage,
ChatMessage,
HumanMessage,
SystemMessage,
)
from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, ChatResult
from langchain_core.pydantic_v1 import Field, root_validator
from langchain_core.utils import get_pydantic_field_names
Expand Down
12 changes: 10 additions & 2 deletions src/funcchain/chain/creation.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,11 @@
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.schema import AIMessage, BaseMessage, BaseOutputParser, HumanMessage
from langchain_core.schema.chat_history import BaseChatMessageHistory
from langchain_core.schema.runnable import RunnableSequence, RunnableSerializable, RunnableWithFallbacks
from langchain_core.schema.runnable import (
RunnableSequence,
RunnableSerializable,
RunnableWithFallbacks,
)
from PIL import Image
from pydantic import BaseModel

Expand All @@ -27,7 +31,11 @@
pydantic_to_grammar,
univeral_model_selector,
)
from .prompt import HumanImageMessagePromptTemplate, create_chat_prompt, create_instruction_prompt
from .prompt import (
HumanImageMessagePromptTemplate,
create_chat_prompt,
create_instruction_prompt,
)

ChainOutput = TypeVar("ChainOutput")

Expand Down
5 changes: 4 additions & 1 deletion src/funcchain/chain/prompt.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,10 @@
from langchain_core.chat_history import BaseChatMessageHistory
from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.prompts.chat import BaseStringMessagePromptTemplate, MessagePromptTemplateT
from langchain_core.prompts.chat import (
BaseStringMessagePromptTemplate,
MessagePromptTemplateT,
)
from langchain_core.prompts.prompt import PromptTemplate
from PIL import Image
from pydantic import BaseModel
Expand Down
7 changes: 6 additions & 1 deletion src/funcchain/utils/model_defaults.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,12 @@
from pathlib import Path
from typing import Any

from langchain.chat_models import AzureChatOpenAI, ChatAnthropic, ChatGooglePalm, ChatOpenAI
from langchain.chat_models import (
AzureChatOpenAI,
ChatAnthropic,
ChatGooglePalm,
ChatOpenAI,
)
from langchain_core.language_models import BaseChatModel

from .._llms import ChatLlamaCpp
Expand Down

0 comments on commit 6276d2f

Please sign in to comment.