|
6 | 6 | from pydantic_ai import Agent |
7 | 7 | from pydantic_ai.models import Model |
8 | 8 |
|
9 | | -from slackbot.modules import display_signature |
10 | 9 | from slackbot.search import ( |
| 10 | + display_callable_signature, |
11 | 11 | explore_module_offerings, |
12 | 12 | get_latest_prefect_release_notes, |
13 | 13 | review_common_3x_gotchas, |
14 | | - review_top_level_prefect_api, |
15 | 14 | search_marvin_docs, |
16 | 15 | search_prefect_2x_docs, |
17 | 16 | search_prefect_3x_docs, |
18 | | - verify_import_statements, |
19 | 17 | ) |
20 | 18 |
|
21 | 19 |
|
@@ -49,31 +47,28 @@ def create_research_agent( |
49 | 47 | deps_type=ResearchContext, |
50 | 48 | result_type=ResearchFindings, |
51 | 49 | system_prompt="""You are a specialized research agent for Prefect documentation and knowledge. |
52 | | -Your job is to thoroughly research topics by using ALL available tools to gather comprehensive, accurate information. |
| 50 | +Your job is to thoroughly research topics by using available tools to gather comprehensive, accurate information. |
53 | 51 |
|
54 | 52 | Your research process: |
55 | | -1. Start with broad searches to understand the topic context |
| 53 | +1. Start with broad documentation searches to understand the topic context |
56 | 54 | 2. Use multiple search queries with different keywords - don't stop at first result |
57 | | -3. For code examples: ALWAYS verify imports with verify_import_statements |
58 | | -4. Focus on Prefect 3.x documentation unless explicitly asked about 2.x or older versions |
59 | | -5. Review gotchas and release notes for recent changes |
60 | | -6. Explore relevant modules for deeper understanding |
| 55 | +3. Use explore_module_offerings to understand what's available in relevant modules |
| 56 | +4. Use display_callable_signature to get detailed signatures of functions, classes, and methods when needed |
| 57 | +5. Focus on Prefect 3.x documentation unless explicitly asked about 2.x or older versions |
| 58 | +6. Review gotchas and release notes for recent changes |
61 | 59 |
|
62 | 60 | Remember: You are the research specialist. The main agent relies on you for accurate, comprehensive information. |
63 | 61 | Be thorough - use tools repeatedly until you have complete information. |
64 | 62 | Default to Prefect 3.x unless the user explicitly asks about 2.x or version compatibility. |
65 | | -You don't need to use all the tools all the time, but use relevant ones repeatedly if needed. |
66 | 63 | """, |
67 | 64 | tools=[ |
68 | 65 | get_latest_prefect_release_notes, |
69 | 66 | search_prefect_2x_docs, |
70 | | - display_signature, |
| 67 | + display_callable_signature, |
71 | 68 | search_prefect_3x_docs, |
72 | 69 | search_marvin_docs, |
73 | | - review_top_level_prefect_api, |
74 | 70 | explore_module_offerings, |
75 | 71 | review_common_3x_gotchas, |
76 | | - verify_import_statements, |
77 | 72 | ], |
78 | 73 | ) |
79 | 74 |
|
|
0 commit comments