|
|
@@ -66,8 +66,12 @@ async def run_agent(query: Optional[str] = None, stream_output: bool = True) ->
|
|
|
prompt_path = Path(__file__).parent / "content_finder.prompt"
|
|
|
prompt = SimplePrompt(prompt_path)
|
|
|
|
|
|
- # 构建消息(替换 %query%)
|
|
|
- messages = prompt.build_messages(query=query)
|
|
|
+ # output 目录
|
|
|
+ output_dir = os.getenv("OUTPUT_DIR", ".output")
|
|
|
+
|
|
|
+ # 构建消息(替换 %query% 和 $output$)
|
|
|
+ messages = prompt.build_messages(query=query, output=output_dir)
|
|
|
+
|
|
|
|
|
|
# 初始化配置
|
|
|
api_key = os.getenv("OPEN_ROUTER_API_KEY")
|