middlewares.py 695 B

1234567891011121314151617181920212223
  1. from langchain.agents.middleware import AgentMiddleware
  2. class MyMiddleware(AgentMiddleware):
  3. def before_agent(self, state, runtime):
  4. print("1. before_agent 只跑一次")
  5. return state
  6. def before_model(self, state, runtime):
  7. print("2. before_model 每次模型前都跑")
  8. return state
  9. def wrap_model_call(self, request, handler):
  10. print("3. wrap_model_call 每次模型都会走")
  11. return handler(request)
  12. def after_model(self, state, runtime):
  13. print("4. after_model 每次模型后都跑")
  14. return state
  15. def after_agent(self, state, runtime):
  16. print("5. after_agent 只跑一次")
  17. return state