Implement system in text completion API (#137)

* Add system prompt to LLM invocation

* Added system parameter to LLMs

* Added to Bedrock and VertexAI
This commit is contained in:
cybermaggedon 2024-11-05 22:46:17 +00:00 committed by GitHub
parent 53c958aaff
commit 65cda7b276
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 24 additions and 13 deletions

View file

@ -111,7 +111,14 @@ class Processor(ConsumerProducer):
print(f"Handling prompt {id}...", flush=True)
prompt = v.prompt
# FIXME: There's a system prompt above. Maybe if system changes,
# then reset self.llm? It shouldn't do, because system prompt
# is set system wide?
# Or... could keep different LLM structures for different system
# prompts?
prompt = v.system + "\n\n" + v.prompt
try: