diff --git a/poetry.lock b/poetry.lock index 0608a7c..f8e444b 100644 --- a/poetry.lock +++ b/poetry.lock @@ -221,13 +221,13 @@ files = [ [[package]] name = "pluggy" -version = "1.4.0" +version = "1.5.0" description = "plugin and hook calling mechanisms for python" optional = false python-versions = ">=3.8" files = [ - {file = "pluggy-1.4.0-py3-none-any.whl", hash = "sha256:7db9f7b503d67d1c5b95f59773ebb58a8c1c288129a88665838012cfb07b8981"}, - {file = "pluggy-1.4.0.tar.gz", hash = "sha256:8c85c2876142a764e5b7548e7d9a0e0ddb46f5185161049a79b7e974454223be"}, + {file = "pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669"}, + {file = "pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1"}, ] [package.extras] diff --git a/pyproject.toml b/pyproject.toml index 09585cd..685b711 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "superagent-py" -version = "v0.2.32" +version = "v0.2.33" description = "" readme = "README.md" authors = [] diff --git a/src/superagent/core/client_wrapper.py b/src/superagent/core/client_wrapper.py index 1a1a7ff..52713b5 100644 --- a/src/superagent/core/client_wrapper.py +++ b/src/superagent/core/client_wrapper.py @@ -16,7 +16,7 @@ def get_headers(self) -> typing.Dict[str, str]: headers: typing.Dict[str, str] = { "X-Fern-Language": "Python", "X-Fern-SDK-Name": "superagent-py", - "X-Fern-SDK-Version": "v0.2.32", + "X-Fern-SDK-Version": "v0.2.33", } token = self._get_token() if token is not None: diff --git a/src/superagent/types/llm_provider.py b/src/superagent/types/llm_provider.py index 4def0f0..6e035c8 100644 --- a/src/superagent/types/llm_provider.py +++ b/src/superagent/types/llm_provider.py @@ -20,6 +20,7 @@ class LlmProvider(str, enum.Enum): BEDROCK = "BEDROCK" GROQ = "GROQ" MISTRAL = "MISTRAL" + COHERE_CHAT = "COHERE_CHAT" def visit( self, @@ -32,6 +33,7 @@ def visit( bedrock: typing.Callable[[], T_Result], groq: typing.Callable[[], T_Result], mistral: typing.Callable[[], T_Result], + cohere_chat: typing.Callable[[], T_Result], ) -> T_Result: if self is LlmProvider.OPENAI: return openai() @@ -51,3 +53,5 @@ def visit( return groq() if self is LlmProvider.MISTRAL: return mistral() + if self is LlmProvider.COHERE_CHAT: + return cohere_chat()