Remove dataclass and write __init__ method.

Only call add_message in Client classes.

charla/chat.py

@@ -108,9 +108,6 @@ def run(argv: argparse.Namespace) -> None:
             if not (user_input := session.prompt()):
                 continue
 
-            client.add_message(role='user', text=user_input)
-            #output.append(f'{session.message}{user_input}\n')
-
             # Handle OPEN command input and continue to next prompt.
             if session.message == ui.t_open:
                 open_location = user_input.strip()

charla/client/__init__.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from dataclasses import dataclass, field
 from typing import Any, NamedTuple
 
 
@@ -8,12 +7,13 @@ class ModelInfo(NamedTuple):
     context_length: int
 
 
-@dataclass
 class Client(ABC):
-    model: str
-    context: Any = field(default_factory=list)
-    message_history: list[dict] = field(default_factory=list)
-    system: str = ''
+    def __init__(self, model: str, system: str = ''):
+        self.model = model
+        self.system = system
+        # For chatting with memory and writing output.
+        self.context: Any = []
+        self.message_history: list[dict] = []
 
     @abstractmethod
     def generate(self, prompt: str):

charla/client/github.py

@@ -53,3 +53,4 @@ def generate(self, prompt: str):
 
         self.context.append(AssistantMessage(content=text))
         self.add_message(role='assistant', text=text)
+

charla/client/ollama.py

@@ -31,6 +31,8 @@ def set_info(self):
 
 
     def generate(self, prompt: str):
+        self.add_message(role='user', text=prompt)
+
         response = self.client.generate(
             model=self.model, prompt=prompt, context=self.context, stream=True, system=self.system)