changed class interface with iterator

interface.py

@@ -11,7 +11,7 @@ from pydantic import Field, field_validator
 class GemmaLLMInterface(CustomLLM):
     def __init__(self, model_id: str = "google/gemma-2-2b-it", **kwargs):
         super().__init__(**kwargs)
-        self.model_id = model_id
+        object.__setattr__(self, "model_id", model_id)  # Use object.__setattr__ to bypass Pydantic restrictions
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_id,
             device_map="auto",
@@ -32,9 +32,10 @@ class GemmaLLMInterface(CustomLLM):
         return LLMMetadata(
             context_window=self.context_window,
             num_output=self.num_output,
-            model_name=self.model_id,
+            model_name=self.model_id,  # Passing the correct model ID here
         )
-
+        
+        
     @llm_completion_callback()
     def complete(self, prompt: str, **kwargs: Any) -> CompletionResponse:
         prompt = self._format_prompt(prompt)