Skip to main content

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

langchain_mistralai-0.1.0rc0.tar.gz (9.8 kB view details)

Uploaded Source

Built Distribution

langchain_mistralai-0.1.0rc0-py3-none-any.whl (10.7 kB view details)

Uploaded Python 3

File details

Details for the file langchain_mistralai-0.1.0rc0.tar.gz.

File metadata

File hashes

Hashes for langchain_mistralai-0.1.0rc0.tar.gz
Algorithm Hash digest
SHA256 bcf937b78c02541a098c14d0d97726ccd614183f93042ada7b778a720f2dd561
MD5 f1442865c6204ed69f312794bf935fab
BLAKE2b-256 cb95f854b06f23398b2d5bf934e77419516a5bed6b61c0072584dd4c3194d593

See more details on using hashes here.

File details

Details for the file langchain_mistralai-0.1.0rc0-py3-none-any.whl.

File metadata

File hashes

Hashes for langchain_mistralai-0.1.0rc0-py3-none-any.whl
Algorithm Hash digest
SHA256 3c64b1a9e5c6a23090cbfc0bc74d7501558363a8bdbd02f2eadcdd51cf39d61a
MD5 18d34f419bf543207057f633f9d2c968
BLAKE2b-256 fe0986d7f67c93a4ff2a943195cf65280b27ded4908374f886ed02ec1492546e

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page