Skip to content
Snippets Groups Projects
Unverified Commit 6bc63928 authored by Thomas-AH-Heller's avatar Thomas-AH-Heller Committed by GitHub
Browse files

Fix async issue when streaming with Mistral AI (#13292)

parent 48e0a267
Branches
Tags
No related merge requests found
...@@ -324,7 +324,7 @@ class MistralAI(FunctionCallingLLM): ...@@ -324,7 +324,7 @@ class MistralAI(FunctionCallingLLM):
messages = to_mistral_chatmessage(messages) messages = to_mistral_chatmessage(messages)
all_kwargs = self._get_all_kwargs(**kwargs) all_kwargs = self._get_all_kwargs(**kwargs)
response = await self._aclient.chat_stream(messages=messages, **all_kwargs) response = self._aclient.chat_stream(messages=messages, **all_kwargs)
async def gen() -> ChatResponseAsyncGen: async def gen() -> ChatResponseAsyncGen:
content = "" content = ""
......
...@@ -27,7 +27,7 @@ exclude = ["**/BUILD"] ...@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT" license = "MIT"
name = "llama-index-llms-mistralai" name = "llama-index-llms-mistralai"
readme = "README.md" readme = "README.md"
version = "0.1.11" version = "0.1.12"
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.9,<4.0" python = ">=3.9,<4.0"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment