mirror of
https://github.com/microsoft/graphrag.git
synced 2025-08-05 23:32:42 +00:00
0e1a6e3770
* unified search app added to graphrag repository * ignore print statements * update words for unified-search * fix lint errors * fix lint error * fix module name --------- Co-authored-by: Gaudy Blanco <gaudy-microsoft@MacBook-Pro-m4-Gaudy-For-Work.local>
29 lines
553 B
Python
29 lines
553 B
Python
# Copyright (c) 2024 Microsoft Corporation.
|
|
# Licensed under the MIT License
|
|
|
|
"""Typing module."""
|
|
|
|
from dataclasses import dataclass
|
|
from enum import Enum
|
|
|
|
import pandas as pd
|
|
|
|
|
|
class SearchType(Enum):
|
|
"""SearchType class definition."""
|
|
|
|
Basic = "basic"
|
|
Local = "local"
|
|
Global = "global"
|
|
Drift = "drift"
|
|
|
|
|
|
@dataclass
|
|
class SearchResult:
|
|
"""SearchResult class definition."""
|
|
|
|
# create a dataclass to store the search result of each algorithm
|
|
search_type: SearchType
|
|
response: str
|
|
context: dict[str, pd.DataFrame]
|