List restate as an optional dependency

Author: igalshilman

docs/durable_execution/restate.md

@@ -58,6 +58,7 @@ pip/uv-add "pydantic-ai-slim[openai]"
 * `a2a` - installs `fasta2a` [PyPI ↗](https://pypi.org/project/fasta2a){:target="_blank"}
 * `ag-ui` - installs `ag-ui-protocol` [PyPI ↗](https://pypi.org/project/ag-ui-protocol){:target="_blank"} and `starlette` [PyPI ↗](https://pypi.org/project/starlette){:target="_blank"}
 * `dbos` - installs [`dbos`](durable_execution/dbos.md) [PyPI ↗](https://pypi.org/project/dbos){:target="_blank"}
+* `restate` - installs [`restate`](durable_execution/restate.md) [PyPI ↗](https://pypi.org/project/restate-sdk){:target="_blank"}
 
 See the [models](models/overview.md) documentation for information on which optional dependencies are required for each model.
 

docs/install.md

@@ -45,7 +45,6 @@ def create_agent(self, context: Context) -> AbstractAgent[AgentDepsT, OutputData
         Note that the agent will automatically wrap tool calls with restate's `ctx.run()`.
 
         Example:
-        ```python
            ...
            agent_provider = RestateAgentProvider(weather_agent)
 
@@ -57,7 +56,6 @@ async def get_weather(ctx: restate.Context, city: str):
                 result = await agent.run(f'What is the weather in {city}?')
                 return result.output
            ...
-        ```
 
         Args:
             context: The Restate context to use for the agent.
@@ -86,7 +84,6 @@ def create_agent_with_advanced_tools(
         since the tools may use the context in different ways.
 
         Example:
-        ```python
            ...
 
            @dataclass
@@ -112,7 +109,6 @@ async def get_weather(ctx: restate.Context, city: str):
                 result = await agent.run(f'What is the weather in {city}?', deps=WeatherDeps(restate_context=ctx, ...))
                 return result.output
            ...
-        ```
 
         Args:
             get_context: A callable that extracts the Restate context from the agent's dependencies.