An integration package connecting Mistral and LangChain
Project description
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)
Embeddings
With MistralAIEmbeddings
, you can directly use the default model 'mistral-embed', or set a different one if available.
Choose model
embedding.model = 'mistral-embed'
Simple query
res_query = embedding.embed_query("The test information")
Documents
res_document = embedding.embed_documents(["test1", "another test"])
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for langchain_mistralai-0.1.0rc2.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | c100f474b74c5447b0ff6cd6e2e71d067392b4218d505adc79e8c8b3398059f7 |
|
MD5 | 0d95e330db8114b6ccfd13bb4a6cfc86 |
|
BLAKE2b-256 | 3bcdf502b790403e9d63da4c91945e1e9865abe31306f33a7b1725cc36772e3f |
Hashes for langchain_mistralai-0.1.0rc2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 96106e16ba7fa8530b979e6eaa8c242adbc2bea87e0a537096b512e353e6dcf1 |
|
MD5 | 7c68bc0358f7c127af2cb26e63744ad6 |
|
BLAKE2b-256 | 52d82b30dcd70e1ff9efdb8907ca49e128e6ab0fa6a7a8a27b6041f5aac0e567 |