flock-core 0.5.0b5__py3-none-any.whl → 0.5.0b6__py3-none-any.whl

flock/tools/web_tools.py

@@ -1,90 +0,0 @@
-
-import importlib
-import os
-from typing import Literal
-
-from flock.core.logging.trace_and_logged import traced_and_logged
-
-
-@traced_and_logged
-def web_search_tavily(query: str):
-    if importlib.util.find_spec("tavily") is not None:
-        from tavily import TavilyClient
-
-        client = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
-        try:
-            response = client.search(query, include_answer=True)  # type: ignore
-            return response
-        except Exception:
-            raise
-    else:
-        raise ImportError(
-            "Optional tool dependencies not installed. Install with 'pip install flock-core[tools]'."
-        )
-
-
-@traced_and_logged
-def web_search_duckduckgo(
-    keywords: str, search_type: Literal["news", "web"] = "web"
-):
-    try:
-        if importlib.util.find_spec("duckduckgo_search") is not None:
-            from duckduckgo_search import DDGS
-
-            if search_type == "news":
-                response = DDGS().news(keywords)
-            else:
-                response = DDGS().text(keywords)
-
-                return response
-        else:
-            raise ImportError(
-                "Optional tool dependencies not installed. Install with 'pip install flock-core[tools]'."
-            )
-    except Exception:
-        raise
-
-
-@traced_and_logged
-def web_search_bing(keywords: str):
-    try:
-        import httpx
-
-        subscription_key = os.environ["BING_SEARCH_V7_SUBSCRIPTION_KEY"]
-        endpoint = "https://api.bing.microsoft.com/v7.0/search"
-
-        # Query term(s) to search for.
-        query = keywords
-
-        # Construct a request
-        mkt = "en-US"
-        params = {"q": query, "mkt": mkt}
-        headers = {"Ocp-Apim-Subscription-Key": subscription_key}
-
-        response = httpx.get(endpoint, headers=headers, params=params)
-        response.raise_for_status()
-        search_results = response.json()
-        return search_results["webPages"]
-    except Exception:
-        raise
-
-@traced_and_logged
-def web_content_as_markdown(url: str) -> str:
-    if (
-        importlib.util.find_spec("httpx") is not None
-        and importlib.util.find_spec("markdownify") is not None
-    ):
-        import httpx
-        from markdownify import markdownify as md
-
-        try:
-            response = httpx.get(url)
-            response.raise_for_status()
-            markdown = md(response.text)
-            return markdown
-        except Exception:
-            raise
-    else:
-        raise ImportError(
-            "Optional tool dependencies not installed. Install with 'pip install flock-core[tools]'."
-        )

flock/tools/zendesk_tools.py

