Test

Exécutez un agent avec un modèle scripté dans vos tests : agents.testing.

class ScriptedModel(steps: Sequence[ToolCallStep | ReplyStep])

class ScriptedModel(steps: Sequence[ToolCallStep | ReplyStep])

Répond à chaque appel au modèle avec l'étape suivante du script, dans l'ordre.

Transmettez-le à Agent.override, sous la forme agent.override(model=model), dans un test. Un appel au modèle au-delà de la dernière étape échoue avec une erreur qui indique la longueur du script, et une étape d'appel d'outil fait échouer son appel au modèle lorsque cet appel ne propose pas l'outil. Il ne répond que dans le processus de test : agents serve et agents deploy refusent un agent surchargé avec l'erreur UnsupportedAgentFeatureError.

Par exemple, dans un test asynchrone d'un agent avec un outil get_weather, le modèle appelle l'outil depuis run_typescript, puis répond :

model = agents.testing.ScriptedModel([
    agents.testing.run_typescript("return await tools.client.get_weather({city: 'Paris'})"),
    agents.testing.reply("Sunny."),
])
state = await agent.override(model=model).run("Weather in Paris?")
assert state.tool_calls[0].arguments == {"city": "Paris"}
assert state.text == "Sunny."
model.assert_consumed()

model = agents.testing.ScriptedModel([
    agents.testing.run_typescript("return await tools.client.get_weather({city: 'Paris'})"),
    agents.testing.reply("Sunny."),
])
state = await agent.override(model=model).run("Weather in Paris?")
assert state.tool_calls[0].arguments == {"city": "Paris"}
assert state.text == "Sunny."
model.assert_consumed()

def assert_consumed() -> None

def assert_consumed() -> None

Échoue si une étape du script n'a jamais été utilisée.

def testing.reply(text: str) -> ReplyStep

def testing.reply(text: str) -> ReplyStep

Scripte un appel au modèle qui répond avec text.

def testing.call(name: str, /, **arguments: JsonValue) -> ToolCallStep

def testing.call(name: str, /, **arguments: JsonValue) -> ToolCallStep

Scripte un appel au modèle qui demande l'outil name, le nom que voit le modèle, avec arguments.

Le modèle doit pouvoir appeler name directement. Un outil avec le model_access par défaut (un agents.ToolModelAccess) est appelé depuis run_typescript : scriptez-le avec run_typescript.

def testing.run_typescript(code: str) -> ToolCallStep

def testing.run_typescript(code: str) -> ToolCallStep

Scripte un appel au modèle qui exécute code dans run_typescript.

Le programme appelle un outil sous la forme tools.<namespace>.<name>(...), comme le ferait un vrai modèle.