mirror of
https://github.com/lfnovo/open-notebook.git
synced 2026-04-29 03:50:04 +00:00
refactor transformation, add graph and admin
This commit is contained in:
parent
e3fa445fcc
commit
4a5d47d934
26 changed files with 326 additions and 384 deletions
|
|
@ -1,10 +1,8 @@
|
|||
from langchain_core.language_models.chat_models import BaseChatModel
|
||||
from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
|
||||
from loguru import logger
|
||||
|
||||
from open_notebook.domain.models import model_manager
|
||||
from open_notebook.models.llms import LanguageModel
|
||||
from open_notebook.prompter import Prompter
|
||||
from open_notebook.utils import token_count
|
||||
|
||||
|
||||
|
|
@ -31,24 +29,3 @@ def provision_langchain_model(
|
|||
|
||||
assert isinstance(model, LanguageModel), f"Model is not a LanguageModel: {model}"
|
||||
return model.to_langchain()
|
||||
|
||||
|
||||
# todo: turn into a graph
|
||||
def run_pattern(
|
||||
pattern_name: str,
|
||||
config,
|
||||
state: dict = {},
|
||||
parser=None,
|
||||
) -> BaseMessage:
|
||||
system_prompt = Prompter(prompt_template=pattern_name, parser=parser).render(
|
||||
data=state
|
||||
)
|
||||
payload = [SystemMessage(content=system_prompt)] + [
|
||||
HumanMessage(content=state["input_text"])
|
||||
]
|
||||
chain = provision_langchain_model(
|
||||
str(payload), config.get("configurable", {}).get("model_id"), "transformation"
|
||||
)
|
||||
|
||||
response = chain.invoke(payload)
|
||||
return response
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue