Update services/prompt_builder.py

services/prompt_builder.py

@@ -2,13 +2,13 @@
 from typing import Protocol, List, Tuple
 from transformers import AutoTokenizer
 
-@observe()
+
 class PromptTemplate(Protocol):
     """Protocol for prompt templates."""
     def format(self, context: str, user_input: str, chat_history: List[Tuple[str, str]], **kwargs) -> str:
         pass
 
-@observe()
+
 class LlamaPromptTemplate:
     def format(self, context: str, user_input: str, chat_history: List[Tuple[str, str]], max_history_turns: int = 1) -> str:
         system_message = f"Please assist based on the following context: {context}"
@@ -22,7 +22,7 @@ class LlamaPromptTemplate:
         prompt += "<|start_header_id|>assistant<|end_header_id|>\n\n"
         return prompt
         
-@observe()
+
 class TransformersPromptTemplate:
     def __init__(self, model_path: str):
         self.tokenizer = AutoTokenizer.from_pretrained(model_path)