updated

Dockerfile

@@ -1,31 +1,23 @@
 FROM python:3.11
 
-# Create a non-root user
+WORKDIR /app
+
+COPY ./requirements.txt /app/requirements.txt
+    
+RUN pip3 install --no-cache-dir -r /app/requirements.txt
+
+# User
 RUN useradd -m -u 1000 user
-RUN chown -R user:user $HOME
 USER user
-
-# Set environment variables
-ENV HOME=/home/user
-ENV PATH=$HOME/.local/bin:$PATH
-ENV NLTK_DATA=$HOME/nltk_data
-ENV PYTHONUSERBASE=$HOME/.local
+ENV HOME /home/user
+ENV PATH $HOME/.local/bin:$PATH
 
 WORKDIR $HOME
-RUN mkdir -p app
-RUN mkdir -p $NLTK_DATA
-
-# Install dependencies
-COPY ./requirements.txt $HOME/app/requirements.txt
-RUN pip3 install --no-cache-dir -r $HOME/app/requirements.txt
-
+RUN mkdir app
+WORKDIR $HOME/app
 COPY . $HOME/app
-RUN python3 -c "import nltk; nltk.download('stopwords', download_dir='$NLTK_DATA')"
 
 EXPOSE 8501
-
-WORKDIR $HOME/app
-
 CMD streamlit run app.py \
     --server.headless true \
     --server.enableCORS false \

README.md

@@ -1,5 +1,5 @@
 ---
-title: Cona Assistant
+title: SuperMicro Demo
 emoji: 🐨
 colorFrom: indigo
 colorTo: indigo
@@ -7,7 +7,7 @@ sdk: docker
 app_port: 8501
 pinned: false
 license: apache-2.0
-short_description: Ask questions about Cona Services
+short_description: Ask questions about Supermicro documents
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

agent.py

@@ -1,9 +1,7 @@
 import os
 from typing import Optional
-import json
 from pydantic import Field, BaseModel
 from omegaconf import OmegaConf
-import requests
 
 from vectara_agentic.agent import Agent
 from vectara_agentic.tools import VectaraToolFactory, ToolsFactory
@@ -15,41 +13,65 @@ initial_prompt = "How can I help you today?"
 
 def create_assistant_tools(cfg):
 
-    class QueryCona(BaseModel):
+    class QueryTIProducts(BaseModel):
         query: str = Field(description="The user query.")
-
+        name: Optional[str] = Field(
+            default="",
+            description="The server name.",
+            examples=['SuperServer SYS-821GE-TNHR', 'A+ Server AS -4125GS-TNRT']
+        )
     vec_factory = VectaraToolFactory(
         vectara_api_key=cfg.api_key,
         vectara_corpus_key=cfg.corpus_key
     )
     
     summarizer = 'vectara-summary-table-md-query-ext-jan-2025-gpt-4o'
-    ask_cona = vec_factory.create_rag_tool(
-        tool_name = "ask_cona",
+    ask_supermicro = vec_factory.create_rag_tool(
+        tool_name = "ask_supermicro",
         tool_description = """
         Given a user query, 
-        returns a response to a user question about bottling companies.
+        returns a response to a user question about Supermicro servers.
         """,
-        tool_args_schema = QueryCona,
-        reranker = "slingshot", rerank_k = 100, 
-        n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.01,
+        tool_args_schema = QueryTIProducts,
+        reranker = "slingshot", rerank_k = 100, rerank_cutoff = 0.5,
+        n_sentences_before = 2, n_sentences_after = 2, lambda_val = 0.005,
         vectara_summarizer = summarizer,
         summary_num_results = 20,
         include_citations = True,
-        verbose = False
+        verbose = True,
+        save_history = True
     )
-    return [ask_cona] + ToolsFactory().guardrail_tools()
+
+    search_supermicro = vec_factory.create_search_tool(
+        tool_name = "search_supermicro",
+        tool_description = """
+        Given a user query, 
+        returns a list of Supermicro documents matching the query, along with metadata.
+        """,
+        tool_args_schema = QueryTIProducts,
+        reranker = "slingshot", rerank_k = 100, rerank_cutoff = 0.5,
+    )
+
+    return [ask_supermicro, search_supermicro]
 
 def initialize_agent(_cfg, agent_progress_callback=None):
-    bot_instructions = """
-    - You are a helpful assistant, with expertise in products from coca cola and other bottling companies.
-    - Use the ask_cona tool to answer most questions about any products related to coca cola.
+    ti_bot_instructions = """
+    - You are a helpful assistant, with expertise in products from SuperMicro.
+    - Always use the 'ask_supermicro' tool to get the information you need to respond to the user query, 
+      and never use your internal knoweldge.
+    - When the 'ask_supermicro' tool does not provide an adequare response, 
+      plan a sequence of queries that will help you get the information you need. 
+      Run each query in sequence, and use the results to refine the next query. 
+      Then use all this information to form a response.
+    - If the user does not specify a server type, ask the user to specify it before attempting a response.
     """
+    # can try "When filtering by the device_name, if the tool cannot provide an answer or does not have enough information
+    # try calling the tool again with the device_family and include the device_name in the query string itself."
 
     agent = Agent(
         tools=create_assistant_tools(_cfg),
-        topic="Cona services and coca cola",
-        custom_instructions=bot_instructions,
+        topic="Supermicro and its products",
+        custom_instructions=ti_bot_instructions,
         agent_progress_callback=agent_progress_callback,
     )
     agent.report()
@@ -61,8 +83,8 @@ def get_agent_config() -> OmegaConf:
         'corpus_key': str(os.environ['VECTARA_CORPUS_KEY']),
         'api_key': str(os.environ['VECTARA_API_KEY']),
         'examples': os.environ.get('QUERY_EXAMPLES', None),
-        'demo_name': "Cona Demo",
-        'demo_welcome': "Cona Assistant.",
-        'demo_description': "This assistant can help you with any questions about Cona Serices."
+        'demo_name': "Supermicro Demo",
+        'demo_welcome': "Supermicro Assistant.",
+        'demo_description': "This assistant can help you with any questions about SuperMicro servers."
     })
     return cfg

app.py

@@ -13,6 +13,6 @@ if "feedback_key" not in st.session_state:
     st.session_state.feedback_key = 0
 
 if __name__ == "__main__":
-   st.set_page_config(page_title="Cona Assistant", layout="wide")
+   st.set_page_config(page_title="Supermicro Assistant", layout="wide")
    nest_asyncio.apply()
    asyncio.run(launch_bot())