adding smolagent

app.py

@@ -11,6 +11,9 @@ from sentence_transformers import SentenceTransformer, util, CrossEncoder
 from langchain.llms.base import LLM
 import google.generativeai as genai
 
+# Import smolagents components
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, ManagedAgent
+
 ###############################################################################
 # 1) Logging Setup
 ###############################################################################
@@ -199,7 +202,36 @@ class QuestionSanityChecker:
 sanity_checker = QuestionSanityChecker(llm)
 
 ###############################################################################
-# 7) Answer Expansion
+# 7) smolagents Integration: GROQ Model and Web Search
+###############################################################################
+# Initialize the smolagents' LiteLLMModel with GROQ model
+smol_model = LiteLLMModel("groq/llama3-8b-8192")
+
+# Instantiate the DuckDuckGo search tool
+search_tool = DuckDuckGoSearchTool()
+
+# Create the web agent with the search tool
+web_agent = CodeAgent(
+    tools=[search_tool],
+    model=smol_model
+)
+
+# Define the managed web agent
+managed_web_agent = ManagedAgent(
+    agent=web_agent,
+    name="web_search",
+    description="Runs a web search for you. Provide your query as an argument."
+)
+
+# Create the manager agent with managed web agent and additional tools if needed
+manager_agent = CodeAgent(
+    tools=[],  # Add additional tools here if required
+    model=smol_model,
+    managed_agents=[managed_web_agent]
+)
+
+###############################################################################
+# 8) Answer Expansion
 ###############################################################################
 class AnswerExpander:
     def __init__(self, llm: GeminiLLM):
@@ -230,7 +262,7 @@ class AnswerExpander:
 answer_expander = AnswerExpander(llm)
 
 ###############################################################################
-# 8) Query Handling
+# 9) Query Handling
 ###############################################################################
 def handle_query(query: str) -> str:
     if not query or not isinstance(query, str) or len(query.strip()) == 0:
@@ -247,14 +279,26 @@ def handle_query(query: str) -> str:
         if not retrieved:
             return "I'm sorry, I couldn't find an answer to your question."
 
-        # Optional: Check similarity threshold (if still desired)
+        # Check similarity threshold
         top_score = retrieved[0][1]  # Assuming the list is sorted descending
         similarity_threshold = 0.3  # Adjust this threshold based on empirical results
 
         if top_score < similarity_threshold:
-            return "I'm sorry, I didn't understand your question. Could you please rephrase it?"
+            # Perform web search using manager_agent
+            logger.info("Similarity score below threshold. Performing web search.")
+            web_search_response = manager_agent.run(query)
+            logger.debug(f"Web search response: {web_search_response}")
+
+            # Optionally, process the web_search_response if needed
+            # For simplicity, return the web search response directly
+            return (
+                f"**Daily Wellness AI**\n\n"
+                f"{web_search_response}\n\n"
+                "Disclaimer: This information is retrieved from the web and is not a substitute for professional medical advice.\n\n"
+                "Wishing you a calm and wonderful day!"
+            )
 
-        # Proceed with answer expansion
+        # Proceed with answer expansion using retrieved_answers
         responses = [ans[0] for ans in retrieved]
         expanded_answer = answer_expander.expand(query, responses)
         return expanded_answer
@@ -264,18 +308,12 @@ def handle_query(query: str) -> str:
         return "An error occurred while processing your request."
 
 ###############################################################################
-# 9) Gradio Interface
+# 10) Gradio Interface
 ###############################################################################
 def gradio_interface(query: str):
     try:
         response = handle_query(query)
-        formatted_response = (
-            f"**Daily Wellness AI**\n\n"
-            f"{response}\n\n"
-            "Disclaimer: This is general wellness information, "
-            "not a substitute for professional medical advice.\n\n"
-            "Wishing you a calm and wonderful day!"
-        )
+        formatted_response = response  # Response is already formatted
         return formatted_response
     except Exception as e:
         logger.error(f"Error in Gradio interface: {e}")
@@ -296,13 +334,14 @@ interface = gr.Interface(
     examples=[
         "What is box breathing and how does it help reduce anxiety?",
         "Provide a daily wellness schedule incorporating box breathing techniques.",
-        "What are some tips for maintaining good posture while working at a desk?"
+        "What are some tips for maintaining good posture while working at a desk?",
+        "Who is the CEO of Hugging Face?"  # Example of an out-of-context question
     ],
     allow_flagging="never"
 )
 
 ###############################################################################
-# 10) Launch Gradio
+# 11) Launch Gradio
 ###############################################################################
 if __name__ == "__main__":
     try: