add custom tools

wikipedia and dictionary

app.py

@@ -87,6 +87,113 @@ def get_ip_info(ip: str = "") -> str:
     except Exception as e:
         return f"Error getting IP information: {str(e)}"
 
+@tool
+def dictionary_lookup(word: str) -> str:
+    """A tool that provides definitions and information about English words using a free dictionary API.
+    
+    Args:
+        word: The English word to look up.
+    """
+    try:
+        word = word.lower().strip()
+        response = requests.get(f"https://api.dictionaryapi.dev/api/v2/entries/en/{word}")
+        
+        if response.status_code == 200:
+            data = response.json()
+            result = f"Definitions for '{word}':\n\n"
+            
+            # Process the first entry
+            entry = data[0]
+            
+            # Include phonetics if available
+            if 'phonetic' in entry and entry['phonetic']:
+                result += f"Pronunciation: {entry['phonetic']}\n\n"
+                
+            # Process meanings
+            for meaning in entry.get('meanings', [])[:3]:  # Limit to first 3 parts of speech
+                part_of_speech = meaning.get('partOfSpeech', 'unknown')
+                result += f"• {part_of_speech.capitalize()}:\n"
+                
+                # Add definitions
+                for i, definition in enumerate(meaning.get('definitions', [])[:2], 1):  # Limit to first 2 definitions
+                    result += f"  {i}. {definition.get('definition', 'No definition available')}\n"
+                    
+                    # Add example if available
+                    if 'example' in definition and definition['example']:
+                        result += f"     Example: \"{definition['example']}\"\n"
+                        
+                result += "\n"
+                
+            # Add synonyms if available
+            synonyms = []
+            for meaning in entry.get('meanings', []):
+                for synonym in meaning.get('synonyms', []):
+                    synonyms.append(synonym)
+                    
+            if synonyms:
+                result += f"Synonyms: {', '.join(synonyms[:5])}\n"  # Limit to first 5 synonyms
+                
+            return result
+        elif response.status_code == 404:
+            return f"Word '{word}' not found in the dictionary."
+        else:
+            return f"Error looking up word: HTTP Status {response.status_code}"
+    except Exception as e:
+        return f"Error performing dictionary lookup: {str(e)}"
+
+@tool
+def wikipedia_search(query: str, sentences: int = 3) -> str:
+    """A tool that searches Wikipedia and returns a summary of the topic.
+    
+    Args:
+        query: The topic to search for on Wikipedia.
+        sentences: Number of sentences to include in the summary (default: 3).
+    """
+    try:
+        # Sanitize input
+        query = query.strip()
+        sentences = max(1, min(10, int(sentences)))  # Ensure between 1-10 sentences
+        
+        # Use Wikipedia's API
+        search_url = f"https://en.wikipedia.org/w/api.php?action=query&list=search&srsearch={query}&format=json"
+        search_response = requests.get(search_url)
+        
+        if search_response.status_code != 200:
+            return f"Error: Failed to search Wikipedia. Status code: {search_response.status_code}"
+            
+        search_data = search_response.json()
+        search_results = search_data.get('query', {}).get('search', [])
+        
+        if not search_results:
+            return f"No Wikipedia articles found for '{query}'."
+            
+        # Get the page ID of the first result
+        page_id = search_results[0]['pageid']
+        
+        # Get the summary using the page ID
+        summary_url = f"https://en.wikipedia.org/w/api.php?action=query&prop=extracts&exintro=true&explaintext=true&pageids={page_id}&format=json"
+        summary_response = requests.get(summary_url)
+        
+        if summary_response.status_code != 200:
+            return f"Error: Failed to get Wikipedia summary. Status code: {summary_response.status_code}"
+            
+        summary_data = summary_response.json()
+        page_data = summary_data.get('query', {}).get('pages', {}).get(str(page_id), {})
+        title = page_data.get('title', 'Unknown')
+        
+        # Get the full extract
+        extract = page_data.get('extract', 'No summary available.')
+        
+        # Split into sentences and limit
+        extract_sentences = extract.split('. ')
+        limited_extract = '. '.join(extract_sentences[:sentences])
+        if not limited_extract.endswith('.'):
+            limited_extract += '.'
+            
+        return f"Wikipedia: {title}\n\n{limited_extract}\n\nSource: Wikipedia"
+    except Exception as e:
+        return f"Error searching Wikipedia: {str(e)}"
+        
 
 @tool
 def string_utilities(action: str, text: str, additional_param: str = "") -> str:
@@ -163,7 +270,8 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[convert_currency, 
-           get_ip_info, 
+           dictionary_lookup,
+           wikipedia_search,
            string_utilities, 
            final_answer],
     max_steps=6,