Add Cohere LLM (#397)

This commit is contained in:
Gustavo Schaedler 2023-05-29 20:57:43 +01:00 committed by GitHub
commit 10cbed6eef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 34 additions and 245 deletions

276
poetry.lock generated

File diff suppressed because it is too large Load diff

View file

@ -56,6 +56,7 @@ weaviate-client = "^3.19.2"
jina = "3.15.2"
sentence-transformers = "^2.2.2"
ctransformers = "^0.2.2"
cohere = "^4.6.0"
[tool.poetry.group.dev.dependencies]

View file

@ -46,6 +46,7 @@ documentloaders:
embeddings:
- OpenAIEmbeddings
- HuggingFaceEmbeddings
- CohereEmbeddings
llms:
- OpenAI
# - AzureOpenAI
@ -53,6 +54,7 @@ llms:
- HuggingFaceHub
- LlamaCpp
- CTransformers
- Cohere
memories:
- ConversationBufferMemory
- ConversationSummaryMemory