diff --git a/examples/scaffolds/example_pipeline_scaffold.py b/examples/scaffolds/example_pipeline_scaffold.py index 11e4f0a..cb0ec11 100644 --- a/examples/scaffolds/example_pipeline_scaffold.py +++ b/examples/scaffolds/example_pipeline_scaffold.py @@ -56,6 +56,10 @@ class Pipeline: # This is where you can add your custom pipelines like RAG. print(f"pipe:{__name__}") + # If you'd like to check for title generation, you can add the following check + if body.get("title", False): + print("Title Generation Request") + print(messages) print(user_message) print(body) diff --git a/examples/scaffolds/filter_pipeline_scaffold.py b/examples/scaffolds/filter_pipeline_scaffold.py index 562ced3..cc08434 100644 --- a/examples/scaffolds/filter_pipeline_scaffold.py +++ b/examples/scaffolds/filter_pipeline_scaffold.py @@ -58,6 +58,10 @@ class Pipeline: # This filter is applied to the form data before it is sent to the OpenAI API. print(f"inlet:{__name__}") + # If you'd like to check for title generation, you can add the following check + if body.get("title", False): + print("Title Generation Request") + print(body) print(user) diff --git a/examples/scaffolds/manifold_pipeline_scaffold.py b/examples/scaffolds/manifold_pipeline_scaffold.py index 623a027..eaff91e 100644 --- a/examples/scaffolds/manifold_pipeline_scaffold.py +++ b/examples/scaffolds/manifold_pipeline_scaffold.py @@ -48,6 +48,10 @@ class Pipeline: # This is where you can add your custom pipelines like RAG. print(f"pipe:{__name__}") + # If you'd like to check for title generation, you can add the following check + if body.get("title", False): + print("Title Generation Request") + print(messages) print(user_message) print(body)