Upload 6 files

agent.py

@@ -0,0 +1,151 @@
+"""Tool definitions and utility functions for the agent"""
+from typing import List, Dict, Any
+import os
+from dotenv import load_dotenv
+
+from llama_index.core.tools import BaseTool, FunctionTool
+from llama_index.readers.wikipedia import WikipediaReader
+from llama_index.readers.web import SimpleWebPageReader
+from llama_index.core.schema import Document
+
+# Load environment variables
+load_dotenv()
+
+# --- Text Processing Tools ---
+def text_reverser(text: str) -> str:
+    """
+    Reverse the given text. Useful for answering questions that are written backwards.
+    
+    Args:
+        text: The text to reverse
+    
+    Returns:
+        The reversed text
+    """
+    return text[::-1]
+
+# --- Math Tools ---
+def simple_calculator(operation: str, a: float, b: float) -> float:
+    """
+    Perform a simple calculation.
+    
+    Args:
+        operation: One of 'add', 'subtract', 'multiply', 'divide'
+        a: First number
+        b: Second number
+    
+    Returns:
+        The result of the calculation
+    """
+    if operation == "add":
+        return a + b
+    elif operation == "subtract":
+        return a - b
+    elif operation == "multiply":
+        return a * b
+    elif operation == "divide":
+        if b == 0:
+            raise ValueError("Cannot divide by zero")
+        return a / b
+    else:
+        raise ValueError(f"Unknown operation: {operation}")
+
+# --- Information Retrieval Tools ---
+def wikipedia_search(query: str, num_results: int = 2) -> str:
+    """
+    Search Wikipedia for information.
+    
+    Args:
+        query: The search query
+        num_results: Number of results to return (default: 2)
+    
+    Returns:
+        A formatted string with the search results
+    """
+    try:
+        reader = WikipediaReader()
+        docs = reader.load_data(query=query, max_docs=num_results)
+        if not docs:
+            return f"No Wikipedia results found for '{query}'."
+        
+        results = []
+        for i, doc in enumerate(docs, 1):
+            title = doc.metadata.get("title", "Unknown Title")
+            content = doc.text[:1000] + "..." if len(doc.text) > 1000 else doc.text
+            results.append(f"Result {i}: {title}\n{content}\n")
+        
+        return "\n".join(results)
+    except Exception as e:
+        return f"Error searching Wikipedia: {str(e)}"
+
+def web_search(url: str) -> str:
+    """
+    Fetch and extract content from a specific web page.
+    
+    Args:
+        url: The URL of the web page to search
+    
+    Returns:
+        The extracted content from the web page
+    """
+    try:
+        reader = SimpleWebPageReader()
+        docs = reader.load_data(urls=[url])
+        if not docs:
+            return f"No content found for URL: {url}"
+        
+        # Just return the content of the first document
+        return docs[0].text
+    except Exception as e:
+        return f"Error retrieving web page: {str(e)}"
+
+# --- Tool Selection and Routing ---
+def get_tools() -> List[BaseTool]:
+    """Create and return a list of tools for the agent."""
+    
+    text_reverser_tool = FunctionTool.from_defaults(
+        fn=text_reverser,
+        name="text_reverser",
+        description="Reverses the given text. Useful for processing reversed questions or text.",
+    )
+    
+    calculator_tool = FunctionTool.from_defaults(
+        fn=simple_calculator,
+        name="calculator",
+        description="Performs simple calculations: add, subtract, multiply, divide.",
+    )
+    
+    wikipedia_tool = FunctionTool.from_defaults(
+        fn=wikipedia_search,
+        name="wikipedia_search",
+        description="Searches Wikipedia for information on a topic.",
+    )
+    
+    web_tool = FunctionTool.from_defaults(
+        fn=web_search,
+        name="web_search",
+        description="Fetches and extracts content from a specific web page.",
+    )
+    
+    return [
+        text_reverser_tool,
+        calculator_tool,
+        wikipedia_tool,
+        web_tool
+    ]
+
+# Import BasicAgent class from app.py
+# The build_agent function is still exposed for compatibility
+from app import BasicAgent
+
+def build_agent():
+    """Build and return a BasicAgent instance."""
+    return BasicAgent()
+
+if __name__ == "__main__":
+    # Test the agent with a simple question
+    agent = build_agent()
+    test_question = "What is the capital of France?"
+    answer = agent(test_question)
+    print(f"Question: {test_question}")
+    print(f"Answer: {answer}") 

gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

requirements.txt

@@ -1,2 +1,28 @@
 gradio
-requests
+requests
+pandas
+llama-index
+llama-index-llms-gemini
+llama-index-llms-huggingface
+llama-index-readers-web
+llama-index-readers-wikipedia
+python-dotenv
+
+# The following are commented out as they're not needed for our simplified approach
+# llama-index-vector-stores-chroma
+# llama-index-postprocessor-colbert-rerank
+# llama_index.embeddings.huggingface
+# chromadb
+
+# llama-index>=0.9.0
+# llama-index-llms-gemini>=0.1.0
+# llama-index-llms-huggingface>=0.1.0
+# llama-index-readers-web>=0.1.0
+# llama-index-readers-wikipedia>=0.1.0
+# llama-index-vector-stores-chroma>=0.1.0
+# llama-index-postprocessor-colbert-rerank>=0.1.0
+# chromadb>=0.4.18
+# python-dotenv>=1.0.0
+# gradio>=4.0.0
+# pandas>=2.0.0
+# requests>=2.31.0

system_prompt.txt

@@ -0,0 +1,31 @@
+You are an intelligent assistant designed to handle a wide variety of questions and tasks. Your purpose is to answer questions accurately and helpfully.
+
+Here are your capabilities and instructions:
+
+1. INFORMATION RETRIEVAL: 
+   - Use the wikipedia_search tool for factual information, historical data, or general knowledge
+   - Use the web_search tool when you have a specific URL to extract content from
+
+2. MATHEMATICAL REASONING: 
+   - Use the calculator tool for basic arithmetic (add, subtract, multiply, divide)
+   - Show your work for complex problems and explain your reasoning
+
+3. TEXT PROCESSING:
+   - Always be alert for questions that might be in unusual formats such as reversed, requiring translation etc.
+   - Use the text_reverser tool to manually reverse text when needed
+   - After processing unusual formats, make sure to answer the actual question
+
+4. GENERAL KNOWLEDGE:
+   - Rely on your built-in knowledge for questions about specialized topics
+   - Use your reasoning capabilities for logical problems
+   - Apply your understanding of various subjects without requiring specialized tools
+
+When using tools:
+- Choose the most appropriate tool for each question
+- For factual questions, prefer Wikipedia for well-established facts
+- For recent events or specialized topics, use web search
+- If a question is unclear, ask for clarification
+
+Always provide concise, accurate answers that directly address the question. Format your response according to any specific instructions in the question. If you're uncertain, state your confidence level and what additional information would help.
+
+Remember: The quality of your answers will be evaluated based on accuracy, relevance, and adherence to specified formats.