suse-documentation-mcp-server 0.30.0__tar.gz → 0.31.0__tar.gz

{suse_documentation_mcp_server-0.30.0 → suse_documentation_mcp_server-0.31.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: suse-documentation-mcp-server
-Version: 0.30.0
+Version: 0.31.0
 Summary: MCP server implementation for web search using Open WebUI Web search
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: httpx>=0.28.1
 Requires-Dist: mcp[cli]>=1.4.1
+Requires-Dist: requests>=2.28

{suse_documentation_mcp_server-0.30.0 → suse_documentation_mcp_server-0.31.0}/pyproject.toml

@@ -1,13 +1,13 @@
 [project]
 name = "suse-documentation-mcp-server"
-version = "0.30.0"
+version = "0.31.0"
 description = "MCP server implementation for web search using Open WebUI Web search"
 readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "python-dotenv>=1.0.1",
-    "httpx>=0.28.1",
     "mcp[cli]>=1.4.1",
+    "requests>=2.28"
 ]
 
 [project.scripts]

{suse_documentation_mcp_server-0.30.0 → suse_documentation_mcp_server-0.31.0}/server/search_suse_documentation.py

@@ -1,8 +1,6 @@
-import asyncio
+import requests
 import os
-import sys
-import time
-
+import re
 from dotenv import load_dotenv
 from mcp.server.fastmcp import FastMCP
 
@@ -11,42 +9,6 @@ load_dotenv()
 base_url = os.getenv("OI_API_BASE", "") + "/api/v1/retrieval/process/web"
 authorization_token = os.getenv("OI_API_TOKEN", "")
 
-# Initialize FastMCP server
-mcp = FastMCP(
-    "web-search",
-    prompt="""
-# Web Search MCP Server
-
-This server provides tools for searching the web using Open WebUI Web search API.
-It allows you to search web for given search query.
-
-## Available Tools
-
-### 1. get_web_search_results
-Use this tool to search web to gather information about SUSE, Rancher products and projects.
-
-Example: "HOw to setup Nvidia GPU in RKE2?" or
-"Does Rancher Desktop support Wasm"
-
-## Guidelines for Use
-
-- Always perform a web search even if you think you know the answer
-- Use the most relevant and specific keywords for your search
-- Keep queries concise and specific for best results
-
-## Output Format
-
-All search results will be formatted as text with clear sections for each
-result item, including:
-
-- URL: The link to the source of the information
-- Content: A brief summary or excerpt from the page
-
-If the API call fails for any reason, appropriate error messages will be
-returned.
-""",
-)
-
 def clean_content(content):
     # Remove excessive blank lines
     cleaned_content = re.sub(r'\n\s*\n+', '\n\n', content.strip())
@@ -55,6 +17,13 @@ def clean_content(content):
     return cleaned_content
 
 def get_web_search_results_from_oi(query):
+    
+    if not base_url or not authorization_token:
+        return (
+            "Error: OI_API_BASE and/or  not OI_API_TOKEN not set. Please set the "
+            "environment variables."
+        )
+        
     headers = {
         'Content-Type': 'application/json',
         'Authorization': f'Bearer {authorization_token}'
@@ -62,8 +31,7 @@ def get_web_search_results_from_oi(query):
 
     # Step 1: Get web search results (list of URLs)
     search_payload = {
-        "query": query,
-        "collection_name": "your_collection_name"
+        "queries": [query]
     }
 
     print("Inside get_web_search_results_from_oi:", query)
@@ -85,8 +53,7 @@ def get_web_search_results_from_oi(query):
     # Step 2: Loop through URLs to get page content
     for filename in filenames:
         process_payload = {
-            "url": filename,
-            "collection_name": search_data["collection_name"]
+            "url": filename
         }
 
         process_response = requests.post(base_url, headers=headers, json=process_payload)
@@ -107,31 +74,53 @@ def get_web_search_results_from_oi(query):
 
     return combined_response
 
-@mcp.tool()
-async def get_web_search_results(
-    query: str
+# Initialize FastMCP server
+server = FastMCP(
+    "open-webui-web-search",
+    prompt="""
+# Open WebUI Search MCP Server
+
+This server provides tools for searching the web using Microsoft Open WebUI's API.
+It allows you to search for web pages, news articles, and images.
+
+## Available Tools
+
+### 1. open_webui_web_search
+Use this tool for general web searches. Best for finding information,
+websites, articles, and general content.
+
+Example: "What is the capital of France?" or
+"recipe for chocolate chip cookies"
+
+## Guidelines for Use
+
+- Keep queries concise and specific for best results
+
+## Output Format
+
+All search results will be formatted as text with clear sections for each
+result item, including:
+
+- Web search: Title, URL, and Description
+
+If the API key is missing or invalid, appropriate error messages will be
+returned.
+""",
+)
+
+@server.tool()
+async def get_web_search_results_from_oi(
+    query: str, count: int = 10, offset: int = 0, market: str = "en-US"
 ) -> str:
-    """Performs a web search using the Open WebUI web search API.
+    """Performs a web search using the Open WebUI Web Search API for general information
+    and websites.
 
     Args:
         query: Search query (required)
     """
-  
-    try:      
+
+    try:       
         # Process the query
-        result = get_web_search_results_from_oi(query)
-        
-        return result
+        return get_web_search_results_from_oi(query)
     except Exception as e:
-        return f"Error performing web search: {str(e)}\n\nQuery: {query}"    
-
-def main():
-    """Main entry point for the script."""
-    # Initialize and run the server
-    print("Starting SUSE Documentation Search Server...")
-    mcp.run()
-    
-if __name__ == "__main__":
-    # Initialize and run the server
-    main()
-    
+        return f"Error performing web search: {str(e)}\n\nPayload: {query}"    

{suse_documentation_mcp_server-0.30.0 → suse_documentation_mcp_server-0.31.0}/suse_documentation_mcp_server.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: suse-documentation-mcp-server
-Version: 0.30.0
+Version: 0.31.0
 Summary: MCP server implementation for web search using Open WebUI Web search
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: httpx>=0.28.1
 Requires-Dist: mcp[cli]>=1.4.1
+Requires-Dist: requests>=2.28

{suse_documentation_mcp_server-0.30.0 → suse_documentation_mcp_server-0.31.0}/suse_documentation_mcp_server.egg-info/requires.txt

@@ -1,3 +1,3 @@
 python-dotenv>=1.0.1
-httpx>=0.28.1
 mcp[cli]>=1.4.1
+requests>=2.28