Checking and Refining the Agent's Reasoning › Building Verification Into Your Agent

from openai import OpenAI

client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))


class PersonalAssistant:
    def __init__(self, client):
        self.client = client

    def solve_with_verification(self, problem, verify=True):
        """Solve a problem with optional verification."""

        # Initial solution
        solution = self._solve(problem)

        if not verify:
            return solution

        # Verification step
        verified = self._verify(problem, solution)

        return verified

    def _solve(self, problem):
        """Generate initial solution."""
        response = self.client.chat.completions.create(
            model="gpt-5",
            messages=[{"role": "user", "content": f"{problem}\n\nSolve step by step."}],
        )
        return response.choices[0].message.content

    def _verify(self, problem, solution):
        """Verify and potentially refine the solution."""
        response = self.client.chat.completions.create(
            model="gpt-5",
            messages=[
                {
                    "role": "user",
                    "content": f"""Problem: {problem}

Solution: {solution}

Verify this solution. If correct, return it as-is. 
If you find errors, return a corrected version.""",
                }
            ],
        )
        return response.choices[0].message.content