diff --git a/docs/griptape-framework/engines/audio-engines.md b/docs/griptape-framework/engines/audio-engines.md index e857bf77b..8f637dca0 100644 --- a/docs/griptape-framework/engines/audio-engines.md +++ b/docs/griptape-framework/engines/audio-engines.md @@ -21,7 +21,7 @@ from griptape.engines import TextToSpeechEngine driver = ElevenLabsTextToSpeechDriver( api_key=os.getenv("ELEVEN_LABS_API_KEY"), model="eleven_multilingual_v2", - voice="Rachel", + voice="Laura", ) engine = TextToSpeechEngine( diff --git a/griptape/drivers/sql/snowflake_sql_driver.py b/griptape/drivers/sql/snowflake_sql_driver.py index 43d147663..656bc4b99 100644 --- a/griptape/drivers/sql/snowflake_sql_driver.py +++ b/griptape/drivers/sql/snowflake_sql_driver.py @@ -59,7 +59,7 @@ def execute_query_raw(self, query: str) -> Optional[list[dict[str, Any]]]: if results is not None: if results.returns_rows: - return [dict(result.items()) for result in results] + return [dict(result._mapping) for result in results] else: return None else: diff --git a/tests/unit/drivers/sql/test_snowflake_sql_driver.py b/tests/unit/drivers/sql/test_snowflake_sql_driver.py index 055b1e744..a2887cb12 100644 --- a/tests/unit/drivers/sql/test_snowflake_sql_driver.py +++ b/tests/unit/drivers/sql/test_snowflake_sql_driver.py @@ -1,4 +1,5 @@ from dataclasses import dataclass +from typing import Any from unittest import mock import pytest @@ -36,8 +37,8 @@ def mock_snowflake_engine(self, mocker): items_mock = mocker.MagicMock(name="items") items_mock_2 = mocker.MagicMock(name="items2") - items_mock.items.return_value = [("first_name", "Tony"), ("last_name", "Hawk")] - items_mock_2.items.return_value = [("first_name", "Bob"), ("last_name", "Ross")] + items_mock._mapping = [("first_name", "Tony"), ("last_name", "Hawk")] + items_mock_2._mapping = [("first_name", "Bob"), ("last_name", "Ross")] result_mock.return_value.returns_rows = True result_mock.__iter__.return_value = iter([items_mock, items_mock_2]) @@ -71,7 +72,7 @@ def get_connection(): return new_driver def test_connection_function_wrong_return_type(self): - def get_connection(): + def get_connection() -> Any: return object with pytest.raises(ValueError):