fix: better logging

Signed-off-by: Marco Trinelli (mtrinell) <[email protected]>

Author: mtrinell

Diff for: src/agent_workflow_server/agents/load.py

@@ -71,6 +71,14 @@ def _read_manifest(path: str) -> AgentACPDescriptor:
 
 
 def _resolve_agent(name: str, path: str) -> AgentInfo:
+    if ":" not in path:
+        raise ValueError(
+            f"""Invalid format for AGENTS_REF environment variable. \
+Value must be a module:var pair. \
+Example: "agent1_module:agent1_var" or "path/to/file.py:agent1_var"
+Got: {path}"""
+        )
+
     module_or_file, export_symbol = path.split(":", 1)
     if not os.path.isfile(module_or_file):
         # It's a module (name), try to import it

Diff for: src/agent_workflow_server/services/queue.py

@@ -126,13 +126,13 @@ async def worker(worker_id: int):
             DB.update_run_info(run_id, run_info)
 
             try:
-                validate_output(run_id, run["agent_id"], last_message.data)
                 log_run(
                     worker_id,
                     run_id,
                     "got message",
                     message_data=json.dumps(last_message.data),
                 )
+                validate_output(run_id, run["agent_id"], last_message.data)
                 DB.add_run_output(run_id, last_message.data)
                 await Runs.Stream.publish(run_id, Message(type="control", data="done"))
                 if last_message.type == "interrupt":