chore: prepare for using pytest to run examples

guillaq · guillaq · commit d9f2b4d646d6 · 2025-04-10T14:50:20.000+02:00
diff --git a/Makefile b/Makefile
@@ -31,3 +31,6 @@ test:
 .PHONY: lock
 lock:
 	poetry lock --no-update
+
+test.examples:
+	pytest examples -n=5
diff --git a/examples/06_streaming_summary.py b/examples/06_streaming_summary.py
@@ -28,7 +28,7 @@ class TranslationOutput(BaseModel):
     )
 
 
-@workflowai.agent(id="french-translator", model=Model.CLAUDE_3_5_SONNET_LATEST)
+@workflowai.agent(id="french-translator", model=Model.GEMINI_2_0_FLASH_LATEST)
 async def translate_to_english(_: TranslationInput) -> TranslationOutput:
     """
     Translate French text into natural, fluent English.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,6 +16,7 @@ httpx = ">=0.27"
 pyright = "^1.1.390"
 pytest = "^8.2.2"
 pytest-asyncio = "^0.25.3"
+pytest-xdist = "^3.6.1"
 ruff = "^0.9.6"
 freezegun = "^1.5.1"
 pre-commit = "^4.0.1"

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ class TranslationOutput(BaseModel):`
`28`	`28`	`)`
`29`	`29`
`30`	`30`
`31`		`-@workflowai.agent(id="french-translator", model=Model.CLAUDE_3_5_SONNET_LATEST)`
	`31`	`+@workflowai.agent(id="french-translator", model=Model.GEMINI_2_0_FLASH_LATEST)`
`32`	`32`	`async def translate_to_english(_: TranslationInput) -> TranslationOutput:`
`33`	`33`	`"""`
`34`	`34`	`Translate French text into natural, fluent English.`