From 09b16a25651af0a3350c0d7f6df8c7fb57c7f37d Mon Sep 17 00:00:00 2001 From: Pavlo Paliychuk Date: Tue, 24 Sep 2024 16:36:24 -0400 Subject: [PATCH] refactor: remove unnecessary type casting in `search()` function (#153) fix: Mypy error --- server/graph_service/routers/retrieve.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/server/graph_service/routers/retrieve.py b/server/graph_service/routers/retrieve.py index 333ca0b4..733766e4 100644 --- a/server/graph_service/routers/retrieve.py +++ b/server/graph_service/routers/retrieve.py @@ -1,5 +1,4 @@ from datetime import datetime -from typing import List, Optional, cast from fastapi import APIRouter, status @@ -18,9 +17,7 @@ router = APIRouter() @router.post('/search', status_code=status.HTTP_200_OK) async def search(query: SearchQuery, graphiti: ZepGraphitiDep): relevant_edges = await graphiti.search( - group_ids=cast( - Optional[List[Optional[str]]], query.group_ids - ), # Cast query.group_ids to match the expected type in graphiti.search + group_ids=query.group_ids, query=query.query, num_results=query.max_facts, )