search-api-webui 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

search_api_webui/providers/querit.py

@@ -18,32 +18,32 @@
 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 # DEALINGS IN THE SOFTWARE.
 
-import time
 import json
+import time
+
 from querit import QueritClient
-from querit.models.request import SearchRequest
 from querit.errors import QueritError
-from .base import BaseProvider
+from querit.models.request import SearchRequest
+
+from .base import BaseProvider, parse_server_latency
+
 
 class QueritSdkProvider(BaseProvider):
-    """
+    '''
     Specialized provider implementation using the official Querit Python SDK.
-    """
+    '''
 
     def __init__(self, config):
         self.config = config
 
     def search(self, query, api_key, **kwargs):
-        """
+        '''
         Executes a search using the Querit SDK.
         Handles the 'Bearer' prefix logic internally within the SDK.
-        """
+        '''
         try:
             # Initialize client with the raw API key
-            client = QueritClient(
-                api_key=api_key.strip(),
-                timeout=30
-            )
+            client = QueritClient(api_key=api_key.strip(), timeout=30)
 
             limit = int(kwargs.get('limit', 10))
 
@@ -55,10 +55,10 @@ class QueritSdkProvider(BaseProvider):
             print(f'[Querit SDK] Searching: {query} (Limit: {limit})')
 
             start_time = time.time()
-            
+
             # Execute search via SDK
             response = client.search(request_model)
-            
+
             end_time = time.time()
 
             # Normalize results to standard format
@@ -66,35 +66,44 @@ class QueritSdkProvider(BaseProvider):
             if response.results:
                 for item in response.results:
                     # Use getattr to safely access SDK object attributes
-                    normalized_results.append({
-                        "title": getattr(item, 'title', ''),
-                        "url": getattr(item, 'url', ''),
-                        # Fallback to description if snippet is missing
-                        "snippet": getattr(item, 'snippet', '') or getattr(item, 'description', '')
-                    })
+                    normalized_results.append(
+                        {
+                            'title': getattr(item, 'title', ''),
+                            'url': getattr(item, 'url', ''),
+                            # Fallback to description if snippet is missing
+                            'snippet': getattr(item, 'snippet', '') or getattr(item, 'description', ''),
+                        },
+                    )
 
             # Calculate estimated size for metrics (approximate JSON size)
-            estimated_size = len(json.dumps([r for r in normalized_results]))
+            estimated_size = len(json.dumps(list(normalized_results)))
+
+            # Extract server latency from SDK response if available
+            server_latency_ms = None
+            # The SDK may provide the server's 'took' field from the response
+            if hasattr(response, 'took') and response.took is not None:
+                server_latency_ms = parse_server_latency(response.took)
 
             return {
-                "results": normalized_results,
-                "metrics": {
-                    "latency_ms": round((end_time - start_time) * 1000, 2),
-                    "size_bytes": estimated_size
-                }
+                'results': normalized_results,
+                'metrics': {
+                    'latency_ms': round((end_time - start_time) * 1000, 2),
+                    'server_latency_ms': server_latency_ms,
+                    'size_bytes': estimated_size,
+                },
             }
 
         except QueritError as e:
-            print(f"Querit SDK Error: {e}")
+            print(f'Querit SDK Error: {e}')
             return {
-                "error": f"Querit SDK Error: {str(e)}",
-                "results": [],
-                "metrics": {"latency_ms": 0, "size_bytes": 0}
+                'error': f'Querit SDK Error: {str(e)}',
+                'results': [],
+                'metrics': {'latency_ms': 0, 'server_latency_ms': None, 'size_bytes': 0},
             }
         except Exception as e:
-            print(f"Unexpected Error: {e}")
+            print(f'Unexpected Error: {e}')
             return {
-                "error": f"Error: {str(e)}",
-                "results": [],
-                "metrics": {"latency_ms": 0, "size_bytes": 0}
+                'error': f'Error: {str(e)}',
+                'results': [],
+                'metrics': {'latency_ms': 0, 'server_latency_ms': None, 'size_bytes': 0},
             }

search_api_webui/providers.yaml

@@ -1,7 +1,22 @@
 querit:
-  type: "querit_sdk"
-  description: "Official Querit Search via Python SDK"
-  default_limit: 10
+  url: "https://api.querit.ai/v1/search"
+  method: "POST"
+  headers:
+    "Accept": "application/json"
+    "Authorization": "Bearer {api_key}"
+    "Content-Type": "application/json"
+  payload:
+    query: "{query}"
+  response_mapping:
+    root_path: "results.result"
+    server_latency_path: "took"
+    fields:
+      title: "title"
+      url: "url"
+      snippet: "snippet"
+      site_name: "site_name"
+      site_icon: "site_icon"
+      page_age: "page_age"
 
 you:
   url: "https://ydc-index.io/v1/search"
@@ -12,8 +27,51 @@ you:
     query: "{query}"
   payload: {}
   response_mapping:
-    root_path: "results.web" 
+    root_path: "results.web"
+    server_latency_path: "metadata.latency"
     fields:
       title: "title"
       url: "url"
       snippet: "snippets[0] || description"
+      site_icon: "favicon_url"
+      page_age: "page_age"
+
+brave:
+  url: "https://api.search.brave.com/res/v1/web/search"
+  method: "GET"
+  headers:
+    X-Subscription-Token: "{api_key}"
+    Accept: "application/json"
+  params:
+    q: "{query}"
+  payload: {}
+  response_mapping:
+    root_path: "web.results"
+    fields:
+      title: "title"
+      url: "url"
+      snippet: "description"
+      page_age: "page_age"
+
+exa:
+  url: "https://api.exa.ai/search"
+  method: "POST"
+  headers:
+    "Content-Type": "application/json"
+    "x-api-key": "{api_key}"
+  params: {}
+  payload:
+    query: "{query}"
+  response_mapping:
+    root_path: "results"
+    fields:
+      title: "title"
+      url: "url"
+      snippet: "text"
+      page_age: "publishedDate"
+      site_icon: "favicon"
+
+querit_sdk:
+  type: "querit_sdk"
+  description: "Official Querit Search via Python SDK"
+  default_limit: 10

search_api_webui/ruff.toml

@@ -0,0 +1,27 @@
+# Ruff configuration
+# Ruff replaces Black, flake8, and isort
+
+line-length = 120
+target-version = "py37"
+
+[format]
+quote-style = "single"
+indent-style = "space"
+docstring-code-format = false
+
+[lint]
+select = [
+    "E",      # pycodestyle errors
+    "W",      # pycodestyle warnings
+    "F",      # pyflakes
+    "I",      # isort
+    "N",      # pep8-naming
+    "UP",     # pyupgrade
+    "B",      # flake8-bugbear
+    "C4",     # flake8-comprehensions
+    "SIM",    # flake8-simplify
+    "COM",    # flake8-commas (trailing commas)
+]
+ignore = [
+    "E203",   # whitespace before ':' - conflicts with Black
+]