agent config so they can be fast, i dont care about the accuracy

Bakery_Shop_Tools.py

@@ -3,11 +3,13 @@ from chromadb.utils import embedding_functions
 from langchain.docstore.document import Document
 from smolagents import Tool, tool
 from sqlalchemy import (
-  Engine,  text
+    Engine, text
 )
 import os
 from smolagents import HfApiModel
-ENGINE:Engine|None = None
+
+ENGINE: Engine | None = None
+
 
 @tool
 def bakery_ingredient_order_sql_database(query: str) -> str:
@@ -86,8 +88,8 @@ class RetrieverTool(Tool):
     }
     output_type = "string"
 
-    def __init__(self, docs: list[Document], model:HfApiModel,name_to_be_inserted:str,
-                 description_to_be_inserted:str,inputs_to_be_inserted:dict[dict],
+    def __init__(self, docs: list[Document], model: HfApiModel, name_to_be_inserted: str,
+                 description_to_be_inserted: str, inputs_to_be_inserted: dict[dict],
                  collection_name: str = "baking_recipes", enable_summary: bool = False,
                  **kwargs):
         super().__init__(**kwargs)
@@ -112,6 +114,7 @@ class RetrieverTool(Tool):
         name = name_to_be_inserted
         description = description_to_be_inserted
         inputs = inputs_to_be_inserted
+
     def forward(self, query: str) -> str:
         assert isinstance(query, str), "Your search query must be a string"
         docs = self.collection.query(query_texts=[query], n_results=20)

Constants.py

@@ -39,7 +39,7 @@ LEGAL_RAG_TOOL_INPUTS = {
     }
 }
 
-#Data
+# Data
 BAKERY_ORDERS_DATA = [
     {
         "order_id": 1,

Data_Initialisation.py

@@ -27,6 +27,7 @@ from Constants import (
 )
 from Bakery_Shop_Tools import RetrieverTool
 
+
 def init_db():
     _engine = create_engine("sqlite:///sqlite3.sqlite")
     metadata_obj = MetaData()

app.py

@@ -20,52 +20,18 @@ from Constants import (
     LAW_QUERY1, SQL_QUERY2, )
 
 from Bakery_Shop_Tools import bakery_ingredient_order_sql_database
-from Data_Initialisation import init_rag,init_db
-
-# # Get your own keys from https://cloud.langfuse.com
-# LANGFUSE_PUBLIC_KEY = "pk-lf-..."
-# LANGFUSE_SECRET_KEY = "sk-lf-..."
-# os.environ["LANGFUSE_PUBLIC_KEY"] = LANGFUSE_PUBLIC_KEY
-# os.environ["LANGFUSE_SECRET_KEY"] = LANGFUSE_SECRET_KEY
-# os.environ["LANGFUSE_HOST"] = "https://cloud.langfuse.com"  # 🇪🇺 EU region example
-#
-# LANGFUSE_AUTH = base64.b64encode(
-#     f"{LANGFUSE_PUBLIC_KEY}:{LANGFUSE_SECRET_KEY}".encode()
-# ).decode()
-#
-# os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = os.environ.get("LANGFUSE_HOST") + "/api/public/otel"
-# os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
-# from opentelemetry.sdk.trace import TracerProvider
-# from openinference.instrumentation.smolagents import SmolagentsInstrumentor
-# from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
-# from opentelemetry.sdk.trace.export import SimpleSpanProcessor
-#
-# # Create a TracerProvider for OpenTelemetry
-# trace_provider = TracerProvider()
-#
-# # Add a SimpleSpanProcessor with the OTLPSpanExporter to send traces
-# trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
-#
-# # Set the global default tracer provider
-# from opentelemetry import trace
-#
-# trace.set_tracer_provider(trace_provider)
-# tracer = trace.get_tracer(__name__)
-#
-# # Instrument smolagents with the configured provider
-# SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
+from Data_Initialisation import init_rag, init_db
+
+
 @spaces.GPU
 def dummy():
     pass
 
 
-
 if __name__ == "__main__":
-    # preparing tools
     engine = init_db()
     Bakery_Shop_Tools.ENGINE = engine
 
-    # model
     model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
     # model_id="meta-llama/Llama-3.2-3B-Instruct",
 
@@ -106,9 +72,9 @@ if __name__ == "__main__":
     baking_recipes_retriever_agent = CodeAgent(
         tools=[baking_recipes_retriever_tool],
         model=model,
-        max_steps=3,
+        max_steps=2,
         verbosity_level=1,
-        planning_interval=1,
+        # planning_interval=1,
         name="baking_recipe_retriever_agent",
         description=
         '''
@@ -117,13 +83,13 @@ if __name__ == "__main__":
         not json or python or any other programming codes.
         '''
     )
-    # more complicated so there are bit more max steps and planning intervals
+    # more complicated so there are a bit more max steps and planning intervals
     legals_retriever_agent = CodeAgent(
         tools=[legals_retriever_tool],
         model=model,
-        max_steps=3,
-        verbosity_level=2,
-        planning_interval=2,
+        max_steps=2,
+        verbosity_level=1,
+        planning_interval=1,
         name="legals_retriever_agent",
         description=
         '''
@@ -132,11 +98,11 @@ if __name__ == "__main__":
         not json or python or any other programming codes.
         '''
     )
-    # more complicated so there are bit more max steps and planning intervals
+    # more complicated so there are a bit more max steps and planning intervals
     retriever_agent = CodeAgent(
         tools=[],
         model=model,
-        max_steps=3,
+        max_steps=2,
         verbosity_level=2,
         planning_interval=1,
         managed_agents=[baking_recipes_retriever_agent, legals_retriever_agent],
@@ -198,7 +164,7 @@ if __name__ == "__main__":
         time.sleep(2)
 
         message = message + " " + transcriber({"sampling_rate": sr, "raw": y})["text"]
-        print("new msseage:", message)
+        print("new message:", message)
         return message
 
 
@@ -249,7 +215,6 @@ if __name__ == "__main__":
             rag_q_button = gr.Button(RAG_QUESTION)
             sql_q1_button = gr.Button(SQL_QUERY)
             sql_q2_button = gr.Button(SQL_QUERY2)
-
             combi_button = gr.Button(RAG_QUESTION + " " + SQL_QUERY)
             legal_q_button = gr.Button(LAW_QUERY1)
         gr.Markdown("This gradio app has many accordion UI components that you can click to expand the UI component")
@@ -261,9 +226,10 @@ if __name__ == "__main__":
         mbox_submit_event = message_box.submit(enter_message,
                                                [message_box, chatbot],
                                                [message_box, chatbot])
-
         audio_stream = audio_interface.change(transcribe, inputs=[audio_interface, message_box],
                                               outputs=[message_box])
+
+        # query examples events
         rag_q_click_event = rag_q_button.click(enter_message,
                                                [rag_q_button, chatbot],
                                                [message_box, chatbot])
@@ -278,6 +244,7 @@ if __name__ == "__main__":
             [combi_button, chatbot],
             [message_box, chatbot])
 
+        # cancels
         legal_q_click_event = legal_q_button.click(enter_message,
                                                    [legal_q_button, chatbot],
                                                    [message_box, chatbot])