diff --git a/trustgraph-flow/scripts/text-completion-mistral b/trustgraph-flow/scripts/text-completion-mistral new file mode 100755 index 00000000..91ef2279 --- /dev/null +++ b/trustgraph-flow/scripts/text-completion-mistral @@ -0,0 +1,6 @@ +#!/usr/bin/env python3 + +from trustgraph.model.text_completion.mistral import run + +run() + diff --git a/trustgraph-flow/setup.py b/trustgraph-flow/setup.py index fe167e90..504499f1 100644 --- a/trustgraph-flow/setup.py +++ b/trustgraph-flow/setup.py @@ -50,6 +50,7 @@ setuptools.setup( "langchain-core", "langchain-text-splitters", "minio", + "mistralai" "neo4j", "ollama", "openai", @@ -107,6 +108,7 @@ setuptools.setup( "scripts/text-completion-cohere", "scripts/text-completion-googleaistudio", "scripts/text-completion-llamafile", + "scripts/text-completion-mistral", "scripts/text-completion-ollama", "scripts/text-completion-openai", "scripts/triples-query-cassandra",