Spaces:
Running
Running
File size: 531 Bytes
0646b18 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from langchain_core.runnables import Runnable
from cuga.backend.cuga_graph.nodes.shared.base_agent import BaseAgent
from cuga.backend.llm.models import LLMManager
from cuga.backend.llm.utils.helpers import load_prompt_simple
llm_manager = LLMManager()
def reflection_task(llm, enable_format=False) -> Runnable:
prompt_template = load_prompt_simple(
"./prompts/reflection_system.jinja2",
"./prompts/reflection_user.jinja2",
)
return BaseAgent.get_chain(prompt_template, llm, wx_json_mode="no_format")
|