New PUT calls to non verbose

NVIDIA · Jan 17, 2025 · a5f7d5a · a5f7d5a
1 parent ab21571
commit a5f7d5a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/nemo_skills/inference/server/serve_trt.py b/nemo_skills/inference/server/serve_trt.py
@@ -776,7 +776,10 @@ def main():
 
     class LogFilter(logging.Filter):
         def filter(self, record):
-            filter_strings = ("PUT /generate HTTP/1.1",)
+            filter_strings = ("PUT /generate HTTP/1.1", 
+                              "PUT /get_generation HTTP/1.1", 
+                              "PUT /generate_async HTTP/1.1", 
+                              "PUT /cancel_generation HTTP/1.1", )
             return all(filter_string not in record.getMessage() for filter_string in filter_strings)
 
     logging.getLogger('uvicorn.access').addFilter(LogFilter())