Dynamic Tool Selection Services

from fastapi import FastAPI
import asyncpg

app = FastAPI()

@app.post("/mcp/router/select_tools")
async def select_relevant_tools(user_query: str, top_k: int = 5):
  # Generate query vector embedding
  # query_vec = await embed_model.aembed(user_query)
  
  # Execute pgvector cosine similarity search across OpenAPI tool registry
  conn = await asyncpg.connect("postgresql://mcp_router@localhost/tools_db")
  rows = await conn.fetch("""
      SELECT tool_name, schema_json, 1 - (embedding <=> $1) as similarity
      FROM tool_registry
      ORDER BY embedding <=> $1 LIMIT $2
  """, query_vec, top_k)
  
  return {
      "selected_tools": [r["tool_name"] for r in rows],
      "schemas": [r["schema_json"] for r in rows]
  }