Update app.py

app.py

@@ -1,4 +1,4 @@
-# unsloth_doc_app.py
+# Koyeb Agent.py
 import streamlit as st
 import os
 from smolagents import (
@@ -18,7 +18,7 @@ from typing import List, Dict, Any
 
 # Configuration
 PAGE_CONFIG = {
-    "page_title": "Unsloth Documentation Search",
+    "page_title": "Koyeb Documentation Search",
     "page_icon": "📘",
     "layout": "wide"
 }
@@ -31,7 +31,7 @@ AUTHORIZED_IMPORTS: List[str] = [
 
 # Agent Initialization
 def initialize_unsloth_agent(model=None):
-    """Initialize the Unsloth Documentation Agent"""
+    """Initialize the Koyeb Documentation Agent"""
     tools = [
         DuckDuckGoSearchTool(),
         VisitWebpageTool()
@@ -44,8 +44,8 @@ def initialize_unsloth_agent(model=None):
         )
     
     return CodeAgent(
-        name="unsloth_agent",
-        description="Unsloth Documentation AI Agent",
+        name="koyeb_agent",
+        description="Koyeb Documentation AI Agent",
         tools=tools,
         model=model,
         max_steps=12,
@@ -57,10 +57,10 @@ def setup_ui():
     """Setup Streamlit UI components"""
     st.set_page_config(**PAGE_CONFIG)
     
-    st.title("📘 Unsloth Documentation Search")
+    st.title("💻 Koyeb Documentation Search")
     st.markdown("""
-    This tool uses an AI agent to help you explore Unsloth documentation.
-    Ask questions about Unsloth features, usage, or troubleshooting.
+    This tool uses an AI agent to help you explore Koyeb documentation.
+    Ask questions about Koyeb features, usage, or troubleshooting.
     """)
 
 def display_results(results: str):
@@ -73,14 +73,14 @@ def setup_sidebar():
     with st.sidebar:
         st.markdown("### About This Tool")
         st.markdown("""
-        This agent specializes in Unsloth documentation, providing:
+        This agent specializes in Koyeb documentation, providing:
         - 📚 Quick answers from web searches
         - 🌐 Webpage content retrieval
         - 🤖 AI-powered insights
         """)
         st.markdown("### Tips")
         st.markdown("""
-        - Use specific queries like "How to install Unsloth" or "Unsloth fine-tuning guide".
+        - Use specific queries like "How to install Koyeb cli" or "Koyeb deepseek model deployment".
         - Results may include raw data from searches or webpages.
         """)
 
@@ -95,20 +95,20 @@ def main():
     setup_sidebar()
     
     if 'agent' not in st.session_state:
-        with st.spinner("Initializing Unsloth Documentation Agent..."):
+        with st.spinner("Initializing Koyeb Documentation Agent..."):
             st.session_state.agent = initialize_app()
 
     search_query = st.text_input(
         "🔍 Search Unsloth Documentation",
-        placeholder="E.g., How to fine-tune with Unsloth"
+        placeholder="E.g., How to use Terraform with the koyeb provider ?"
     )
 
     # Single button to handle both cases
     if st.button("Search", type="primary", key="search_button"):
         if search_query:
-            with st.spinner("Searching Unsloth documentation..."):
+            with st.spinner("Searching Koyeb documentation..."):
                 try:
-                    results = st.session_state.agent.run(search_query, additional_args={"sources" : ["https://docs.unsloth.ai/"],"instructions": "Response to the user answer as a structured documentation format."})
+                    results = st.session_state.agent.run(search_query, additional_args={"reference_websites": ['https://www.koyeb.com/docs', 'https://www.koyeb.com/tutorials', 'https://www.koyeb.com/deploy'],"instructions": "Response to the user answer as a structured documentation format."})
                     display_results(results)
                     
                     st.markdown("---")
@@ -123,7 +123,7 @@ def main():
             st.warning("Please enter a search query.")
 
     st.markdown("---")
-    st.caption("Powered by SmolAgents and Unsloth")
+    st.caption("Powered by SmolAgents and Koyeb")
 
 if __name__ == "__main__":
     main()