-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy pathsql_pro.py
More file actions
38 lines (29 loc) · 1.21 KB
/
Copy pathsql_pro.py
File metadata and controls
38 lines (29 loc) · 1.21 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
from textwrap import dedent
from agno.agent import Agent
from rich.console import Console
from rich.panel import Panel
from devops_agent.utils.model_provider import get_model
from devops_agent.utils.prompt_generator_from_poml import prompt_from_poml
db_sql_pro_prompt = prompt_from_poml('db_sql_pro.poml')
console = Console()
def execute_db_sql_pro_agent(provider: str, model: str, debug_mode: bool = False, reasoning: bool = False) -> Agent:
console.print(Panel.fit(
"[bold cyan]Sql Pro Agent Invoking...[/bold cyan]",
border_style="cyan"
))
model = get_model(provider=provider, model_str=model)
db_optmization_assist = Agent(
name="Sql Pro Agent",
model=model,
description=dedent("""\
You are Master modern SQL with cloud-native databases, OLTP/OLAP optimization, and advanced query techniques.
Expert in performance tuning, data modeling, and hybrid analytical systems. Use PROACTIVELY for database
optimization or complex analysis.\
"""),
instructions=db_sql_pro_prompt,
stream_intermediate_steps=True,
markdown=True,
debug_mode=debug_mode,
reasoning=reasoning
)
return db_optmization_assist