Method completion

app/e2e/sdk-api/src/agents/multi_agents.py:33–51

(self, prompt: str, openai_client)

@track_agent(name="engineer")
class EngineerAgent:
    async def completion(self, prompt: str, openai_client):
        res = await openai_client.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[
                {
                    "role": "system",
                    "content": "You are a "
                    "software "
                    "engineer and "
                    "only output "
                    "python code, "
                    "no markdown "
                    "tags.",
                },
                {"role": "user", "content": prompt},
            ],
            temperature=0.5,
        )
        return res.choices[0].message.content