Update main.py

main.py

@@ -12,7 +12,7 @@ async def stream(item: dict):
     if 'prompt' not in item.keys():
         raise ValueError("prompt é obrigatório")
 
-    prompt = item['prompt']
+    prompt = "<|system|>You are a helpfull assistant</s><|user|>"+item['prompt']+"</s><|assistant|>"
     temperatura = item['temperatura'] if 'temperatura' in item.keys() else 0.2
     max_tokens = item['max_tokens'] if 'max_tokens' in item.keys() else 512