[IMP][FIX] fixes summarization common class and removed duplicate code.

app.py

@@ -90,19 +90,6 @@ model_providers = {
 }
 
 
-selected_provider = "HuggingFace"
-
-model_id = model_providers[selected_provider]["model_id"]
-
-# model = HfApiModel(
-#     max_tokens=1000,
-#     temperature=0.5,
-#     model_id=model_id,
-#     custom_role_conversions=None,
-#     api_key=os.environ.get(model_providers[selected_provider]["api_key_env_var"]) if model_providers[selected_provider]["api_key_env_var"] else None
-# )
-
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -137,11 +124,6 @@ def launch_gradio_ui(additional_args=None):
 
     model = HfApiModel(**model_kwargs)
 
-    odoo_documentation_search_tool = OdooDocumentationSearchTool()
-    odoo_code_agent_16_tool = OdooCodeAgent16(prompt_templates["system_prompt"])
-    odoo_code_agent_17_tool = OdooCodeAgent17(prompt_templates["system_prompt"])
-    odoo_code_agent_18_tool = OdooCodeAgent18(prompt_templates["system_prompt"])
-
     agent = CodeAgent(
         model=model,
         tools=[final_answer, visit_webpage, web_search, image_generation_tool, get_current_time_in_timezone, job_search_tool, odoo_documentation_search_tool, odoo_code_agent_16_tool, odoo_code_agent_17_tool, odoo_code_agent_18_tool],

tools/odoo_code_agent_16.py

@@ -2,19 +2,19 @@ from smolagents.tools import Tool
 import requests
 from typing import List, Dict
 from bs4 import BeautifulSoup
-from transformers import pipeline  # Requires transformers
 import tempfile
 from tools.odoo_documentation_search import OdooDocumentationSearchTool
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.summarizer_service import SummarizerService
+
 
 class OdooCodeAgent16(Tool):
     name = "odoo_code_agent_16"
     description = "Generates Odoo code for version 16 based on the user's query and Odoo documentation, using prompt templates."
 
     def __init__(self, system_prompt: str):
-        # Load the summarization pipeline
-        self.summarizer = pipeline("summarization", model="google/flan-t5-small")
+        self.summarizer_service = SummarizerService()
         self.is_initialized = True
         self.system_prompt = system_prompt
     
@@ -68,7 +68,6 @@ class OdooCodeAgent16(Tool):
             prompt += "# No documentation found.\n"
 
         truncated_prompt = prompt[:500]
-        max_length = int(len(truncated_prompt) / 2)
-        code = self.summarizer(truncated_prompt, max_length=max_length, min_length=30, do_sample=False)[0]['summary_text']
+        code = self.summarizer_service.summarize(truncated_prompt)
 
         return FinalAnswerTool(answer=f"```py\n{code}\n```", include_download=True)

tools/odoo_code_agent_17.py

@@ -2,19 +2,18 @@ from smolagents.tools import Tool
 import requests
 from typing import List, Dict
 from bs4 import BeautifulSoup
-from transformers import pipeline  # Requires transformers
 import tempfile
 from tools.odoo_documentation_search import OdooDocumentationSearchTool
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.summarizer_service import SummarizerService
 
 class OdooCodeAgent17(Tool):
     name = "odoo_code_agent_17"
     description = "Generates Odoo code for version 17 based on the user's query and Odoo documentation, using prompt templates."
 
     def __init__(self, system_prompt: str):
-        # Load the summarization pipeline
-        self.summarizer = pipeline("summarization", model="google/flan-t5-small")
+        self.summarizer_service = SummarizerService()
         self.is_initialized = True
         self.system_prompt = system_prompt
     
@@ -68,7 +67,6 @@ class OdooCodeAgent17(Tool):
             prompt += "# No documentation found.\n"
 
         truncated_prompt = prompt[:500]
-        max_length = int(len(truncated_prompt) / 2)
-        code = self.summarizer(truncated_prompt, max_length=max_length, min_length=30, do_sample=False)[0]['summary_text']
+        code = self.summarizer_service.summarize(truncated_prompt)
 
         return FinalAnswerTool(answer=f"```py\n{code}\n```", include_download=True)

tools/odoo_code_agent_18.py

@@ -2,19 +2,18 @@ from smolagents.tools import Tool
 import requests
 from typing import List, Dict
 from bs4 import BeautifulSoup
-from transformers import pipeline  # Requires transformers
 import tempfile
 from tools.odoo_documentation_search import OdooDocumentationSearchTool
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.summarizer_service import SummarizerService
 
 class OdooCodeAgent18(Tool):
     name = "odoo_code_agent_18"
     description = "Generates Odoo code for version 18 based on the user's query and Odoo documentation, using prompt templates."
 
     def __init__(self, system_prompt: str):
-        # Load the summarization pipeline
-        self.summarizer = pipeline("summarization", model="google/flan-t5-small")
+        self.summarizer_service = SummarizerService()
         self.is_initialized = True
         self.system_prompt = system_prompt
     
@@ -68,7 +67,6 @@ class OdooCodeAgent18(Tool):
             prompt += "# No documentation found.\n"
 
         truncated_prompt = prompt[:500]
-        max_length = int(len(truncated_prompt) / 2)
-        code = self.summarizer(truncated_prompt, max_length=max_length, min_length=30, do_sample=False)[0]['summary_text']
+        code = self.summarizer_service.summarize(truncated_prompt)
 
         return FinalAnswerTool(answer=f"```py\n{code}\n```", include_download=True)