From 79c696c0a1c1f73ee5fb946a274b42048d59988d Mon Sep 17 00:00:00 2001 From: Nathan Date: Wed, 12 Feb 2025 01:05:37 -0600 Subject: [PATCH] simple chat implimentation --- src/siren.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/src/siren.py b/src/siren.py index b5f2f69..89168a3 100644 --- a/src/siren.py +++ b/src/siren.py @@ -1,9 +1,23 @@ import ollama -c = ollama.Client( +client = ollama.Client( host='https://ollama.blunkall.us' ) -r = c.generate(model='llama3.2', prompt='say hello')['response'] +messages = [] + +response = '' + +model = 'llama3.2' + +userInput = '' + +print('Enter a prompt!') + +while(userInput != 'stop'): + userInput = input() + if(userInput != 'stop'): + messages += [{ 'role': 'user', 'content': userInput, 'stream': False }] + print(client.chat(model=model, messages=messages)['message']['content']) + -print(r)