mirror of
https://github.com/Azure-Samples/graphrag-accelerator.git
synced 2025-08-18 14:07:11 +00:00
34 lines
796 B
Python
34 lines
796 B
Python
|
# Copyright (c) Microsoft Corporation.
|
||
|
|
# Licensed under the MIT License.
|
||
|
|
|
||
|
|
from enum import Enum
|
||
|
|
|
||
|
|
|
||
|
|
class PromptKeys(Enum):
|
||
|
|
ENTITY = "entity_extraction"
|
||
|
|
SUMMARY = "summarize_descriptions"
|
||
|
|
COMMUNITY = "community_report"
|
||
|
|
|
||
|
|
|
||
|
|
class PromptFileNames(Enum):
|
||
|
|
ENTITY = "entity_extraction_prompt.txt"
|
||
|
|
SUMMARY = "summarize_descriptions_prompt.txt"
|
||
|
|
COMMUNITY = "community_report_prompt.txt"
|
||
|
|
|
||
|
|
|
||
|
|
class PromptTextAreas(Enum):
|
||
|
|
ENTITY = "entity_text_area"
|
||
|
|
SUMMARY = "summary_text_area"
|
||
|
|
COMMUNITY = "community_text_area"
|
||
|
|
|
||
|
|
|
||
|
|
class StorageIndexVars(Enum):
|
||
|
|
SELECTED_STORAGE = "selected_storage"
|
||
|
|
INPUT_STORAGE = "input_storage"
|
||
|
|
SELECTED_INDEX = "selected_index"
|
||
|
|
|
||
|
|
|
||
|
|
class EnvVars(Enum):
|
||
|
|
APIM_SUBSCRIPTION_KEY = "APIM_SUBSCRIPTION_KEY"
|
||
|
|
DEPLOYMENT_URL = "DEPLOYMENT_URL"
|