initial commit

Author: samuelcolvin

.gitignore

@@ -0,0 +1,11 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+*.svg

.python-version

@@ -0,0 +1 @@
+3.12

01-logfire-hello-world/main.py

@@ -0,0 +1,5 @@
+import logfire
+
+logfire.configure(service_name='hello-world')
+
+logfire.info('hello {place}', place='world')

02-pai-mcp-sampling/client.py

@@ -0,0 +1,25 @@
+from pathlib import Path
+
+import logfire
+from pydantic_ai import Agent
+from pydantic_ai.mcp import MCPServerStdio
+
+logfire.configure(service_name='mcp-sampling-client')
+logfire.instrument_pydantic_ai()
+logfire.instrument_mcp()
+
+THIS_DIR = Path(__file__).parent
+server = MCPServerStdio(command='python', args=[str(THIS_DIR / 'generate_svg.py')])
+agent = Agent('openai:gpt-4.1-mini', mcp_servers=[server])
+
+
+async def main():
+    async with agent.run_mcp_servers():
+        result = await agent.run('Create an image of a robot in a punk style.')
+    print(result.output)
+
+
+if __name__ == '__main__':
+    import asyncio
+
+    asyncio.run(main())

02-pai-mcp-sampling/generate_svg.py

@@ -0,0 +1,40 @@
+import re
+from pathlib import Path
+
+from mcp import SamplingMessage
+from mcp.server.fastmcp import Context, FastMCP
+from mcp.server.session import ServerSessionT
+from mcp.shared.context import LifespanContextT, RequestT
+from mcp.types import TextContent
+
+app = FastMCP(log_level='WARNING')
+
+import logfire
+
+logfire.configure(service_name='mcp-sampling-server', console=False)
+logfire.instrument_mcp()
+
+
+@app.tool()
+async def image_generator(ctx: Context[ServerSessionT, LifespanContextT, RequestT], subject: str, style: str) -> str:
+    prompt = f'{subject=} {style=}'
+    # `ctx.session.create_message` is the sampling call
+    result = await ctx.session.create_message(
+        [SamplingMessage(role='user', content=TextContent(type='text', text=prompt))],
+        max_tokens=1_024,
+        system_prompt='Generate an SVG image as per the user input',
+    )
+    assert isinstance(result.content, TextContent)
+
+    path = Path(f'{subject}_{style}.svg')
+    # remove triple backticks if the svg was returned within markdown
+    if m := re.search(r'^```\w*$(.+?)```$', result.content.text, re.S | re.M):
+        path.write_text(m.group(1))
+    else:
+        path.write_text(result.content.text)
+    return f'See {path}'
+
+
+if __name__ == '__main__':
+    # run the server via stdio
+    app.run()

README.md

@@ -0,0 +1 @@
+# Demo of the Pydantic Stack

pyproject.toml

@@ -0,0 +1,28 @@
+[project]
+name = "pydantic-demo"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "logfire[httpx]>=3.21.1",
+    "pydantic-ai>=0.3.4",
+]
+
+[tool.ruff]
+line-length = 120
+target-version = "py39"
+
+[tool.ruff.lint]
+extend-select = ["Q", "RUF100", "C90", "UP", "I"]
+flake8-quotes = { inline-quotes = "single", multiline-quotes = "double" }
+isort = { combine-as-imports = true }
+mccabe = { max-complexity = 15 }
+
+[tool.ruff.lint.pydocstyle]
+convention = "google"
+
+[tool.ruff.format]
+# don't format python in docstrings, pytest-examples takes care of it
+docstring-code-format = false
+quote-style = "single"