Skip to content

Commit 4938cd7

Browse files
committed
fix: improve query agent tool descriptions and prompt clarity
1 parent febc8c9 commit 4938cd7

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

openkb/agent/query.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@
2727
- PageIndex documents (doc_type: pageindex): use get_page_content(doc_name, pages)
2828
with tight page ranges. The summary shows document tree structure with page
2929
ranges to help you target. Never fetch the whole document.
30-
5. Synthesize a clear, well-cited answer grounded in wiki content.
30+
5. Synthesize a clear, concise, well-cited answer grounded in wiki content.
3131
32-
Answer based only on wiki content. Before each tool call, briefly state what you
33-
are about to do. Be concise.
32+
Answer based only on wiki content. Be concise.
33+
Before each tool call, briefly state what you are about to do.
3434
3535
If you cannot find relevant information, say so clearly.
3636
"""
@@ -52,9 +52,9 @@ def read_file(path: str) -> str:
5252

5353
@function_tool
5454
def get_page_content_tool(doc_name: str, pages: str) -> str:
55-
"""Get text content of specific pages from a long document.
56-
Use this when you need detailed content from a document. The summary
57-
page shows document tree structure with page ranges.
55+
"""Get text content of specific pages from a PageIndex (long) document.
56+
Only use for documents with doc_type: pageindex. For short documents,
57+
use read_file instead.
5858
Args:
5959
doc_name: Document name (e.g. 'attention-is-all-you-need').
6060
pages: Page specification (e.g. '3-5,7,10-12').

openkb/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def init():
257257

258258
# Interactive prompts
259259
model = click.prompt(
260-
f"Model (e.g. gpt-5.4, anthropic/claude-sonnet-4-6) [default: {DEFAULT_CONFIG['model']}]",
260+
f"Model (e.g. gpt-5.4-mini, anthropic/claude-sonnet-4-6) [default: {DEFAULT_CONFIG['model']}]",
261261
default=DEFAULT_CONFIG["model"],
262262
show_default=False,
263263
)

0 commit comments

Comments
 (0)