@@ -1,147 +0,0 @@
-"""Tools for interacting with Zendesk."""
-
-import os
-
-import httpx
-from mcp.server.fastmcp import FastMCP
-
-mcp = FastMCP("ZendeskTools")
-
-
-ZENDESK_BEARER_TOKEN = os.getenv("ZENDESK_BEARER_TOKEN")
-
-HEADERS = {
-    "Authorization": f"Bearer {ZENDESK_BEARER_TOKEN}",
-    "Accept": "application/json",
-}
-
-
-@mcp.tool()
-def zendesk_get_tickets(number_of_tickets: int = 10) -> list[dict]:
-    """Get all tickets."""
-    ZENDESK_SUBDOMAIN = os.getenv("ZENDESK_SUBDOMAIN_TICKET")
-    BASE_URL = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url = f"{BASE_URL}/api/v2/tickets.json"
-    all_tickets = []
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        while url and len(all_tickets) < number_of_tickets:
-            response = client.get(url)
-            response.raise_for_status()
-
-            data = response.json()
-            tickets = data.get("tickets", [])
-            all_tickets.extend(tickets)
-
-            url = data.get("next_page")
-    return all_tickets
-
-@mcp.tool()
-def zendesk_get_ticket_by_id(ticket_id: str) -> dict:
-    """Get a ticket by ID."""
-    ZENDESK_SUBDOMAIN = os.getenv("ZENDESK_SUBDOMAIN_TICKET")
-    BASE_URL = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url = f"{BASE_URL}/api/v2/tickets/{ticket_id}"
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        response = client.get(url)
-        response.raise_for_status()
-        return response.json()["ticket"]
-
-@mcp.tool()
-def zendesk_get_comments_by_ticket_id(ticket_id: str) -> list[dict]:
-    """Get all comments for a ticket."""
-    ZENDESK_SUBDOMAIN = os.getenv("ZENDESK_SUBDOMAIN_TICKET")
-    BASE_URL = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url = f"{BASE_URL}/api/v2/tickets/{ticket_id}/comments"
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        response = client.get(url)
-        response.raise_for_status()
-        return response.json()["comments"]
-
-@mcp.tool()
-def zendesk_get_article_by_id(article_id: str) -> dict:
-    """Get an article by ID."""
-    ZENDESK_LOCALE = os.getenv("ZENDESK_ARTICLE_LOCALE")
-    ZENDESK_SUBDOMAIN = os.getenv("ZENDESK_SUBDOMAIN_ARTICLE")
-    BASE_URL = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url = (
-        f"{BASE_URL}/api/v2/help_center/{ZENDESK_LOCALE}/articles/{article_id}"
-    )
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        response = client.get(url)
-        response.raise_for_status()
-        return response.json()["article"]
-
-@mcp.tool()
-def zendesk_get_articles() -> list[dict]:
-    """Get all articles."""
-    ZENDESK_LOCALE = os.getenv("ZENDESK_ARTICLE_LOCALE")
-    ZENDESK_SUBDOMAIN = os.getenv("ZENDESK_SUBDOMAIN_ARTICLE")
-    BASE_URL = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url = f"{BASE_URL}/api/v2/help_center/{ZENDESK_LOCALE}/articles.json"
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        response = client.get(url)
-        response.raise_for_status()
-        return response.json()["articles"]
-
-@mcp.tool()   
-def zendesk_get_articles_count() -> int:
-    """
-    Count every Help-Center article in the configured locale.
-
-    Uses cursor pagination (page[size]=100) because it’s faster and
-    has no 10 000-record ceiling. Falls back to offset pagination
-    if the account hasn’t been migrated yet.
-    """
-    ZENDESK_LOCALE     = os.getenv("ZENDESK_ARTICLE_LOCALE")  # e.g. "en-us"
-    ZENDESK_SUBDOMAIN  = os.getenv("ZENDESK_SUBDOMAIN_ARTICLE")
-    BASE_URL           = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url                = (
-        f"{BASE_URL}/api/v2/help_center/{ZENDESK_LOCALE}/articles.json"
-        "?page[size]=100"            # max page size for HC APIs
-    )
-
-    total = 0
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        while url:
-            resp = client.get(url)
-            resp.raise_for_status()
-            data = resp.json()
-
-            total += len(data.get("articles", []))
-            print(f"Locale: {ZENDESK_LOCALE}")
-            print(f"Number of articles: {total}")
-
-            # Cursor pagination (preferred)
-            if data.get("meta", {}).get("has_more"):
-                url = data.get("links", {}).get("next")
-                continue
-
-            # Offset pagination fallback
-            url = data.get("next_page")
-
-    return total
-
-@mcp.tool()
-def zendesk_search_articles(query: str) -> list[dict]:
-    """Search Zendesk Help Center articles using a query string."""
-    ZENDESK_LOCALE = os.getenv("ZENDESK_ARTICLE_LOCALE")  # e.g., "en-us"
-    ZENDESK_SUBDOMAIN = os.getenv("ZENDESK_SUBDOMAIN_ARTICLE")
-    BASE_URL = f"https://{ZENDESK_SUBDOMAIN}.zendesk.com"
-    url = f"{BASE_URL}/api/v2/help_center/articles/search.json"
-
-    params = {
-        "query": query,
-        "locale": ZENDESK_LOCALE,
-        "sort_by": "updated_at",
-        "sort_order": "desc",
-    }
-
-    with httpx.Client(headers=HEADERS, timeout=30.0) as client:
-        response = client.get(url, params=params)
-        response.raise_for_status()
-        return response.json().get("results", [])
-
-
-if __name__ == "__main__":
-    transport = os.getenv("ZENDESK_MCP_TRANSPORT", "stdio")
-    mcp.run(transport=transport)