suse-documentation-mcp-server 0.2.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

server/search_suse_documentation.py

@@ -0,0 +1,101 @@
+import requests
+import re
+import os
+from dotenv import load_dotenv
+from typing import Any
+from mcp.server.fastmcp import FastMCP
+from pydantic import BaseModel, Field
+
+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("suse-documentation")
+
+def clean_content(content):
+    # Remove excessive blank lines
+    cleaned_content = re.sub(r'\n\s*\n+', '\n\n', content.strip())
+    # Replace multiple spaces with a single space
+    cleaned_content = re.sub(r'[ \t]+', ' ', cleaned_content)
+    return cleaned_content
+
+def get_web_search_results_from_oi(query):
+    headers = {
+        'Content-Type': 'application/json',
+        'Authorization': f'Bearer {authorization_token}'
+    }
+
+    # Step 1: Get web search results (list of URLs)
+    search_payload = {
+        "query": query,
+        "collection_name": "your_collection_name"
+    }
+
+    search_response = requests.post(f"{base_url}/search", headers=headers, json=search_payload)
+    if search_response.status_code != 200:
+        raise Exception(f"Search API call failed: {search_response.status_code} - {search_response.text}")
+
+    search_data = search_response.json()
+
+    if not search_data.get("status"):
+        raise Exception(f"Search API response indicates failure: {search_data}")
+
+    filenames = search_data.get("filenames", [])
+    if not filenames:
+        return "No filenames found in the search response."
+
+    combined_response = ""
+
+    # Step 2: Loop through URLs to get page content
+    for filename in filenames:
+        process_payload = {
+            "url": filename,
+            "collection_name": search_data["collection_name"]
+        }
+
+        process_response = requests.post(base_url, headers=headers, json=process_payload)
+        if process_response.status_code != 200:
+            print(f"Failed to process URL {filename}: {process_response.status_code} - {process_response.text}")
+            continue
+
+        process_data = process_response.json()
+        if not process_data.get("status"):
+            print(f"Processing failed for URL {filename}: {process_data}")
+            continue
+
+        content = process_data.get("file", {}).get("data", {}).get("content", "No content available")
+
+        # Append to get combined response
+        cleaned_content = clean_content(content)
+        combined_response += f"Source: {filename}\n\nContent:\n{cleaned_content}\n\n"
+
+    return combined_response
+    
+class WebSearchToolInput(BaseModel):
+    """Input schema for WebSearchTool."""
+    query: str = Field(..., description="search query")
+@mcp.tool()
+async def get_web_search_results(query: str) -> str:
+    """Get web search results for a given query.
+
+    Args:
+        query: Web search query
+    """
+    args_schema: Type[BaseModel] = WebSearchToolInput
+    try:
+        return get_web_search_results_from_oi(query)
+    except Exception as e:
+        return f"Error performing web search: {str(e)}"
+    
+
+def main():
+    """Main entry point for the script."""
+    # Initialize and run the server
+    mcp.run(transport='stdio')
+    
+if __name__ == "__main__":
+    # Initialize and run the server
+    mcp.run(transport='stdio')
+        

{suse_documentation_mcp_server-0.2.0.dist-info → suse_documentation_mcp_server-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: suse-documentation-mcp-server
-Version: 0.2.0
+Version: 0.4.0
 Summary: Add your description here
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

suse_documentation_mcp_server-0.4.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+server/search_suse_documentation.py,sha256=ZKznYKFSr2VZm1aHbjKlHaJG3rBCSV_DiUZYT1gZcJk,3359
+suse_documentation_mcp_server-0.4.0.dist-info/METADATA,sha256=I6BIWipFsdVs3T52YahcNCtiB9Y7B0nlXPQKAwxAoko,343
+suse_documentation_mcp_server-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+suse_documentation_mcp_server-0.4.0.dist-info/entry_points.txt,sha256=IEM7kH8YFWMMJCcp4hJGv5tHzHWxiR5lZ7jOyzRCHQA,88
+suse_documentation_mcp_server-0.4.0.dist-info/top_level.txt,sha256=StKOSmRhvWS5IPcvhsDRbtxUTEofJgYFGOu5AAJdSWo,7
+suse_documentation_mcp_server-0.4.0.dist-info/RECORD,,

suse_documentation_mcp_server-0.4.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+suse-documentation-mcp-server = server.search_suse_documentation:main

suse_documentation_mcp_server-0.4.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+server

suse_documentation_mcp_server-0.2.0.dist-info/RECORD

@@ -1,4 +0,0 @@
-suse_documentation_mcp_server-0.2.0.dist-info/METADATA,sha256=0fy3O3OTkesfR66aUbXsdXjUWZodR92Z3P8tp_luFnY,343
-suse_documentation_mcp_server-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-suse_documentation_mcp_server-0.2.0.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-suse_documentation_mcp_server-0.2.0.dist-info/RECORD,,

suse_documentation_mcp_server-0.2.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-