Fix enhance title param, repair test.sh, pin dependencies

Three pre-existing issues found while testing #2:

enhance's `title` parameter never reached clients. FastMCP's schema
compressor treats the properties map as a schema node when it contains a
`type` key, then strips the sibling `title` as a JSON Schema annotation.
The parameter is now `job_title` and is still sent upstream as `title`,
so the Diffbot KG request is unchanged.

test.sh silently exited 0 no matter what. It posted to /mcp/, which now
307-redirects to /mcp, and curl was not following redirects, so every run
produced no output and looked like a pass. It now follows redirects, fails
when no session is returned or when the tool call comes back isError, and
takes URL/TOKEN overrides. Also adds the missing shebang.

Dependencies were unpinned within their major, and every push to main
rebuilds the image and redeploys, so an untested fastmcp or aiohttp
release could reach production on an unrelated commit. Both are pinned to
the patch level of the versions verified here. requires-python was also
wrong: both dependencies need 3.10 or newer, not 3.8.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

Author: jeromechoo

diffbot_mcp.egg-info/PKG-INFO

@@ -0,0 +1,149 @@
+Metadata-Version: 2.4
+Name: diffbot-mcp
+Version: 0.1.0
+Summary: A Diffbot MCP server with a variety of helpful web data handling tools for your agent or LLM pipeline.
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: fastmcp~=2.14.7
+Requires-Dist: aiohttp~=3.14.3
+Dynamic: license-file
+
+# Diffbot MCP Server
+A Diffbot MCP server with a variety of helpful web data handling tools for your agent or LLM pipeline.
+
+**What is Diffbot?**
+Diffbot is a small research company building AI that structure the web. Our products connect apps to the structured web automatically, like an API for all of the public web.
+
+## Tools
+
+#### 🧬 extract
+**Web fetch tool alternative.** Conventional fetch tools return either 500k+ tokens of HTML or a markdown approximation of the page's content. Extract classifies the page it is fetching and extracts content into a meaningful and consistent JSON structure. Responses are optimized for minimal token usage while maintaining data fidelity. Powered by [Diffbot Extract API](https://docs.diffbot.com/docs/getting-started-with-diffbot-extract).
+
+#### 🔎 search_web
+**Web search tool that ranks accuracy, not popularity.** Built on Diffbot's first party web index. Ranking model heavily favors primary sources over secondary sources (e.g. press release > news piece on said press release). Returns markdown formatted content chunks for source citation. Optimized token management built-in. 
+
+#### 🪄 enhance
+**Web lookup tool for organizations and people.** Unlike web search, `enhance` will return structured data from the Diffbot Knowledge Graph on an organization or person (by name, url, or other acceptable input). Helpful for verifying facts, market research, or sales intelligence workflows.
+
+## Authentication
+A free Diffbot token is required for tool use. [Get one here.](https://app.diffbot.com/get-started)
+
+## Get Started
+
+### Quick Start
+
+The easiest way to get started is to connect your MCP client directly to the remote MCP server hosted by Diffbot. 
+
+```
+https://mcp.diffbot.com/mcp?token=<YOUR_DIFFBOT_TOKEN>
+```
+
+This repo deploys directly to the remote server. 
+
+### Local Installation
+
+Clone the repository
+```bash
+git clone git@github.com:diffbot/diffbot-mcp.git
+```
+
+Install requirements
+```bash
+pip install .
+```
+
+You can now run the server with `python3 diffbot_mcp_server.py`, or skip this step if you will be setting this server up with an MCP client.
+
+#### (Optional) Docker Install
+
+Build the image
+```bash
+docker build -t diffbot-mcp .
+```
+
+### Setup with MCP Clients
+
+To setup Claude Desktop, follow the [official installation guide](https://modelcontextprotocol.io/docs/develop/connect-local-servers) and use this example  `claude_desktop_config.json` configuration (replace variables as needed).
+
+This standard config works with most IDEs and environments.
+
+```json
+{
+    "mcpServers": {
+        "diffbot-mcp": {
+            "command": "python3",
+            "args": [
+                "/<YOUR_PROJECT_FOLDER>/diffbot-mcp/diffbot_mcp_server.py"
+            ],
+            "env": {
+                "DIFFBOT_TOKEN": "<YOUR_DIFFBOT_TOKEN>"
+            }
+        }
+    }
+}
+```
+For Docker installs, follow this configuration.
+
+```json
+{
+    "mcpServers": {
+        "diffbot-mcp": {
+            "command": "docker",
+            "args": [
+                "run",
+                "--rm",
+                "-i",
+                "-e", "DIFFBOT_TOKEN=<YOUR_DIFFBOT_TOKEN>",
+                "diffbot-mcp"
+            ]
+        }
+    }
+}
+```
+
+<details>
+<summary>Visual Studio Code / Copilot</summary>
+
+For remote setup, simply enter the remote server URL and include "http" as the transport type.
+
+```json
+{
+  "servers": {
+    "diffbot-mcp": {
+      "type": "http",
+      "url": "https://mcp.diffbot.com/mcp/?token=<YOUR_DIFFBOT_TOKEN>"
+    }
+  }
+}
+```
+
+For local setup, try this modified config for better secrets management.
+
+```json
+{
+	"servers": {
+        "diffbot-mcp": {
+            "command": "python3",
+            "args": [
+                "/<YOUR_PROJECT_FOLDER>/diffbot-mcp/diffbot_mcp_server.py"
+            ],
+            "env": {
+                "DIFFBOT_TOKEN": "${input:diffbot-api-key}"
+            }
+        }
+	},
+	"inputs": [
+		{
+			"password": true,
+			"id": "diffbot-api-key",
+			"type": "promptString",
+			"description": "Diffbot API Key"
+		}
+	]
+}
+```
+</details>
+
+## Help & Support
+This project is built and maintained by [@jeromechoo](https://github.com/jeromechoo). For support, file an issue and he'll get back to you as soon as he can. For faster answers, write to [sales@diffbot.com](mailto:sales@diffbot.com) 😜. 

diffbot_mcp.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+LICENSE
+README.md
+diffbot_mcp_server.py
+pyproject.toml
+diffbot_mcp.egg-info/PKG-INFO
+diffbot_mcp.egg-info/SOURCES.txt
+diffbot_mcp.egg-info/dependency_links.txt
+diffbot_mcp.egg-info/requires.txt
+diffbot_mcp.egg-info/top_level.txt

diffbot_mcp.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

diffbot_mcp.egg-info/requires.txt

@@ -0,0 +1,2 @@
+fastmcp~=2.14.7
+aiohttp~=3.14.3

diffbot_mcp.egg-info/top_level.txt

@@ -0,0 +1 @@
+diffbot_mcp_server

diffbot_mcp_server.py

@@ -95,7 +95,11 @@ async def enhance(
 	location: Annotated[Optional[str], "The location (e.g. Houston, Texas, United States) of the entity to look up. Do not specify this key unless a value is provided."] = None,
 	email: Annotated[Optional[List[str]], "The email(s) of the entity to look up. Can only be used with type 'Person'. Do not specify this key unless a value is provided."] = None,
 	employer: Annotated[Optional[str], "The employer name of the entity to look up. Can only be used with type 'Person'. Do not specify this key unless a value is provided."] = None,
-	title: Annotated[Optional[str], "The current position/title/role of the entity to look up. Can only be used with type 'Person'. Do not specify this key unless a value is provided."] = None,
+	# Named job_title rather than title: a parameter named "title" alongside one
+	# named "type" is stripped from the advertised tool schema, because the schema
+	# compressor reads the properties map as a schema node and treats the sibling
+	# "title" as the JSON Schema annotation. Sent upstream as "title" regardless.
+	job_title: Annotated[Optional[str], "The current position/title/role of the entity to look up. Can only be used with type 'Person'. Do not specify this key unless a value is provided."] = None,
 	school: Annotated[Optional[str], "Any previous educational institution associated with the entity to look up. Can only be used with type 'Person'. Do not specify this key unless a value is provided."] = None
 ) -> dict:
 
@@ -116,8 +120,8 @@ async def enhance(
 		params["email"] = email
 	if employer and type == "Person":
 		params["employer"] = employer
-	if title and type == "Person":
-		params["title"] = title
+	if job_title and type == "Person":
+		params["title"] = job_title
 	if school and type == "Person":
 		params["school"] = school
 

pyproject.toml

@@ -3,8 +3,11 @@ name = "diffbot-mcp"
 version = "0.1.0"
 description = "A Diffbot MCP server with a variety of helpful web data handling tools for your agent or LLM pipeline."
 readme = "README.md"
-requires-python = ">=3.8"
+requires-python = ">=3.10"
+# Pinned to the patch level: every push to main rebuilds the image and redeploys,
+# so an unpinned minor lets an untested dependency release reach production on a
+# commit that never touched it.
 dependencies = [
-    "fastmcp~=2.0",
-    "aiohttp~=3.0"
+    "fastmcp~=2.14.7",
+    "aiohttp~=3.14.3"
 ]

test.sh

@@ -1,27 +1,53 @@
+#!/usr/bin/env bash
+# Smoke test for a locally running server: initialize a session, then call search_web.
+# Override the defaults as needed, e.g. TOKEN=<your-diffbot-token> ./test.sh
+set -uo pipefail
+
+URL="${URL:-http://127.0.0.1:8000/mcp}"
+TOKEN="${TOKEN:-TEST}"
+ENDPOINT="$URL?token=$TOKEN"
+
+# -L so the request survives the trailing-slash redirect fastmcp applies to /mcp/
+curl_mcp() {
+	curl -sL \
+		-H 'Content-Type: application/json' \
+		-H 'Accept: application/json, text/event-stream' \
+		"$@"
+}
+
 # initialize and capture session ID
-SESSION=$(curl -sD - \
-  -H 'Content-Type: application/json' \
-  -H 'Accept: application/json, text/event-stream' \
-  -d '{"jsonrpc":"2.0","id":0,"method":"initialize",
-       "params":{"protocolVersion":"2025-06-18",
-                 "capabilities":{"tools":{}},
-                 "clientInfo":{"name":"curl","version":"1"}}}' \
-  http://127.0.0.1:8000/mcp/?token=TEST |
-  grep -i mcp-session-id | awk '{print $2}' | tr -d '\r')
+SESSION=$(curl_mcp -D - \
+	-d '{"jsonrpc":"2.0","id":0,"method":"initialize",
+	     "params":{"protocolVersion":"2025-06-18",
+	               "capabilities":{"tools":{}},
+	               "clientInfo":{"name":"curl","version":"1"}}}' \
+	"$ENDPOINT" |
+	grep -i '^mcp-session-id' | awk '{print $2}' | tr -d '\r')
+
+if [ -z "$SESSION" ]; then
+	echo "FAIL: no session ID returned from $URL — is the server running?" >&2
+	exit 1
+fi
 
 # send initialized notification
-curl -s \
-  -H 'Content-Type: application/json' \
-  -H 'Accept: application/json, text/event-stream' \
-  -H "Mcp-Session-Id: $SESSION" \
-  -d '{"jsonrpc":"2.0","method":"notifications/initialized"}' \
-  http://127.0.0.1:8000/mcp/?token=TEST
+curl_mcp \
+	-H "Mcp-Session-Id: $SESSION" \
+	-d '{"jsonrpc":"2.0","method":"notifications/initialized"}' \
+	"$ENDPOINT"
 
 # call the search tool
-curl -s \
-  -H 'Content-Type: application/json' \
-  -H 'Accept: application/json, text/event-stream' \
-  -H "Mcp-Session-Id: $SESSION" \
-  -d '{"jsonrpc":"2.0","id":1,"method":"tools/call",
-       "params":{"name":"search_web","arguments":{"query":"diffbot"}}}' \
-  http://127.0.0.1:8000/mcp/?token=TEST
+RESPONSE=$(curl_mcp \
+	-H "Mcp-Session-Id: $SESSION" \
+	-d '{"jsonrpc":"2.0","id":1,"method":"tools/call",
+	     "params":{"name":"search_web","arguments":{"query":"diffbot"}}}' \
+	"$ENDPOINT")
+
+echo "$RESPONSE"
+
+# the server reports tool failures in-band, so a 200 alone does not mean success
+if [ -z "$RESPONSE" ] || echo "$RESPONSE" | grep -q '"isError":true'; then
+	echo "FAIL: search_web did not return a successful result" >&2
+	exit 1
+fi
+
+echo "PASS: search_web returned a result"