Skip to content

Fixing Extract and Enhance #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jeromechoo merged 3 commits into from
Aug 10, 2026
Merged

Fixing Extract and Enhance #2

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
7191ae9
fix: resolve request token instead of silently falling back to None
claude Aug 10, 2026
2f979b5
fix(enhance): give optional parameters None defaults so they are actu…
claude Aug 10, 2026
3801901
Merge pull request #1 from Alicorn-Max-S/claude/diffbot-mcp-token-fix…
Alicorn-Max-S Aug 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 17 additions & 18 deletions diffbot_mcp_server.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
import aiohttp
from fastmcp import FastMCP, Context
from fastmcp.server.dependencies import get_http_request
from starlette.requests import Request
from typing import Annotated, Literal, Optional, List

mcp = FastMCP(name="Diffbot MCP Server")
Expand All @@ -13,17 +12,17 @@ class DiffbotAPI:
def __init__(self):
self.token = os.getenv('DIFFBOT_TOKEN')
try:
request = Request(get_http_request())
auth_header = request.headers.get('Authorization', '')
if auth_header.lower().startswith('bearer '):
self.token = auth_header[7:]
else:
token = request.query_params.get('token')
if token:
self.token = token
except Exception:
request = get_http_request()
except RuntimeError:
# Not an http request, use token in env
pass
return
auth_header = request.headers.get('Authorization', '')
if auth_header.lower().startswith('bearer '):
self.token = auth_header[7:]
else:
token = request.query_params.get('token')
if token:
self.token = token

@mcp.tool(
name="extract",
Expand Down Expand Up @@ -91,13 +90,13 @@ async def search_web(
async def enhance(
type: Annotated[Literal["Person", "Organization"],
"Select an entity type to look up. Required."],
name: Annotated[Optional[List[str]], "The name(s) of the entity to look up. Do not specify this key unless a value is provided."],
url: Annotated[Optional[List[str]], "The URL(s) of the entity to look up. Do not specify this key unless a value is provided."],
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."],
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."],
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."],
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."],
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."]
name: Annotated[Optional[List[str]], "The name(s) of the entity to look up. Do not specify this key unless a value is provided."] = None,
url: Annotated[Optional[List[str]], "The URL(s) of the entity to look up. Do not specify this key unless a value is provided."] = None,
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,
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:

diffbot = DiffbotAPI()
Expand Down