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.2rc1.tar.gz (10.3 kB view details)

Uploaded Source

Built Distribution

langchain_mistralai-0.1.2rc1-py3-none-any.whl (11.1 kB view details)

Uploaded Python 3

File details

Details for the file langchain_mistralai-0.1.2rc1.tar.gz.

File metadata

File hashes

Hashes for langchain_mistralai-0.1.2rc1.tar.gz
Algorithm Hash digest
SHA256 476d07c28689ea027b1fc7513044102ecd2290d2b74d65e55e2120d39f131ae5
MD5 14aa3fbc71f6817051e27b78c07de775
BLAKE2b-256 92cf7a48e785f3483eb2718d1db57719ba959df23365de83152c1f0906b1a15c

See more details on using hashes here.

File details

Details for the file langchain_mistralai-0.1.2rc1-py3-none-any.whl.

File metadata

File hashes

Hashes for langchain_mistralai-0.1.2rc1-py3-none-any.whl
Algorithm Hash digest
SHA256 422915ee5096d9d0b37c2ba6cd004330034a6d83625fef477e6e0d1de03c0a7f
MD5 9f57b333e97863d3099cbc89b7108a46
BLAKE2b-256 63321e0203443a67a362f75834727bdaf3e6302fa0d0f9871909930680f2f7ab

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