Add pre-commit to ensure consistent formatting

This commit is contained in:
Kacper Łukawski
2025-03-05 22:47:12 +01:00
parent 7c1d05ae17
commit e673084c5e
8 changed files with 149 additions and 20 deletions

View File

@@ -1,5 +1,6 @@
from typing import Optional
from qdrant_client import AsyncQdrantClient, models
from qdrant_client import AsyncQdrantClient
class QdrantConnector:
@@ -27,7 +28,9 @@ class QdrantConnector:
# For the time being, FastEmbed models are the only supported ones.
# A list of all available models can be found here:
# https://qdrant.github.io/fastembed/examples/Supported_Models/
self._client = AsyncQdrantClient(location=qdrant_url, api_key=qdrant_api_key, path=qdrant_local_path)
self._client = AsyncQdrantClient(
location=qdrant_url, api_key=qdrant_api_key, path=qdrant_local_path
)
self._client.set_model(fastembed_model_name)
async def store_memory(self, information: str):

View File

@@ -1,12 +1,11 @@
import asyncio
from typing import Optional
from mcp.server import Server, NotificationOptions
from mcp.server.models import InitializationOptions
import click
import mcp.types as types
import asyncio
import mcp
import mcp.types as types
from mcp.server import NotificationOptions, Server
from mcp.server.models import InitializationOptions
from .qdrant import QdrantConnector
@@ -29,7 +28,11 @@ def serve(
server = Server("qdrant")
qdrant = QdrantConnector(
qdrant_url, qdrant_api_key, collection_name, fastembed_model_name, qdrant_local_path
qdrant_url,
qdrant_api_key,
collection_name,
fastembed_model_name,
qdrant_local_path,
)
@server.list_tools()
@@ -151,7 +154,9 @@ def main(
):
# XOR of url and local path, since we accept only one of them
if not (bool(qdrant_url) ^ bool(qdrant_local_path)):
raise ValueError("Exactly one of qdrant-url or qdrant-local-path must be provided")
raise ValueError(
"Exactly one of qdrant-url or qdrant-local-path must be provided"
)
async def _run():
async with mcp.server.stdio.stdio_server() as (read_stream, write_stream):