The configuration is now complete

(help is not quite optimal, but whatever)

Author: Neverbolt

src/hackingBuddyGPT/usecases/agents.py

@@ -4,7 +4,7 @@
 from mako.template import Template
 from typing import Dict
 
-from hackingBuddyGPT.utils.logging import log_conversation, GlobalLogger
+from hackingBuddyGPT.utils.logging import log_conversation, Logger, log_param
 from hackingBuddyGPT.capabilities.capability import (
     Capability,
     capabilities_to_simple_text_handler,
@@ -15,7 +15,7 @@
 
 @dataclass
 class Agent(ABC):
-    log: GlobalLogger = None
+    log: Logger = log_param
 
     _capabilities: Dict[str, Capability] = field(default_factory=dict)
     _default_capability: Capability = None

src/hackingBuddyGPT/usecases/base.py

@@ -3,7 +3,7 @@
 import argparse
 from dataclasses import dataclass
 
-from hackingBuddyGPT.utils.logging import GlobalLogger
+from hackingBuddyGPT.utils.logging import Logger, log_param
 from typing import Dict, Type, TypeVar, Generic
 
 from hackingBuddyGPT.utils.configurable import Transparent, configurable
@@ -20,7 +20,7 @@ class UseCase(abc.ABC):
     so that they can be automatically discovered and run from the command line.
     """
 
-    log: GlobalLogger
+    log: Logger = log_param
 
     def init(self):
         """

src/hackingBuddyGPT/usecases/viewer.py

@@ -18,6 +18,7 @@
 from starlette.templating import Jinja2Templates
 
 from hackingBuddyGPT.usecases.base import UseCase, use_case
+from hackingBuddyGPT.utils.configurable import parameter
 from hackingBuddyGPT.utils.db_storage import DbStorage
 from hackingBuddyGPT.utils.db_storage.db_storage import (
     Message,
@@ -205,8 +206,8 @@ class Viewer(UseCase):
     TODOs:
     - [ ] This server needs to be as async as possible to allow good performance, but the database accesses are not yet, might be an issue?
     """
-    log: GlobalLocalLogger
-    log_db: DbStorage
+    log: GlobalLocalLogger = None
+    log_db: DbStorage = None
     log_server_address: str = "127.0.0.1:4444"
     save_playback_dir: str = ""
 
@@ -231,7 +232,7 @@ async def save_message(self, message: ControlMessage):
         with open(file_path, "a") as f:
             f.write(ReplayMessage(datetime.datetime.now(), message).to_json() + "\n")
 
-    def run(self):
+    def run(self, config):
         @asynccontextmanager
         async def lifespan(app: FastAPI):
             app.state.db = self.log_db
@@ -358,8 +359,8 @@ def get_name(self) -> str:
 
 @use_case("Tool to replay the .jsonl logs generated by the Viewer (not well tested)")
 class Replayer(UseCase):
-    log: GlobalRemoteLogger
-    replay_file: str
+    log: GlobalRemoteLogger = None
+    replay_file: str = None
     pause_on_message: bool = False
     pause_on_tool_calls: bool = False
     playback_speed: float = 1.0