11 lines
235 B
Python
11 lines
235 B
Python
from neo4j_graphrag.llm import MistralAILLM
|
|
|
|
# set api key here on in the MISTRAL_API_KEY env var
|
|
api_key = None
|
|
|
|
with MistralAILLM(
|
|
model_name="mistral-small-latest",
|
|
api_key=api_key,
|
|
) as llm:
|
|
llm.invoke("say something")
|