admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

model fixes

Browse source
This commit is contained in:
LUIS NOVO 2024-11-01 22:43:33 -03:00
parent 223f1bdaf5
commit 7dc37a3ac7
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
open_notebook/graphs/chat.py
View file

@ -26,7 +26,7 @@ def call_model_with_messages(state: ThreadState, config: RunnableConfig) -> dict
system_prompt = Prompter(prompt_template="chat").render(data=state)
payload = [system_prompt] + state.get("messages", [])
model = provision_model(str(payload), config, "chat")
ai_message = model.invoke(payload, [])
ai_message = model.invoke(payload)
return {"messages": ai_message}

4
open_notebook/graphs/utils.py
View file

@ -22,7 +22,9 @@ def provision_model(content, config, default_type):
)
return model_manager.get_default_model("large_context").to_langchain()
elif config.get("configurable", {}).get("model_id"):
return model_manager.get_model(config.get("configurable", {}).get("model_id"))
return model_manager.get_model(
config.get("configurable", {}).get("model_id")
).to_langchain()
else:
return model_manager.get_default_model(default_type).to_langchain()