From 21c9d21c0db01f964088e7337387b81a9e7c7418 Mon Sep 17 00:00:00 2001 From: billytrend-cohere <144115527+billytrend-cohere@users.noreply.github.com> Date: Thu, 5 Sep 2024 16:32:10 -0500 Subject: [PATCH] Revert "Filter out cohere's pydantic warning (#1081)" (#1147) --- griptape/drivers/prompt/cohere_prompt_driver.py | 4 ---- poetry.lock | 6 +++--- 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/griptape/drivers/prompt/cohere_prompt_driver.py b/griptape/drivers/prompt/cohere_prompt_driver.py index 05be5b7f2..ff1a8b482 100644 --- a/griptape/drivers/prompt/cohere_prompt_driver.py +++ b/griptape/drivers/prompt/cohere_prompt_driver.py @@ -1,6 +1,5 @@ from __future__ import annotations -import warnings from typing import TYPE_CHECKING, Any from attrs import Factory, define, field @@ -25,9 +24,6 @@ from griptape.tokenizers import BaseTokenizer, CohereTokenizer from griptape.utils import import_optional_dependency -# TODO Remove once https://github.com/cohere-ai/cohere-python/issues/559 is resolved -warnings.filterwarnings("ignore", module="pydantic") - if TYPE_CHECKING: from collections.abc import Iterator diff --git a/poetry.lock b/poetry.lock index 640b3bab0..ad6effafd 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1107,13 +1107,13 @@ colorama = {version = "*", markers = "platform_system == \"Windows\""} [[package]] name = "cohere" -version = "5.9.0" +version = "5.9.1" description = "" optional = true python-versions = "<4.0,>=3.8" files = [ - {file = "cohere-5.9.0-py3-none-any.whl", hash = "sha256:7c70cc9e6ade3355e00aa4a77fcb5662b32261a3237e00975d92b97bb5f3c0c9"}, - {file = "cohere-5.9.0.tar.gz", hash = "sha256:74e5b6e1fed0f617c26dfb8ef1cfccf8334321a51cc886c37374047916d71568"}, + {file = "cohere-5.9.1-py3-none-any.whl", hash = "sha256:8e1e1dde0e1a5ee5a3f22b890e0c927989f9326bab57b6b4be812a40e2565c3a"}, + {file = "cohere-5.9.1.tar.gz", hash = "sha256:de9a828b91481882bf554e94a61ccc52843a09fbc58f53b8e1fa940b17ce1dc9"}, ] [package.dependencies]