suse-documentation-mcp-server 0.9.0__tar.gz → 0.11.0__tar.gz

{suse_documentation_mcp_server-0.9.0 → suse_documentation_mcp_server-0.11.0}/PKG-INFO

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

{suse_documentation_mcp_server-0.9.0 → suse_documentation_mcp_server-0.11.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "suse-documentation-mcp-server"
-version = "0.9.0"
+version = "0.11.0"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.12"

{suse_documentation_mcp_server-0.9.0 → suse_documentation_mcp_server-0.11.0}/server/search_suse_documentation.py

@@ -32,6 +32,7 @@ def get_web_search_results_from_oi(query):
         "collection_name": "your_collection_name"
     }
 
+    print("Inside get_web_search_results_from_oi:", query)
     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}")
@@ -73,23 +74,40 @@ def get_web_search_results_from_oi(query):
     return combined_response
     
 @mcp.tool()
-async def get_web_search_results(query: str) -> str:
+async def get_web_search_results(input_data: Any) -> str:
     """
     Search SUSE documentation using web search and return results.
     
     This function is designed to perform a web search on the SUSE documentation 
     for a given query and return the results. The function takes a search query as
-    a string, performs the web search, and returns a formatted string containing 
-    the content from the search results.
+    a string, JSON object, or dictionary, normalizes it, performs the web search, 
+    and returns a formatted string containing the content from the search results.
     
     Args:
-        query (str): Web search query
-        Examples: "sriov in rke2", "how to configure suse manager", "latest rancher version"
+        input_data (Any): Web search query, either as a string, JSON object, or dictionary.
+        Examples: 
+            - "sriov in rke2"
+            - {"query": "how to configure suse manager"}
+            - {"queries": "latest rancher version"}
     
     Returns:
         str: Combined content from the search results, cleaned and formatted
     """
     try:
+        # Normalize input to extract the query string
+        if isinstance(input_data, str):
+            query = input_data
+        elif isinstance(input_data, dict):
+            query = input_data.get("query") or input_data.get("queries")
+            if not query:
+                raise ValueError("Input dictionary must contain 'query' or 'queries' key.")
+        else:
+            raise TypeError("Input must be a string or a dictionary containing 'query' or 'queries'.")
+
+        # Ensure query is a string
+        if not isinstance(query, str):
+            raise ValueError("The extracted query must be a string.")
+        print("Before web search:", query)
         return get_web_search_results_from_oi(query)
     except Exception as e:
         return f"Error performing web search: {str(e)}\n\n{query}"

{suse_documentation_mcp_server-0.9.0 → suse_documentation_mcp_server-0.11.0}/suse_documentation_mcp_server.egg-info/PKG-INFO

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