mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-08 03:10:24 +00:00
560e746cea
* reformat code, support flowllm 0.1.9 * Update README.md add FLOW_USE_FRAMEWORK=true * Update README_ZH.md add FLOW_USE_FRAMEWORK=true * Update index.md add FLOW_USE_FRAMEWORK=true * add env FLOW_APP_NAME=ReMe
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
from flowllm import C, BaseAsyncOp
|
|
from flowllm.utils.llm_utils import merge_messages_content
|
|
from loguru import logger
|
|
|
|
from reme_ai.schema import Message, Role
|
|
|
|
|
|
@C.register_op()
|
|
class BuildQueryOp(BaseAsyncOp):
|
|
file_path: str = __file__
|
|
|
|
async def async_execute(self):
|
|
if "query" in self.context:
|
|
query = self.context.query
|
|
|
|
elif "messages" in self.context:
|
|
if self.op_params.get("enable_llm_build", True):
|
|
execution_process = merge_messages_content(self.context.messages)
|
|
prompt = self.prompt_format(prompt_name="query_build", execution_process=execution_process)
|
|
message = await self.llm.achat(messages=[Message(role=Role.USER, content=prompt)])
|
|
query = message.content
|
|
|
|
else:
|
|
context_parts = []
|
|
message_summaries = []
|
|
for message in self.context.messages[-3:]: # Last 3 messages
|
|
content = message.content[:200] + "..." if len(message.content) > 200 else message.content
|
|
message_summaries.append(f"- {message.role.value}: {content}")
|
|
if message_summaries:
|
|
context_parts.append("Recent messages:\n" + "\n".join(message_summaries))
|
|
|
|
query = "\n\n".join(context_parts)
|
|
|
|
else:
|
|
raise RuntimeError("query or messages is required!")
|
|
|
|
logger.info(f"build.query={query}")
|
|
self.context.query = query
|