Pi Agent: Prompt Templates

from pi_agent import Agent

agent = Agent(name="reviewer")
result = agent.run(template="code_review", code="def add(a,b): return a+b")

name: code_explain
description: "Explain code functionality and principles"
system: |
  You are a code explanation expert. Explain code in simple language,
  including: 1) Function overview 2) Key logic 3) Potential issues
template: |
  Please explain the following {{ language }} code:

  ```{{ language }}
  {{ code }}

### (2) تعريف القالب في Python

```python
from pi_agent import PromptTemplate

explain_template = PromptTemplate(
    name="code_explain",
    description="Explain code functionality and principles",
    system="You are a code explanation expert",
    template="Please explain the following {language} code:\n{code}",
    variables=["language", "code"]
)

agent = Agent(name="explainer")
agent.add_template(explain_template)

result = agent.run(
    template="code_explain",
    language="python",
    code="def fib(n): return n if n < 2 else fib(n-1) + fib(n-2)"
)

template: |
  Translate the following text from {{ source_lang }} to {{ target_lang }}:
  {{ text }}

template: |
  Please review the following code:
  {{ code }}

  {% if focus_area %}
  Focus on {{ focus_area }} aspects.
  {% endif %}

  {% if severity == "strict" %}
  Please review strictly, list all issues.
  {% else %}
  Only list critical issues.
  {% endif %}

template: |
  Check if the following files conform to coding standards:
  {% for file in files %}
  - {{ file }}
  {% endfor %}

from pi_agent import Agent, PromptTemplate

review_template = PromptTemplate(
    name="review",
    template="Review code: {code}\nFocus: {focus}"
)

fix_template = PromptTemplate(
    name="fix",
    template="Fix code based on review: {review_result}\nOriginal: {code}"
)

agent = Agent(name="pipeline")
code = "def add(a, b): return a + b"
review_result = agent.run(template="review", code=code, focus="security")
fix_result = agent.run(template="fix", review_result=review_result, code=code)

from pi_agent import Agent

template_a = "Review this code: {code}"
template_b = "As a senior engineer, strictly review this code for security, performance, and readability: {code}"

agent = Agent(name="ab_test")
result_a = agent.run(template=template_a, code=sample_code)
result_b = agent.run(template=template_b, code=sample_code)