diff --git a/docs/docs/pages/agent-integration.mdx b/docs/docs/pages/agent-integration.mdx
index f8473314..ed111dd4 100644
--- a/docs/docs/pages/agent-integration.mdx
+++ b/docs/docs/pages/agent-integration.mdx
@@ -93,7 +93,7 @@ class LiteLLMAgent(scenario.AgentAdapter):
model=self.model,
messages=input.messages
)
- return response.choices[0].message
+ return response.choices[0].message # type: ignore
```
### 3. Stateful Agents
@@ -157,4 +157,4 @@ Now that you understand agent integration, explore specific integration guides:
Or dive deeper into advanced topics:
- [Scenario Basics](/scenario) - Master scenario creation and control
-- Check the [AgentAdapter API Reference](/reference/python/scenario/agent_adapter.html) for detailed documentation */}
\ No newline at end of file
+- Check the [AgentAdapter API Reference](/reference/python/scenario/agent_adapter.html) for detailed documentation */}
diff --git a/docs/docs/pages/basics/cache.mdx b/docs/docs/pages/basics/cache.mdx
index 26f3c545..57e7c652 100644
--- a/docs/docs/pages/basics/cache.mdx
+++ b/docs/docs/pages/basics/cache.mdx
@@ -69,7 +69,7 @@ def my_agent_function(messages) -> scenario.AgentReturnTypes:
model="openai/gpt-4o-mini",
messages=messages
)
- return response.choices[0].message
+ return response.choices[0].message # type: ignore
```
### 3. Run Tests
diff --git a/docs/docs/pages/introduction/getting-started.mdx b/docs/docs/pages/introduction/getting-started.mdx
index 7f18a50c..8370a61d 100644
--- a/docs/docs/pages/introduction/getting-started.mdx
+++ b/docs/docs/pages/introduction/getting-started.mdx
@@ -83,7 +83,7 @@ def vegetarian_recipe_agent(messages) -> scenario.AgentReturnTypes:
*messages,
],
)
- return response.choices[0].message
+ return response.choices[0].message # type: ignore
```
### 2. Set up your environment
@@ -102,7 +102,10 @@ uv run pytest -s test_my_agent.py
You should see output showing the conversation between the simulated user and your agent, followed by the judge's evaluation:
-
+
## What happened?
@@ -125,4 +128,4 @@ In this example:
- Learn about [Scenario Basics](../basics/concepts) to understand the framework deeply
- Explore [Agent Integration](../agent-integration) to connect your existing agents
-- Check out more [examples](https://github.com/langwatch/scenario/tree/main/examples) on GitHub
\ No newline at end of file
+- Check out more [examples](https://github.com/langwatch/scenario/tree/main/examples) on GitHub
diff --git a/python/scenario/cache.py b/python/scenario/cache.py
index cfd770d6..e1704137 100644
--- a/python/scenario/cache.py
+++ b/python/scenario/cache.py
@@ -86,7 +86,7 @@ def invoke(self, message: str, context: dict) -> str:
model="gpt-4",
messages=[{"role": "user", "content": message}]
)
- return response.choices[0].message.content
+ return response.choices[0].message.content # type: ignore
# Usage in tests
scenario.configure(cache_key="my-test-suite-v1")