18 lines
483 B
Python
18 lines
483 B
Python
from typing import List, Dict
|
|
|
|
from .vector_store import QdrantVectorStore
|
|
|
|
|
|
class MCPTool:
|
|
"""
|
|
A simple tool that uses the vector store to answer queries.
|
|
"""
|
|
|
|
def __init__(self, vector_store: QdrantVectorStore):
|
|
self.vector_store = vector_store
|
|
|
|
def answer(self, query: str, top_k: int = 3) -> List[Dict]:
|
|
"""
|
|
Return the top_k most relevant documents for the query.
|
|
"""
|
|
return self.vector_store.search(query, top_k=top_k) |