Files
simplemind/simplemind/conversation.py
T
Kurt Heiden f2777a013d Build ollama integration
Based on functionality defined in the top level README.md
2024-10-28 12:44:40 -06:00

12 lines
398 B
Python

class Conversation:
def __init__(self, ai_client):
self.messages = []
self.ai_client = ai_client
def say(self, message):
self.messages.append({'role': 'user', 'content': message})
def get_reply(self):
reply = self.ai_client.message(messages=self.messages)
self.messages.append({'role': 'system', 'content': reply.text})
return reply