langchain/libs/partners/mistralai
2024-01-12 10:09:02 -08:00
..
langchain_mistralai mistralai[patch]: warn on stop token, fix on_llm_new_token (#15787) 2024-01-09 16:27:20 -08:00
scripts
tests
.gitignore
LICENSE
Makefile
poetry.lock mistralai[patch]: release 0.0.2 (#15912) 2024-01-11 13:42:04 -08:00
pyproject.toml multiple[patch]: .post releases and pyproject metadata (#15962) 2024-01-12 10:09:02 -08:00
README.md

langchain-mistralai

This package contains the LangChain integrations for MistralAI through their mistralai SDK.

Installation

pip install -U langchain-mistralai

Chat Models

This package contains the ChatMistralAI class, which is the recommended way to interface with MistralAI models.

To use, install the requirements, and configure your environment.

export MISTRAL_API_KEY=your-api-key

Then initialize

from langchain_core.messages import HumanMessage
from langchain_mistralai.chat_models import ChatMistralAI

chat = ChatMistralAI(model="mistral-small")
messages = [HumanMessage(content="say a brief hello")]
chat.invoke(messages)

ChatMistralAI also supports async and streaming functionality:

# For async...
await chat.ainvoke(messages)

# For streaming...
for chunk in chat.stream(messages):
    print(chunk.content, end="", flush=True)