feat(max): Include query description in formatted query result

[Twixes]

Problem

The assistant root didn't have a query title associated with query results, the only context being the earlier message history. Usually this has been fine, but e.g. if the previous message was an assistant message based on docs retrieval, root's summarization was confused what the query is (e.g. with a question like "tell me about key features of PostHog trends and show me an example of such a insight", root wasn't sure that it's looking at an example insight).

Changes

With this change, we're using the query_description arg of create_and_query_insight in query executor results formatting. This is a straightforward way to give root the right context.

How did you test this code?

TODO

[greptile-apps]

PR Summary

This PR enhances the assistant's context by including query descriptions in formatted query results, improving clarity when previous messages lack sufficient context.

• Modified QUERY_RESULTS_PROMPT in ee/hogai/query_executor/prompts.py to replace generic phrasing with specific query descriptions
• Added logic in ee/hogai/query_executor/nodes.py to find the original tool call and extract its query description
• Updated documentation for WorldMap visualization in multiple files to clarify it MUST be used with $geoip_country_name breakdown
• Implemented backward search through message history in QueryExecutorNode.run() to locate the relevant tool call ID
• Added proper type handling with Optional import for cases where tool calls might not be found

_{5 file(s) reviewed, 1 comment(s)
Edit PR Review Bot Settings | Greptile}

[greptile-apps]

style: Consider adding a break after finding the relevant tool call to exit both loops for better performance.

[github-actions]

Size Change: 0 B

Total Size: 9.72 MB

ℹ️ View Unchanged

Filename	Size
frontend/dist/toolbar.js	9.72 MB

_{compressed-size-action}