From 1c08c2b2c2ea27d02f1543fcfe6da0c6dbb61e6a Mon Sep 17 00:00:00 2001
From: MisluNotFound <108245592+MisluNotFound@users.noreply.github.com>
Date: Fri, 18 Apr 2025 09:28:38 +0800
Subject: [PATCH] feat: Add page params. (#17)

---
 .../apo_select/tools/query_full_logs.py       |  6 +++--
 .../apo_select/tools/query_full_logs.yaml     | 24 +++++++++++++++++++
 2 files changed, 28 insertions(+), 2 deletions(-)

diff --git a/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.py b/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.py
index dad1d81ee8..fde5ef2dee 100644
--- a/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.py
+++ b/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.py
@@ -21,13 +21,15 @@ class QueryFullLogsTool(BuiltinTool):
         start_time = tool_parameters.get("startTime")
         end_time = tool_parameters.get("endTime")
         query = tool_parameters.get("query")
+        page_num = tool_parameters.get('pageNum', 1)
+        page_size = tool_parameters.get('pageSize', 100)
 
         params = {
             "dataBase": "apo",
             "endTime": end_time,
             "isExternal": False,
-            "pageNum" : 1,
-            "pageSize" : 100,
+            "pageNum" : page_num,
+            "pageSize" : page_size,
             "query" : query,
             "startTime" : start_time,
             "tableName" : "raw_logs",
diff --git a/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.yaml b/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.yaml
index ce26874129..2962667c5c 100644
--- a/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.yaml
+++ b/api/core/tools/builtin_tool/providers/apo_select/tools/query_full_logs.yaml
@@ -26,6 +26,30 @@ parameters:
     llm_description: Query condition(clickhouse where clause).
     form: llm
 
+  - name: pageNum
+    type: number
+    required: true
+    label: 
+      en_US: Current page
+      zh_Hans: 当前页
+    human_description:
+      en_US: Current page
+      zh_Hans: 当前页
+    llm_description: Current page
+    form: llm
+
+  - name: pageSize
+    type: number
+    required: true
+    label: 
+      en_US: Page size
+      zh_Hans: 页大小
+    human_description:
+      en_US: Page size
+      zh_Hans: 页大小
+    llm_description: Page size
+    form: llm
+
   - name: startTime
     type: number
     required: true