From 96b1bafcde582dd468bd35c26e9fcf2c89bcce53 Mon Sep 17 00:00:00 2001
From: Dinesh Yeduguru <dineshyv@fb.com>
Date: Wed, 13 Nov 2024 12:26:31 -0800
Subject: [PATCH] remove self in construction

---
 llama_stack/providers/remote/inference/ollama/ollama.py | 5 +----
 llama_stack/providers/remote/inference/vllm/vllm.py     | 5 +----
 2 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/llama_stack/providers/remote/inference/ollama/ollama.py b/llama_stack/providers/remote/inference/ollama/ollama.py
index a5ea8e0db..32825e153 100644
--- a/llama_stack/providers/remote/inference/ollama/ollama.py
+++ b/llama_stack/providers/remote/inference/ollama/ollama.py
@@ -73,10 +73,7 @@ model_aliases = [
 
 class OllamaInferenceAdapter(Inference, ModelsProtocolPrivate):
     def __init__(self, url: str) -> None:
-        self.model_register_helper = ModelRegistryHelper(
-            self,
-            model_aliases=model_aliases,
-        )
+        self.model_register_helper = ModelRegistryHelper(model_aliases)
         self.url = url
         self.formatter = ChatFormat(Tokenizer.get_instance())
 
diff --git a/llama_stack/providers/remote/inference/vllm/vllm.py b/llama_stack/providers/remote/inference/vllm/vllm.py
index 7fad85e8c..657b5b576 100644
--- a/llama_stack/providers/remote/inference/vllm/vllm.py
+++ b/llama_stack/providers/remote/inference/vllm/vllm.py
@@ -47,10 +47,7 @@ def build_model_aliases():
 
 class VLLMInferenceAdapter(Inference, ModelsProtocolPrivate):
     def __init__(self, config: VLLMInferenceAdapterConfig) -> None:
-        self.model_register_helper = ModelRegistryHelper(
-            self,
-            model_aliases=build_model_aliases(),
-        )
+        self.model_register_helper = ModelRegistryHelper(build_model_aliases())
         self.config = config
         self.formatter = ChatFormat(Tokenizer.get_instance())
         self.client = None