import os
import sys

# Add this directory to sys.path so that agent_pb2_grpc.py 
# can find agent_pb2.py via absolute import 'import agent_pb2'.
# This is a common workaround for gRPC-generated code in Python packages.
proto_dir = os.path.dirname(os.path.abspath(__file__))
if proto_dir not in sys.path:
    sys.path.insert(0, proto_dir)