Improve error handling of agent

agents/native_restate/__main__.py

@@ -1,10 +1,16 @@
 import hypercorn
 import asyncio
 import restate
+import logging
 
 from agent import agent
 from account import account
 
+logging.basicConfig(
+    level=logging.INFO,
+    format="[%(asctime)s] [%(process)d] [%(levelname)s] - %(message)s",
+)
+
 
 def main():
     app = restate.app(

agents/openai_sdk/__main__.py

@@ -1,9 +1,15 @@
 import hypercorn
 import asyncio
 import restate
+import logging
 
 from agent import agent
 
+logging.basicConfig(
+    level=logging.INFO,
+    format="[%(asctime)s] [%(process)d] [%(levelname)s] - %(message)s",
+)
+
 
 def main():
     app = restate.app(services=[agent])

common/a2a_middleware.py

@@ -5,7 +5,7 @@
 
 from collections.abc import AsyncIterable, Iterable
 from datetime import datetime
-from pydantic import BaseModel
+from pydantic import BaseModel, ValidationError
 from restate.serde import PydanticJsonSerde
 
 from .a2a_agent import GenericRestateAgent
@@ -313,7 +313,7 @@ async def process_request(
 
             try:
                 json_rpc_request = A2ARequest.validate_python(req.model_dump())
-            except Exception as e:
+            except ValidationError as e:
                 logger.error("Error validating request: %s", e)
                 return JSONRPCResponse(
                     id=req.id,