apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-20 21:05:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Added quick Cohere fix

Browse source
This commit is contained in:
JackColquitt 2024-08-11 13:42:24 -07:00
parent d5a19e8fb1
commit 4584464e69
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph/model/text_completion/cohere/llm.py
View file

@ -62,7 +62,7 @@ class Processor(ConsumerProducer):
output = self.cohere.chat( output = self.cohere.chat(
model=self.model, model=self.model,
message=prompt, message=prompt,
preamble = "You are a helpful AI-assistant." preamble = "You are a helpful AI-assistant.",
temperature=0.0, temperature=0.0,
chat_history=[], chat_history=[],
prompt_truncation='auto', prompt_truncation='auto',