cavepedia/cavepediav2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

optimize mcp
All checks were successful
Build and Push Agent Docker Image / build (push) Successful in 1m6s
Details

Browse Source
This commit is contained in:
Paul Walko
2025-12-18 18:37:30 +01:00
parent 46171a9a2b
commit 1af194f05d
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
mcp/server.py
View File

@@ -59,11 +59,10 @@ def embed(text, input_type):
assert resp.embeddings.float_ is not None assert resp.embeddings.float_ is not None
return resp.embeddings.float_[0] return resp.embeddings.float_[0]
def search(query, roles: list[str], limit: int = 5) -> list[dict]: def search(query, roles: list[str], limit: int = 3, max_content_length: int = 1500) -> list[dict]:
query_embedding = embed(query, 'search_query') query_embedding = embed(query, 'search_query')
if not roles: if not roles:
# No roles = no results
return [] return []
rows = conn.execute( rows = conn.execute(
@@ -71,7 +70,13 @@ def search(query, roles: list[str], limit: int = 5) -> list[dict]:
(roles, query_embedding, limit) (roles, query_embedding, limit)
).fetchall() ).fetchall()
return [{'key': row['key'], 'content': row['content']} for row in rows] docs = []
for row in rows:
content = row['content'] or ''
if len(content) > max_content_length:
content = content[:max_content_length] + '...[truncated, use get_document_page for full text]'
docs.append({'key': row['key'], 'content': content})
return docs
@mcp.tool @mcp.tool
def get_cave_location(cave: str, state: str, county: str) -> list[dict]: def get_cave_location(cave: str, state: str, county: str) -> list[dict]: