Merge pull request #102 from ray-project/shrekris/track_usage_serve_c…

…onfigs Record telemetry when RayLLM is launched using a Serve config
ray-project · Dec 6, 2023 · 78e076e · 78e076e
2 parents fa3a766 + d2518d3
commit 78e076e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rayllm/backend/server/run.py b/rayllm/backend/server/run.py
@@ -111,6 +111,7 @@ def router_deployment(
 
 
 def router_application(args):
+    ray._private.usage.usage_lib.record_library_usage("ray-llm")
     router_args = RouterArgs.parse_obj(args)
     llm_apps = parse_args(router_args.models, llm_app_cls=VLLMApp)
     return router_deployment(llm_apps, enable_duplicate_models=False)
@@ -125,7 +126,6 @@ def run(
         models: The paths of the model yamls to deploy
 
     """
-    ray._private.usage.usage_lib.record_library_usage("aviary")
     router_app = router_application({"models": vllm_base_args})
 
     host = "0.0.0.0"