Update app.py

app.py

@@ -439,17 +439,35 @@ class LlamaGenerator(BaseGenerator):
         strategy: str = "default",
         **kwargs
     ) -> str:
+        """
+        Generate text based on a given strategy.
+    
+        Args:
+            prompt (str): Input prompt for text generation.
+            model_kwargs (Dict[str, Any]): Additional arguments for model generation.
+            strategy (str): The generation strategy to use (default: "default").
+            **kwargs: Additional arguments passed to the strategy.
+    
+        Returns:
+            str: Generated text response.
+    
+        Raises:
+            ValueError: If the specified strategy is not available.
+        """
+        # Validate that the strategy exists
         if strategy not in self.strategies:
-            raise ValueError(f"Unknown strategy: {strategy}")
-            
+            raise ValueError(f"Unknown strategy: {strategy}. Available strategies are: {list(self.strategies.keys())}")
+    
+        # Call the selected strategy with the provided arguments
         return self.strategies[strategy].generate(
-            self,
-            prompt,
-            model_kwargs,
-            **kwargs
+            generator=self,            # The generator instance
+            prompt=prompt,             # The input prompt
+            model_kwargs=model_kwargs, # Arguments for the model
+            **kwargs                   # Any additional strategy-specific arguments
         )
 
 
+
     def generate_with_context(
         self,
         context: str,
@@ -486,14 +504,16 @@ class LlamaGenerator(BaseGenerator):
             max_history_turns
         )
         return self.generate(
-            prompt,
-            model_kwargs,
-            strategy,
-            num_samples,
-            depth,
-            breadth
+            generator=self,
+            prompt=prompt,
+            model_kwargs=model_kwargs,
+            strategy=strategy,
+            num_samples=num_samples,
+            depth=depth,
+            breadth=breadth
         )
 
+
     
     def check_health(self) -> HealthStatus:
         """Check the health status of the generator."""