Check vLLM registration

llama_stack/providers/remote/inference/vllm/vllm.py

@@ -45,27 +45,25 @@ class VLLMInferenceAdapter(Inference, ModelsProtocolPrivate):
         self.client = OpenAI(base_url=self.config.url, api_key=self.config.api_token)
 
     async def register_model(self, model: Model) -> None:
-        raise ValueError("Model registration is not supported for vLLM models")
+        for running_model in self.client.models.list():
-
+            repo = running_model.id
-    async def shutdown(self) -> None:
-        pass
-
-    async def list_models(self) -> List[Model]:
-        models = []
-        for model in self.client.models.list():
-            repo = model.id
             if repo not in self.huggingface_repo_to_llama_model_id:
                 print(f"Unknown model served by vllm: {repo}")
                 continue
 
             identifier = self.huggingface_repo_to_llama_model_id[repo]
-            models.append(
+            if identifier == model.provider_resource_id:
-                Model(
+                print(
-                    identifier=identifier,
+                    f"Verified that model {model.provider_resource_id} is being served by vLLM"
-                    llama_model=identifier,
                 )
-            )
+                return
-        return models
+
+        raise ValueError(
+            f"Model {model.provider_resource_id} is not being served by vLLM"
+        )
+
+    async def shutdown(self) -> None:
+        pass
 
     async def completion(
         self,