1.0 Release

src/retrieve.py

@@ -13,6 +13,7 @@ RETRIEVE_MODEL = CFG["models"]["retrieval"]
 API_BASE = CFG["api"]["base_url"]
 API_VERSION = CFG["api"]["api_version"]
 
+
 class CallbackHandler(BaseCallback):
     """Custom callback class for logging agent interactions."""
 
@@ -47,6 +48,7 @@ class CallbackHandler(BaseCallback):
     def _is_reasoning_output(self, outputs):
         return any(k.startswith("Thought") for k in outputs)
 
+
 def setup_logging():
     """Set up logging configuration for Merlin."""
     # Create a custom logger
@@ -60,15 +62,11 @@ def setup_logging():
     console_handler.setLevel(logging.INFO)
 
     # Create a file handler with rotation every 5MB
-    file_handler = RotatingFileHandler(
-        "dmv.log", maxBytes=5 * 1024 * 1024, backupCount=3
-    )
+    file_handler = RotatingFileHandler("data/dmv.log", maxBytes=5 * 1024 * 1024, backupCount=3)
     file_handler.setLevel(logging.DEBUG)
 
     # Create a formatter
-    formatter = logging.Formatter(
-        "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
-    )
+    formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
 
     # Set the formatter for the handler
     console_handler.setFormatter(formatter)
@@ -129,5 +127,6 @@ def main():
         except Exception as e:
             print(f"\n⚠️ An error occurred: {e}")
 
+
 if __name__ == "__main__":
-    main()
+    main()